import { useCallback } from 'react'

import { ExtensionTypeEnum, ConversationalExtension } from '@janhq/core'

import { useAtom, useSetAtom } from 'jotai'

import { currentPromptAtom } from '@/containers/Providers/Jotai'

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

import useSetActiveThread from './useSetActiveThread'

import { extensionManager } from '@/extension/ExtensionManager'

import { deleteChatMessageAtom as deleteChatMessagesAtom } from '@/helpers/atoms/ChatMessage.atom'
import {
  threadsAtom,
  setActiveThreadIdAtom,
  deleteThreadStateAtom,
  updateThreadAtom,
} from '@/helpers/atoms/Thread.atom'

export default function useDeleteThread() {
  const [threads, setThreads] = useAtom(threadsAtom)
  const updateThread = useSetAtom(updateThreadAtom)

  const setCurrentPrompt = useSetAtom(currentPromptAtom)
  const setActiveThreadId = useSetAtom(setActiveThreadIdAtom)
  const deleteMessages = useSetAtom(deleteChatMessagesAtom)

  const deleteThreadState = useSetAtom(deleteThreadStateAtom)
  const { setActiveThread } = useSetActiveThread()

  const cleanThread = useCallback(
    async (threadId: string) => {
      const messages = await extensionManager
        .get<ConversationalExtension>(ExtensionTypeEnum.Conversational)
        ?.listMessages(threadId)
        .catch(console.error)
      if (messages) {
        messages.forEach((message) => {
          extensionManager
            .get<ConversationalExtension>(ExtensionTypeEnum.Conversational)
            ?.deleteMessage(threadId, message.id)
            .catch(console.error)
        })
        const thread = threads.find((e) => e.id === threadId)
        if (thread) {
          const updatedThread = {
            ...thread,
            title: 'New Thread',
            metadata: {
              ...thread.metadata,
              title: 'New Thread',
              lastMessage: '',
            },
          }
          extensionManager
            .get<ConversationalExtension>(ExtensionTypeEnum.Conversational)
            ?.modifyThread(updatedThread)
            .catch(console.error)
          updateThread(updatedThread)
        }
      }
      deleteMessages(threadId)
    },
    [deleteMessages, threads, updateThread]
  )

  const deleteThread = async (threadId: string) => {
    if (!threadId) {
      alert('No active thread')
      return
    }
    await extensionManager
      .get<ConversationalExtension>(ExtensionTypeEnum.Conversational)
      ?.deleteThread(threadId)
      .catch(console.error)
    const availableThreads = threads.filter((c) => c.id !== threadId)
    setThreads(availableThreads)

    // delete the thread state
    deleteThreadState(threadId)

    deleteMessages(threadId)
    setCurrentPrompt('')
    toaster({
      title: 'Thread successfully deleted.',
      description: `Thread ${threadId} has been successfully deleted.`,
      type: 'success',
    })
    if (availableThreads.length > 0) {
      setActiveThread(availableThreads[0])
    } else {
      setActiveThreadId(undefined)
    }
  }

  return {
    cleanThread,
    deleteThread,
  }
}