import fs from 'fs';
import { v4 as uuidv4 } from 'uuid';

import { ToolNote, ToolCall } from './api-defs.js';
import * as chat from './chat.js';
import * as u from './utils.js';

export type Note = ToolNote;

export function run( toolCall: ToolCall, notes: Note[], uuid: string ) : chat.Msg | null {
  if( !toolCall.parameters )
    return null;

  let note = toolCall.parameters.contents;
  if( note.length > 1 ) {
    save( note, notes, uuid );
    console.log( "saved note: " + note );
  }

  return { timestamp: u.getTimestamp(), role: "tool", content: "the note has been saved" };
}

export function getPromptStr( notes: Note[] ) : string {
  if( !notes.length )
    return '';

  let promptStr = "here is a list of your (the assistant's) saved notes:\n";
  for( let i = 0; i < notes.length; ++i ) {
    promptStr += `${i+1}. ${notes[i].content}\n`
  }

  return promptStr;
}

export function load( uuid: string ) : Note[] {
  try {
    let file = `../data/notes/${uuid}-notes.json`;
    const data = fs.readFileSync( file, 'utf8' );
    return JSON.parse( data );
  } catch( err ) {
    return [];
  }
}

export function save( newNote: string, notes: Note[], uuid: string ) {
  let txt = newNote.slice( 0, 75 );
  let foundNote = false;
  for( let [key, savedNote] of Object.entries( notes ) ) {
    let split = savedNote.content.split( ' : ' );
    if( split.length < 2 )
      continue;

    let savedWords = split[1].split( " " );
    let words = txt.split( " " );

    let totalWords = savedWords.length;
    let matchingWords = 0;
    for( let i = 0; i < savedWords.length; ++i ) {
      if( words.find( ( e ) => e === savedWords[i] ) )
        ++matchingWords;
    }

    if( matchingWords >= Math.floor( totalWords * 0.95 ) ) {
      notes[key].content = `${u.getTimestamp()} : ${txt}`;
      foundNote = true;
      break;
    }
  }
  if( !foundNote )
    notes.push( { id: uuidv4(), content: `${u.getTimestamp()} : ${newNote.slice( 0, 75 )}` } );
  if( notes.length > 25 )
    notes = notes.slice( 1 );

  let file = `../data/notes/${uuid}-notes.json`;
  fs.writeFileSync( file, JSON.stringify( notes ) );
}