62 lines
1.9 KiB
TypeScript
62 lines
1.9 KiB
TypeScript
import { currentPromptAtom } from '@/_helpers/JotaiWrapper'
|
|
import { execute } from '@/_services/pluginService'
|
|
import { useAtom, useAtomValue, useSetAtom } from 'jotai'
|
|
import { DataService } from '@janhq/core'
|
|
import { deleteConversationMessage } from '@/_helpers/atoms/ChatMessage.atom'
|
|
import {
|
|
userConversationsAtom,
|
|
getActiveConvoIdAtom,
|
|
setActiveConvoIdAtom,
|
|
} from '@/_helpers/atoms/Conversation.atom'
|
|
import {
|
|
showingProductDetailAtom,
|
|
showingAdvancedPromptAtom,
|
|
} from '@/_helpers/atoms/Modal.atom'
|
|
import {
|
|
MainViewState,
|
|
setMainViewStateAtom,
|
|
} from '@/_helpers/atoms/MainView.atom'
|
|
|
|
export default function useDeleteConversation() {
|
|
const [userConversations, setUserConversations] = useAtom(
|
|
userConversationsAtom
|
|
)
|
|
const setCurrentPrompt = useSetAtom(currentPromptAtom)
|
|
const setShowingProductDetail = useSetAtom(showingProductDetailAtom)
|
|
const setShowingAdvancedPrompt = useSetAtom(showingAdvancedPromptAtom)
|
|
const activeConvoId = useAtomValue(getActiveConvoIdAtom)
|
|
|
|
const setActiveConvoId = useSetAtom(setActiveConvoIdAtom)
|
|
const deleteMessages = useSetAtom(deleteConversationMessage)
|
|
const setMainViewState = useSetAtom(setMainViewStateAtom)
|
|
|
|
const deleteConvo = async () => {
|
|
if (activeConvoId) {
|
|
try {
|
|
await execute(DataService.DeleteConversation, activeConvoId)
|
|
const currentConversations = userConversations.filter(
|
|
(c) => c._id !== activeConvoId
|
|
)
|
|
setUserConversations(currentConversations)
|
|
deleteMessages(activeConvoId)
|
|
|
|
if (currentConversations.length > 0) {
|
|
setActiveConvoId(currentConversations[0]._id)
|
|
} else {
|
|
setMainViewState(MainViewState.Welcome)
|
|
setActiveConvoId(undefined)
|
|
}
|
|
setCurrentPrompt('')
|
|
setShowingProductDetail(false)
|
|
setShowingAdvancedPrompt(false)
|
|
} catch (err) {
|
|
console.error(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return {
|
|
deleteConvo,
|
|
}
|
|
}
|