// Written by retoor@molodetz.nl

// This source code initializes a command-line application that uses OpenAI for chat interactions, handles user inputs, and can start a simple HTTP server with CGI support. The code allows command execution, markdown parsing, and OpenAI chat integration.

// External imports used in this code:
// - openai.h
// - markdown.h
// - plugin.h
// - line.h

// MIT License
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.

#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <signal.h>


#include "openai.h"
#include "markdown.h"
#include "line.h"
#include <locale.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "utils.h"
#include "r.h"
#include "db_utils.h"

volatile sig_atomic_t sigint_count = 0;
time_t first_sigint_time = 0;
bool SYNTAX_HIGHLIGHT_ENABLED = true;
bool API_MODE = false;
void help();
void render(char *);
bool openai_include(char *path);
char * strreplace(char * content, char * what, char * with);

char * get_prompt_from_stdin(char * prompt) {
    int index = 0; 
    prompt[index] = '\0';
    char c = 0;
    while ((c = getchar()) != EOF) {
        prompt[index++] = c;
    }
    prompt[index++] = '\0';
    return prompt;
}

char *get_prompt_from_args(int c, char **argv) {
    char *prompt = (char *)malloc(1024 * 1024 * 10 + 1);
    char *system = (char *)malloc(1024*1024);

    system[0] = 0;
    prompt[0] = 0;
    bool get_from_std_in = false; 
    for (int i = 1; i < c; i++) {
        if (!strcmp(argv[i],"--stdin")){
            fprintf(stderr, "%s\n", "Reading from stdin.");
            get_from_std_in = true;
        }else if(!strcmp(argv[i],"--verbose")){
           is_verbose = true; 
        }

        else if(!strcmp(argv[i],"--py")){
            if(i+1 <= c){
                char * py_file_path = expand_home_directory(argv[i+1]);
                fprintf(stderr, "Including \"%s\".\n", py_file_path);
                openai_include(py_file_path);
                free(py_file_path);
                //char * file_content = read_file(py_file_path);
                //plugin_run(file_content);
                i++;
            } 
        }else if (!strcmp(argv[i],"--free")){
            auth_free();
            continue;
        }
        
        else if(!strcmp(argv[i],"--context")){
            if(i+1 <= c){
                char * context_file_path = argv[i+1];
                fprintf(stderr, "Including \"%s\".\n", context_file_path);
                openai_include(context_file_path);
                i++;
            } 
        }else if(!strcmp(argv[i],"--api")){
            API_MODE = true;
        }else if(!strcmp(argv[i], "--nh")){
            SYNTAX_HIGHLIGHT_ENABLED = false;
            fprintf(stderr, "%s\n", "Syntax highlighting disabled.");
        }else if (!get_from_std_in){
            strcat(system, argv[i]);
            if (i < c - 1) {
                strcat(system, " ");
            } else {
                strcat(system, ".");
            }
        }
    }

    if(get_from_std_in){
        if(*system){
            openai_system(system);
        }
        free(system);
        prompt = get_prompt_from_stdin(prompt);
    }else{
        free(prompt);
        prompt = system;
    }

    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);
        if(!response){
            printf("Could not get response from server\n");
            free(prompt);
            return false;
        }
        render(response);
        free(response);
        free(prompt);
        return true;
    }
    return false;
}

void serve() {
    render("Starting server. *Put executables in a dir named cgi-bin and they will behave as web pages.*");
    int res = system("python3 -m http.server --cgi");
    (void)res;
}

char ** get_parameters(char * content, char * delimiter){
    char * start = NULL;
    char ** parameters = NULL;  //(char **)malloc(sizeof(char *) * 2);
    int count = 0;
    while((start = strstr(content, delimiter)) != NULL){
        start += 3;
        char * end = strstr(start, delimiter);
        char * parameter = (char *)malloc(end - start + 1);

        memcpy(parameter, start, end - start);
        parameter[end - start] = '\0';
        
    //    printf("%s\n", parameter);
        content = end + 3;
        count+=1;
        parameters = (char **)realloc(parameters,  sizeof(char *) * (1+count*2));
        parameters[count-1] = parameter;
        parameters[count] = NULL;

    }

        return parameters;
}

