From bbafb1ec2ddfcf45f19cc2e70c32b9042db167a7 Mon Sep 17 00:00:00 2001
From: RugerMc <550279039@qq.com>
Date: Wed, 29 Mar 2023 22:09:52 +0800
Subject: [PATCH] feat: display the balance of current key

---
 app/api/credit-summary/route.ts | 22 ++++++++++++++++++++++
 app/components/settings.tsx     | 18 ++++++++++++++++++
 app/locales/cn.ts               |  5 +++++
 app/locales/en.ts               |  5 +++++
 app/locales/tw.ts               |  5 +++++
 app/requests.ts                 | 18 +++++++++++++++++-
 6 files changed, 72 insertions(+), 1 deletion(-)
 create mode 100644 app/api/credit-summary/route.ts

diff --git a/app/api/credit-summary/route.ts b/app/api/credit-summary/route.ts
new file mode 100644
index 000000000..6c7c3c399
--- /dev/null
+++ b/app/api/credit-summary/route.ts
@@ -0,0 +1,22 @@
+import { NextRequest } from "next/server";
+export async function POST(req: NextRequest) {
+    try {
+      let apiKey = process.env.OPENAI_API_KEY;
+      const userApiKey = req.headers.get("token");
+      if (userApiKey) {
+        apiKey = userApiKey;
+      }
+  
+      const res = await fetch("https://api.openai.com/dashboard/billing/credit_grants", {
+          headers: {
+            "Content-Type": "application/json",
+            Authorization: `Bearer ${apiKey}`,
+          },
+          method: "GET"
+      });
+      console.log(res.body)
+      return new Response(res.body);
+    } catch (error) {
+      console.error("[TOKEN]", error);
+    }
+  }
\ No newline at end of file
diff --git a/app/components/settings.tsx b/app/components/settings.tsx
index 241cebae6..baf3c8fdc 100644
--- a/app/components/settings.tsx
+++ b/app/components/settings.tsx
@@ -27,6 +27,7 @@ import { getCurrentCommitId } from "../utils";
 import Link from "next/link";
 import { UPDATE_URL } from "../constant";
 import { SearchService, usePromptStore } from "../store/prompt";
+import { requestAccountBalance, requestCreditSummary, requestWithPrompt } from "../requests";
 
 function SettingItem(props: {
   title: string;
@@ -72,6 +73,7 @@ export function Settings(props: { closeSettings: () => void }) {
 
   useEffect(() => {
     checkUpdate();
+    requestAccountBalance().then((res)=> setAccountBalance(res))
   }, []);
 
   const accessStore = useAccessStore();
@@ -84,6 +86,8 @@ export function Settings(props: { closeSettings: () => void }) {
   const builtinCount = SearchService.count.builtin;
   const customCount = promptStore.prompts.size ?? 0;
 
+  const [accountBalance, setAccountBalance] = useState(-1);
+
   return (
     <>
       <div className={styles["window-header"]}>
@@ -320,10 +324,24 @@ export function Settings(props: { closeSettings: () => void }) {
               placeholder={Locale.Settings.Token.Placeholder}
               onChange={(e) => {
                 accessStore.updateToken(e.currentTarget.value);
+                requestAccountBalance().then((res)=> setAccountBalance(res))
               }}
             ></input>
           </SettingItem>
 
+          <SettingItem
+            title={Locale.Settings.AccountBalance.Title}
+            subTitle={Locale.Settings.AccountBalance.SubTitle}
+          >
+            <input
+              value={accountBalance}
+              type="number"
+              style={{maxWidth: '15%', textAlign: 'center', display: 'flex', justifyContent: 'center', alignItems: 'center' }}
+              placeholder={Locale.Settings.AccountBalance.Placeholder}
+              disabled= {true}
+            ></input>
+          </SettingItem>
+
           <SettingItem
             title={Locale.Settings.HistoryCount.Title}
             subTitle={Locale.Settings.HistoryCount.SubTitle}
diff --git a/app/locales/cn.ts b/app/locales/cn.ts
index b93ec859b..8fc0a0a45 100644
--- a/app/locales/cn.ts
+++ b/app/locales/cn.ts
@@ -97,6 +97,11 @@ const cn = {
       SubTitle: "使用自己的 Key 可绕过受控访问限制",
       Placeholder: "OpenAI API Key",
     },
+    AccountBalance: {
+      Title: "账户余额",
+      SubTitle: "查看当前 Key 账户余额",
+      Placeholder: "-1"
+    },
     AccessCode: {
       Title: "访问码",
       SubTitle: "现在是受控访问状态",
diff --git a/app/locales/en.ts b/app/locales/en.ts
index d16893725..ee8d479e2 100644
--- a/app/locales/en.ts
+++ b/app/locales/en.ts
@@ -100,6 +100,11 @@ const en: LocaleType = {
       SubTitle: "Use your key to ignore access code limit",
       Placeholder: "OpenAI API Key",
     },
+    AccountBalance: {
+      Title: "Account Balance",
+      SubTitle: "Query balance of current key",
+      Placeholder: "-1"
+    },
     AccessCode: {
       Title: "Access Code",
       SubTitle: "Access control enabled",
diff --git a/app/locales/tw.ts b/app/locales/tw.ts
index 29f5ec22a..01e8dff7c 100644
--- a/app/locales/tw.ts
+++ b/app/locales/tw.ts
@@ -98,6 +98,11 @@ const tw: LocaleType = {
       SubTitle: "使用自己的 Key 可規避受控訪問限制",
       Placeholder: "OpenAI API Key",
     },
+    AccountBalance: {
+      Title: "賬戶余額",
+      SubTitle: "查詢當前 Key 的賬戶余額",
+      Placeholder: "-1"
+    },
     AccessCode: {
       Title: "訪問碼",
       SubTitle: "現在是受控訪問狀態",
diff --git a/app/requests.ts b/app/requests.ts
index e9da87084..ba0fd944c 100644
--- a/app/requests.ts
+++ b/app/requests.ts
@@ -38,7 +38,7 @@ function getHeaders() {
   if (accessStore.token && accessStore.token.length > 0) {
     headers["token"] = accessStore.token;
   }
-
+ 
   return headers;
 }
 
@@ -57,6 +57,22 @@ export async function requestChat(messages: Message[]) {
   return (await res.json()) as ChatReponse;
 }
 
+export async function requestCreditSummary() {
+  const res = await fetch("/api/credit-summary", {
+    method: "POST",
+    headers: {
+      "Content-Type": "application/json",
+      ...getHeaders(),
+    },
+  });
+  return (await res.json())
+} 
+
+export async function requestAccountBalance() {
+  const res = await requestCreditSummary();
+  return res?.total_available ?? -1
+}
+
 export async function requestChatStream(
   messages: Message[],
   options?: {