/* * Copyright (C) 2023 Yomitan Authors * Copyright (C) 2020-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/>. */ import {deferPromise, generateId, isObject} from '../core.js'; export class FrameClient { constructor() { /** @type {?string} */ this._secret = null; /** @type {?string} */ this._token = null; /** @type {?number} */ this._frameId = null; } /** @type {number} */ get frameId() { if (this._frameId === null) { throw new Error('Not connected'); } return this._frameId; } /** * @param {import('extension').HtmlElementWithContentWindow} frame * @param {string} targetOrigin * @param {number} hostFrameId * @param {import('frame-client').SetupFrameFunction} setupFrame * @param {number} [timeout] */ async connect(frame, targetOrigin, hostFrameId, setupFrame, timeout=10000) { const {secret, token, frameId} = await this._connectInternal(frame, targetOrigin, hostFrameId, setupFrame, timeout); this._secret = secret; this._token = token; this._frameId = frameId; } /** * @returns {boolean} */ isConnected() { return (this._secret !== null); } /** * @template T * @param {T} data * @returns {import('frame-client').Message<T>} * @throws {Error} */ createMessage(data) { if (!this.isConnected()) { throw new Error('Not connected'); } return { token: /** @type {string} */ (this._token), secret: /** @type {string} */ (this._secret), data }; } /** * @param {import('extension').HtmlElementWithContentWindow} frame * @param {string} targetOrigin * @param {number} hostFrameId * @param {(frame: import('extension').HtmlElementWithContentWindow) => void} setupFrame * @param {number} timeout * @returns {Promise<{secret: string, token: string, frameId: number}>} */ _connectInternal(frame, targetOrigin, hostFrameId, setupFrame, timeout) { return new Promise((resolve, reject) => { const tokenMap = new Map(); /** @type {?import('core').Timeout} */ let timer = null; const deferPromiseDetails = /** @type {import('core').DeferredPromiseDetails<void>} */ (deferPromise()); const frameLoadedPromise = deferPromiseDetails.promise; let frameLoadedResolve = /** @type {?() => void} */ (deferPromiseDetails.resolve); let frameLoadedReject = /** @type {?(reason?: import('core').RejectionReason) => void} */ (deferPromiseDetails.reject); /** * @param {string} action * @param {import('core').SerializableObject} params * @throws {Error} */ const postMessage = (action, params) => { const contentWindow = frame.contentWindow; if (contentWindow === null) { throw new Error('Frame missing content window'); } let validOrigin = true; try { validOrigin = (contentWindow.location.origin === targetOrigin); } catch (e) { // NOP } if (!validOrigin) { throw new Error('Unexpected frame origin'); } contentWindow.postMessage({action, params}, targetOrigin); }; /** @type {import('extension').ChromeRuntimeOnMessageCallback<import('extension').ChromeRuntimeMessageWithFrameId>} */ const onMessage = (message) => { onMessageInner(message); return false; }; /** * @param {import('extension').ChromeRuntimeMessageWithFrameId} message */ const onMessageInner = async (message) => { try { if (!isObject(message)) { return; } const {action, params} = message; if (!isObject(params)) { return; } await frameLoadedPromise; if (timer === null) { return; } // Done switch (action) { case 'frameEndpointReady': { const {secret} = /** @type {import('frame-client').FrameEndpointReadyDetails} */ (params); const token = generateId(16); tokenMap.set(secret, token); postMessage('frameEndpointConnect', {secret, token, hostFrameId}); } break; case 'frameEndpointConnected': { const {secret, token} = /** @type {import('frame-client').FrameEndpointConnectedDetails} */ (params); const frameId = message.frameId; const token2 = tokenMap.get(secret); if (typeof token2 !== 'undefined' && token === token2 && typeof frameId === 'number') { cleanup(); resolve({secret, token, frameId}); } } break; } } catch (e) { cleanup(); reject(e); } }; const onLoad = () => { if (frameLoadedResolve === null) { cleanup(); reject(new Error('Unexpected load event')); return; } if (FrameClient.isFrameAboutBlank(frame)) { return; } frameLoadedResolve(); frameLoadedResolve = null; frameLoadedReject = null; }; const cleanup = () => { if (timer === null) { return; } // Done clearTimeout(timer); timer = null; frameLoadedResolve = null; if (frameLoadedReject !== null) { frameLoadedReject(new Error('Terminated')); frameLoadedReject = null; } chrome.runtime.onMessage.removeListener(onMessage); frame.removeEventListener('load', onLoad); }; // Start timer = setTimeout(() => { cleanup(); reject(new Error('Timeout')); }, timeout); chrome.runtime.onMessage.addListener(onMessage); frame.addEventListener('load', onLoad); // Prevent unhandled rejections frameLoadedPromise.catch(() => {}); // NOP try { setupFrame(frame); } catch (e) { cleanup(); reject(e); } }); } /** * @param {import('extension').HtmlElementWithContentWindow} frame * @returns {boolean} */ static isFrameAboutBlank(frame) { try { const contentDocument = frame.contentDocument; if (contentDocument === null) { return false; } const url = contentDocument.location.href; return /^about:blank(?:[#?]|$)/.test(url); } catch (e) { return false; } } }