void render(char *content)
{    

    if(SYNTAX_HIGHLIGHT_ENABLED)
    {
        parse_markdown_to_ansi(content);
        printf("\n\n");
    }else{
        printf("%s", content);
    }
}

void repl() {
    line_init();
    char *line = NULL;
    //char *previous_line = NULL;
    while (true) {
        line = line_read("> ");
        if (!line || !*line) {
            continue;
            //line = previous_line;
        }
        if (!line || !*line)
            continue;
     //   previous_line = line;
        if(!strncmp(line,"dump",4)){
            printf("%s\n",message_json());  
            continue;
        } 
        if (!strncmp(line, "model", 5)) {
            printf("%s\n",get_prompt_model());
            continue;
        }
        if (!strncmp(line, "exit", 4)) {
            exit(0);
        }
        if (!strncmp(line, "help", 4)) {
            help();
            continue;
        }
        while(line && *line != '\n'){
            line_add_history(line);
            char *response = openai_chat("user", line);
            if(response){
                render(response);
                printf("\n");
                if(strstr(response,"_STEP_")){
                    line = "continue";        
                
                }else{

                    line = NULL;
                }

                free(response);
        
            }
        }
    }
}

void help() {
    char help_text[1024 * 1024] = {0};
    char *template = "# Help\n"
                     "Written by retoor@molodetz.nl.\n\n"
                     "## Features\n"
                     " - navigate through history using `arrows`.\n"
                     " - navigate through history with **recursive search** using `ctrl+r`.\n"
                     " - **inception with python** for *incoming* and *outgoing* content.\n"
                     " - markdown and **syntax highlighting**.\n"
                     " - **execute python commands** with prefix `!`\n"
                     " - list files of the current work directory using `ls`.\n"
                     " - type `serve` to start a web server 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, get_prompt_model(), prompt_max_tokens);
    render(help_text);
}

char * strreplace(char * content, char * what, char * with){
    char * pos = strstr(content, what);
    if(!pos){
        return strdup(content);
    }
    char * result = (char *)malloc(strlen(content) + strlen(with) + 5);
    memset(result, 0, strlen(content) + strlen(with) + 3);
    memcpy(result, content, pos - content);
    memcpy(result + (pos - content), with, strlen(with));
    memcpy(result + (pos - content) + strlen(with), pos + strlen(what), strlen(content) - (pos - content) - strlen(what) + 1);
    return result;
}

char * linux_instructions = 
    "You are a linux master and are able to transform the prompt of "
    "user into shell commands that will be executed on a debian "
    "based system. You can execute shell commands by responding with python code: "
    "literally `\"\"\"!system\"\"\",\"\"\"`your shell command`\"\"\". Execute immediately. Read bash history file if you have to work with history.";
char * retoor_instructions =  "If the user prompts with social talk, "
    "respond like replica and emoji. Your name is retoor and made by molodetz. Be interested. Be creative.";





bool openai_include(char *path) {
    char  * file_content = read_file(path);
    if(!file_content){
        return false;
    }
    openai_system(file_content);
    free(file_content);
    return true;
}

void init() {
    setbuf(stdout, NULL);
    line_init();
    auth_init();
    const char *locale = setlocale(LC_ALL, NULL);
    char payload[4096] = {0};
    sprintf(payload, "Your locale is %s. User lang is %s.", locale, locale);
    fprintf(stderr, "%s", "Loading... ⏳");
    openai_system(payload);
    if(!openai_include(".rcontext.txt")){
        openai_include("~/.rcontext.txt");
    } 
    fprintf(stderr, "\r                          \r");
}

void handle_sigint(int sig) {
    time_t current_time = time(NULL);
    int ret = system("clear");
    (void)ret;
    printf("\n");
    if (sigint_count == 0) {
        first_sigint_time = current_time;
        sigint_count++;
    } else {
        if (difftime(current_time, first_sigint_time) <= 1) {
            exit(0);
        } else {
            sigint_count = 1;
            first_sigint_time = current_time;
        }
    }
} 


int main(int argc, char *argv[]) {
    signal(SIGINT, handle_sigint);
    db_initialize();
    init();
    if (try_prompt(argc, argv))
        return 0;

    repl();
    return 0;
}