You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
keys/routes/ping.js

50 lines
1.4 KiB
JavaScript

let express = require("express");
let ipaddr = require("ipaddr.js");
let db = require("../libs/Database").instance();
let router = express.Router();
const IP_TYPES = ["ipv4", "ipv6"];
function combineToList(a, b) {
return a + "\n" + b;
}
function toList(array) {
return array.reduce(combineToList, "").trim();
}
function testIP(req, res, next) {
let addr = ipaddr.parse(req.ip);
if (!IP_TYPES.includes(addr.kind())) return res.status(400).send("Invalid ip address");
req.params.addr = addr;
next();
}
router.use((req, res, next) => {
res.type('text/plain');
next();
});
router.get("/", testIP, (req, res) => {
res.send(req.params.addr.toString());
});
router.get("/:fqdn", (req, res) => {
db.q.select(IP_TYPES).from("addresses").filter("fqdn", req.params.fqdn).first((e, d) => {
if (e) return res.status(400).send("Error");
if (!d) return res.status(404).send("Unknown FQDN");
res.send(d.ipv4 ? (d.ipv6 ? d.ipv4 + "\n" + d.ipv6 : d.ipv4) : (d.ipv6 ? d.ipv6 : ""));
});
});
router.get("/:fqdn/:code", testIP, (req, res) => {
let addr = req.params.addr;
db.query("UPDATE addresses SET " + addr.kind() + " = ?, " + addr.kind() + "LastPing = CURRENT_TIMESTAMP WHERE fqdn = ? AND code = ?", [addr.toString(), req.params.fqdn, req.params.code], (e, d) => {
if (e) return res.status(400).send("Error");
if (d.affectedRows <= 0) return res.status(400).send("Invalid token");
res.send(addr.toString());
});
});
module.exports = router;