feat: 临时data

src/service/models/data.ts

@@ -0,0 +1,27 @@
+import { Schema, model, models } from 'mongoose';
+
+const DataSchema = new Schema({
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  docId: {
+    type: String,
+    required: true
+  },
+  createTime: {
+    type: Date,
+    default: () => new Date()
+  },
+  q: {
+    type: String,
+    required: true
+  },
+  a: {
+    type: String,
+    required: true
+  }
+});
+
+export const Data = models['data'] || model('data', DataSchema);

src/service/mongo.ts

@@ -32,3 +32,4 @@ export * from './models/user';
 export * from './models/training';
 export * from './models/bill';
 export * from './models/pay';
+export * from './models/data';

src/service/utils/tools.ts

@@ -2,6 +2,8 @@ import crypto from 'crypto';
 import jwt from 'jsonwebtoken';
 import { User } from '../models/user';
 import tunnel from 'tunnel';
+import type { UserModelSchema } from '@/types/mongoSchema';
+import { formatPrice } from '@/utils/user';
 
 /* 密码加密 */
 export const hashPassword = (psw: string) => {
@@ -49,7 +51,37 @@ export const getUserOpenaiKey = async (userId: string) => {
     return Promise.reject('缺少ApiKey, 无法请求');
   }
 
-  return Promise.resolve(userApiKey);
+  return Promise.resolve(userApiKey as string);
+};
+
+/* 获取key，如果没有就用平台的，用平台记得加账单 */
+export const getOpenApiKey = async (authorization?: string) => {
+  const userId = await authToken(authorization);
+  const user = await User.findById<UserModelSchema>(userId);
+
+  if (!user) return Promise.reject('用户不存在');
+
+  const userApiKey = user.accounts?.find((item: any) => item.type === 'openai')?.value;
+
+  // 有自己的key， 直接使用
+  if (userApiKey) {
+    return {
+      userId,
+      userApiKey: await getUserOpenaiKey(userId),
+      systemKey: ''
+    };
+  }
+
+  // 余额校验
+  if (formatPrice(user.balance) <= 0) {
+    return Promise.reject('该账号余额不足');
+  }
+
+  return {
+    userId,
+    userApiKey: '',
+    systemKey: process.env.OPENAIKEY as string
+  };
 };
 
 /* 代理 */