components/buttons_binder.js

import { ChatCompletionsRequester } from "../networks/llm_requester.js";
import {
    stop_latest_message_animation,
    start_latest_message_animation,
    create_new_chat_session,
    get_latest_message_content_displayer,
    get_selected_llm_model,
    create_messager,
} from "./chat_operator.js";

import { NewAgentModalWidget } from "../widgets/new_agent_modal_widget.js";

import { screen_scroller } from "./screen_scroller.js";
import { chat_history_storage } from "../storages/chat_history_storage.js";
import { endpoint_storage } from "../storages/endpoint_storage.js";
import { agent_storage } from "../storages/agent_storage.js";

export class ButtonsBinder {
    constructor() {}
    bind() {
        let send_user_input_binder = new SendUserInputButtonBinder();
        send_user_input_binder.bind();
        let new_chat_binder = new NewChatButtonBinder();
        new_chat_binder.bind();
        let toggle_endpoint_and_api_key_items_button_binder =
            new ToggleEndpointAndApiKeyItemsButtonBinder();
        toggle_endpoint_and_api_key_items_button_binder.bind();
        let add_endpoint_and_api_key_item_button_binder =
            new AddEndpointAndApiKeyItemButtonBinder();
        add_endpoint_and_api_key_item_button_binder.bind();
        let scroll_to_bottom_binder = new ScrollToBottomButtonBinder();
        scroll_to_bottom_binder.bind();
        let screenshot_button_binder = new ScreenshotButtonBinder();
        screenshot_button_binder.bind();
        let chat_history_sidebar_toggle_button_binder =
            new ChatHistorySidebarToggleButtonBinder();
        chat_history_sidebar_toggle_button_binder.bind();
        let chat_agents_sidebar_toggle_button_binder =
            new ChatAgentsSidebarToggleButtonBinder();
        chat_agents_sidebar_toggle_button_binder.bind();
        let new_agent_button_binder = new NewAgentButtonBinder();
        new_agent_button_binder.bind();
        let clear_chat_history_button_binder =
            new ClearChatHistoryButtonBinder();
        clear_chat_history_button_binder.bind();
        let clear_chat_agents_button_binder = new ClearChatAgentsButtonBinder();
        clear_chat_agents_button_binder.bind();
        let available_models_select_binder = new AvailableModelsSelectBinder();
        available_models_select_binder.bind();
        let agents_select_binder = new AgentsSelectBinder();
        agents_select_binder.bind();
        let dark_theme_toggle_button_binder = new DarkThemeToggleButtonBinder();
        dark_theme_toggle_button_binder.bind();
    }
}

class SendUserInputButtonBinder {
    constructor() {
        this.requester = null;
    }
    bind() {
        const button = $("#send-user-input");
        button.attr("status", "send").attr("title", "Send");
        button.click(() => {
            this.handle_user_input(button);
        });

        $("#user-input").keypress((event) => {
            if (
                event.key === "Enter" &&
                !event.shiftKey &&
                button.attr("status") === "send"
            ) {
                event.preventDefault();
                this.handle_user_input(button);
            }
        });
    }
    handle_user_input(button) {
        let user_input_content = $("#user-input").val();
        let status = button.attr("status");
        if (status === "send") {
            if (user_input_content === "") {
                return;
            } else {
                this.send(button);
            }
        } else if (status === "stop") {
            this.stop(button);
        } else {
            console.log("No action");
        }
    }

    send(button) {
        let button_icon = button.find("i");
        button.attr("status", "stop").attr("title", "Stop");
        button_icon.removeClass().addClass("fa fa-circle-pause fa-fade-fast");
        let user_input_content = $("#user-input").val();
        console.log(user_input_content);
        // empty user input and reset height
        $("#user-input").val("");
        $("#user-input").css("height", "auto");
        if (get_selected_llm_model() == "notes") {
            create_messager("user", user_input_content);
        } else {
            this.requester = new ChatCompletionsRequester({
                prompt: user_input_content,
            });
            this.requester.create_messager_components();
            start_latest_message_animation();
            this.requester.post().then(() => {
                this.stop(button);
            });
        }
    }

