import { useCallback } from 'react'

import { ExtensionTypeEnum, ModelExtension, ModelFile } from '@janhq/core'

import { useSetAtom } from 'jotai'

import { toaster } from '@/containers/Toast'

import { extensionManager } from '@/extension/ExtensionManager'
import { removeDownloadedModelAtom } from '@/helpers/atoms/Model.atom'

export default function useDeleteModel() {
  const removeDownloadedModel = useSetAtom(removeDownloadedModelAtom)

  const deleteModel = useCallback(
    async (model: ModelFile) => {
      await localDeleteModel(model)
      removeDownloadedModel(model.id)
      toaster({
        title: 'Model Deletion Successful',
        description: `Model ${model.name} has been successfully deleted.`,
        type: 'success',
      })
    },
    [removeDownloadedModel]
  )

  return { deleteModel }
}

const localDeleteModel = async (model: ModelFile) =>
  extensionManager
    .get<ModelExtension>(ExtensionTypeEnum.Model)
    ?.deleteModel(model)