aboutsummaryrefslogtreecommitdiff
path: root/background/storage_server.mjs
blob: 3a141ffb5ed868c53dac8ce3bcdddf736b773039 (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
/**
* Myext storage through connection (server side)
*
* Copyright (C) 2021 Wojtek Kosior
*
* Dual-licensed under:
*   - 0BSD license
*   - GPLv3 or (at your option) any later version
*/

import listen_for_connection from './message_server.mjs';
import get_storage from './storage.mjs';
import {TYPE_PREFIX} from '/common/stored_types.mjs';
import CONNECTION_TYPE from '/common/connection_types.mjs';

var storage;

async function handle_remote_call(port, message)
{
    let [call_id, func, args] = message;

    try {
	let result = await Promise.resolve(storage[func](...args));
	port.postMessage({call_id, result});
    } catch (error) {
	error = error + '';
	port.postMessage({call_id, error});
    }
}

function remove_storage_listener(cb) {
    storage.remove_change_listener(cb);
}

function new_connection(port)
{
    console.log("new remote storage connection!");

    port.postMessage({
	[TYPE_PREFIX.SCRIPT] : storage.get_all(TYPE_PREFIX.SCRIPT),
	[TYPE_PREFIX.BUNDLE] : storage.get_all(TYPE_PREFIX.BUNDLE),
	[TYPE_PREFIX.PAGE] : storage.get_all(TYPE_PREFIX.PAGE)
    });

    let handle_change = change => port.postMessage(change);

    storage.add_change_listener(handle_change);

    port.onMessage.addListener(m => handle_remote_call(port, m));
    port.onDisconnect.addListener(() => remove_storage_listener(handle_change));
}

export default async function start()
{
    storage = await get_storage();

    listen_for_connection(CONNECTION_TYPE.REMOTE_STORAGE, new_connection);
}