<?php

namespace AIIFY;

use Orhanerday\OpenAi\OpenAi;
use Youniwemi\StringTemplate\Engine;

function render(string $template, $values = [])
{
    static $tpl;
    if ($tpl === null) {
        $tpl = new Engine();
    }
    return $tpl->render($template, $values);
}

/**
 * Main AI query action
 */
add_action('wp_ajax_open_ai', function () {
    check_ajax_referer('secure-nonce', 'openai_nonce');
    require_once AIIFY_VENDOR. '/autoload.php';


    $open_ai = new OpenAi(AIIFY_OPEN_AI_KEY);
    // Make sur style and tone are in our list of styles and tones
    $style = isset($_GET['style']) && isset(AIIFY_STYLES[ $_GET['style'] ]) ? $_GET['style'] : AIIFY_WRITING_STYLE;
    $tone = isset($_GET['tone']) && isset(AIIFY_TONES[ $_GET['tone'] ]) ? $_GET['tone'] : AIIFY_WRITING_TONE;
    $words = isset($_GET['maxWords']) ? intval($_GET['maxWords']) : AIIFY_WRITING_MAX_WORDS;

    $tpl = new Engine();
    //$words = intval($maxTokens * AIIFY_TOKEN_WORD_RATIO); //
    // Prepare context for style, tone, formating and length
    //$setup = sprintf(AIIFY_EDIT_INSTRUCTION_HEADER, $style, $tone, $words);
    $languages = get_languages();
    $language = isset($_GET['language']) && isset($languages[$_GET['language']]) ? $_GET['language'] : AIIFY_WRITING_LANGUAGE;
    $header = render(AIIFY_SYSTEM_INSTRUCTION_HEADER, compact('style', 'tone', 'words', 'language'));

    // don't need slashes, sani
    $prompt = isset($_GET['prompt']) ? wp_kses_post(wp_unslash($_GET['prompt'])) : null;
    if (isset($_GET['edit'])) {
        $edit = wp_kses_post($_GET['edit']) ;
        $commads = array_merge(AIIFY_EDIT_PROMPTS, AIIFY_GENERATE_AFTER_PROMPTS, AIIFY_GENERATE_BEFORE_PROMPTS);

        $command = isset($commads[ $prompt ]) ? $commads[$prompt] : $prompt ;
        if (isset(AIIFY_EDIT_PROMPTS[$prompt])) {
            // Do not change structure
            $command .= " Do not change its structure (if the input text is a paragraph, please respond with a paragraph)." ;
        }

        $prompt = render(AIIFY_SYSTEM_EDIT_STRUCTURE, compact('header', 'command', 'edit'));
    } else {

        $context = isset($_GET['context']) ? wp_kses_post(wp_unslash($_GET['context'])) : null;
        $keywords = isset($_GET['keywords']) ? wp_kses_post(wp_unslash($_GET['keywords'])) : null;
        $prompt = rtrim($prompt, '.');

        $prompt = render(AIIFY_SYSTEM_PROMPT_STRUCTURE, compact('header', 'language', 'prompt', 'context', 'keywords'));

    }


    // Total input words
    // $count_words_input = str_word_count(AIIFY_SYSTEM_PROMPT) + str_word_count($prompt);
    // total allowed response
    // $max_response = $words - $count_words_input;

    $isStream = !isset($_GET['nostream']);
    $method =  substr(AIIFY_CHAT_MODEL, 0, 3) == 'gpt' ? 'chat' : 'completion';
    $opts = [
       'model' => AIIFY_CHAT_MODEL ,
       'temperature' => (float) AIIFY_TEMPERATURE,
       //'max_tokens' => $maxTokens + intval($count_words_input/AIIFY_TOKEN_WORD_RATIO) ,
       'frequency_penalty' => (float) AIIFY_FREQUENCY_PENALTY,
       'presence_penalty' => (float) AIIFY_PRESENCE_PENALTY,
    ];
    if ($method == 'chat') {
        $opts['messages'] = [
           [
               "role" => "system",
               "content" => AIIFY_SYSTEM_PROMPT
           ],
           [
               "role" => "system",
               "content" => AIIFY_SYSTEM_PROMPT_FORMATING
           ],
           [
               "role" => "user",
               "content" => $prompt
           ]
       ];
    } else {
        $opts['prompt'] = AIIFY_SYSTEM_PROMPT. "\n\n".AIIFY_SYSTEM_PROMPT_FORMATING."\n\n" .  $prompt;
    }

    if ($isStream) {
        $opts["stream"] = true;
        header('Content-type: text/event-stream');
        header('Cache-Control: no-cache');
    }


    $complete = $open_ai->$method($opts, function ($curl_info, $data) use ($opts) {
        static $sentDebug;
        // add error
        if ($obj = json_decode($data) and $obj->error->message != "") {
            $opts['error'] = $obj->error;
        }
        if ($sentDebug === null) {
            $debug = json_encode($opts).PHP_EOL.PHP_EOL;
            echo "data: ".wp_kses_post($debug);
            $sentDebug = true;
            ob_flush();
        }

        if (isset($opts['error'])) {
            echo "error: ". wp_kses_post($obj->error->message);
        } else {
            echo wp_kses_post($data);
        }
        echo PHP_EOL;
        ob_flush();
        flush();
        return strlen($data);
    });
});