aboutsummaryrefslogtreecommitdiff
path: root/browser/GMPolyfill.js
blob: 6f05ca0bf2cc15d91a673ba224c7484eff003f25 (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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
/*
 * Vencord, a modification for Discord's desktop app
 * Copyright (c) 2022 Vendicated and contributors
 *
 * 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/>.
*/

function fetchOptions(url) {
    return new Promise((resolve, reject) => {
        const opt = {
            method: "OPTIONS",
            url: url,
        };
        opt.onload = resp => resolve(resp.responseHeaders);
        opt.ontimeout = () => reject("fetch timeout");
        opt.onerror = () => reject("fetch error");
        opt.onabort = () => reject("fetch abort");
        GM_xmlhttpRequest(opt);
    });
}

function parseHeaders(headers) {
    if (!headers)
        return {};
    const result = {};
    const headersArr = headers.trim().split("\n");
    for (var i = 0; i < headersArr.length; i++) {
        var row = headersArr[i];
        var index = row.indexOf(":")
            , key = row.slice(0, index).trim().toLowerCase()
            , value = row.slice(index + 1).trim();

        if (result[key] === undefined) {
            result[key] = value;
        } else if (Array.isArray(result[key])) {
            result[key].push(value);
        } else {
            result[key] = [result[key], value];
        }
    }
    return result;
}

// returns true if CORS permits request
async function checkCors(url, method) {
    const headers = parseHeaders(await fetchOptions(url));

    const origin = headers["access-control-allow-origin"];
    if (origin !== "*" && origin !== window.location.origin) return false;

    const methods = headers["access-control-allow-methods"]?.split(/,\s/g);
    if (methods && !methods.includes(method)) return false;

    return true;
}

function blobTo(to, blob) {
    if (to === "arrayBuffer" && blob.arrayBuffer) return blob.arrayBuffer();
    return new Promise((resolve, reject) => {
        var fileReader = new FileReader();
        fileReader.onload = event => resolve(event.target.result);
        if (to === "arrayBuffer") fileReader.readAsArrayBuffer(blob);
        else if (to === "text") fileReader.readAsText(blob, "utf-8");
        else reject("unknown to");
    });
}

function GM_fetch(url, opt) {
    return new Promise((resolve, reject) => {
        checkCors(url, opt?.method || "GET")
            .then(can => {
                if (can) {
                    // https://www.tampermonkey.net/documentation.php?ext=dhdg#GM_xmlhttpRequest
                    const options = opt || {};
                    options.url = url;
                    options.data = options.body;
                    options.responseType = "blob";
                    options.onload = resp => {
                        var blob = resp.response;
                        resp.blob = () => Promise.resolve(blob);
                        resp.arrayBuffer = () => blobTo("arrayBuffer", blob);
                        resp.text = () => blobTo("text", blob);
                        resp.json = async () => JSON.parse(await blobTo("text", blob));
                        resp.headers = new Headers(parseHeaders(resp.responseHeaders));
                        resolve(resp);
                    };
                    options.ontimeout = () => reject("fetch timeout");
                    options.onerror = () => reject("fetch error");
                    options.onabort = () => reject("fetch abort");
                    GM_xmlhttpRequest(options);
                } else {
                    reject("CORS issue");
                }
            });
    });
}
export const fetch = GM_fetch;