import { useEffect } from 'react'
import { PluginType } from '@janhq/core'
import { useAtom } from 'jotai'
import { downloadedModelAtom } from '@helpers/atoms/DownloadedModel.atom'
import { pluginManager } from '@plugin/PluginManager'
import { ModelPlugin } from '@janhq/core/lib/plugins'
import { Model } from '@janhq/core/lib/types'

export function useGetDownloadedModels() {
  const [downloadedModels, setDownloadedModels] = useAtom(downloadedModelAtom)

  useEffect(() => {
    getDownloadedModels().then((downloadedModels) => {
      setDownloadedModels(downloadedModels)
    })
  }, [setDownloadedModels])

  return { downloadedModels }
}

export async function getDownloadedModels(): Promise<Model[]> {
  const models =
    ((await pluginManager
      .get<ModelPlugin>(PluginType.Model)
      ?.getDownloadedModels()) as Model[]) ?? []
  return models
}