import { defHttp } from '/@/utils/http/axios';
import { useMessage } from '/@/hooks/web/useMessage';
import { useUserStore } from '/@/store/modules/user'; // 引入用户store获取token

const { createMessage } = useMessage();

enum Api {
  send = '/airag/zdyRag/send',
  buttonList = '/airagbutton/airagButton/buttonList',
}

/**
 * 发送消息（流式处理）
 * @param params
 */
export const sendMessage = async (params: { questionText: string }) => {
  try {
    const userStore = useUserStore();
    const token = userStore.getToken;

    const response = await fetch(
      `/jeecgboot/airag/zdyRag/sendStream?questionText=${encodeURIComponent(params.questionText)}`,
      {
        method: 'GET',
        headers: {
          'x-access-token': token,
          'Accept': 'text/event-stream'
        },
        credentials: 'include',
      }
    );

    if (!response.ok || !response.body) {
      throw new Error(`请求失败: ${response.status} ${response.statusText}`);
    }

    // 返回异步生成器函数
    return (async function* () {
      const reader = response.body.getReader();
      const decoder = new TextDecoder('utf-8');
      let buffer = '';

      try {
        while (true) {
          const { done, value } = await reader.read();
          if (done) break;

          const chunk = decoder.decode(value, { stream: true });
          buffer += chunk;

          const lines = buffer.split('\n');
          buffer = lines.pop() || '';

          for (const line of lines) {
            if (line.startsWith('data:')) {
              try {
                const jsonStr = line.substring(5).trim();
                if (jsonStr) {
                  yield JSON.parse(jsonStr); // 直接 yield 解析后的数据
                }
              } catch (e) {
                console.error('JSON解析错误:', e);
              }
            }
          }
        }
      } finally {
        reader.releaseLock();
      }
    })();
  } catch (error) {
    console.error('Error sending message:', error);
    createMessage.error('发送消息失败');
    throw error; // 抛出错误让调用方处理
  }
};
export const getButtonList = async () => {
  const res = await defHttp.get({
    url: Api.buttonList,
  });
  if (res) {
    return res;
  }
  return null;
}