dify/web/service/knowledge/use-metadata.ts

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) => {
      // 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('/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'}`)
    },
  })
}