♻️ refactor: split into testable modules, add test suite

This commit is contained in:
2026-07-01 13:03:49 +07:00
parent 56de27161e
commit 57e27ea87f
20 changed files with 860 additions and 478 deletions
+1
View File
@@ -31,6 +31,7 @@ jobs:
check:
- typecheck
- format
- test
steps:
- name: Checkout
+5
View File
@@ -11,6 +11,7 @@
"winston": "3.19.0",
},
"devDependencies": {
"@types/bun": "^1.3.14",
"@types/node": "^26.0.1",
"@types/validator": "^13.15.10",
"prettier": "^3.9.4",
@@ -25,6 +26,8 @@
"@so-ric/colorspace": ["@so-ric/colorspace@1.1.6", "", { "dependencies": { "color": "^5.0.2", "text-hex": "1.0.x" } }, "sha512-/KiKkpHNOBgkFJwu9sh48LkHSMYGyuTcSFK/qMBdnOAlrRJzRSXAOFB5qwzaVQuDl8wAvHVMkaASQDReTahxuw=="],
"@types/bun": ["@types/bun@1.3.14", "", { "dependencies": { "bun-types": "1.3.14" } }, "sha512-h1hFqFVcvAvD9j9K7ZW7vd82aSA+rTdznZa+5bwvCwqSB1jmmfLcbIWhOLx1/+boy/xmjgCs/OMUL8hRJSmnPw=="],
"@types/node": ["@types/node@26.0.1", "", { "dependencies": { "undici-types": "~8.3.0" } }, "sha512-fc3KiUoBt6kie0N9bIW3E47vZsuaMf0PM2AaUpLCLT0s/LvX1nxAim6Fc049cNxODPpGm6qRAuUOB86SkRuPQw=="],
"@types/triple-beam": ["@types/triple-beam@1.3.5", "", {}, "sha512-6WaYesThRMCl19iryMYP7/x2OVgCtbIVflDGFpWnb9irXI3UjYE4AzmYuiUKY1AJstGijoY+MgUszMgRxIYTYw=="],
@@ -39,6 +42,8 @@
"axios": ["axios@1.18.1", "", { "dependencies": { "follow-redirects": "^1.16.0", "form-data": "^4.0.5", "https-proxy-agent": "^5.0.1", "proxy-from-env": "^2.1.0" } }, "sha512-3nTvFlvpn9Zu/RkHUqtc7/+al4UpRW5az71ap5zccp6e8RAYEzhMTecX8Dz1wWDYrPpUoB1HAQEGEAEvUr7S9g=="],
"bun-types": ["bun-types@1.3.14", "", { "dependencies": { "@types/node": "*" } }, "sha512-4N0ig0fEomHt5R0KCFWjovxow98rIoRwKolrYdCcknNwMekCXRnWEUvgu5soYV8QXtVsrUD8B95MBOZGPvr6KQ=="],
"call-bind-apply-helpers": ["call-bind-apply-helpers@1.0.2", "", { "dependencies": { "es-errors": "^1.3.0", "function-bind": "^1.1.2" } }, "sha512-Sp1ablJ0ivDkSzjcaJdxEunN5/XvksFJ2sMBFfq6x0ryhQV/2b/KwFe21cMpmHtPOSij8K99/wSfoEuTObmuMQ=="],
"color": ["color@5.0.3", "", { "dependencies": { "color-convert": "^3.1.3", "color-string": "^2.1.3" } }, "sha512-ezmVcLR3xAVp8kYOm4GS45ZLLgIE6SPAFoduLr6hTDajwb3KZ2F46gulK3XpcwRFb5KKGCSezCBAY4Dw4HsyXA=="],
+3 -1
View File
@@ -13,6 +13,7 @@
"winston": "3.19.0"
},
"devDependencies": {
"@types/bun": "^1.3.14",
"@types/node": "^26.0.1",
"@types/validator": "^13.15.10",
"prettier": "^3.9.4",
@@ -22,7 +23,8 @@
"typecheck": "tsc --noEmit",
"format": "prettier --check .",
"format:fix": "prettier --write .",
"check": "bun run typecheck && bun run format",
"check": "bun run typecheck && bun run format && bun test",
"test": "bun test",
"dev": "bun --watch src/index.ts",
"start": "bun src/index.ts",
"build": "bun run typecheck",
+2 -2
View File
@@ -1,7 +1,7 @@
import App from "./providers/App";
import App from "./providers/app";
App.loadConfig();
App.loadENV();
App.loadDaemon();
App.loadUpdater();
export default App;
+102
View File
@@ -0,0 +1,102 @@
import { beforeEach, describe, expect, mock, test } from "bun:test";
const axiosGet = mock();
const axiosPost = mock();
const axiosPut = mock();
mock.module("axios", () => ({
default: {
get: axiosGet,
post: axiosPost,
put: axiosPut,
},
}));
const { default: CloudflareAPI } = await import("./cloudflare-api");
describe("CloudflareAPI", () => {
beforeEach(() => {
axiosGet.mockReset();
axiosPost.mockReset();
axiosPut.mockReset();
});
test("getRecord returns the result on success", async () => {
axiosGet.mockResolvedValue({ data: { success: true, result: [{ id: "record-1" }] } });
const api = new CloudflareAPI("token", "zone-id");
const result = await api.getRecord({ type: "A", name: "example.com" });
expect(result).toEqual([{ id: "record-1" }]);
expect(axiosGet).toHaveBeenCalledWith(
"https://api.cloudflare.com/client/v4/zones/zone-id/dns_records",
expect.objectContaining({
headers: expect.objectContaining({ Authorization: "Bearer token" }),
}),
);
});
test("getRecord throws when the API reports failure", async () => {
axiosGet.mockResolvedValue({ data: { success: false, errors: [{ message: "bad request" }] } });
const api = new CloudflareAPI("token", "zone-id");
await expect(api.getRecord({})).rejects.toThrow("Unable to fetch record: bad request");
});
test("createRecord posts the record payload and returns the result", async () => {
axiosPost.mockResolvedValue({ data: { success: true, result: { id: "new-record" } } });
const api = new CloudflareAPI("token", "zone-id");
const result = await api.createRecord({
type: "A",
name: "example.com",
content: "1.2.3.4",
ttl: 300,
proxied: false,
});
expect(result).toEqual({ id: "new-record" });
expect(axiosPost).toHaveBeenCalledWith(
"https://api.cloudflare.com/client/v4/zones/zone-id/dns_records",
expect.objectContaining({ type: "A", name: "example.com", content: "1.2.3.4" }),
expect.anything(),
);
});
test("createRecord throws when the API reports failure", async () => {
axiosPost.mockResolvedValue({ data: { success: false, errors: [{ message: "nope" }] } });
const api = new CloudflareAPI("token", "zone-id");
await expect(api.createRecord({ ttl: 300 })).rejects.toThrow("Unable to create record: nope");
});
test("updateRecord puts the record payload and returns the result", async () => {
axiosPut.mockResolvedValue({ data: { success: true, result: { id: "record-1" } } });
const api = new CloudflareAPI("token", "zone-id");
const result = await api.updateRecord({
record_id: "record-1",
type: "A",
name: "example.com",
content: "1.2.3.4",
ttl: 300,
proxied: false,
});
expect(result).toEqual({ id: "record-1" });
expect(axiosPut).toHaveBeenCalledWith(
"https://api.cloudflare.com/client/v4/zones/zone-id/dns_records/record-1",
expect.objectContaining({ type: "A", name: "example.com" }),
expect.anything(),
);
});
test("updateRecord throws when the API reports failure", async () => {
axiosPut.mockResolvedValue({ data: { success: false, errors: [{ message: "denied" }] } });
const api = new CloudflareAPI("token", "zone-id");
await expect(api.updateRecord({ record_id: "record-1", ttl: 300 })).rejects.toThrow(
"Unable to update record: denied",
);
});
});
+122
View File
@@ -0,0 +1,122 @@
import axios from "axios";
export default class CloudflareAPI {
private readonly token: string;
private readonly zoneId: string;
constructor(token: string, zoneId: string) {
this.token = token;
this.zoneId = zoneId;
}
public async getRecord({
name,
type,
content,
proxied,
page,
}: {
name?: string;
type?: string;
content?: string;
proxied?: boolean;
page?: number;
}) {
const response = await axios.get(
`https://api.cloudflare.com/client/v4/zones/${this.zoneId}/dns_records`,
{
params: {
name,
type,
perPage: 5000,
content,
proxied,
page,
},
headers: {
Authorization: `Bearer ${this.token}`,
"Content-Type": "application/json",
},
},
);
if (response.data.success !== true)
throw new Error(`Unable to fetch record: ${response.data.errors[0].message}`);
return response.data.result;
}
public async createRecord({
name,
type,
content,
ttl,
proxied,
}: {
name?: string;
type?: string;
content?: string;
ttl: number;
proxied?: boolean;
}) {
const response = await axios.post(
`https://api.cloudflare.com/client/v4/zones/${this.zoneId}/dns_records`,
{
type,
name,
content,
ttl,
proxied,
},
{
headers: {
Authorization: `Bearer ${this.token}`,
"Content-Type": "application/json",
},
},
);
if (response.data.success !== true)
throw new Error(`Unable to create record: ${response.data.errors[0].message}`);
return response.data.result;
}
public async updateRecord({
record_id,
name,
type,
content,
ttl,
proxied,
}: {
record_id: string;
name?: string;
type?: string;
content?: string;
ttl: number;
proxied?: boolean;
}) {
const response = await axios.put(
`https://api.cloudflare.com/client/v4/zones/${this.zoneId}/dns_records/${record_id}`,
{
type,
name,
content,
ttl,
proxied,
},
{
headers: {
Authorization: `Bearer ${this.token}`,
"Content-Type": "application/json",
},
},
);
if (response.data.success !== true)
throw new Error(`Unable to update record: ${response.data.errors[0].message}`);
return response.data.result;
}
}
+118
View File
@@ -0,0 +1,118 @@
import { describe, expect, test } from "bun:test";
import {
arraysEqual,
isCloudflareConfig,
isEnviromentTokenPlaceholder,
isZoneConfig,
parseEnvironmentTokenPlaceholderName,
} from "./config-validation";
describe("arraysEqual", () => {
test("returns true for identical arrays", () => {
expect(arraysEqual(["a", "b"], ["a", "b"])).toBe(true);
});
test("returns false for different length arrays", () => {
expect(arraysEqual(["a"], ["a", "b"])).toBe(false);
});
test("returns false for different order", () => {
expect(arraysEqual(["a", "b"], ["b", "a"])).toBe(false);
});
test("returns false when either input is nullish", () => {
expect(arraysEqual(null, ["a"])).toBe(false);
expect(arraysEqual(["a"], undefined)).toBe(false);
});
});
describe("isEnviromentTokenPlaceholder", () => {
test("recognizes a well formed placeholder", () => {
expect(isEnviromentTokenPlaceholder("{ENV_TOKEN:MY_TOKEN}")).toBe(true);
});
test("rejects a plain string", () => {
expect(isEnviromentTokenPlaceholder("plain-token")).toBe(false);
});
});
describe("parseEnvironmentTokenPlaceholderName", () => {
test("extracts the env var name", () => {
expect(parseEnvironmentTokenPlaceholderName("{ENV_TOKEN:MY_TOKEN}")).toBe("MY_TOKEN");
});
test("returns null for a non-placeholder token", () => {
expect(parseEnvironmentTokenPlaceholderName("plain-token")).toBeNull();
});
});
describe("isZoneConfig", () => {
const validZone = {
id: "zone-id",
type: "A",
name: "example.com",
content: "{CURRENT_IPv4}",
ttl: 300,
proxied: true,
};
test("accepts a well formed zone config", () => {
expect(isZoneConfig(validZone)).toBe(true);
});
test("rejects a zone config missing a key", () => {
const { proxied, ...rest } = validZone;
expect(isZoneConfig(rest)).toBe(false);
});
test("rejects a zone config with wrong types", () => {
expect(isZoneConfig({ ...validZone, ttl: "300" })).toBe(false);
});
});
describe("isCloudflareConfig", () => {
const validConfig = {
token: "some-token",
updateInterval: 60,
zone: [
{
id: "zone-id",
type: "A",
name: "example.com",
content: "{CURRENT_IPv4}",
ttl: 300,
proxied: true,
},
],
};
test("accepts a well formed config", () => {
expect(isCloudflareConfig(validConfig)).toBe(true);
});
test("rejects a config with an invalid zone", () => {
expect(isCloudflareConfig({ ...validConfig, zone: [{ id: "only-id" }] })).toBe(false);
});
test("rejects an env token placeholder whose variable is unset", () => {
delete process.env.MISSING_TOKEN_VAR;
const onMissing = () => {};
expect(
isCloudflareConfig({ ...validConfig, token: "{ENV_TOKEN:MISSING_TOKEN_VAR}" }, onMissing),
).toBe(false);
});
test("accepts an env token placeholder whose variable is set", () => {
process.env.SET_TOKEN_VAR = "resolved-value";
expect(isCloudflareConfig({ ...validConfig, token: "{ENV_TOKEN:SET_TOKEN_VAR}" })).toBe(true);
delete process.env.SET_TOKEN_VAR;
});
test("invokes the onMissingEnvToken callback with the variable name", () => {
let received: string | undefined;
isCloudflareConfig({ ...validConfig, token: "{ENV_TOKEN:CALLBACK_VAR}" }, (name) => {
received = name;
});
expect(received).toBe("CALLBACK_VAR");
});
});
+89
View File
@@ -0,0 +1,89 @@
export interface ZoneConfig {
id: string;
type: string;
name: string;
content: string;
ttl: number;
proxied: boolean;
}
export interface CloudflareConfig {
token: string;
updateInterval: number;
zone: Array<ZoneConfig>;
}
export function arraysEqual(a: any, b: any): boolean {
if (a === b) return true;
if (a == null || b == null) return false;
if (a.length !== b.length) return false;
for (var i = 0; i < a.length; ++i) {
if (a[i] !== b[i]) return false;
}
return true;
}
export function isEnviromentTokenPlaceholder(token: string): boolean {
return token.startsWith("{ENV_TOKEN:") && token.endsWith("}");
}
export function parseEnvironmentTokenPlaceholderName(token: string): string | null {
if (!isEnviromentTokenPlaceholder(token)) return null;
return token.split("{ENV_TOKEN:")[1].slice(0, -1);
}
export function isZoneConfig(object: any): object is ZoneConfig {
if (!arraysEqual(Object.keys(object), ["id", "type", "name", "content", "ttl", "proxied"]))
return false;
return (
object &&
object.id &&
typeof object.id == "string" &&
object.type &&
typeof object.type == "string" &&
object.name &&
typeof object.name == "string" &&
object.content &&
typeof object.content == "string" &&
object.ttl &&
typeof object.ttl == "number" &&
typeof object.proxied == "boolean"
);
}
export function isCloudflareConfig(
object: any,
onMissingEnvToken?: (envTokenName: string) => void,
): object is CloudflareConfig {
if (!arraysEqual(Object.keys(object), ["token", "updateInterval", "zone"])) return false;
const res =
object &&
object.token &&
typeof object.token == "string" &&
object.updateInterval &&
typeof object.updateInterval == "number";
for (let zone of object.zone) {
if (!isZoneConfig(zone)) return false;
}
if (!res) return false;
const token = object.token as string;
// Process the env token
if (isEnviromentTokenPlaceholder(token)) {
const envTokenName = parseEnvironmentTokenPlaceholderName(token)!;
const envValue = process.env[envTokenName];
if (!envValue) {
onMissingEnvToken?.(envTokenName);
return false;
}
}
return true;
}
+47
View File
@@ -0,0 +1,47 @@
import { beforeEach, describe, expect, mock, test } from "bun:test";
const axiosGet = mock();
mock.module("axios", () => ({
default: { get: axiosGet },
}));
const { getCurrentIPv4, getCurrentIPv6 } = await import("./public-ip");
describe("getCurrentIPv4", () => {
beforeEach(() => axiosGet.mockReset());
test("returns the ip when a valid IPv4 address is returned", async () => {
axiosGet.mockResolvedValue({ data: { ip: "1.2.3.4" } });
expect(await getCurrentIPv4()).toBe("1.2.3.4");
});
test("throws when no ip is returned", async () => {
axiosGet.mockResolvedValue({ data: {} });
await expect(getCurrentIPv4()).rejects.toThrow("Unable to fetch ip address");
});
test("throws when the returned ip is not a valid IPv4 address", async () => {
axiosGet.mockResolvedValue({ data: { ip: "not-an-ip" } });
await expect(getCurrentIPv4()).rejects.toThrow("Invalid IP");
});
});
describe("getCurrentIPv6", () => {
beforeEach(() => axiosGet.mockReset());
test("returns the ip when a valid IPv6 address is returned", async () => {
axiosGet.mockResolvedValue({ data: { ip: "::1" } });
expect(await getCurrentIPv6()).toBe("::1");
});
test("returns null when the returned value is not a valid IPv6 address", async () => {
axiosGet.mockResolvedValue({ data: { ip: "1.2.3.4" } });
expect(await getCurrentIPv6()).toBeNull();
});
test("throws when no ip is returned", async () => {
axiosGet.mockResolvedValue({ data: {} });
await expect(getCurrentIPv6()).rejects.toThrow("Unable to fetch ip address");
});
});
+22
View File
@@ -0,0 +1,22 @@
import axios from "axios";
import validator from "validator";
export async function getCurrentIPv4(): Promise<string> {
const response = await axios.get("https://api.ipify.org?format=json");
if (!response.data.ip) throw new Error("Unable to fetch ip address");
if (!validator.isIP(response.data.ip, 4)) throw new Error("Invalid IP");
return response.data.ip;
}
export async function getCurrentIPv6(): Promise<string | null> {
const response = await axios.get("https://api64.ipify.org/?format=json");
if (!response.data.ip) throw new Error("Unable to fetch ip address");
if (!validator.isIP(response.data.ip, 6)) return null;
return response.data.ip;
}
-44
View File
@@ -1,44 +0,0 @@
import fs from "fs";
import path from "path";
const ConfigurationData: any = [];
class Configuration {
public init(): void {
const dir = path.join(process.cwd(), "configs/");
if (!fs.existsSync(dir)) {
fs.mkdirSync(dir);
}
this.copyExampleIfNotExists("UpdaterConfig.json");
this.loadConfig("UpdaterConfig.json");
}
public loadConfig(configFileName: string): void {
const dir = path.join(process.cwd(), "configs/");
if (!fs.existsSync(path.join(dir, configFileName)))
throw new Error(`Config file ${configFileName} does not exist`);
const config = JSON.parse(fs.readFileSync(path.join(dir, configFileName)).toString());
ConfigurationData[configFileName.replace(/\.[^/.]+$/, "")] = config;
}
public getConfig(key?: string) {
if (!key) return ConfigurationData;
else {
if (ConfigurationData[key]) return ConfigurationData[key];
else throw new Error(`No configuration found for ${key}`);
}
}
private copyExampleIfNotExists(file: string): void {
const dir = path.join(process.cwd(), "configs/");
if (!fs.existsSync(path.join(dir, file))) {
fs.copyFileSync(
path.join(process.cwd(), "configs_example/", `${file.replace(".json", ".example.json")}`),
path.join(dir, file),
);
}
}
}
export default new Configuration();
-408
View File
@@ -1,408 +0,0 @@
import Logger from "../libs/Logger";
import Environment from "./Environment";
import Configuration from "./Configuration";
import axios from "axios";
import validator from "validator";
interface CloudflareConfig {
token: string;
updateInterval: number;
zone: Array<ZoneConfig>;
}
interface ZoneConfig {
id: string;
type: string;
name: string;
content: string;
ttl: number;
proxied: boolean;
}
class Deamon {
private config: Array<CloudflareConfig> = [];
constructor() {}
public init(): void {
const _config = Configuration.getConfig("UpdaterConfig");
if (!_config) {
Logger.error("No configuration found");
return;
}
for (let cloudflareConfig of _config) {
if (!this.isCloudflareConfig(cloudflareConfig)) {
Logger.error(`Invalid configuration for cloudflare: ${JSON.stringify(cloudflareConfig)}`);
continue;
}
this.config.push(cloudflareConfig);
}
Logger.info(`Loaded ${this.config.length} cloudflare configs`);
this.start();
}
public start(): void {
Logger.info("Starting deamon");
for (const cloudflareConfig of this.config) {
this.update(cloudflareConfig);
setInterval(() => this.update(cloudflareConfig), cloudflareConfig.updateInterval * 1000);
}
}
private async update(cloudflareConfig: CloudflareConfig) {
let token = cloudflareConfig.token;
// Replace placeholders env with value
if (this.isEnviromentTokenPlaceholder(token)) {
const envTokenName = this.parseEnvironmentTokenPlaceholderName(token)!;
if (process.env[envTokenName]) {
token = process.env[envTokenName]!;
} else {
Logger.error(`Environment variable ${envTokenName} not found`);
return;
}
}
let IPv4, IPv6;
try {
IPv4 = await this.getCurrentIPv4();
IPv6 = await this.getCurrentIPv6();
} catch (err) {
Logger.error(`Unable to fetch ip address: ${err}`);
return;
}
IPv4 && Logger.info(`Current IPv4 address: ${IPv4}`);
IPv6 && Logger.info(`Current IPv6 address: ${IPv6}`);
for (const zone of cloudflareConfig.zone) {
const api = new CloudflareAPI(token, zone.id);
const records = await api
.getRecord({
type: zone.type,
name: zone.name,
})
.catch((err) => {
Logger.error(`Unable to get records for zone ${zone.id}`);
return err;
});
if (records instanceof Error) {
continue;
}
// No records found, create it
if (!records || records.length === 0) {
const newContent = zone.content
.replaceAll("{CURRENT_IPv4}", IPv4)
.replaceAll("{CURRENT_IPv6}", IPv6);
const result = await api
.createRecord({
type: zone.type,
name: zone.name,
content: IPv4,
ttl: zone.ttl,
proxied: zone.proxied,
})
.catch((err) => {
Logger.error(`Unable to create record: ${err.message}`);
return err;
});
if (records instanceof Error) {
continue;
}
if (result) Logger.info(`Created [${zone.type}] (${zone.name} -> ${newContent})`);
}
// Only 1 matching records found
else if (records.length === 1) {
const record = records[0];
const newContent = zone.content
.replaceAll("{CURRENT_IPv4}", IPv4)
.replaceAll("{CURRENT_IPv6}", IPv6);
// Check if the ip is the same
if (record.content !== IPv4 && record.content !== IPv6) {
const updateResult = await api
.updateRecord({
record_id: record.id,
type: zone.type,
name: zone.name,
ttl: zone.ttl,
content: newContent,
proxied: zone.proxied,
})
.catch((err) => {
Logger.error(`Unable to update record: ${err.message}`);
return err;
});
if (records instanceof Error) {
continue;
}
if (updateResult)
Logger.info(
`[${zone.type}] (${zone.name} -> ${record.content}) updated to (${zone.name} -> ${newContent})`,
);
} else if (record.ttl !== zone.ttl || record.proxied !== zone.proxied) {
const updateResult = await api
.updateRecord({
record_id: record.id,
type: zone.type,
name: zone.name,
ttl: zone.ttl,
content: newContent,
proxied: zone.proxied,
})
.catch((err) => {
Logger.error(`Unable to update record: ${err.message}`);
console.log(err);
return;
});
if (updateResult)
Logger.info(
`[${zone.type}] (${zone.name} -> ${record.content}) updated proxy status/ttl (TTL: ${record.ttl} -> ${zone.ttl}) (Proxy status: ${record.proxied} -> ${zone.proxied})`,
);
} else {
Logger.info(`[${zone.type}] (${zone.name} -> ${record.content}) already up to date`);
continue;
}
}
// Many records found
else if (records.length > 1) {
Logger.error(
`Multiple records found for ${zone.type} (${zone.name}) (Multiple records are not supported right now)`,
);
continue;
}
}
}
private isCloudflareConfig(object: any): object is CloudflareConfig {
if (!this.arraysEqual(Object.keys(object), ["token", "updateInterval", "zone"])) return false;
const res =
object &&
object.token &&
typeof object.token == "string" &&
object.updateInterval &&
typeof object.updateInterval == "number";
for (let zone of object.zone) {
if (!this.isZoneConfig(zone)) return false;
}
if (!res) return false;
const token = object.token as string;
// Process the env token
if (this.isEnviromentTokenPlaceholder(token)) {
const envTokenName = this.parseEnvironmentTokenPlaceholderName(token)!;
const envValue = process.env[envTokenName];
if (!envValue) {
Logger.error(`Environment variable ${envTokenName} not found`);
return false;
}
}
return true;
}
private isZoneConfig(object: any): object is ZoneConfig {
if (!this.arraysEqual(Object.keys(object), ["id", "type", "name", "content", "ttl", "proxied"]))
return false;
const res =
object &&
object.id &&
typeof object.id == "string" &&
object.type &&
typeof object.type == "string" &&
object.name &&
typeof object.name == "string" &&
object.content &&
typeof object.content == "string" &&
object.ttl &&
typeof object.ttl == "number" &&
typeof object.proxied == "boolean";
if (!res) return false;
return true;
}
private isEnviromentTokenPlaceholder(token: string) {
return token.startsWith("{ENV_TOKEN:") && token.endsWith("}");
}
private parseEnvironmentTokenPlaceholderName(token: string) {
if (!this.isEnviromentTokenPlaceholder(token)) return null;
return token.split("{ENV_TOKEN:")[1].slice(0, -1);
}
private async getCurrentIPv4() {
const response = await axios.get("https://api.ipify.org?format=json");
if (!response.data.ip) throw new Error("Unable to fetch ip address");
if (!validator.isIP(response.data.ip, 4)) throw new Error("Invalid IP");
return response.data.ip;
}
private async getCurrentIPv6() {
const response = await axios.get("https://api64.ipify.org/?format=json");
if (!response.data.ip) throw new Error("Unable to fetch ip address");
if (!validator.isIP(response.data.ip, 6)) return null;
return response.data.ip;
}
private arraysEqual(a: any, b: any) {
if (a === b) return true;
if (a == null || b == null) return false;
if (a.length !== b.length) return false;
for (var i = 0; i < a.length; ++i) {
if (a[i] !== b[i]) return false;
}
return true;
}
}
class CloudflareAPI {
private token: string;
private zoneId: string;
constructor(token: string, zoneId: string) {
this.token = token;
this.zoneId = zoneId;
}
public async getRecord({
name,
type,
content,
proxied,
page,
}: {
name?: string;
type?: string;
content?: string;
proxied?: boolean;
page?: number;
}) {
const response = await axios.get(
`https://api.cloudflare.com/client/v4/zones/${this.zoneId}/dns_records`,
{
params: {
name,
type,
perPage: 5000,
content,
proxied,
page,
},
headers: {
Authorization: `Bearer ${this.token}`,
"Content-Type": "application/json",
},
},
);
if (response.data.success !== true)
throw new Error(`Unable to fetch record: ${response.data.errors[0].message}`);
return response.data.result;
}
public async createRecord({
name,
type,
content,
ttl,
proxied,
}: {
name?: string;
type?: string;
content?: string;
ttl: number;
proxied?: boolean;
}) {
const response = await axios.post(
`https://api.cloudflare.com/client/v4/zones/${this.zoneId}/dns_records`,
{
type,
name,
content,
ttl,
proxied,
},
{
headers: {
Authorization: `Bearer ${this.token}`,
"Content-Type": "application/json",
},
},
);
if (response.data.success !== true)
throw new Error(`Unable to create record: ${response.data.errors[0].message}`);
return response.data.result;
}
public async updateRecord({
record_id,
name,
type,
content,
ttl,
proxied,
}: {
record_id: string;
name?: string;
type?: string;
content?: string;
ttl: number;
proxied?: boolean;
}) {
const response = await axios.put(
`https://api.cloudflare.com/client/v4/zones/${this.zoneId}/dns_records/${record_id}`,
{
type,
name,
content,
ttl,
proxied,
},
{
headers: {
Authorization: `Bearer ${this.token}`,
"Content-Type": "application/json",
},
},
);
if (response.data.success !== true)
throw new Error(`Unable to update record: ${response.data.errors[0].message}`);
return response.data.result;
}
}
export default new Deamon();
@@ -1,8 +1,8 @@
import Logger from "../libs/Logger";
import Logger from "../libs/logger";
import Environment from "./Environment";
import Configuration from "./Configuration";
import Daemon from "./Daemon";
import Environment from "./environment";
import Configuration from "./configuration";
import Updater from "./updater";
class App {
public loadConfig(): void {
@@ -14,9 +14,9 @@ class App {
Logger.log("info", "Loading environment");
Environment.init();
}
public loadDaemon(): void {
Logger.log("info", "Loading daemon");
Daemon.init();
public loadUpdater(): void {
Logger.log("info", "Loading updater");
Updater.init();
}
}
+65
View File
@@ -0,0 +1,65 @@
import { afterEach, beforeEach, describe, expect, test } from "bun:test";
import fs from "fs";
import os from "os";
import path from "path";
describe("Configuration", () => {
let tmpDir: string;
let originalCwd: string;
beforeEach(() => {
originalCwd = process.cwd();
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), "ddns-config-test-"));
fs.mkdirSync(path.join(tmpDir, "configs_example"), { recursive: true });
fs.writeFileSync(
path.join(tmpDir, "configs_example", "UpdaterConfig.example.json"),
JSON.stringify([{ token: "example-token", updateInterval: 60, zone: [] }]),
);
process.chdir(tmpDir);
});
afterEach(() => {
process.chdir(originalCwd);
fs.rmSync(tmpDir, { recursive: true, force: true });
});
test("copies the example config and loads it on init", async () => {
const { default: Configuration } = await import(`./configuration?t=${Date.now()}-1`);
Configuration.init();
expect(Configuration.getConfig("UpdaterConfig")).toEqual([
{ token: "example-token", updateInterval: 60, zone: [] },
]);
});
test("does not overwrite an existing config file", async () => {
fs.mkdirSync(path.join(tmpDir, "configs"), { recursive: true });
fs.writeFileSync(
path.join(tmpDir, "configs", "UpdaterConfig.json"),
JSON.stringify([{ token: "custom-token", updateInterval: 30, zone: [] }]),
);
const { default: Configuration } = await import(`./configuration?t=${Date.now()}-2`);
Configuration.init();
expect(Configuration.getConfig("UpdaterConfig")).toEqual([
{ token: "custom-token", updateInterval: 30, zone: [] },
]);
});
test("getConfig throws for an unknown key", async () => {
const { default: Configuration } = await import(`./configuration?t=${Date.now()}-3`);
Configuration.init();
expect(() => Configuration.getConfig("Unknown")).toThrow("No configuration found for Unknown");
});
test("loadConfig throws when the file does not exist", async () => {
const { default: Configuration } = await import(`./configuration?t=${Date.now()}-4`);
expect(() => Configuration.loadConfig("Missing.json")).toThrow(
"Config file Missing.json does not exist",
);
});
});
+49
View File
@@ -0,0 +1,49 @@
import fs from "fs";
import path from "path";
const ConfigurationData: Record<string, unknown> = {};
function configDir(): string {
return path.join(process.cwd(), "configs");
}
function configExampleDir(): string {
return path.join(process.cwd(), "configs_example");
}
class Configuration {
public init(): void {
const dir = configDir();
if (!fs.existsSync(dir)) {
fs.mkdirSync(dir);
}
this.copyExampleIfNotExists("UpdaterConfig.json");
this.loadConfig("UpdaterConfig.json");
}
public loadConfig(configFileName: string): void {
const filePath = path.join(configDir(), configFileName);
if (!fs.existsSync(filePath)) throw new Error(`Config file ${configFileName} does not exist`);
const config = JSON.parse(fs.readFileSync(filePath, "utf-8"));
ConfigurationData[configFileName.replace(/\.[^/.]+$/, "")] = config;
}
public getConfig<T = unknown>(key?: string): T {
if (!key) return ConfigurationData as T;
if (ConfigurationData[key]) return ConfigurationData[key] as T;
throw new Error(`No configuration found for ${key}`);
}
private copyExampleIfNotExists(file: string): void {
const target = path.join(configDir(), file);
if (!fs.existsSync(target)) {
const example = path.join(configExampleDir(), file.replace(".json", ".example.json"));
fs.copyFileSync(example, target);
}
}
}
export default new Configuration();
+38
View File
@@ -0,0 +1,38 @@
import { afterEach, beforeEach, describe, expect, mock, test } from "bun:test";
mock.module("dotenv", () => ({
config: () => ({}),
}));
describe("Environment", () => {
const originalEnv = { ...process.env };
beforeEach(() => {
process.env = { ...originalEnv };
});
afterEach(() => {
process.env = { ...originalEnv };
});
test("throws when NODE_ENV is missing", async () => {
delete process.env.NODE_ENV;
const { default: Environment } = await import(`./environment?t=${Date.now()}-1`);
expect(() => Environment.init()).toThrow(".env NODE_ENV is undefined");
});
test("throws when NODE_ENV is an invalid value", async () => {
process.env.NODE_ENV = "staging";
const { default: Environment } = await import(`./environment?t=${Date.now()}-2`);
expect(() => Environment.init()).toThrow(
'.env NODE_ENV must be either "production" or "development"',
);
});
test("succeeds and exposes NODE_ENV when valid", async () => {
process.env.NODE_ENV = "development";
const { default: Environment } = await import(`./environment?t=${Date.now()}-3`);
Environment.init();
expect(Environment.get().NODE_ENV).toBe("development");
});
});
@@ -1,7 +1,11 @@
import * as path from "path";
import * as dotenv from "dotenv";
import Logger from "../libs/Logger";
import Logger from "../libs/logger";
interface EnvironmentVariables {
NODE_ENV: string | undefined;
}
const requiredENV = ["NODE_ENV"];
@@ -9,34 +13,26 @@ class Environment {
public init(): void {
dotenv.config({ path: path.resolve(__dirname, "../../.env") });
for (let param of requiredENV) {
for (const param of requiredENV) {
if (this.isUndefinedOrEmpty(process.env[param]))
throw new Error(`.env ${param} is undefined`);
}
// NODE_ENV Checks
if (this.get().NODE_ENV != "production" && this.get().NODE_ENV != "development")
if (this.get().NODE_ENV !== "production" && this.get().NODE_ENV !== "development")
throw new Error('.env NODE_ENV must be either "production" or "development"');
Logger.log("info", `Running in ${process.env.NODE_ENV} environment`);
}
public get(): any {
const NODE_ENV = process.env.NODE_ENV;
public get(): EnvironmentVariables {
return {
NODE_ENV,
NODE_ENV: process.env.NODE_ENV,
};
}
private isUndefinedOrEmpty(value: String | undefined): boolean {
if (typeof value === "undefined") return true;
if (value === undefined) return true;
if (value === "") return true;
return false;
private isUndefinedOrEmpty(value: string | undefined): boolean {
return value === undefined || value === "";
}
}
+176
View File
@@ -0,0 +1,176 @@
import Logger from "../libs/logger";
import Configuration from "./configuration";
import CloudflareAPI from "../libs/cloudflare-api";
import { getCurrentIPv4, getCurrentIPv6 } from "../libs/public-ip";
import {
CloudflareConfig,
isCloudflareConfig,
isEnviromentTokenPlaceholder,
parseEnvironmentTokenPlaceholderName,
} from "../libs/config-validation";
class Updater {
private readonly config: Array<CloudflareConfig> = [];
public init(): void {
const _config = Configuration.getConfig<unknown[]>("UpdaterConfig");
if (!_config) {
Logger.error("No configuration found");
return;
}
for (const cloudflareConfig of _config) {
if (
!isCloudflareConfig(cloudflareConfig, (envTokenName) =>
Logger.error(`Environment variable ${envTokenName} not found`),
)
) {
Logger.error(`Invalid configuration for cloudflare: ${JSON.stringify(cloudflareConfig)}`);
continue;
}
this.config.push(cloudflareConfig);
}
Logger.info(`Loaded ${this.config.length} cloudflare configs`);
this.start();
}
public start(): void {
Logger.info("Starting updater");
for (const cloudflareConfig of this.config) {
this.update(cloudflareConfig);
setInterval(() => this.update(cloudflareConfig), cloudflareConfig.updateInterval * 1000);
}
}
private async update(cloudflareConfig: CloudflareConfig) {
let token = cloudflareConfig.token;
// Replace placeholders env with value
if (isEnviromentTokenPlaceholder(token)) {
const envTokenName = parseEnvironmentTokenPlaceholderName(token)!;
if (process.env[envTokenName]) {
token = process.env[envTokenName]!;
} else {
Logger.error(`Environment variable ${envTokenName} not found`);
return;
}
}
let IPv4, IPv6;
try {
IPv4 = await getCurrentIPv4();
IPv6 = await getCurrentIPv6();
} catch (err) {
Logger.error(`Unable to fetch ip address: ${err}`);
return;
}
IPv4 && Logger.info(`Current IPv4 address: ${IPv4}`);
IPv6 && Logger.info(`Current IPv6 address: ${IPv6}`);
for (const zone of cloudflareConfig.zone) {
const api = new CloudflareAPI(token, zone.id);
const records = await api
.getRecord({
type: zone.type,
name: zone.name,
})
.catch((err) => {
Logger.error(`Unable to get records for zone ${zone.id}`);
return err;
});
if (records instanceof Error) {
continue;
}
// No records found, create it
if (!records || records.length === 0) {
const newContent = zone.content
.replaceAll("{CURRENT_IPv4}", IPv4)
.replaceAll("{CURRENT_IPv6}", IPv6!);
const result = await api
.createRecord({
type: zone.type,
name: zone.name,
content: IPv4,
ttl: zone.ttl,
proxied: zone.proxied,
})
.catch((err) => {
Logger.error(`Unable to create record: ${err.message}`);
return err;
});
if (result) Logger.info(`Created [${zone.type}] (${zone.name} -> ${newContent})`);
}
// Only 1 matching records found
else if (records.length === 1) {
const record = records[0];
const newContent = zone.content
.replaceAll("{CURRENT_IPv4}", IPv4)
.replaceAll("{CURRENT_IPv6}", IPv6!);
// Check if the ip is the same
if (record.content !== IPv4 && record.content !== IPv6) {
const updateResult = await api
.updateRecord({
record_id: record.id,
type: zone.type,
name: zone.name,
ttl: zone.ttl,
content: newContent,
proxied: zone.proxied,
})
.catch((err) => {
Logger.error(`Unable to update record: ${err.message}`);
return err;
});
if (updateResult)
Logger.info(
`[${zone.type}] (${zone.name} -> ${record.content}) updated to (${zone.name} -> ${newContent})`,
);
} else if (record.ttl !== zone.ttl || record.proxied !== zone.proxied) {
const updateResult = await api
.updateRecord({
record_id: record.id,
type: zone.type,
name: zone.name,
ttl: zone.ttl,
content: newContent,
proxied: zone.proxied,
})
.catch((err) => {
Logger.error(`Unable to update record: ${err.message}`);
console.log(err);
return;
});
if (updateResult)
Logger.info(
`[${zone.type}] (${zone.name} -> ${record.content}) updated proxy status/ttl (TTL: ${record.ttl} -> ${zone.ttl}) (Proxy status: ${record.proxied} -> ${zone.proxied})`,
);
} else {
Logger.info(`[${zone.type}] (${zone.name} -> ${record.content}) already up to date`);
continue;
}
}
// Many records found
else if (records.length > 1) {
Logger.error(
`Multiple records found for ${zone.type} (${zone.name}) (Multiple records are not supported right now)`,
);
continue;
}
}
}
}
export default new Updater();
+3 -1
View File
@@ -25,7 +25,9 @@
// "useDefineForClassFields": true, /* Emit ECMAScript-standard-compliant class fields. */
/* Modules */
"module": "commonjs",
"module": "esnext",
"moduleResolution": "bundler",
"types": ["bun"],
/* Specify what module code is generated. */
// "rootDir": "./", /* Specify the root folder within your source files. */
// "moduleResolution": "node", /* Specify how TypeScript looks up a file from a given module specifier. */