feat: 前端页面设置ApiKey

fix: 登录之后默认无法发消息

Author: Kerwin

README.en.md

@@ -91,6 +91,8 @@ For all parameter variables, check [here](#docker-parameter-example) or see:
 
 [✓] Login or Register
 
+[✓] Set API key and other information on the front-end page.
+
 [✓] Data import and export
 
 [✓] Save message to local image

README.md

@@ -87,6 +87,8 @@
 
 [✓] 支持用户登录注册
 
+[✓] 前端页面设置 apikey 等信息
+
 [✓] 数据导入、导出
 
 [✓] 保存消息到本地图片

service/src/chatgpt/index.ts

@@ -6,6 +6,7 @@ import { SocksProxyAgent } from 'socks-proxy-agent'
 import { HttpsProxyAgent } from 'https-proxy-agent'
 import fetch from 'node-fetch'
 import axios from 'axios'
+import { getCacheConfig, getOriginConfig } from '../storage/config'
 import { sendResponse } from '../utils'
 import { isNotEmptyString } from '../utils/is'
 import type { ApiModel, ChatContext, ChatGPTUnofficialProxyAPIOptions, ModelConfig } from '../types'
@@ -21,57 +22,57 @@ const ErrorCodeMessage: Record<string, string> = {
 
 dotenv.config()
 
-const timeoutMs: number = !isNaN(+process.env.TIMEOUT_MS) ? +process.env.TIMEOUT_MS : 30 * 1000
-
 let apiModel: ApiModel
 
-if (!process.env.OPENAI_API_KEY && !process.env.OPENAI_ACCESS_TOKEN)
-  throw new Error('Missing OPENAI_API_KEY or OPENAI_ACCESS_TOKEN environment variable')
-
 let api: ChatGPTAPI | ChatGPTUnofficialProxyAPI
 
-(async () => {
+export async function initApi() {
   // More Info: https://github.com/transitive-bullshit/chatgpt-api
 
-  if (process.env.OPENAI_API_KEY) {
-    const OPENAI_API_MODEL = process.env.OPENAI_API_MODEL
+  const config = await getCacheConfig()
+  if (!config.apiKey && !config.accessToken)
+    throw new Error('Missing OPENAI_API_KEY or OPENAI_ACCESS_TOKEN environment variable')
+
+  if (config.apiKey) {
+    const OPENAI_API_MODEL = config.apiModel
     const model = isNotEmptyString(OPENAI_API_MODEL) ? OPENAI_API_MODEL : 'gpt-3.5-turbo'
 
     const options: ChatGPTAPIOptions = {
-      apiKey: process.env.OPENAI_API_KEY,
+      apiKey: config.apiKey,
       completionParams: { model },
       debug: true,
     }
 
-    if (isNotEmptyString(process.env.OPENAI_API_BASE_URL))
-      options.apiBaseUrl = process.env.OPENAI_API_BASE_URL
+    if (isNotEmptyString(config.apiBaseUrl))
+      options.apiBaseUrl = config.apiBaseUrl
 
-    setupProxy(options)
+    await setupProxy(options)
 
     api = new ChatGPTAPI({ ...options })
     apiModel = 'ChatGPTAPI'
   }
   else {
     const options: ChatGPTUnofficialProxyAPIOptions = {
-      accessToken: process.env.OPENAI_ACCESS_TOKEN,
+      accessToken: config.accessToken,
       debug: true,
     }
 
-    if (isNotEmptyString(process.env.API_REVERSE_PROXY))
-      options.apiReverseProxyUrl = process.env.API_REVERSE_PROXY
+    if (isNotEmptyString(config.reverseProxy))
+      options.apiReverseProxyUrl = config.reverseProxy
 
-    setupProxy(options)
+    await setupProxy(options)
 
     api = new ChatGPTUnofficialProxyAPI({ ...options })
     apiModel = 'ChatGPTUnofficialProxyAPI'
   }
-})()
+}
 
 async function chatReplyProcess(
   message: string,
   lastContext?: { conversationId?: string; parentMessageId?: string },
   process?: (chat: ChatMessage) => void,
 ) {
+  const timeoutMs = (await getCacheConfig()).timeoutMs
   try {
     let options: SendMessageOptions = { timeoutMs }
 
@@ -101,8 +102,9 @@ async function chatReplyProcess(
 }
 
 async function fetchBalance() {
-  const OPENAI_API_KEY = process.env.OPENAI_API_KEY
-  const OPENAI_API_BASE_URL = process.env.OPENAI_API_BASE_URL
+  const config = await getCacheConfig()
+  const OPENAI_API_KEY = config.apiKey
+  const OPENAI_API_BASE_URL = config.apiBaseUrl
 
   if (!isNotEmptyString(OPENAI_API_KEY))
     return Promise.resolve('-')
@@ -123,31 +125,28 @@ async function fetchBalance() {
 }
 
 async function chatConfig() {
-  const balance = await fetchBalance()
-  const reverseProxy = process.env.API_REVERSE_PROXY ?? '-'
-  const httpsProxy = (process.env.HTTPS_PROXY || process.env.ALL_PROXY) ?? '-'
-  const socksProxy = (process.env.SOCKS_PROXY_HOST && process.env.SOCKS_PROXY_PORT)
-    ? (`${process.env.SOCKS_PROXY_HOST}:${process.env.SOCKS_PROXY_PORT}`)
-    : '-'
+  const config = await getOriginConfig() as ModelConfig
+  config.balance = await fetchBalance()
   return sendResponse<ModelConfig>({
     type: 'Success',
-    data: { apiModel, reverseProxy, timeoutMs, socksProxy, httpsProxy, balance },
+    data: config,
   })
 }
 
-function setupProxy(options: ChatGPTAPIOptions | ChatGPTUnofficialProxyAPIOptions) {
-  if (process.env.SOCKS_PROXY_HOST && process.env.SOCKS_PROXY_PORT) {
+async function setupProxy(options: ChatGPTAPIOptions | ChatGPTUnofficialProxyAPIOptions) {
+  const config = await getCacheConfig()
+  if (config.socksProxy) {
     const agent = new SocksProxyAgent({
-      hostname: process.env.SOCKS_PROXY_HOST,
-      port: process.env.SOCKS_PROXY_PORT,
+      hostname: config.socksProxy.split(':')[0],
+      port: parseInt(config.socksProxy.split(':')[1]),
     })
     options.fetch = (url, options) => {
       return fetch(url, { agent, ...options })
     }
   }
   else {
-    if (process.env.HTTPS_PROXY || process.env.ALL_PROXY) {
-      const httpsProxy = process.env.HTTPS_PROXY || process.env.ALL_PROXY
+    if (config.httpsProxy || process.env.ALL_PROXY) {
+      const httpsProxy = config.httpsProxy || process.env.ALL_PROXY
       if (httpsProxy) {
         const agent = new HttpsProxyAgent(httpsProxy)
         options.fetch = (url, options) => {
@@ -162,6 +161,8 @@ function currentModel(): ApiModel {
   return apiModel
 }
 
+initApi()
+
 export type { ChatContext, ChatMessage }
 
 export { chatReplyProcess, chatConfig, currentModel }

service/src/index.ts

@@ -2,11 +2,13 @@ import express from 'express'
 import jwt from 'jsonwebtoken'
 import { ObjectId } from 'mongodb'
 import type { ChatContext, ChatMessage } from './chatgpt'
-import { chatConfig, chatReplyProcess, currentModel } from './chatgpt'
+import { chatConfig, chatReplyProcess, currentModel, initApi } from './chatgpt'
 import { auth } from './middleware/auth'
-import type { ChatOptions, UserInfo } from './storage/model'
+import { clearConfigCache, getCacheConfig, getOriginConfig } from './storage/config'
+import type { ChatOptions, Config, MailConfig, SiteConfig, UserInfo } from './storage/model'
 import { Status } from './storage/model'
-import { clearChat, createChatRoom, createUser, deleteChat, deleteChatRoom, existsChatRoom, getChat, getChatRooms, getChats, getUser, getUserById, insertChat, renameChatRoom, updateChat, updateUserInfo, verifyUser } from './storage/mongo'
+import { clearChat, createChatRoom, createUser, deleteChat, deleteChatRoom, existsChatRoom, getChat, getChatRooms, getChats, getUser, getUserById, insertChat, renameChatRoom, updateChat, updateConfig, updateUserInfo, verifyUser } from './storage/mongo'
+import { isNotEmptyString } from './utils/is'
 import { sendMail } from './utils/mail'
 import { checkUserVerify, getUserVerifyUrl, md5 } from './utils/security'
 
@@ -175,14 +177,14 @@ router.post('/chat-process', auth, async (req, res) => {
 
 router.post('/user-register', async (req, res) => {
   const { username, password } = req.body as { username: string; password: string }
-
-  if (process.env.REGISTER_ENABLED !== 'true') {
+  const config = await getCacheConfig()
+  if (config.siteConfig.registerEnabled) {
     res.send({ status: 'Fail', message: '注册账号功能未启用 | Register account is disabled!', data: null })
     return
   }
-  if (typeof process.env.REGISTER_MAILS === 'string' && process.env.REGISTER_MAILS.length > 0) {
+  if (isNotEmptyString(config.siteConfig.registerMails)) {
     let allowSuffix = false
-    const emailSuffixs = process.env.REGISTER_MAILS.split(',')
+    const emailSuffixs = config.siteConfig.registerMails.split(',')
     for (let index = 0; index < emailSuffixs.length; index++) {
       const element = emailSuffixs[index]
       allowSuffix = username.toLowerCase().endsWith(element)
@@ -207,13 +209,19 @@ router.post('/user-register', async (req, res) => {
     res.send({ status: 'Success', message: '注册成功 | Register success', data: null })
   }
   else {
-    sendMail(username, getUserVerifyUrl(username))
+    await sendMail(username, await getUserVerifyUrl(username))
     res.send({ status: 'Success', message: '注册成功, 去邮箱中验证吧 | Registration is successful, you need to go to email verification', data: null })
   }
 })
 
-router.post('/config', async (req, res) => {
+router.post('/config', auth, async (req, res) => {
   try {
+    const userId = new ObjectId(req.headers.userId.toString())
+
+    const user = await getUserById(userId)
+    if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
+      throw new Error('无权限 | No permission.')
+
     const response = await chatConfig()
     res.send(response)
   }
@@ -226,7 +234,7 @@ router.post('/session', async (req, res) => {
   try {
     const AUTH_SECRET_KEY = process.env.AUTH_SECRET_KEY
     const hasAuth = typeof AUTH_SECRET_KEY === 'string' && AUTH_SECRET_KEY.length > 0
-    const allowRegister = process.env.REGISTER_ENABLED === 'true'
+    const allowRegister = (await getCacheConfig()).siteConfig.registerEnabled
     res.send({ status: 'Success', message: '', data: { auth: hasAuth, allowRegister, model: currentModel() } })
   }
   catch (error) {
@@ -293,6 +301,77 @@ router.post('/verify', async (req, res) => {
   }
 })
 
+router.post('/setting-base', auth, async (req, res) => {
+  try {
+    const { apiKey, apiModel, apiBaseUrl, accessToken, timeoutMs, socksProxy, httpsProxy } = req.body as Config
+    const userId = new ObjectId(req.headers.userId.toString())
+
+    if (apiKey == null && accessToken == null)
+      throw new Error('Missing OPENAI_API_KEY or OPENAI_ACCESS_TOKEN environment variable.')
+
+    const user = await getUserById(userId)
+    if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
+      throw new Error('无权限 | No permission.')
+
+    const thisConfig = await getOriginConfig()
+    thisConfig.apiKey = apiKey
+    thisConfig.apiModel = apiModel
+    thisConfig.apiBaseUrl = apiBaseUrl
+    thisConfig.accessToken = accessToken
+    thisConfig.timeoutMs = timeoutMs
+    thisConfig.socksProxy = socksProxy
+    thisConfig.httpsProxy = httpsProxy
+    await updateConfig(thisConfig)
+    clearConfigCache()
+    initApi()
+    const response = await chatConfig()
+    res.send({ status: 'Success', message: '操作成功 | Successfully', data: response.data })
+  }
+  catch (error) {
+    res.send({ status: 'Fail', message: error.message, data: null })
+  }
+})
+
+router.post('/setting-site', auth, async (req, res) => {
+  try {
+    const config = req.body as SiteConfig
+    const userId = new ObjectId(req.headers.userId.toString())
+
+    const user = await getUserById(userId)
+    if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
+      throw new Error('无权限 | No permission.')
+
+    const thisConfig = await getOriginConfig()
+    thisConfig.siteConfig = config
+    const result = await updateConfig(thisConfig)
+    clearConfigCache()
+    res.send({ status: 'Success', message: '操作成功 | Successfully', data: result.siteConfig })
+  }
+  catch (error) {
+    res.send({ status: 'Fail', message: error.message, data: null })
+  }
+})
+
+router.post('/setting-mail', auth, async (req, res) => {
+  try {
+    const config = req.body as MailConfig
+    const userId = new ObjectId(req.headers.userId.toString())
+
+    const user = await getUserById(userId)
+    if (user == null || user.status !== Status.Normal || user.email.toLowerCase() !== process.env.ROOT_USER)
+      throw new Error('无权限 | No permission.')
+
+    const thisConfig = await getOriginConfig()
+    thisConfig.mailConfig = config
+    const result = await updateConfig(thisConfig)
+    clearConfigCache()
+    res.send({ status: 'Success', message: '操作成功 | Successfully', data: result.mailConfig })
+  }
+  catch (error) {
+    res.send({ status: 'Fail', message: error.message, data: null })
+  }
+})
+
 app.use('', router)
 app.use('/api', router)
 

service/src/storage/config.ts

@@ -0,0 +1,52 @@
+import { ObjectId } from 'mongodb'
+import { Config, MailConfig, SiteConfig } from './model'
+import { getConfig } from './mongo'
+
+let cachedConfig: Config | undefined
+let cacheExpiration = 0
+
+export async function getCacheConfig(): Promise<Config> {
+  const now = Date.now()
+  if (cachedConfig && cacheExpiration > now)
+    return Promise.resolve(cachedConfig)
+
+  const loadedConfig = await getOriginConfig()
+
+  cachedConfig = loadedConfig
+  cacheExpiration = now + 10 * 60 * 1000
+
+  return Promise.resolve(cachedConfig)
+}
+
+export async function getOriginConfig() {
+  let config = await getConfig()
+  if (config == null) {
+    config = new Config(new ObjectId(),
+      !isNaN(+process.env.TIMEOUT_MS) ? +process.env.TIMEOUT_MS : 30 * 1000,
+      process.env.OPENAI_API_KEY,
+      process.env.OPENAI_ACCESS_TOKEN,
+      process.env.OPENAI_API_BASE_URL,
+      process.env.OPENAI_API_MODEL || 'gpt-3.5-turbo',
+      process.env.API_REVERSE_PROXY,
+      (process.env.SOCKS_PROXY_HOST && process.env.SOCKS_PROXY_PORT)
+        ? (`${process.env.SOCKS_PROXY_HOST}:${process.env.SOCKS_PROXY_PORT}`)
+        : '',
+      process.env.HTTPS_PROXY,
+      new SiteConfig(
+        process.env.SITE_TITLE || 'ChatGpt Web',
+        process.env.REGISTER_ENABLED === 'true',
+        process.env.REGISTER_MAILS,
+        process.env.SITE_DOMAIN),
+      new MailConfig(process.env.SMTP_HOST,
+        !isNaN(+process.env.SMTP_PORT) ? +process.env.SMTP_PORT : 465,
+        process.env.SMTP_TSL === 'true',
+        process.env.SMTP_USERNAME,
+        process.env.SMTP_PASSWORD))
+  }
+  return config
+}
+
+export function clearConfigCache() {
+  cacheExpiration = 0
+  cachedConfig = null
+}