From 95724845cf7429b3ba3df43a7e862d0ebecca5bd Mon Sep 17 00:00:00 2001
From: ChenZhaoYu <790348264@qq.com>
Date: Fri, 17 Mar 2023 09:36:12 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20session=20=E8=BF=94=E5=9B=9E=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E6=96=B9=E5=BC=8F=E5=81=9A=E6=98=BE=E7=A4=BA=E5=88=A4?=
 =?UTF-8?q?=E6=96=AD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 service/src/index.ts                    |  4 ++--
 src/components/common/Setting/About.vue | 15 ++++++++++++---
 src/store/modules/auth/index.ts         | 15 +++++++++++++--
 3 files changed, 27 insertions(+), 7 deletions(-)

diff --git a/service/src/index.ts b/service/src/index.ts
index a14b843..467c2b3 100644
--- a/service/src/index.ts
+++ b/service/src/index.ts
@@ -1,6 +1,6 @@
 import express from 'express'
 import type { ChatContext, ChatMessage } from './chatgpt'
-import { chatConfig, chatReplyProcess } from './chatgpt'
+import { chatConfig, chatReplyProcess, currentModel } from './chatgpt'
 import { auth } from './middleware/auth'
 import { isNotEmptyString } from './utils/is'
 
@@ -50,7 +50,7 @@ router.post('/session', async (req, res) => {
   try {
     const AUTH_SECRET_KEY = process.env.AUTH_SECRET_KEY
     const hasAuth = isNotEmptyString(AUTH_SECRET_KEY)
-    res.send({ status: 'Success', message: '', data: { auth: hasAuth } })
+    res.send({ status: 'Success', message: '', data: { auth: hasAuth, model: currentModel() } })
   }
   catch (error) {
     res.send({ status: 'Fail', message: error.message, data: null })
diff --git a/src/components/common/Setting/About.vue b/src/components/common/Setting/About.vue
index 4314415..3edb317 100644
--- a/src/components/common/Setting/About.vue
+++ b/src/components/common/Setting/About.vue
@@ -1,8 +1,9 @@
 <script setup lang='ts'>
-import { onMounted, ref } from 'vue'
+import { computed, onMounted, ref } from 'vue'
 import { NSpin } from 'naive-ui'
 import { fetchChatConfig } from '@/api'
 import pkg from '@/../package.json'
+import { useAuthStore } from '@/store'
 
 interface ConfigState {
   timeoutMs?: number
@@ -13,10 +14,14 @@ interface ConfigState {
   balance?: string
 }
 
+const authStore = useAuthStore()
+
 const loading = ref(false)
 
 const config = ref<ConfigState>()
 
+const isChatGPTAPI = computed<boolean>(() => !!authStore.isChatGPTAPI)
+
 async function fetchConfig() {
   try {
     loading.value = true
@@ -56,8 +61,12 @@ onMounted(() => {
         </p>
       </div>
       <p>{{ $t("setting.api") }}：{{ config?.apiModel ?? '-' }}</p>
-      <p>{{ $t("setting.balance") }}：{{ config?.balance ?? '-' }}</p>
-      <p>{{ $t("setting.reverseProxy") }}：{{ config?.reverseProxy ?? '-' }}</p>
+      <p v-if="isChatGPTAPI">
+        {{ $t("setting.balance") }}：{{ config?.balance ?? '-' }}
+      </p>
+      <p v-if="!isChatGPTAPI">
+        {{ $t("setting.reverseProxy") }}：{{ config?.reverseProxy ?? '-' }}
+      </p>
       <p>{{ $t("setting.timeout") }}：{{ config?.timeoutMs ?? '-' }}</p>
       <p>{{ $t("setting.socks") }}：{{ config?.socksProxy ?? '-' }}</p>
       <p>{{ $t("setting.httpsProxy") }}：{{ config?.httpsProxy ?? '-' }}</p>
diff --git a/src/store/modules/auth/index.ts b/src/store/modules/auth/index.ts
index 5b8c8d8..8edde9c 100644
--- a/src/store/modules/auth/index.ts
+++ b/src/store/modules/auth/index.ts
@@ -3,9 +3,14 @@ import { getToken, removeToken, setToken } from './helper'
 import { store } from '@/store'
 import { fetchSession } from '@/api'
 
+interface SessionResponse {
+  auth: boolean
+  model: 'ChatGPTAPI' | 'ChatGPTUnofficialProxyAPI'
+}
+
 export interface AuthState {
   token: string | undefined
-  session: { auth: boolean } | null
+  session: SessionResponse | null
 }
 
 export const useAuthStore = defineStore('auth-store', {
@@ -14,10 +19,16 @@ export const useAuthStore = defineStore('auth-store', {
     session: null,
   }),
 
+  getters: {
+    isChatGPTAPI(state): boolean {
+      return state.session?.model === 'ChatGPTAPI'
+    },
+  },
+
   actions: {
     async getSession() {
       try {
-        const { data } = await fetchSession<{ auth: boolean }>()
+        const { data } = await fetchSession<SessionResponse>()
         this.session = { ...data }
         return Promise.resolve(data)
       }