From 4f6e56fd22e78b7d8df8ac2fe37fd7cf91d769ef Mon Sep 17 00:00:00 2001
From: Pas <74743263+Pasithea0@users.noreply.github.com>
Date: Fri, 19 Dec 2025 15:58:54 -0700
Subject: [PATCH] show febbox usage

---
 src/assets/locales/en.json                   |  7 +++--
 src/pages/parts/settings/ConnectionsPart.tsx | 33 ++++++++++++++++----
 src/pages/parts/settings/SetupPart.tsx       | 27 ++++++++++++++++
 3 files changed, 58 insertions(+), 9 deletions(-)

diff --git a/src/assets/locales/en.json b/src/assets/locales/en.json
index b7cb3005..04c10f8e 100644
--- a/src/assets/locales/en.json
+++ b/src/assets/locales/en.json
@@ -1276,8 +1276,7 @@
         "2": "2. Open DevTools or inspect the page",
         "3": "3. Go to Application tab → Cookies",
         "4": "4. Copy the 'ui' cookie's value.",
-        "5": "5. Close the tab, but do NOT logout!",
-        "warning": "(Do not share this token!)"
+        "5": "5. Close the tab, but do NOT logout!"
       },
       "tokenLabel": "Token",
       "tokenExample": {
@@ -1286,7 +1285,9 @@
         "description": "This is what a Febbox UI token looks like:",
         "warning": "Don't try to use, it's fake.",
         "close": "Got it"
-      }
+      },
+      "traffic": "{{used}} / {{limit}} High-speed Traffic • Resets in {{reset}}",
+      "trafficExplanation": "Febbox gives you 100GB/month of high-speed traffic, the streams might buffer more after you've used up your quota. Depends on your internet speed and the quality of the stream."
     },
     "status": {
       "success": "success",
diff --git a/src/pages/parts/settings/ConnectionsPart.tsx b/src/pages/parts/settings/ConnectionsPart.tsx
index c0888c3d..443b916a 100644
--- a/src/pages/parts/settings/ConnectionsPart.tsx
+++ b/src/pages/parts/settings/ConnectionsPart.tsx
@@ -24,6 +24,7 @@ import { Heading1, Heading2, Paragraph } from "@/components/utils/Text";
 import {
   SetupPart,
   Status,
+  fetchFebboxQuota,
   testFebboxKey,
   testTorboxToken,
   testdebridToken,
@@ -237,9 +238,10 @@ function BackendEdit({ backendUrl, setBackendUrl }: BackendEditProps) {
 async function getFebboxKeyStatus(febboxKey: string | null) {
   if (febboxKey) {
     const status: Status = await testFebboxKey(febboxKey);
-    return status;
+    const quota = await fetchFebboxQuota(febboxKey);
+    return { status, quota };
   }
-  return "unset";
+  return { status: "unset" as Status, quota: null };
 }
 
 interface FebboxSetupProps extends FebboxKeyProps {
@@ -282,6 +284,7 @@ export function FebboxSetup({
   }, [user.account, febboxKey, preferences.febboxKey, setFebboxKey, mode]);
 
   const [status, setStatus] = useState<Status>("unset");
+  const [quota, setQuota] = useState<any>(null);
   const statusMap: Record<Status, StatusCircleProps["type"]> = {
     error: "error",
     success: "success",
@@ -293,7 +296,8 @@ export function FebboxSetup({
   useEffect(() => {
     const checkTokenStatus = async () => {
       const result = await getFebboxKeyStatus(febboxKey);
-      setStatus(result);
+      setStatus(result.status);
+      setQuota(result.quota);
     };
     checkTokenStatus();
   }, [febboxKey]);
@@ -395,9 +399,6 @@ export function FebboxSetup({
                   <br />
                   <Trans i18nKey="fedapi.setup.step.5" />
                 </p>
-                <p className="text-type-danger mt-2">
-                  <Trans i18nKey="fedapi.setup.step.warning" />
-                </p>
               </div>
 
               <Divider marginClass="my-6 px-8 box-content -mx-8" />
@@ -438,6 +439,26 @@ export function FebboxSetup({
                   {t("fedapi.status.invalid_token")}
                 </p>
               )}
+              {status === "success" &&
+                quota &&
+                (() => {
+                  if (!quota?.data?.flow) return null;
+                  const {
+                    traffic_usage: used,
+                    traffic_limit: limit,
+                    reset_at: reset,
+                  } = quota.data.flow;
+                  return (
+                    <>
+                      <p className="text-sm text-green-500 mt-2">
+                        {t("fedapi.setup.traffic", { used, limit, reset })}
+                      </p>
+                      <p className="max-w-[30rem] text-xs opacity-70 mt-2">
+                        {t("fedapi.setup.trafficExplanation")}
+                      </p>
+                    </>
+                  );
+                })()}
             </>
           ) : null}
         </SettingsCard>
diff --git a/src/pages/parts/settings/SetupPart.tsx b/src/pages/parts/settings/SetupPart.tsx
index 0dd9ca3c..897ca4ef 100644
--- a/src/pages/parts/settings/SetupPart.tsx
+++ b/src/pages/parts/settings/SetupPart.tsx
@@ -58,6 +58,33 @@ function testProxy(url: string) {
   });
 }
 
+export async function fetchFebboxQuota(febboxKey: string | null): Promise<any> {
+  if (!febboxKey) {
+    return null;
+  }
+
+  console.log("SetupPart.tsx: Fetching Febbox quota");
+  try {
+    const response = await fetch("https://fed-api.pstream.mov/quota", {
+      headers: {
+        "ui-token": febboxKey,
+      },
+    });
+
+    if (!response.ok) {
+      console.error("Febbox quota API failed with status:", response.status);
+      return null;
+    }
+
+    const data = await response.json();
+    console.log("SetupPart.tsx: Febbox quota fetched successfully");
+    return data;
+  } catch (error) {
+    console.error("SetupPart.tsx: Error fetching Febbox quota:", error);
+    return null;
+  }
+}
+
 export async function testFebboxKey(febboxKey: string | null): Promise<Status> {
   const febboxApiTestUrl = `https://fed-api.pstream.mov/movie/tt0325980`;