    stop(button) {
        this.requester.stop();
        let button_icon = button.find("i");
        stop_latest_message_animation();
        button.attr("status", "send").attr("title", "Send");
        button_icon
            .removeClass()
            .addClass("fa fa-paper-plane")
            .addClass("icon-success");
        hljs.highlightAll();
        console.log(get_latest_message_content_displayer().data("raw_content"));
        screen_scroller.set_user_scrolling(false);
    }
}

class NewChatButtonBinder {
    constructor() {}
    bind() {
        const button = $("#new-chat-session");
        button.attr("status", "new").attr("title", "New Chat");
        button.click(() => {
            chat_history_storage.save_current_chat_session();
            create_new_chat_session();
        });
    }
}

class ToggleEndpointAndApiKeyItemsButtonBinder {
    constructor() {}
    bind() {
        const button = $("#toggle-endpoint-and-api-key-items-button");
        button.attr("title", "Toggle endpoint and api key items");
        button.click(() => {
            $("#endpoint-and-api-key-items").toggle();
            $("#add-endpoint-and-api-key-item-button").parent().toggle();
        });
    }
}
class AddEndpointAndApiKeyItemButtonBinder {
    constructor() {}
    bind() {
        const button = $("#add-endpoint-and-api-key-item-button");
        button.attr("title", "Add endpoint and api key item");
        button.click(() => {
            if (!$("#endpoint-and-api-key-items").is(":visible")) {
                $("#endpoint-and-api-key-items").toggle();
            }
            endpoint_storage.add_endpoint_and_api_key_item();
        });
        button.parent().hide();
    }
}

class ScrollToBottomButtonBinder {
    constructor() {}
    bind() {
        const button = $("#scroll-to-bottom-button");
        button.attr("title", "Scroll to bottom");
        button.click(() => {
            screen_scroller.set_user_scrolling(false);
            screen_scroller.scroll_to_bottom(true);
        });
    }
}

class ScreenshotButtonBinder {
    constructor() {}
    bind() {
        const button = $("#screenshot-button");
        button.attr("title", "Take screenshot for whole chat");
        button.click(() => {
            let screenshot_padding = 0;
            // default padding is 0.75em (12px)
            // p-1 (4px)(0.25em); p-2 (8px)(0.5em); p-3 (16px)(1em);
            let container_padding = 12;
            let right_offset = 20;
            html2canvas($("#messagers-container")[0], {
                x: -(container_padding + screenshot_padding),
                width:
                    $("#messagers-container").width() +
                    container_padding * 2 +
                    screenshot_padding * 2 +
                    right_offset,
            }).then((canvas) => {
                var link = document.createElement("a");
                let date = new Date();
                let date_string = date.toISOString().split("T")[0];
                let time_string = date
                    .toTimeString()
                    .split(" ")[0]
                    .replace(/:/g, "-");
                let datetime_string = `${date_string}_${time_string}`;
                link.download = `chat_${datetime_string}.png`;
                link.href = canvas.toDataURL("image/png");
                link.click();
            });
        });
    }
}

class ChatHistorySidebarToggleButtonBinder {
    constructor() {
        this.storage_key = "show_chat_history_sidebar";
        this.sidebar_name = "chat-history";
    }
    get_show_history_sidebar_storage() {
        return localStorage.getItem(this.storage_key);
    }
    bind() {
        const sidebar = $(`#${this.sidebar_name}-sidebar`);
        // this line is not to check value as false,
        // but to check item not existed in localStorage
        if (!this.get_show_history_sidebar_storage()) {
            localStorage.setItem(this.storage_key, "true");
        }
        if (this.get_show_history_sidebar_storage() === "true") {
            sidebar.addClass("show");
        }
        const toggle_button = $(`#${this.sidebar_name}-sidebar-toggle-button`);
        toggle_button.attr("title", "Toggle chat history sidebar");
        toggle_button.click(() => {
            sidebar.toggleClass("show");
            localStorage.setItem(
                this.storage_key,
                sidebar.hasClass("show").toString()
            );
        });
        const close_button = $(`#${this.sidebar_name}-sidebar-close-button`);
        close_button.click(() => {
            sidebar.removeClass("show");
            localStorage.setItem(this.storage_key, "false");
        });
    }
}

