Revert "sub plan page (#885)" (#886)

This reverts commit 443ad37b6a.

packages/service/support/permission/auth/dataset.ts

@@ -107,7 +107,7 @@ export async function authDatasetCollection({
     collection: CollectionWithDatasetType;
   }
 > {
-  const { teamId, tmbId } = await parseHeaderCert(props);
+  const { userId, teamId, tmbId } = await parseHeaderCert(props);
   const { role } = await getTmbInfoByTmbId({ tmbId });
 
   const { collection, isOwner, canWrite } = await (async () => {
@@ -143,6 +143,7 @@ export async function authDatasetCollection({
   })();
 
   return {
+    userId,
     teamId,
     tmbId,
     collection,
@@ -162,7 +163,7 @@ export async function authDatasetFile({
     file: DatasetFileSchema;
   }
 > {
-  const { teamId, tmbId } = await parseHeaderCert(props);
+  const { userId, teamId, tmbId } = await parseHeaderCert(props);
 
   const [file, collection] = await Promise.all([
     getFileById({ bucketName: BucketNameEnum.dataset, fileId }),
@@ -189,6 +190,7 @@ export async function authDatasetFile({
     });
 
     return {
+      userId,
       teamId,
       tmbId,
       file,
@@ -198,4 +200,4 @@ export async function authDatasetFile({
   } catch (error) {
     return Promise.reject(DatasetErrEnum.unAuthDatasetFile);
   }
-}
+}

packages/service/support/permission/auth/user.ts

@@ -12,7 +12,7 @@ export async function authUserNotVisitor(props: AuthModeType): Promise<
     role: `${TeamMemberRoleEnum}`;
   }
 > {
-  const { teamId, tmbId } = await parseHeaderCert(props);
+  const { userId, teamId, tmbId } = await parseHeaderCert(props);
   const team = await getTmbInfoByTmbId({ tmbId });
 
   if (team.role === TeamMemberRoleEnum.visitor) {
@@ -20,6 +20,7 @@ export async function authUserNotVisitor(props: AuthModeType): Promise<
   }
 
   return {
+    userId,
     teamId,
     tmbId,
     team,

packages/service/support/permission/controller.ts

@@ -94,10 +94,10 @@ export async function parseHeaderCert({
     })();
 
     // auth apikey
-    const { teamId, tmbId, appId: apiKeyAppId = '' } = await authOpenApiKey({ apikey });
+    const { userId, teamId, tmbId, appId: apiKeyAppId = '' } = await authOpenApiKey({ apikey });
 
     return {
-      uid: '',
+      uid: userId,
       teamId,
       tmbId,
       apikey,
@@ -217,4 +217,4 @@ export const authFileToken = (token?: string) =>
         fileId: decoded.fileId
       });
     });
-  });
+  });

packages/service/support/permission/limit/dataset.ts

@@ -0,0 +1,23 @@
+import { StandSubPlanLevelMapType } from '@fastgpt/global/support/wallet/sub/type';
+import { getVectorCountByTeamId } from '../../../common/vectorStore/controller';
+import { getTeamDatasetMaxSize } from '../../wallet/sub/utils';
+
+export const checkDatasetLimit = async ({
+  teamId,
+  insertLen = 0,
+  standardPlans
+}: {
+  teamId: string;
+  insertLen?: number;
+  standardPlans?: StandSubPlanLevelMapType;
+}) => {
+  const [{ maxSize }, usedSize] = await Promise.all([
+    getTeamDatasetMaxSize({ teamId, standardPlans }),
+    getVectorCountByTeamId(teamId)
+  ]);
+
+  if (usedSize + insertLen >= maxSize) {
+    return Promise.reject(`数据库容量不足，无法继续添加。可以在账号页面进行扩容。`);
+  }
+  return;
+};