utils.ts

import {
  ChatMessage,
  Form,
  LLMRequest,
  LLMResponse,
  Response,
  User,
} from '@/types';
import { Session, SupabaseClient } from '@supabase/auth-helpers-nextjs';
import { v4 } from 'uuid';
import { Database, Json } from '../types/supabase';

export const callLLM = async (
export const callLLM = async (
  systemPrompt: string,
  messages: ChatMessage[],
  endpoint: string
) => {
  const data: LLMRequest = {
    completion_create: {
      model: 'gpt-3.5-turbo',
      temperature: 0,
      messages: [{ role: 'system', content: systemPrompt }, ...messages],
    },
  };
  const response = await fetch(`/api/${endpoint}`, {
    method: 'POST',
    body: JSON.stringify(data),
    headers: {
      'Content-Type': 'application/json',
      Authorization: `Bearer ${process.env.NEXT_PUBLIC_LLM_API_KEY}`,
    },
  });
  if (!response.ok) {
    return Error(`LLM API returned ${response.status}: ${response.statusText}`);
  }
  const json: LLMResponse = await response.json();
  return json.completion.choices[0].message;
};
export async function getUserFromSupabase(
  session: Session | null,
  supabase: SupabaseClient<Database>,
  setUser: React.Dispatch<React.SetStateAction<User | null>>
): Promise<User | undefined> {
  if (!session) {
    return;
  }
  const { data, error } = await supabase
    .from('users')
    .select()
    .eq('id', session?.user.id)
    .maybeSingle();
  if (error) {
    console.error(error);
    return;
  } else if (data === null) {
    console.error('No user found');
    return;
  } else {
    setUser(data);
  }
}

export async function getFormsFromSupabase(
  userId: string,
  supabase: SupabaseClient<Database>
): Promise<Form[] | undefined> {
  const { data, error } = await supabase
    .from('forms')
    .select()
    .eq('user_id', userId);
  if (error) {
    console.error(error);
    return;
  }
  return data as Form[];
}

export async function getFormFromSupabase(
  formId: string,
  supabase: SupabaseClient<Database>
): Promise<Form | Error> {
  const { data, error } = await supabase
    .from('forms')
    .select()
    .eq('id', formId)
    .maybeSingle();
  if (error) {
    return Error(error.message, { cause: error });
  } else if (data === null) {
    return Error(`No form found with id '${formId}'`);
  }
  return data;
}

export async function getResponsesFromSupabase(
  formId: string,
  supabase: SupabaseClient<Database>
): Promise<Response[] | undefined> {
  const { data, error } = await supabase
    .from('responses')
    .select()
    .eq('form_id', formId);
  if (error) {
    console.error(error);
    return;
  }
  return data as Response[];
}

export async function submitResponseToSupabase(
  formId: string,
  responseJson: Json,
  supabase: SupabaseClient<Database>
): Promise<Response | Error> {
  const response: Response = {
    id: v4(),
    form_id: formId,
    fields: responseJson,
    created_at: new Date().toISOString(),
    updated_at: new Date().toISOString(),
  };
  console.log('Submitting response to Supabase', { formId, response });
  const { error } = await supabase.from('responses').insert(response);

  if (error) {
    console.error(`Error creating response`, { response, error });
    return Error(error.message, { cause: error });
  } else {
    console.log('Successfully created response', response);
    return response;
  }
}
export async function submitResponseToSupabase(
  formId: string,
  responseJson: Json,
  supabase: SupabaseClient<Database>,
  endpoint: string
): Promise<Response | Error> {
  const response: Response = {
    id: v4(),
    form_id: formId,
    fields: responseJson,
    created_at: new Date().toISOString(),
    updated_at: new Date().toISOString(),
  };
  console.log('Submitting response to Supabase', { formId, response });
  const { error } = await supabase.from('responses').insert(response);

  if (error) {
    console.error(`Error creating response`, { response, error });
    return Error(error.message, { cause: error });
  } else {
    console.log('Successfully created response', response);
    return response;
  }
}
export const removeStartAndEndQuotes = (str: string | null) => {
  if (!str) {
    return str;
  }
  return str.replace(/^"(.*)"$/, '$1');
};