aboutsummaryrefslogtreecommitdiff
path: root/content/page_actions.js
blob: 8057541b8f4ff04252db88cf71a6462162f17db0 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
/**
 * Hachette handling of page actions in content scripts
 *
 * Copyright (C) 2021 Wojtek Kosior
 * Redistribution terms are gathered in the `copyright' file.
 */

/*
 * IMPORTS_START
 * IMPORT CONNECTION_TYPE
 * IMPORT browser
 * IMPORT report_script
 * IMPORT report_settings
 * IMPORT report_content_type
 * IMPORTS_END
 */

let policy_received_callback;
/* 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 = [];
let nonce;

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);
	}
    }
    if (action === "settings") {
	report_settings(data);
	policy_received_callback({url, allow: data[1].allow});
    }
}

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", nonce);
    script._hachette_payload = true;
    document.body.appendChild(script);

    report_script(script_text);
}

function handle_page_actions(script_nonce, policy_received_cb,
			     doc_ready_promise) {
    policy_received_callback = policy_received_cb;
    url = document.URL;
    is_html = /html/.test(document.contentType);
    report_content_type(document.contentType);

    doc_ready_promise.then(document_ready);

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

    nonce = script_nonce;
}

/*
 * EXPORTS_START
 * EXPORT handle_page_actions
 * EXPORTS_END
 */