// mDNS 应答器: 从 mdns.json 加载 { "名字.local": "IPv4或IPv6" }
// 热加载: 每次查询入口懒校验(10 秒节流)
var fs = require("fs");
var Dns = require("multicast-dns");
var cfgFile = require("path").join(__dirname, "mdns.json");

var cfg = {};

function respond(record, rinfo) {
    mdns.respond({ answers: [record] });
    console.log("[" + (rinfo ? rinfo.address + ":" + rinfo.port : "?") + "] " + record.type + " " + record.name + " -> " + record.data);
}

// 加载配置
function loadCfg() {
    try { parseJson(fs.readFileSync(cfgFile)); }
    catch(err) { console.log("无法加载配置文件：" + err.message); }
}

// 解析配置
function parseJson(text) {
    var json = JSON.parse(text);
    var next = { A: {}, AAAA: {} };
    for(var x in json) {
        // 判断是否包含 : 来区分是 AAAA 还是 A 记录
        var type = json[x].indexOf(":") < 0 ? "A" : "AAAA";
        var name = x.toLowerCase();
        next[type][name] = { name, type, data: json[x], ttl: 300 };
    }
    cfg.A = next.A;          // 全部解析成功才整体替换, 避免半途出错配置残缺
    cfg.AAAA = next.AAAA;
    cfg.mtime = fs.statSync(cfgFile).mtime.getTime();
}

// 检查配置文件是否更新(自带 10 秒节流)
function recheck() {
    var now = new Date;
    if(now - cfg.lastCheck < 1e4) return;
    cfg.lastCheck = now;
    try {
        var mtime = fs.statSync(cfgFile).mtime.getTime();
        if(mtime != cfg.mtime) { cfg.mtime = mtime; loadCfg(); }
    } catch(err) { console.log("检查配置文件失败：" + err.message); }
}

var mdns = Dns();

mdns.on("query", (q, rinfo) => {
    if(!q.questions) return;
    recheck();  // 放入口而不是 miss 分支: 否则改已有记录的 IP 永远不生效; 自带 10 秒节流不亏
    q.questions.forEach(item => {
        if(!item || !item.name) return;
        var name = String(item.name).toLowerCase();
        var data = (cfg[item.type] || {})[name];
        // 未找到？可能是其他设备上存在, 也可能是配置刚更新过(上面的 recheck 已兜住)
        if(data) respond(data, rinfo);
    });
});

console.log("mDNS responder started, config=" + cfgFile);