import Perplexity from "./libs/Perplexity";
import "dotenv/config";
import required from "./libs/required";
import readin from "./libs/readin";
import fs from "fs";
import config from "./config";
import { ContextPart, parseCommands } from "./libs/parseCommands";

required(process.env.PPLX_API_KEY);
const PPLX_API_KEY = process.env.PPLX_API_KEY as string;
let perplexity: Perplexity;

// NOTE Helper to load context from file
function inject_context(verbose: boolean = true) {
  let loaded_context = fs.readFileSync("./context.json", { encoding: "utf-8" });
  let list_context = JSON.parse(loaded_context) as ContextPart[];
  for (let i = 0; i < list_context.length; i++) {
    let msg = list_context[i] as ContextPart;
    perplexity.add_to_context(msg);
    if (verbose) console.log("[" + msg.role + "]> " + msg.content);
  }
}

// NOTE Helper to save context long term
function save_in_context(role: string, message: string) {
  let loaded_context = fs.readFileSync("./context.json", { encoding: "utf-8" });
  let list_context = JSON.parse(loaded_context) as Array<{
    role: string;
    content: string;
  }>;
  let new_insertion = {
    role: role,
    content: message,
  };
  list_context.push(new_insertion);
  fs.writeFileSync("./context.json", JSON.stringify(list_context));
}

async function main() {
  console.log("[TCSenPPLX] Working with: " + PPLX_API_KEY);
  // Connecting to perplexity
  perplexity = new Perplexity(
    PPLX_API_KEY,
    "https://api.perplexity.ai",
    "sonar-small-chat",
    false,
    {}
  );
  // Setting our ai personality
  perplexity.add_to_context(config);
  // If any, inject context
  inject_context();
  // Chatting
  await chat();
}

async function chat(loop: boolean = true, save: boolean = true) {
  var proceed = true;
  while (proceed) {
    proceed = loop;
    let question = await readin("[You]> ");
    let parsed = parseCommands(question);
    proceed = parsed.proceed;
    let logpart = parsed.logpart;
    let response = await perplexity.ask(question);
    console.log("[Assistant]> " + response);
    if (save) {
      save_in_context("user", question);
      if (logpart) {
        response += " " + logpart;
      }
      save_in_context("assistant", response as string);
    }
  }
}

main();