// 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 "r.h"
#include <signal.h>
#include <stdbool.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "line.h"
#include "markdown.h"
#include "openai.h"
#include "utils.h"
#include <locale.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.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);
  } 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", 5)) {
      printf("%s\n", message_json());
      continue;
    }
    if (!strncmp(line, "!verbose", 7)) {
      is_verbose = !is_verbose;
      fprintf(stderr, "%s\n",
              is_verbose ? "Verbose mode enabled" : "Verbose mode disabled");
      continue;
    }
    if (line && *line != '\n') {

      line_add_history(line);
    }
    if (!strncmp(line, "!models", 7)) {
      printf("Current model: %s\n", openai_fetch_models());
      continue;
    }
    if (!strncmp(line, "!model", 6)) {
      if (!strncmp(line + 6, " ", 1)) {
        line = line + 7;
        set_prompt_model(line);
      }
      printf("Current model: %s\n", get_prompt_model());
      continue;
    }
    if (!strncmp(line, "exit", 4)) {
      exit(0);
    }
    if (!strncmp(line, "help", 4)) {
      help();
      continue;
    }
    if (!strncmp(line, "!debug", 6)) {
      r_malloc_stats();
      continue;
    }
    while (line && *line != '\n') {

      char *response = openai_chat("user", line);
      if (response) {
        render(response);
        printf("\n");
        if (strstr(response, "_STEP_")) {
          line = "continue";

        } else {

          line = NULL;
        }

        free(response);
      } else {
        exit(0);
      }
    }
  }
}

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();
  db_initialize();
  char *schema = db_get_schema();
  char payload[1024 * 1024] = {0};
  sprintf(payload,
          "Your have a database that you can mutate using the query tool and "
          "the get and set tool. This is the schema in json format: %s. "
          "Dialect is sqlite.",
          schema);
  free(schema);
  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);
  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);

  init();
  if (try_prompt(argc, argv))
    return 0;

  repl();
  return 0;
}