132 lines
3.6 KiB
TypeScript
132 lines
3.6 KiB
TypeScript
import type { BuiltInMetadataItem, MetadataBatchEditToServer, MetadataItemWithValueLength } from '@/app/components/datasets/metadata/types'
|
|
import { del, get, patch, post } from '../base'
|
|
|
|
import { useMutation, useQuery } from '@tanstack/react-query'
|
|
import { useInvalid } from '../use-base'
|
|
|
|
const NAME_SPACE = 'dataset-metadata'
|
|
|
|
// let datasetMetaData = [
|
|
// {
|
|
// id: '1-ae54c',
|
|
// name: 'Doc type',
|
|
// type: 'string',
|
|
// use_count: 1,
|
|
// },
|
|
// {
|
|
// id: '2-fufex',
|
|
// name: 'Title',
|
|
// type: 'string',
|
|
// use_count: 5,
|
|
// },
|
|
// ]
|
|
|
|
export const useDatasetMetaData = (datasetId: string) => {
|
|
return useQuery<{ data: MetadataItemWithValueLength[] }>({
|
|
queryKey: [NAME_SPACE, 'dataset', datasetId],
|
|
queryFn: () => {
|
|
return get(`/datasets/${datasetId}/metadata`)
|
|
// return {
|
|
// data: datasetMetaData,
|
|
// } as {
|
|
// data: MetadataItemWithValueLength[],
|
|
// }
|
|
},
|
|
})
|
|
}
|
|
|
|
export const useInvalidDatasetMetaData = (datasetId: string) => {
|
|
return useInvalid([NAME_SPACE, 'dataset', datasetId])
|
|
}
|
|
|
|
export const useCreateMetaData = (datasetId: string) => {
|
|
const invalidDatasetMetaData = useInvalidDatasetMetaData(datasetId)
|
|
return useMutation({
|
|
mutationFn: async (payload: BuiltInMetadataItem) => {
|
|
// datasetMetaData.push({
|
|
// id: `${Math.random()}`,
|
|
// ...payload,
|
|
// use_count: 0,
|
|
// })
|
|
await post(`/datasets/${datasetId}/metadata`, payload)
|
|
await invalidDatasetMetaData()
|
|
return Promise.resolve(true)
|
|
},
|
|
})
|
|
}
|
|
|
|
export const useRenameMeta = (datasetId: string) => {
|
|
const invalidDatasetMetaData = useInvalidDatasetMetaData(datasetId)
|
|
return useMutation({
|
|
mutationFn: async (payload: MetadataItemWithValueLength) => {
|
|
// datasetMetaData = datasetMetaData.map((item) => {
|
|
// if (item.id === payload.id)
|
|
// return payload
|
|
|
|
// return item
|
|
// })
|
|
await patch(`/datasets/${datasetId}/metadata/${payload.id}`, {
|
|
name: payload.name,
|
|
type: payload.type,
|
|
})
|
|
await invalidDatasetMetaData()
|
|
return Promise.resolve(true)
|
|
},
|
|
})
|
|
}
|
|
|
|
export const useDeleteMetaData = (datasetId: string) => {
|
|
const invalidDatasetMetaData = useInvalidDatasetMetaData(datasetId)
|
|
return useMutation({
|
|
mutationFn: async (metaDataId: string) => {
|
|
// datasetMetaData = datasetMetaData.filter(item => item.id !== metaDataId)
|
|
await del(`/datasets/${datasetId}/metadata/${metaDataId}`)
|
|
await invalidDatasetMetaData()
|
|
return Promise.resolve(true)
|
|
},
|
|
})
|
|
}
|
|
|
|
export const useBuiltInMetaDataFields = () => {
|
|
return useQuery<{ fields: BuiltInMetadataItem[] }>({
|
|
queryKey: [NAME_SPACE, 'built-in'],
|
|
queryFn: () => {
|
|
return get('/metadata/built-in')
|
|
// return {
|
|
// fields: [
|
|
// {
|
|
// name: 'OriginalfileNmae',
|
|
// type: 'string',
|
|
// },
|
|
// {
|
|
// name: 'Title',
|
|
// type: 'string',
|
|
// },
|
|
// ],
|
|
// } as {
|
|
// fields: BuiltInMetadataItem[],
|
|
// }
|
|
},
|
|
})
|
|
}
|
|
|
|
export const useBatchUpdateDocMetadata = () => {
|
|
return useMutation({
|
|
mutationFn: (payload: {
|
|
dataset_id: string
|
|
metadata_list: MetadataBatchEditToServer
|
|
}) => {
|
|
// /console/api/datasets/{dataset_id}/documents/metadata
|
|
return post(`/datasets/${payload.dataset_id}/documents/metadata`, payload.metadata_list)
|
|
},
|
|
})
|
|
}
|
|
|
|
export const useUpdateBuiltInStatus = (datasetId: string) => {
|
|
return useMutation({
|
|
mutationFn: (enabled: boolean) => {
|
|
return post(`/datasets/${datasetId}/metadata/built-in/${enabled ? 'disable' : 'enable'}`)
|
|
},
|
|
})
|
|
}
|