feat: 显示AccessToken过期时间(Close #145)

Author: Kerwin

service/src/chatgpt/index.ts

@@ -6,12 +6,14 @@ import { SocksProxyAgent } from 'socks-proxy-agent'
 import httpsProxyAgent from 'https-proxy-agent'
 import fetch from 'node-fetch'
 import type { AuditConfig } from 'src/storage/model'
+import jwt_decode from 'jwt-decode'
+import dayjs from 'dayjs'
 import type { TextAuditService } from '../utils/textAudit'
 import { textAuditServices } from '../utils/textAudit'
 import { getCacheConfig, getOriginConfig } from '../storage/config'
 import { sendResponse } from '../utils'
 import { isNotEmptyString } from '../utils/is'
-import type { ChatContext, ChatGPTUnofficialProxyAPIOptions, ModelConfig } from '../types'
+import type { ChatContext, ChatGPTUnofficialProxyAPIOptions, JWT, ModelConfig } from '../types'
 import { getChatByMessageId } from '../storage/mongo'
 import type { RequestOptions } from './types'
 
@@ -145,6 +147,15 @@ async function containsSensitiveWords(audit: AuditConfig, text: string): Promise
   }
   return false
 }
+
+async function fetchAccessTokenExpiredTime() {
+  const config = await getCacheConfig()
+  const jwt = jwt_decode(config.accessToken) as JWT
+  if (jwt.exp)
+    return dayjs.unix(jwt.exp).format('YYYY-MM-DD HH:mm:ss')
+  return '-'
+}
+
 let cachedBalance: number | undefined
 let cacheExpiration = 0
 
@@ -230,7 +241,10 @@ function formatDate(date) {
 
 async function chatConfig() {
   const config = await getOriginConfig() as ModelConfig
-  config.balance = await fetchBalance()
+  if (config.apiModel === 'ChatGPTAPI')
+    config.balance = await fetchBalance()
+  else
+    config.accessTokenExpiredTime = await fetchAccessTokenExpiredTime()
   return sendResponse<ModelConfig>({
     type: 'Success',
     data: config,

service/src/types.ts

@@ -35,6 +35,24 @@ export interface ModelConfig {
   httpsProxy?: string
   allowRegister?: boolean
   balance?: string
+  accessTokenExpiredTime?: string
 }
 
 export type ApiModel = 'ChatGPTAPI' | 'ChatGPTUnofficialProxyAPI' | undefined
+
+export interface JWT {
+  'https://api.openai.com/profile': {
+    'email': string
+    'email_verified': boolean
+  }
+  'https://api.openai.com/auth': {
+    'user_id': string
+  }
+  'iss': string
+  'sub': string
+  'aud': []
+  'iat': number
+  'exp': number
+  'azp': string
+  'scope': string
+}

src/components/common/Setting/About.vue

@@ -1,6 +1,6 @@
 <script setup lang='ts'>
 import { onMounted, ref, watch } from 'vue'
-import { NButton, NInput, NSelect, NSpin, useMessage } from 'naive-ui'
+import { NButton, NInput, NSelect, NSpin, NTag, useMessage } from 'naive-ui'
 import { ConfigState } from './model'
 import { fetchChatConfig, fetchUpdateBaseSetting } from '@/api'
 import { t } from '@/locales'
@@ -110,8 +110,17 @@ onMounted(() => {
           <div class="flex-1">
             <NInput :value="config.accessToken" placeholder="" @input="(val) => { config.accessToken = val }" />
           </div>
+        </div>
+        <div v-if="!isChatGPTAPI" class="flex items-center space-x-4">
+          <span class="flex-shrink-0 w-[100px]">{{ $t('setting.accessTokenExpiredTime') }}</span>
+          <div class="flex-1">
+            {{ config.accessTokenExpiredTime }}
+            <NTag v-if="config.accessTokenExpiredTime && config.accessTokenExpiredTime !== '-' && new Date(config.accessTokenExpiredTime as string) < new Date()" :bordered="false" type="warning">
+              {{ new Date(config.accessTokenExpiredTime as string) > new Date() ? '' : 'Expired' }}
+            </NTag>
+          </div>
           <p>
-            <a target="_blank" href="https://chat.openai.com/api/auth/session">Get Token</a>
+            <a target="_blank" href="https://chat.openai.com/api/auth/session">Goto Refresh Token</a>
           </p>
         </div>
         <div v-if="!isChatGPTAPI" class="flex items-center space-x-4">

src/components/common/Setting/model.ts

@@ -2,6 +2,7 @@ export class ConfigState {
   timeoutMs?: number
   apiKey?: string
   accessToken?: string
+  accessTokenExpiredTime?: string
   apiBaseUrl?: string
   apiModel?: ApiModel
   chatModel?: CHATMODEL

src/locales/en-US.ts

@@ -124,6 +124,7 @@ export default {
     auditBaiduLabelLink: 'Goto Label Detail',
     auditCustomizeEnabled: 'Customize',
     auditCustomizeWords: 'Sensitive Words',
+    accessTokenExpiredTime: 'Expired Time',
   },
   store: {
     siderButton: 'Prompt Store',

src/locales/ko-KR.ts

@@ -122,6 +122,7 @@ export default {
     auditBaiduLabelLink: '레이블 세부 정보로 이동',
     auditCustomizeEnabled: '맞춤화하다',
     auditCustomizeWords: '단어 맞춤설정',
+    accessTokenExpiredTime: '만료된 시간',
   },
   store: {
     siderButton: '프롬프트 스토어',

src/locales/zh-CN.ts

@@ -124,6 +124,7 @@ export default {
     auditBaiduLabelLink: '查看细分类型',
     auditCustomizeEnabled: '自定义',
     auditCustomizeWords: '敏感词',
+    accessTokenExpiredTime: '过期时间',
   },
   store: {
     siderButton: '提示词商店',

src/locales/zh-TW.ts

@@ -124,6 +124,7 @@ export default {
     auditBaiduLabelLink: '查看细分类型',
     auditCustomizeEnabled: '自定义',
     auditCustomizeWords: '敏感词',
+    accessTokenExpiredTime: '过期时间',
   },
   store: {
     siderButton: '提示詞商店',