#include "openai.h"
#include "markdown.h"
#include "plugin.h"
#include "line.h"
#include <locale.h>
#include <stdio.h>

char * get_prompt_from_args(int c, char **argv){
    char * prompt = malloc(1024*1024 + 1);
    prompt[0] = 0;
    for(int i = 1; i < c; i++){
        if(argv[i][0] == '-')
            break;
        strncat(prompt, argv[i], 1024*1024);
        if(i < c - 1){
            strncat(prompt, " ", 1024*1024);
        }else{
            strncat(prompt, ".", 1024*1024);
        }
    }
    if(!*prompt){
        free(prompt);
        return NULL;
    }
    return prompt;
}

bool try_prompt(int argc,char*argv[]){
     char * prompt = get_prompt_from_args(argc, argv);
    if(prompt != NULL){
        char * response = openai_chat("user",prompt);
        parse_markdown_to_ansi(response);
        printf("\n");
        free(response);
        free(prompt);
        return true;
    }
    return false;
}

void help();
void render(char *);
void serve(){
    render("Starting server. *Put executables in a dir named cgi-bin and they will behave as webpages.*");
    int res = system("python3 -m http.server --cgi");
    // Thanks tsoding!
    (void)res;
}


void render(char * content){
    parse_markdown_to_ansi(content);
    printf("\n\n");
}

void repl(){
    line_init();
    setbuf(stdout, NULL);
    char *line;
    char *previous_line = NULL;
    while((line = line_read("> "))){
        if(!line || !*line){
            line = previous_line;
        }
        if(!line || !*line)
            continue;
        previous_line = line;
        if(line[0] == '!'){
            plugin_run(line + 1);
            continue;
        }
        if(!strncmp(line,"exit", 4)){
            exit(0);
        }
        if(!strncmp(line,"help",4)){
            help();
            continue;
        }
        if(!strncmp(line,"serve",5)){
            serve();
        }
        if(!strncmp(line,"spar ",5)){
            char * response = line+5;
            while(true){
                render(response);
                sleep(2);
                //line = line_read("> ");
                //if(!*line)
                    response = openai_chat("user",response);
                
            }   
        }
        if(!strncmp(line,"ls",2) || !strncmp(line,"list",4)){
            int offset = 2;
            if(!strncmp(line,"list",4)){
                offset = 4;
            }
            char * command = (char *)malloc(strlen(line) + 42);
            command[0] = 0;
            strcpy(command, "ls ");
            strcat(command, line + offset);
            int res = system(command);
            (void)res;
            free(command);
            continue;
        }

        line_add_history(line);
        char * response = openai_chat("user", line);
        render(response);
        free(response);
    }
}


void help(){
    char help_text[1024*1024] = {0};
    char * template = "# Help\n"
        "Written by retoor@molodetz.nl.\n\n"
        "## Features\n"
        " - navigate trough history using `arrows`.\n"
        " - navigate trough history with **recursive search** using `ctrl+r`.\n"
        " - **inception with python** for *incomming* and *outgoing* content.\n"
        " - markdown and **syntax highlighting**.\n"
        " - **execute python commands** with prefixing `!`\n"
        " - list files of current workdirectory using `ls`.\n"
        " - type `serve` to start a webserver with directory listing. Easy for network transfers.\n\n"
        "## Configuration\n"
        " - model temperature is %f.\n"
        " - model name is %s.\n"
        " - max tokens is %d.\n\n"
        "## In development\n"
        " - **google search** and actions with those results.\n"
        " - **reminders**.\n"
        " - predefined **templates** for **reviewing** / **refactoring** so you can personalize.\n";
        sprintf(help_text,template,prompt_temperature,prompt_model,prompt_max_tokens);
        render(help_text);
}

void openai_include(char * path){
    FILE * file = fopen(path,"r");
    if(file == NULL){
        return;
    }
    fseek(file, 0, SEEK_END);
    long size = ftell(file);
    fseek(file, 0, SEEK_SET);

    char * buffer = (char *)malloc(size);
    size_t read = fread(buffer,1,size,file);
    if(read == 0){
        return;
    }

    fclose(file);
    buffer[read] = 0;
    openai_system(buffer);
    
    free(buffer);
}

void init(){
     line_init();
     const char *locale = setlocale(LC_ALL, NULL);
     char payload[4096] = {0};
     sprintf(payload, "User locale is %s. User lang is %s.\n"
     "You are Retoor. Use a lot of markdown in response.\n"
     "Be confident and short in answers.\n"
     "You divide things by zero if you have to."
     , locale, locale);
     
     printf("%s","Loading...");
     openai_system(payload);
     openai_include("context.txt");
     printf("%s", "\rLoaded! Type help for feautures.\n");
}

int main(int argc, char *argv[]){
    init();
    if(try_prompt(argc,argv))
        return 0;
   
    repl();
    return 0;
}