sourcebot/packages/web/src/features/chat/useCreateNewChatThread.ts

'use client';

import { useCallback, useState } from "react";
import { Descendant } from "slate";
import { createUIMessage, getAllMentionElements } from "./utils";
import { slateContentToString } from "./utils";
import { useDomain } from "@/hooks/useDomain";
import { useToast } from "@/components/hooks/use-toast";
import { useRouter } from "next/navigation";
import { createChat } from "./actions";
import { isServiceError } from "@/lib/utils";
import { createPathWithQueryParams } from "@/lib/utils";
import { SET_CHAT_STATE_QUERY_PARAM, SetChatStatePayload } from "./types";
import { ContextItem } from "./components/chatBox/contextSelector";

export const useCreateNewChatThread = () => {
    const domain = useDomain();
    const [isLoading, setIsLoading] = useState(false);
    const { toast } = useToast();
    const router = useRouter();

    const createNewChatThread = useCallback(async (children: Descendant[], selectedItems: ContextItem[]) => {
        const text = slateContentToString(children);
        const mentions = getAllMentionElements(children);
        
        // Extract repos and contexts from selectedItems
        const selectedRepos = selectedItems.filter(item => item.type === 'repo').map(item => item.value);
        const selectedContexts = selectedItems.filter(item => item.type === 'context').map(item => item.value);
        
        const inputMessage = createUIMessage(text, mentions.map((mention) => mention.data), selectedRepos, selectedContexts);

        setIsLoading(true);
        const response = await createChat(domain);
        if (isServiceError(response)) {
            toast({
                description: `❌ Failed to create chat. Reason: ${response.message}`
            });
            setIsLoading(false);
            return;
        }

        const url = createPathWithQueryParams(`/${domain}/chat/${response.id}`,
            [SET_CHAT_STATE_QUERY_PARAM, JSON.stringify({
                inputMessage,
                selectedRepos,
                selectedContexts,
            } satisfies SetChatStatePayload)],
        );

        router.push(url);
        router.refresh();
    }, [domain, router, toast]);

    return {
        createNewChatThread,
        isLoading,
    };
}