yomichan/ext/js/accessibility/google-docs.js

/*
 * Copyright (C) 2021-2022  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/>.
 */

(async () => {
    // Reentrant check
    if (self.googleDocsAccessibilitySetup) { return; }
    self.googleDocsAccessibilitySetup = true;

    const invokeApi = (action, params) => {
        return new Promise((resolve, reject) => {
            chrome.runtime.sendMessage({action, params}, (response) => {
                void chrome.runtime.lastError;
                if (typeof response !== 'object' || response === null) {
                    reject(new Error('Unexpected response'));
                } else if (typeof response.error !== 'undefined') {
                    reject(new Error('Invalid response'));
                } else {
                    resolve(response.result);
                }
            });
        });
    };

    const optionsContext = {depth: 0, url: location.href};
    let options;
    try {
        options = await invokeApi('optionsGet', {optionsContext});
    } catch (e) {
        return;
    }

    if (!options.accessibility.forceGoogleDocsHtmlRendering) { return; }

    /* eslint-disable */
    const inject = () => {
        const start = Date.now();
        const maxDuration = 10000;
        const updateDocData = () => {
            const target = window._docs_flag_initialData;
            if (typeof target === 'object' && target !== null) {
                try {
                    target['kix-awcp'] = true;
                } catch (e) {
                    // NOP
                }
            } else if (Date.now() - start < maxDuration) {
                setTimeout(updateDocData, 0);
            }
        };
        const params = new URLSearchParams(location.search);
        if (params.get('mode') !== 'html') {
            const url = new URL(location.href);
            params.set('mode', 'html');
            url.search = params.toString();
            try {
                history.replaceState(history.state, '', url.toString());
            } catch (e) {
                // Ignore
            }
        }
        window._docs_force_html_by_ext = true;
        updateDocData();
    };
    /* eslint-enable */

    let parent = document.head;
    if (parent === null) {
        parent = document.documentElement;
        if (parent === null) { return; }
    }
    const script = document.createElement('script');
    script.textContent = `(${inject.toString()})();`;
    parent.appendChild(script);
    parent.removeChild(script);
})();