90 lines
2.8 KiB
TypeScript
90 lines
2.8 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'
|
|
|
|
export const useDatasetMetaData = (datasetId: string) => {
|
|
return useQuery<{ doc_metadata: MetadataItemWithValueLength[], built_in_field_enabled: boolean }>({
|
|
queryKey: [NAME_SPACE, 'dataset', datasetId],
|
|
queryFn: () => {
|
|
return get<{ doc_metadata: MetadataItemWithValueLength[], built_in_field_enabled: boolean }>(`/datasets/${datasetId}/metadata`)
|
|
},
|
|
})
|
|
}
|
|
|
|
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) => {
|
|
await post(`/datasets/${datasetId}/metadata`, {
|
|
body: payload,
|
|
})
|
|
await invalidDatasetMetaData()
|
|
return Promise.resolve(true)
|
|
},
|
|
})
|
|
}
|
|
|
|
export const useRenameMeta = (datasetId: string) => {
|
|
const invalidDatasetMetaData = useInvalidDatasetMetaData(datasetId)
|
|
return useMutation({
|
|
mutationFn: async (payload: MetadataItemWithValueLength) => {
|
|
await patch(`/datasets/${datasetId}/metadata/${payload.id}`, {
|
|
body: {
|
|
name: payload.name,
|
|
},
|
|
})
|
|
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('/datasets/metadata/built-in')
|
|
},
|
|
})
|
|
}
|
|
|
|
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 ? 'enable' : 'disable'}`)
|
|
},
|
|
})
|
|
}
|