onlyoffice.github.io/sdkjs-plugins/content/ai/scripts/customAssistant.js

/*
 * (c) Copyright Ascensio System SIA 2010-2025
 *
 * This program is a free software product. You can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public License (AGPL)
 * version 3 as published by the Free Software Foundation. In accordance with
 * Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
 * that Ascensio System SIA expressly excludes the warranty of non-infringement
 * of any third-party rights.
 *
 * This program is distributed WITHOUT ANY WARRANTY; without even the implied
 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR  PURPOSE. For
 * details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
 *
 * You can contact Ascensio System SIA at 20A-6 Ernesta Birznieka-Upish
 * street, Riga, Latvia, EU, LV-1050.
 *
 * The  interactive user interfaces in modified source and object code versions
 * of the Program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU AGPL version 3.
 *
 * Pursuant to Section 7(b) of the License you must retain the original Product
 * logo when distributing the program. Pursuant to Section 7(e) we decline to
 * grant you any rights under trademark law for use of our trademarks.
 *
 * All the Product's GUI elements, including illustrations and icon sets, as
 * well as technical writing content are licensed under the terms of the
 * Creative Commons Attribution-ShareAlike 4.0 International. See the License
 * terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
 *
 */

// @ts-check

/// <reference path="./utils/theme.js" />
/**
 * @typedef {Object} localStorageCustomAssistantItem
 * @property {string} id
 * @property {string} name
 * @property {string} query
 */

(function (window) {
    const LOCAL_STORAGE_KEY = "onlyoffice_ai_saved_assistants";
    const { form, textarea, inputId, inputName } = getFormElements();
    const mainContainer = document.getElementById("custom_assistant_window");
    if (!mainContainer) {
        console.error("Custom Assistant: required elements are missing");
        return;
    }

    window.Asc.plugin.init = function () {
        window.Asc.plugin.sendToPlugin("onWindowReady", {});

        inputId.value = generateHashedDateString();
        inputName.focus();

        mainContainer.addEventListener("click", function (e) {
            if (e.target instanceof HTMLAnchorElement) {
                e.preventDefault();
                window.open(e.target.href, "_blank");
            }
        });
    };

    window.Asc.plugin.attachEvent("onClickAdd", () => {
        let formFields = null;
        if (form.checkValidity()) {
            formFields = saveCustomAssistantToLocalStorage();
        } else {
            form.reportValidity();
            if (!inputName.value) {
                inputName.focus();
            } else {
                textarea.focus();
            }
        }

        window.Asc.plugin.sendToPlugin("onAddNewAssistant", formFields);
    });

    window.Asc.plugin.attachEvent(
        "onEditAssistant",
        (/** @type {string} */ assistantId) => {
            const savedAssistants = JSON.parse(
                localStorage.getItem(LOCAL_STORAGE_KEY) || "[]"
            );
            const assistant = savedAssistants.find(
                (/** @type {localStorageCustomAssistantItem} */ item) =>
                    item.id === assistantId
            );
            if (assistant) {
                inputId.value = assistant.id;
                inputName.value = assistant.name;
                textarea.value = assistant.query;
            }

            textarea.focus();
        }
    );

    function onThemeChanged(theme) {
        window.Asc.plugin.onThemeChangedBase(theme);
        updateBodyThemeClasses(theme.type, theme.name);
        updateThemeVariables(theme);
    }
    window.Asc.plugin.onTranslate = function () {
        let elements = document.querySelectorAll(".i18n");
        elements.forEach(function (element) {
            if (
                element instanceof HTMLTextAreaElement ||
                element instanceof HTMLInputElement
            ) {
                element.placeholder = window.Asc.plugin.tr(element.placeholder);
            } else if (element instanceof HTMLElement) {
                element.innerText = window.Asc.plugin.tr(element.innerText);
            }
        });
    };
    window.Asc.plugin.onThemeChanged = onThemeChanged;
    window.Asc.plugin.attachEvent("onThemeChanged", onThemeChanged);

    /** @returns {localStorageCustomAssistantItem} */
    function saveCustomAssistantToLocalStorage() {
        const id = inputId.value.trim();
        const name = inputName.value.trim();
        const query = textarea.value.trim();

        /** @type {localStorageCustomAssistantItem[]} */
        const savedAssistants = JSON.parse(
            localStorage.getItem(LOCAL_STORAGE_KEY) || "[]"
        );
        const existingAssistantIndex = savedAssistants.findIndex(
            (item) => item.id === id
        );
        if (existingAssistantIndex !== -1) {
            savedAssistants[existingAssistantIndex] = { id, name, query };
        } else {
            savedAssistants.push({ id, name, query });
        }

        localStorage.setItem(
            LOCAL_STORAGE_KEY,
            JSON.stringify(savedAssistants)
        );

        return { id, name, query };
    }

    /** @returns {{textarea: HTMLTextAreaElement, inputId: HTMLInputElement, inputName: HTMLInputElement, form: HTMLFormElement}} */
    function getFormElements() {
        const form = document.getElementById("input_prompt_wrapper");
        const inputId = document.getElementById("input_prompt_id");
        const inputName = document.getElementById("input_prompt_name");
        const textarea = document.getElementById("input_prompt");
        if (form instanceof HTMLFormElement === false) {
            throw new Error("Custom Assistant: form is not HTMLFormElement");
        }
        if (textarea instanceof HTMLTextAreaElement === false) {
            throw new Error(
                "Custom Assistant: textarea is not HTMLTextAreaElement"
            );
        }
        if (inputId instanceof HTMLInputElement === false) {
            throw new Error(
                "Custom Assistant: input id is not HTMLInputElement"
            );
        }
        if (inputName instanceof HTMLInputElement === false) {
            throw new Error(
                "Custom Assistant: input name is not HTMLInputElement"
            );
        }

        return { form, textarea, inputId, inputName };
    }

    /** @returns {string} */
    function generateHashedDateString() {
        const date = new Date();
        const data = date.toISOString() + Math.random() + performance.now();

        let hash = 0;
        for (let i = 0; i < data.length; i++) {
            const char = data.charCodeAt(i);
            hash = (hash << 5) - hash + char;
            hash = hash & hash;
        }

        return Math.abs(hash).toString(36) + "_" + date.getTime().toString(36);
    }
})(window);