google login and power share link (#292)

client/src/service/common/ipLimit/schema.ts

@@ -0,0 +1,24 @@
+import { Schema, model, models, Model } from 'mongoose';
+import type { IpLimitSchemaType } from '@/types/common/ipLimit';
+
+const IpLimitSchema = new Schema({
+  eventId: {
+    type: String,
+    required: true
+  },
+  ip: {
+    type: String,
+    required: true
+  },
+  account: {
+    type: Number,
+    default: 0
+  },
+  lastMinute: {
+    type: Date,
+    default: () => new Date()
+  }
+});
+
+export const IpLimit: Model<IpLimitSchemaType> =
+  models['ip_limit'] || model('ip_limit', IpLimitSchema);

client/src/service/mongo.ts

@@ -133,7 +133,7 @@ export * from './models/trainingData';
 export * from './models/openapi';
 export * from './models/promotionRecord';
 export * from './models/collection';
-export * from './models/outLink';
 export * from './models/kb';
 export * from './models/inform';
 export * from './models/image';
+export * from './support/outLink/schema';

client/src/service/support/outLink/auth.ts

@@ -0,0 +1,79 @@
+import { PRICE_SCALE } from '@/constants/common';
+import { IpLimit } from '@/service/common/ipLimit/schema';
+import { authBalanceByUid, AuthUserTypeEnum } from '@/service/utils/auth';
+import { OutLinkSchema } from '@/types/support/outLink';
+import { OutLink } from './schema';
+
+export async function authOutLinkChat({ shareId, ip }: { shareId: string; ip?: string | null }) {
+  // get outLink
+  const outLink = await OutLink.findOne({
+    shareId
+  });
+
+  if (!outLink) {
+    return Promise.reject('分享链接无效');
+  }
+
+  const uid = String(outLink.userId);
+
+  // authBalance
+  const user = await authBalanceByUid(uid);
+
+  // limit auth
+  await authOutLinkLimit({ outLink, ip });
+
+  return {
+    user,
+    userId: String(outLink.userId),
+    appId: String(outLink.appId),
+    authType: AuthUserTypeEnum.token,
+    responseDetail: outLink.responseDetail
+  };
+}
+
+export async function authOutLinkLimit({
+  outLink,
+  ip
+}: {
+  outLink: OutLinkSchema;
+  ip?: string | null;
+}) {
+  if (!ip || !outLink.limit) {
+    return;
+  }
+
+  if (outLink.limit.expiredTime && outLink.limit.expiredTime.getTime() < Date.now()) {
+    return Promise.reject('分享链接已过期');
+  }
+
+  if (outLink.limit.credit > -1 && outLink.total > outLink.limit.credit * PRICE_SCALE) {
+    return Promise.reject('链接超出使用限制');
+  }
+
+  const ipLimit = await IpLimit.findOne({ ip, eventId: outLink._id });
+
+  try {
+    if (!ipLimit) {
+      await IpLimit.create({
+        eventId: outLink._id,
+        ip,
+        account: outLink.limit.QPM - 1
+      });
+      return;
+    }
+    // over one minute
+    const diffTime = Date.now() - ipLimit.lastMinute.getTime();
+    if (diffTime >= 60 * 1000) {
+      ipLimit.account = outLink.limit.QPM - 1;
+      ipLimit.lastMinute = new Date();
+      return await ipLimit.save();
+    }
+    if (ipLimit.account <= 0) {
+      return Promise.reject(
+        `每分钟仅能请求 ${outLink.limit.QPM} 次, ${60 - Math.round(diffTime / 1000)}s 后重试~`
+      );
+    }
+    ipLimit.account = ipLimit.account - 1;
+    await ipLimit.save();
+  } catch (error) {}
+}

client/src/service/support/outLink/schema.ts

@@ -1,5 +1,5 @@
 import { Schema, model, models, Model } from 'mongoose';
-import { OutLinkSchema as SchmaType } from '@/types/mongoSchema';
+import { OutLinkSchema as SchemaType } from '@/types/support/outLink';
 import { OutLinkTypeEnum } from '@/constants/chat';
 
 const OutLinkSchema = new Schema({
@@ -26,12 +26,30 @@ const OutLinkSchema = new Schema({
     required: true
   },
   total: {
+    // total amount
     type: Number,
     default: 0
   },
   lastTime: {
     type: Date
+  },
+  responseDetail: {
+    type: Boolean,
+    default: false
+  },
+  limit: {
+    expiredTime: {
+      type: Date
+    },
+    QPM: {
+      type: Number,
+      default: 1000
+    },
+    credit: {
+      type: Number,
+      default: -1
+    }
   }
 });
 
-export const OutLink: Model<SchmaType> = models['outlinks'] || model('outlinks', OutLinkSchema);
+export const OutLink: Model<SchemaType> = models['outlinks'] || model('outlinks', OutLinkSchema);

client/src/service/utils/auth.ts

@@ -208,24 +208,3 @@ export const authKb = async ({ kbId, userId }: { kbId: string; userId: string })
   }
   return Promise.reject(ERROR_ENUM.unAuthKb);
 };
-
-export const authShareChat = async ({ shareId }: { shareId: string }) => {
-  // get shareChat
-  const shareChat = await OutLink.findOne({ shareId });
-
-  if (!shareChat) {
-    return Promise.reject('分享链接已失效');
-  }
-
-  const uid = String(shareChat.userId);
-
-  // authBalance
-  const user = await authBalanceByUid(uid);
-
-  return {
-    user,
-    userId: String(shareChat.userId),
-    appId: String(shareChat.appId),
-    authType: AuthUserTypeEnum.token
-  };
-};