v4.6-3 (#471)

projects/app/src/service/support/permission/auth/dataset.ts

@@ -1,36 +1,43 @@
-import { getDatasetPgData } from '@/service/core/dataset/data/controller';
-import { PgDataItemType } from '@fastgpt/global/core/dataset/type';
+import { DatasetDataItemType, DatasetDataSchemaType } from '@fastgpt/global/core/dataset/type';
 import { AuthResponseType } from '@fastgpt/global/support/permission/type';
+import { MongoDatasetData } from '@fastgpt/service/core/dataset/data/schema';
 import { authDatasetCollection } from '@fastgpt/service/support/permission/auth/dataset';
-import { parseHeaderCert } from '@fastgpt/service/support/permission/controller';
 import { AuthModeType } from '@fastgpt/service/support/permission/type';
 
+/* data permission same of collection */
 export async function authDatasetData({
   dataId,
   ...props
 }: AuthModeType & {
   dataId: string;
-}): Promise<
-  AuthResponseType & {
-    datasetData: PgDataItemType;
-  }
-> {
-  const result = await parseHeaderCert(props);
-  const { tmbId } = result;
+}) {
   // get pg data
-  const datasetData = await getDatasetPgData({ id: dataId });
+  const datasetData = await MongoDatasetData.findById(dataId);
 
-  const isOwner = String(datasetData.tmbId) === tmbId;
-  // data has the same permissions as collection
-  const { canWrite } = await authDatasetCollection({
+  if (!datasetData) {
+    return Promise.reject('Data not found');
+  }
+
+  const result = await authDatasetCollection({
     ...props,
     collectionId: datasetData.collectionId
   });
 
+  const data: DatasetDataItemType = {
+    id: String(datasetData._id),
+    q: datasetData.q,
+    a: datasetData.a,
+    indexes: datasetData.indexes,
+    datasetId: String(datasetData.datasetId),
+    collectionId: String(datasetData.collectionId),
+    sourceName: result.collection.name || '',
+    sourceId: result.collection.metadata?.fileId || result.collection.metadata?.rawLink,
+    isOwner: String(datasetData.tmbId) === result.tmbId,
+    canWrite: result.canWrite
+  };
+
   return {
     ...result,
-    datasetData,
-    isOwner,
-    canWrite
+    datasetData: data
   };
 }