class ClearChatHistoryButtonBinder {
    constructor() {
        this.sidebar_name = "chat-history";
    }
    bind() {
        const button = $(`#clear-${this.sidebar_name}-button`);
        button.attr("title", "Clear chat history");
        button.click(() => {
            if (confirm("Clear chat history?")) {
                chat_history_storage.clear_database();
            } else {
                console.log("Clear chat history canceled.");
            }
        });
    }
}

class ChatAgentsSidebarToggleButtonBinder {
    constructor() {
        this.storage_key = "show_chat_agents_sidebar";
        this.sidebar_name = "chat-agents";
    }
    get_show_sidebar_storage() {
        return localStorage.getItem(this.storage_key);
    }
    bind() {
        const sidebar = $(`#${this.sidebar_name}-sidebar`);
        // this line is not to check value as false,
        // but to check item not existed in localStorage
        if (!this.get_show_sidebar_storage()) {
            localStorage.setItem(this.storage_key, "true");
        }
        if (this.get_show_sidebar_storage() === "true") {
            sidebar.addClass("show");
        }
        const toggle_button = $(`#${this.sidebar_name}-sidebar-toggle-button`);
        toggle_button.attr("title", "Toggle chat agents sidebar");
        toggle_button.click(() => {
            sidebar.toggleClass("show");
            localStorage.setItem(
                this.storage_key,
                sidebar.hasClass("show").toString()
            );
        });

        const close_button = $(`#${this.sidebar_name}-sidebar-close-button`);
        close_button.click(() => {
            sidebar.removeClass("show");
            localStorage.setItem(this.storage_key, "false");
        });
    }
}

class NewAgentButtonBinder {
    constructor() {}
    bind() {
        const button = $("#new-agent-button");
        button.attr("title", "New agent");
        button.click(() => {
            let new_agent_modal_widget_id = "new-agent-modal";
            let new_agent_modal_widget_parent = $(
                `#${new_agent_modal_widget_id}`
            );
            if (new_agent_modal_widget_parent.length <= 0) {
                let new_agent_modal_widget = new NewAgentModalWidget({
                    widget_id: new_agent_modal_widget_id,
                });
                new_agent_modal_widget.spawn();
                new_agent_modal_widget_parent = $(
                    `#${new_agent_modal_widget_id}`
                );
            }
            new_agent_modal_widget_parent.modal("show");
        });
    }
}

class ClearChatAgentsButtonBinder {
    constructor() {
        this.sidebar_name = "chat-agents";
    }
    bind() {
        const button = $(`#clear-${this.sidebar_name}-button`);
        button.attr("title", "Clear agents");
        button.click(() => {
            if (confirm("Clear agents?")) {
                agent_storage.clear_database();
            } else {
                console.log("Clear agents canceled.");
            }
        });
    }
}

class AvailableModelsSelectBinder {
    constructor() {}
    bind() {
        const select = $("#available-models-select");
        select.change(() => {
            localStorage.setItem("default_model", select.val());
            console.log("set default_model:", select.val());
        });
    }
}

class AgentsSelectBinder {
    constructor() {}
    bind() {
        const select = $("#agents-select");
        select.change(() => {
            localStorage.setItem("default_agent", select.val());
            console.log("set default_agent:", select.val());
        });
    }
}

class DarkThemeToggleButtonBinder {
    constructor() {
        this.storage_key = "theme";
        this.toggle_button = $("#dark-theme-toggle-button");
        window.onload = () => this.set_theme();
    }
    bind() {
        this.toggle_button.click(() => {
            let theme = localStorage.getItem(this.storage_key);
            if (theme === "dark") {
                localStorage.setItem(this.storage_key, "light");
            } else {
                localStorage.setItem(this.storage_key, "dark");
            }
            this.set_theme();
        });
        DarkReader.setFetchMethod(window.fetch);
    }
    set_theme() {
        let theme = localStorage.getItem(this.storage_key);
        if (theme === "dark") {
            DarkReader.enable();
            this.toggle_button.attr("title", "Switch to Light theme");
        } else {
            DarkReader.disable();
            this.toggle_button.attr("title", "Switch to Dark theme");
        }
        console.log("set theme:", theme);
    }
}