81 lines
2.6 KiB
JavaScript
81 lines
2.6 KiB
JavaScript
const dns = require("dns");
|
|
const express = require("express");
|
|
const NodeCache = require("node-cache");
|
|
const isValidDomain = require("is-valid-domain");
|
|
|
|
const host = process.env.DNSLINK_API_HOSTNAME || "0.0.0.0";
|
|
const port = Number(process.env.DNSLINK_API_PORT) || 3100;
|
|
const cacheTTL = Number(process.env.DNSLINK_API_CACHE_TTL) || 300; // default to 5 minutes
|
|
|
|
const server = express();
|
|
const cache = new NodeCache({ stdTTL: cacheTTL });
|
|
|
|
const dnslinkNamespace = "skynet-ns";
|
|
const dnslinkRegExp = new RegExp(`^dnslink=/${dnslinkNamespace}/.+$`);
|
|
const dnslinkSkylinkRegExp = new RegExp(`^dnslink=/${dnslinkNamespace}/([a-zA-Z0-9_-]{46}|[a-z0-9]{55})`);
|
|
const hint = `valid example: dnslink=/${dnslinkNamespace}/3ACpC9Umme41zlWUgMQh1fw0sNwgWwyfDDhRQ9Sppz9hjQ`;
|
|
|
|
server.get("/dnslink/:name", async (req, res) => {
|
|
const success = (skylink) => res.set("Skynet-Skylink", skylink).send(skylink);
|
|
const failure = (message) => res.status(400).set("Dnslink-Error", message).send(message);
|
|
|
|
if (!isValidDomain(req.params.name)) {
|
|
return failure(`"${req.params.name}" is not a valid domain`);
|
|
}
|
|
|
|
if (cache.has(req.params.name)) {
|
|
return success(cache.get(req.params.name));
|
|
}
|
|
|
|
const lookup = `_dnslink.${req.params.name}`;
|
|
|
|
dns.resolveTxt(lookup, (error, records) => {
|
|
if (error) {
|
|
if (error.code === "ENOTFOUND") {
|
|
return failure(`ENOTFOUND: ${lookup} TXT record doesn't exist`);
|
|
}
|
|
|
|
if (error.code === "ENODATA") {
|
|
return failure(`ENODATA: ${lookup} dns lookup returned no data`);
|
|
}
|
|
|
|
return failure(`Failed to fetch ${lookup} TXT record: ${error.message}`);
|
|
}
|
|
|
|
if (records.length === 0) {
|
|
return failure(`No TXT record found for ${lookup}`);
|
|
}
|
|
|
|
const dnslinks = records.flat().filter((record) => dnslinkRegExp.test(record));
|
|
|
|
if (dnslinks.length === 0) {
|
|
return failure(`TXT records for ${lookup} found but none of them contained valid skynet dnslink - ${hint}`);
|
|
}
|
|
|
|
if (dnslinks.length > 1) {
|
|
return failure(`Multiple TXT records with valid skynet dnslink found for ${lookup}, only one allowed`);
|
|
}
|
|
|
|
const [dnslink] = dnslinks;
|
|
const matchSkylink = dnslink.match(dnslinkSkylinkRegExp);
|
|
|
|
if (!matchSkylink) {
|
|
return failure(`TXT record with skynet dnslink for ${lookup} contains invalid skylink - ${hint}`);
|
|
}
|
|
|
|
const skylink = matchSkylink[1];
|
|
|
|
cache.set(req.params.name, skylink);
|
|
|
|
console.log(`${req.params.name} => ${skylink}`);
|
|
|
|
return success(skylink);
|
|
});
|
|
});
|
|
|
|
server.listen(port, host, (error) => {
|
|
if (error) throw error;
|
|
|
|
console.info(`Server listening at http://${host}:${port}`);
|
|
});
|