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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
|
/**
* This file is part of Haketilo.
*
* Function: Instantiate the Pattern Tree data structure, filled with mappings
* from IndexedDB.
*
* 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.
*/
#IMPORT common/patterns_query_tree.js AS pqt
#IMPORT common/indexeddb.js AS haketilodb
#FROM common/browser.js IMPORT browser
const tree = pqt.make();
#EXPORT tree
const current_mappings = new Map();
let registered_script = null;
let script_update_occuring = false;
let script_update_needed;
async function update_content_script()
{
if (script_update_occuring)
return;
script_update_occuring = true;
while (script_update_needed) {
script_update_needed = false;
const code = `\
this.haketilo_pattern_tree = ${JSON.stringify(tree)};
if (this.haketilo_content_script_main)
haketilo_content_script_main();`;
const new_script = await browser.contentScripts.register({
"js": [{code}],
"matches": ["<all_urls>"],
"allFrames": true,
"runAt": "document_start"
});
if (registered_script)
registered_script.unregister();
registered_script = new_script;
}
script_update_occuring = false;
}
function register_mapping(mapping)
{
for (const pattern in mapping.payloads)
pqt.register(tree, pattern, mapping.identifier, mapping);
current_mappings.set(mapping.identifier, mapping);
}
function mapping_changed(change)
{
console.log('mapping changes!', arguments);
const old_version = current_mappings.get(change.identifier);
if (old_version !== undefined) {
for (const pattern in old_version.payloads)
pqt.deregister(tree, pattern, change.identifier);
current_mappings.delete(change.identifier);
}
if (change.new_val !== undefined)
register_mapping(change.new_val);
script_update_needed = true;
setTimeout(update_content_script, 0);
}
async function start()
{
const [tracking, initial_mappings] =
await haketilodb.track_mappings(mapping_changed);
initial_mappings.forEach(register_mapping);
script_update_needed = true;
await update_content_script();
}
#EXPORT start
|