yomichan/ext/js/templates/sandbox/template-renderer-frame-api.js

/*
 * Copyright (C) 2020-2021  Yomichan Authors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */

class TemplateRendererFrameApi {
    constructor(templateRenderer) {
        this._templateRenderer = templateRenderer;
        this._windowMessageHandlers = new Map([
            ['render', {async: false, handler: this._onRender.bind(this)}],
            ['renderMulti', {async: false, handler: this._onRenderMulti.bind(this)}],
            ['getModifiedData', {async: false, handler: this._onGetModifiedData.bind(this)}]
        ]);
    }

    prepare() {
        window.addEventListener('message', this._onWindowMessage.bind(this), false);
        this._postMessage(window.parent, 'ready', {}, null);
    }

    // Private

    _onWindowMessage(e) {
        const {source, data: {action, params, id}} = e;
        const messageHandler = this._windowMessageHandlers.get(action);
        if (typeof messageHandler === 'undefined') { return; }

        this._onWindowMessageInner(messageHandler, action, params, source, id);
    }

    async _onWindowMessageInner({handler, async}, action, params, source, id) {
        let response;
        try {
            let result = handler(params);
            if (async) {
                result = await result;
            }
            response = {result};
        } catch (error) {
            response = {error: this._serializeError(error)};
        }

        if (typeof id === 'undefined') { return; }
        this._postMessage(source, `${action}.response`, response, id);
    }

    _onRender({template, data, type}) {
        return this._templateRenderer.render(template, data, type);
    }

    _onRenderMulti({items}) {
        return this._serializeMulti(this._templateRenderer.renderMulti(items));
    }

    _onGetModifiedData({data, type}) {
        const result = this._templateRenderer.getModifiedData(data, type);
        return this._clone(result);
    }

    _serializeError(error) {
        try {
            if (typeof error === 'object' && error !== null) {
                const result = {
                    name: error.name,
                    message: error.message,
                    stack: error.stack
                };
                if (Object.prototype.hasOwnProperty.call(error, 'data')) {
                    result.data = error.data;
                }
                return result;
            }
        } catch (e) {
            // NOP
        }
        return {
            value: error,
            hasValue: true
        };
    }

    _serializeMulti(array) {
        for (let i = 0, ii = array.length; i < ii; ++i) {
            const value = array[i];
            const {error} = value;
            if (typeof error !== 'undefined') {
                value.error = this._serializeError(error);
            }
        }
        return array;
    }

    _clone(value) {
        return JSON.parse(JSON.stringify(value));
    }

    _postMessage(target, action, params, id) {
        return target.postMessage({action, params, id}, '*');
    }
}