117 lines
3.8 KiB
JavaScript
117 lines
3.8 KiB
JavaScript
const url = require("url");
|
|
const express = require("express");
|
|
const proxy = require("express-http-proxy");
|
|
const { NodeClient } = require("hs-client");
|
|
|
|
const host = process.env.HOSTNAME || "0.0.0.0";
|
|
const port = Number(process.env.PORT) || 3100;
|
|
|
|
const hsdNetworkType = process.env.HSD_NETWORK || "regtest";
|
|
const hsdHost = process.env.HSD_HOST || "localhost";
|
|
const hsdPort = Number(process.env.HSD_PORT) || 12037;
|
|
const hsdApiKey = process.env.HSD_API_KEY || "foo";
|
|
|
|
const clientOptions = {
|
|
network: hsdNetworkType,
|
|
host: hsdHost,
|
|
port: hsdPort,
|
|
apiKey: hsdApiKey,
|
|
};
|
|
const client = new NodeClient(clientOptions);
|
|
|
|
// Match both `sia://HASH` and `HASH` links.
|
|
const startsWithSkylinkRegExp = /^(sia:\/\/)?[a-zA-Z0-9_-]{46}/;
|
|
|
|
const getDomainRecords = async (name) => {
|
|
const response = await client.execute("getnameresource", [name]);
|
|
const records = response?.records ?? null;
|
|
|
|
console.log(`${name} => ${JSON.stringify(records)}`);
|
|
|
|
return records;
|
|
};
|
|
|
|
const findSkylinkRecord = (records) => {
|
|
// Find the last one, so people can update their domains in a non-destructive
|
|
// way by simply adding a new link. This will also allow keeping links to
|
|
// older versions for backwards compatibility.
|
|
return records
|
|
?.slice()
|
|
.reverse()
|
|
.find(({ txt }) => txt?.some((entry) => isValidSkylink(entry)));
|
|
};
|
|
|
|
const getSkylinkFromRecord = (record) => {
|
|
return record?.txt?.find((entry) => isValidSkylink(entry));
|
|
};
|
|
|
|
const resolveDomainHandler = async (req, res) => {
|
|
try {
|
|
const records = await getDomainRecords(req.params.name);
|
|
if (!records) return res.status(404).send(`No records found for ${req.params.name}`);
|
|
|
|
const record = findSkylinkRecord(records);
|
|
if (!record) throw new Error(`No skylink found in dns records of ${req.params.name}`);
|
|
|
|
const skylink = getSkylinkFromRecord(record);
|
|
return res.json({ skylink });
|
|
} catch (error) {
|
|
res.status(500).send(`Handshake error: ${error.message}`);
|
|
}
|
|
};
|
|
|
|
// Checks if the given string is a valid Sia link.
|
|
function isValidSkylink(link) {
|
|
if (!link || link.length === 0) {
|
|
return false;
|
|
}
|
|
return Boolean(link.match(startsWithSkylinkRegExp));
|
|
}
|
|
|
|
const server = express();
|
|
|
|
server.use(
|
|
"/hns/:name",
|
|
proxy("nginx", {
|
|
// eslint-disable-next-line no-unused-vars
|
|
userResHeaderDecorator(headers, userReq, userRes, proxyReq, proxyRes) {
|
|
if (headers.location && headers.location.match(startsWithSkylinkRegExp)) {
|
|
headers.location = headers.location.replace(
|
|
startsWithSkylinkRegExp,
|
|
`/hns/${userReq.params.name.replace("sia://", "")}`
|
|
);
|
|
}
|
|
|
|
return headers;
|
|
},
|
|
proxyReqPathResolver: async (req) => {
|
|
const records = await getDomainRecords(req.params.name);
|
|
if (!records) throw new Error(`No records found for ${req.params.name}`);
|
|
|
|
const record = findSkylinkRecord(records);
|
|
if (!record) throw new Error(`No skylink found in dns records of ${req.params.name}`);
|
|
|
|
const skylink = getSkylinkFromRecord(record).replace("sia://", ""); // get skylink and strip sia:// prefix
|
|
const basepath = url.resolve("/", skylink); // make the url absolute
|
|
const subpath = req.url.slice(1); // drop the leading slash
|
|
|
|
// if the record is just a raw skylink, replace baseUrl with /skylink
|
|
if (skylink.length === 46) {
|
|
return req.originalUrl.replace(req.baseUrl, basepath);
|
|
}
|
|
|
|
// if the record contains more than a skylink then it needs to be resolved
|
|
return url.resolve(basepath, subpath);
|
|
},
|
|
})
|
|
);
|
|
|
|
server.get("/hnsres/:name", resolveDomainHandler);
|
|
|
|
server.listen(port, host, (error) => {
|
|
if (error) throw error;
|
|
|
|
console.info(`API will look for HSD Server at ${hsdHost}:${hsdPort}, "${hsdNetworkType}" network.`);
|
|
console.info(`Server listening at http://${host}:${port}`);
|
|
});
|