summaryrefslogtreecommitdiff
path: root/ext/fg/js/frontend-api-receiver.js
blob: 3fa9e8b69b4dfdf4bae80ffe92d19f291cf6dba9 (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
/*
 * Copyright (C) 2019-2020  Yomichan Authors
 *
 * 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */


class FrontendApiReceiver {
    constructor(source, messageHandlers) {
        this._source = source;
        this._messageHandlers = messageHandlers;
    }

    prepare() {
        chrome.runtime.onConnect.addListener(this._onConnect.bind(this));
    }

    _onConnect(port) {
        if (port.name !== 'frontend-api-receiver') { return; }

        port.onMessage.addListener(this._onMessage.bind(this, port));
    }

    _onMessage(port, {id, action, params, target, senderId}) {
        if (target !== this._source) { return; }

        const messageHandler = this._messageHandlers.get(action);
        if (typeof messageHandler === 'undefined') { return; }

        const {handler, async} = messageHandler;

        this._sendAck(port, id, senderId);
        if (async) {
            this._invokeHandlerAsync(handler, params, port, id, senderId);
        } else {
            this._invokeHandler(handler, params, port, id, senderId);
        }
    }

    _invokeHandler(handler, params, port, id, senderId) {
        try {
            const result = handler(params);
            this._sendResult(port, id, senderId, {result});
        } catch (error) {
            this._sendResult(port, id, senderId, {error: errorToJson(error)});
        }
    }

    async _invokeHandlerAsync(handler, params, port, id, senderId) {
        try {
            const result = await handler(params);
            this._sendResult(port, id, senderId, {result});
        } catch (error) {
            this._sendResult(port, id, senderId, {error: errorToJson(error)});
        }
    }

    _sendAck(port, id, senderId) {
        port.postMessage({type: 'ack', id, senderId});
    }

    _sendResult(port, id, senderId, data) {
        port.postMessage({type: 'result', id, senderId, data});
    }
}