import AiService from "@/services/AiService";

export default class KaiService extends AiService {
  getKaiChats = async (userId: string, projectId: string): Promise<{ title: string, threadId: string }[]> => {
    // const r = await this.client.get('/kai/PROJECT_ID/chats');
    const jwt = window.env.KAI_TESTING // this.client.getJwt()
    const r = await fetch(`http://localhost:8700/kai/${projectId}/chats?user_id=${userId}`, {
      headers: new Headers({
        Accept: 'application/json',
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${jwt}`
      }),
    });
    if (!r.ok) {
      throw new Error('Failed to fetch chats');
    }
    const data = await r.json();
    return data;
  }

  deleteKaiChat = async (projectId: string, userId: string, threadId: string): Promise<boolean> => {
    const jwt = window.env.KAI_TESTING // this.client.getJwt()
    const r = await fetch(`http://localhost:8700/kai/${projectId}/chats/${threadId}?user_id=${userId}`, {
      method: 'DELETE',
      headers: new Headers({
        Accept: 'application/json',
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${jwt}`
      }),
    });
    if (!r.ok) {
      throw new Error('Failed to delete chat');
    }
    return true;
  }

  getKaiChat = async (projectId: string, userId: string, threadId: string): Promise<{ role: string, content: string, message_id: any }[]> => {
    const jwt = window.env.KAI_TESTING // this.client.getJwt()
    const r = await fetch(`http://localhost:8700/kai/${projectId}/chats/${threadId}?user_id=${userId}`, {
      method: 'GET',
      headers: new Headers({
        Accept: 'application/json',
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${jwt}`
      }),
    });
    if (!r.ok) {
      throw new Error('Failed to fetch chat');
    }
    const data = await r.json();
    return data;
  }

  createKaiChat = async (projectId: string, userId: string): Promise<number> => {
    const jwt = window.env.KAI_TESTING // this.client.getJwt()
    const r = await fetch(`http://localhost:8700/kai/${projectId}/chat/new?user_id=${userId}`, {
      method: 'GET',
      headers: new Headers({
        Accept: 'application/json',
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${jwt}`
      }),
    })
    if (!r.ok) {
      throw new Error('Failed to create chat');
    }
    const data = await r.json();
    return data;
  }

  feedback = async (positive: boolean | null, messageId: string, projectId: string, userId: string) => {
    const jwt = window.env.KAI_TESTING // this.client.getJwt()
    const r = await fetch(`http://localhost:8700/kai/${projectId}/messages/feedback`, {
      method: 'POST',
      headers: new Headers({
        Accept: 'application/json',
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${jwt}`
      }),
      body: JSON.stringify({
        message_id: messageId,
        value: positive,
        user_id: userId,
      }),
    });
    if (!r.ok) {
      throw new Error('Failed to send feedback');
    }

    return await r.json()
  }

  cancelGeneration = async (projectId: string, threadId: string, userId: string) => {
    const jwt = window.env.KAI_TESTING // this.client.getJwt()
    const r = await fetch(`http://localhost:8700/kai/${projectId}/cancel/${threadId}?user_id=${userId}`, {
      method: 'POST',
      headers: new Headers({
        Accept: 'application/json',
        'Content-Type': 'application/json',
        'Authorization': `Bearer ${jwt}`
      }),
    });
    if (!r.ok) {
      throw new Error('Failed to cancel generation');
    }

    const data = await r.json();
    return data;
  }
}