aboutsummaryrefslogtreecommitdiff
/**
 * This file is part of Haketilo.
 *
 * Function: Handle page actions in a content script.
 *
 * Copyright (C) 2021 Wojtek Kosior
 *
 * 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.
 *
 * As additional permission under GNU GPL version 3 section 7, you
 * may distribute forms of that code without the copy of the GNU
 * GPL normally required by section 4, provided you include this
 * license notice and, in case of non-source distribution, a URL
 * through which recipients can access the Corresponding Source.
 * If you modify file(s) with this exception, you may extend this
 * exception to your version of the file(s), but you are not
 * obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 *
 * As a special exception to the GPL, any HTML file which merely
 * makes function calls to this code, and for that purpose
 * includes it by reference shall be deemed a separate work for
 * copyright law purposes. If you modify this code, you may extend
 * this exception to your version of the code, but you are not
 * obligated to do so. If you do not wish to do so, delete this
 * exception statement from your version.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 *
 * I, Wojtek Kosior, thereby promise not to sue for violation of this file's
 * license. Although I request that you do not make use this code in a
 * proprietary program, I am not going to enforce this in court.
 */

/*
 * IMPORTS_START
 * IMPORT CONNECTION_TYPE
 * IMPORT browser
 * IMPORT report_script
 * IMPORT report_document_type
 * IMPORTS_END
 */

let policy;
/* Snapshot url and content type early; these can be changed by other code. */
let url;
let is_html;
let port;
let loaded = false;
let scripts_awaiting = [];

function handle_message(message)
{
    const [action, data] = message;

    if (action === "inject") {
	for (let script_text of data) {
	    if (loaded)
		add_script(script_text);
	    else
		scripts_awaiting.push(script_text);
	}
    }
    else {
	console.error(`Bad page action '${action}'.`);
    }
}

function document_ready(event)
{
    loaded = true;

    for (let script_text of scripts_awaiting)
	add_script(script_text);

    scripts_awaiting = undefined;
}

function add_script(script_text)
{
    if (!is_html)
	return;

    let script = document.createElement("script");
    script.textContent = script_text;
    script.setAttribute("nonce", policy.nonce);
    script.haketilo_payload = true;
    document.body.appendChild(script);

    report_script(script_text);
}

function handle_page_actions(_policy, doc_ready_promise) {
    policy = _policy;

    url = document.URL;
    is_html = document instanceof HTMLDocument;
    report_document_type(is_html);

    doc_ready_promise.then(document_ready);

    if (policy.payload) {
	port = browser.runtime.connect({name : CONNECTION_TYPE.PAGE_ACTIONS});
	port.onMessage.addListener(handle_message);
	port.postMessage({payload: policy.payload});
    }
}

/*
 * EXPORTS_START
 * EXPORT handle_page_actions
 * EXPORTS_END
 */