/**
 * 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 of 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

#IF MOZILLA || MV3
#FROM common/browser.js IMPORT browser
#ENDIF

let secret;

const tree = pqt.make();
#EXPORT tree

const current_mappings = new Map();

#IF MOZILLA || MV3
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_secret       = ${secret};
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, resource] of Object.entries(mapping.payloads))
	pqt.register(tree, pattern, mapping.identifier, resource);
    current_mappings.set(mapping.identifier, mapping);
}
#ENDIF

function mapping_changed(change)
{
    console.log('mapping changes!', arguments);
    const old_version = current_mappings.get(change.key);
    if (old_version !== undefined) {
	for (const pattern in old_version.payloads)
	    pqt.deregister(tree, pattern, change.key);

	current_mappings.delete(change.key);
    }

    if (change.new_val !== undefined)
	register_mapping(change.new_val);

#IF MOZILLA || MV3
    script_update_needed = true;
    setTimeout(update_content_script, 0);
#ENDIF
}

async function start(secret_)
{
    secret = secret_;

    const [tracking, initial_mappings] =
	  await haketilodb.track.mappings(mapping_changed);

    initial_mappings.forEach(register_mapping);
#IF MOZILLA || MV3
    script_update_needed = true;
    await update_content_script();
#ENDIF
}
#EXPORT start