register

1import t from "ava";
2import fs from "fs-extra";
3import path from "path";
4 
5const requiredRecordsToProxy = new Set(["A", "AAAA", "CNAME"]);
6 
7const domainCache = {};
8 
9function getDomainData(file) {
10    if (domainCache[file]) {
11        return domainCache[file];
12    }
13 
14    try {
15        const data = fs.readJsonSync(path.join(domainsPath, file));
16        domainCache[file] = data;
17        return data;
18    } catch (error) {
19        throw new Error(`Failed to read JSON for ${file}: ${error.message}`);
20    }
21}
22 
23function validateProxiedRecords(t, data, file) {
24    const recordTypes = Array.from(requiredRecordsToProxy).join(", ");
25 
26    // Forcefully stop raw.is-a.dev from being proxied
27    if (file === "raw.json") {
28        t.true(!data.proxied, `${file}: raw.is-a.dev cannot be proxied`);
29        return;
30    }
31 
32    if (data.proxied) {
33        const hasProxiedRecord = Object.keys(data.records).some((key) => requiredRecordsToProxy.has(key));
34 
35        t.true(
36            hasProxiedRecord,
37            `${file}: Proxied is true but there are no records that can be proxied (${recordTypes} expected)`
38        );
39    }
40}
41 
42const domainsPath = path.resolve("domains");
43const files = fs.readdirSync(domainsPath).filter((file) => file.endsWith(".json"));
44 
45t("Domains with proxy enabled must have at least one proxy-able record", (t) => {
46    files.forEach((file) => {
47        const data = getDomainData(file);
48 
49        validateProxiedRecords(t, data, file);
50    });
51});
52 
53const unproxyable = [
54    {
55        type: "CNAME",
56        value: "*.onrender.com"
57    }
58];
59 
60t("Domains with specific records must not have proxy enabled", (t) => {
61    files.forEach((file) => {
62        const data = getDomainData(file);
63 
64        if (data.proxied) {
65            unproxyable.forEach((record) => {
66                let recordExists = false;
67 
68                if (!data.records[record.type]) {
69                    t.pass();
70                    return;
71                } else if (Array.isArray(data.records[record.type])) {
72                    recordExists = data.records[record.type].some((r) =>
73                        record.value.startsWith("*.") ? r.endsWith(record.value.slice(2)) : r === record.value
74                    );
75                } else {
76                    recordExists = record.value.startsWith("*.")
77                        ? data.records[record.type].endsWith(record.value.slice(2))
78                        : data.records[record.type] === record.value;
79                }
80 
81                t.false(
82                    recordExists,
83                    `${file}: Records matching \`${record.type}: "${record.value}"\` cannot be proxied`
84                );
85            });
86        }
87    });
88});
89