Extraction schema (#398)

packages/support/outLink/auth.ts

@@ -0,0 +1,67 @@
+import { AuthUserTypeEnum, authBalanceByUid } from '../user/auth';
+import { MongoOutLink } from './schema';
+import { POST } from '@fastgpt/common/plusApi/request';
+import { OutLinkSchema } from './type.d';
+
+export type AuthLinkProps = { ip?: string | null; authToken?: string; question: string };
+export type AuthLinkLimitProps = AuthLinkProps & { outLink: OutLinkSchema };
+
+export async function authOutLinkChat({
+  shareId,
+  ip,
+  authToken,
+  question
+}: AuthLinkProps & {
+  shareId: string;
+}) {
+  // get outLink
+  const outLink = await MongoOutLink.findOne({
+    shareId
+  });
+
+  if (!outLink) {
+    return Promise.reject('分享链接无效');
+  }
+
+  const uid = String(outLink.userId);
+
+  const [user] = await Promise.all([
+    authBalanceByUid(uid), // authBalance
+    ...(global.feConfigs?.isPlus ? [authOutLinkLimit({ outLink, ip, authToken, question })] : []) // limit auth
+  ]);
+
+  return {
+    user,
+    userId: String(outLink.userId),
+    appId: String(outLink.appId),
+    authType: AuthUserTypeEnum.token,
+    responseDetail: outLink.responseDetail
+  };
+}
+
+export function authOutLinkLimit(data: AuthLinkLimitProps) {
+  return POST('/support/outLink/authLimit', data);
+}
+
+export async function authOutLinkId({ id }: { id: string }) {
+  const outLink = await MongoOutLink.findOne({
+    shareId: id
+  });
+
+  if (!outLink) {
+    return Promise.reject('分享链接无效');
+  }
+
+  return {
+    userId: String(outLink.userId)
+  };
+}
+
+export type AuthShareChatInitProps = {
+  authToken?: string;
+  tokenUrl?: string;
+};
+
+export function authShareChatInit(data: AuthShareChatInitProps) {
+  return POST('/support/outLink/authShareChatInit', data);
+}

packages/support/outLink/constant.ts

@@ -0,0 +1,5 @@
+export enum OutLinkTypeEnum {
+  'share' = 'share',
+  'iframe' = 'iframe',
+  apikey = 'apikey'
+}

packages/support/outLink/schema.ts

@@ -0,0 +1,60 @@
+import { connectionMongo, type Model } from '@fastgpt/common/mongo';
+const { Schema, model, models } = connectionMongo;
+import { OutLinkSchema as SchemaType } from './type.d';
+import { OutLinkTypeEnum } from './constant';
+
+const OutLinkSchema = new Schema({
+  shareId: {
+    type: String,
+    required: true
+  },
+  userId: {
+    type: Schema.Types.ObjectId,
+    ref: 'user',
+    required: true
+  },
+  appId: {
+    type: Schema.Types.ObjectId,
+    ref: 'model',
+    required: true
+  },
+  type: {
+    type: String,
+    default: OutLinkTypeEnum.share
+  },
+  name: {
+    type: String,
+    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
+    },
+    hookUrl: {
+      type: String
+    }
+  }
+});
+
+export const MongoOutLink: Model<SchemaType> =
+  models['outlinks'] || model('outlinks', OutLinkSchema);

packages/support/outLink/tools.ts

@@ -0,0 +1,50 @@
+import axios from 'axios';
+import { MongoOutLink } from './schema';
+
+export const updateOutLinkUsage = async ({
+  shareId,
+  total
+}: {
+  shareId: string;
+  total: number;
+}) => {
+  try {
+    await MongoOutLink.findOneAndUpdate(
+      { shareId },
+      {
+        $inc: { total },
+        lastTime: new Date()
+      }
+    );
+  } catch (err) {
+    console.log('update shareChat error', err);
+  }
+};
+
+export const pushResult2Remote = async ({
+  authToken,
+  shareId,
+  responseData
+}: {
+  authToken?: string;
+  shareId?: string;
+  responseData?: any[];
+}) => {
+  if (!shareId || !authToken) return;
+  try {
+    const outLink = await MongoOutLink.findOne({
+      shareId
+    });
+    if (!outLink?.limit?.hookUrl) return;
+
+    axios({
+      method: 'post',
+      baseURL: outLink.limit.hookUrl,
+      url: '/shareAuth/finish',
+      data: {
+        token: authToken,
+        responseData
+      }
+    });
+  } catch (error) {}
+};

packages/support/outLink/type.d.ts

@@ -0,0 +1,26 @@
+import { OutLinkTypeEnum } from './constant';
+
+export type OutLinkSchema = {
+  _id: string;
+  shareId: string;
+  userId: string;
+  appId: string;
+  name: string;
+  total: number;
+  lastTime: Date;
+  type: `${OutLinkTypeEnum}`;
+  responseDetail: boolean;
+  limit?: {
+    expiredTime?: Date;
+    QPM: number;
+    credit: number;
+    hookUrl?: string;
+  };
+};
+
+export type OutLinkEditType = {
+  _id?: string;
+  name: string;
+  responseDetail: OutLinkSchema['responseDetail'];
+  limit: OutLinkSchema['limit'];
+};