import { fetchApi } from "@/lib/server-api-util";
import { Message } from "@/types/ask";
import { useCallback } from "react";

// 实现一个函数，从两个数组中轮流插入新数组
export const mergeArrays = (arr1: any[], arr2: any[]) => {
    const result: any[] = [];
    const maxLength = Math.max(arr1.length, arr2.length);
    for (let i = 0; i < maxLength; i++) {
        if (i < arr1.length) result.push(arr1[i]);
        if (i < arr2.length) result.push(arr2[i]);
    }
    return result;
};


// 创建新对话并获取消息
export const createNewConversation = useCallback(async (user_text: string) => {
    const data = await fetchApi<string>("/chat/new", {
        method: "POST",
    });
    return data
}, []);

// 获取对话信息
export const getConversation = async ({
    session_id,
    user_text,
    material_ids
}: {
    session_id: string,
    user_text: string,
    material_ids: string[]
}): Promise<Message | undefined> => {
    // 获取对话信息必须要有对话id
    if (!session_id) return undefined;

    try {
        const response = await fetchApi<Message>(`/chat`, {
            method: "POST",
            body: JSON.stringify({
                session_id,
                user_text,
                material_ids
            })
        });
        return response;
    } catch (error) {
        // console.error('Error in getConversation:', error);
        return undefined;
    }
};