import { Messager, MessagerList } from "./messager.js";
import { transform_footnote } from "../networks/stream_jsonizer.js";
import { screen_scroller } from "./screen_scroller.js";

let messagers_container = $("#messagers-container");
let available_models_select = $("#available-models-select");
let temperature_select = $("#temperature-select");

let messager_list = new MessagerList(messagers_container);
let chat_history = [messager_list];
let md_to_html_converter = new showdown.Converter();
md_to_html_converter.setFlavor("github");

export function get_active_messager_list() {
    return chat_history[chat_history.length - 1];
}

export function get_latest_messager() {
    return get_active_messager_list().messagers.slice(-1)[0];
}

export function create_messager(
    role,
    content = "",
    model = "",
    temperature = ""
) {
    let message = {
        role: role,
        content: content,
        model: model,
        temperature: temperature,
    };
    let messager = new Messager(message);
    get_active_messager_list().push(messager);
    screen_scroller.scroll_to_bottom();
}

export function get_selected_llm_model() {
    return available_models_select.val();
}

export function get_selected_temperature() {
    return Number(temperature_select.val());
}

export function get_latest_messager_container() {
    return get_active_messager_list().messagers_container.children().last();
}
export function get_latest_message_content_displayer() {
    return get_latest_messager_container().find(".content-displayer");
}

export function get_latest_user_messager() {
    return $(".message-user").last();
}

export function get_latest_assistant_messager() {
    return $(".message-assistant").last();
}

export function start_latest_message_animation() {
    get_latest_assistant_messager().addClass("inferring");
    get_latest_assistant_messager()
        .find(".button-group")
        .find(".regenerate-button")
        .find("i")
        .addClass("fa-spin-fast");
}

export function stop_latest_message_animation() {
    get_latest_assistant_messager().removeClass("inferring");
    get_latest_assistant_messager()
        .find(".button-group")
        .find(".regenerate-button")
        .find("i")
        .removeClass("fa-spin-fast");
}

export function get_request_messages() {
    return get_active_messager_list().get_request_messages();
}

export function pop_messager(n = 2) {
    return get_active_messager_list().pop(n);
}

export function update_message(json_chunks, content_displayer_updater = null) {
    if (content_displayer_updater === null) {
        content_displayer_updater = new ContentDisplayerUpdater();
    }
    json_chunks.forEach(function (item) {
        let choice = item.choices[0];
        let content = choice.delta.content;
        let role = choice.delta.role;
        let finish_reason = choice.finish_reason;
        if (role) {
            // console.log("role: " + role);
        }
        if (content) {
            content_displayer_updater.update_with_chunk_content(content);
        }
        if (finish_reason === "stop") {
            console.log("[STOP]");
        }
        // console.log(item);
    });
    return json_chunks;
}

export function create_new_chat_session() {
    let new_messager_list = new MessagerList(messagers_container);
    chat_history.push(new_messager_list);
    messagers_container.empty();
}

export class ContentDisplayerUpdater {
    constructor(content_displayer = null) {
        if (content_displayer === null) {
            self.content_displayer = get_latest_message_content_displayer();
        } else {
            self.content_displayer = content_displayer;
        }
    }
    update_with_chunk_content(content) {
        self.content_displayer.data(
            "raw_content",
            self.content_displayer.data("raw_content") + content
        );
        get_active_messager_list().messagers.slice(-1)[0].message.content +=
            content;
        self.content_displayer.html(
            md_to_html_converter.makeHtml(
                transform_footnote(self.content_displayer.data("raw_content"))
            )
        );
        self.content_displayer
            .find("table")
            .addClass("table table-bordered table-hover");
        screen_scroller.scroll_to_bottom();
    }
}