import fp from 'fastify-plugin'
import fs from 'fs';
const failureFile = fs.readFileSync("Checkfailed.html", "utf8");
const LICENSE_SERVER_URL = "https://license.mercurywork.shop/validate?license=";
const whiteListedDomain = ["nebulaproxy.io"];
async function licenseCheck(req, pass) {
    try {
        const resp = await fetch(`${LICENSE_SERVER_URL}${pass}&host=${req.headers.host}`);
        const data = await resp.json();
        if (data.status === "License valid") {
            return true;
        }
        else {
            return false;
        }
    } catch { return false; }
}
const plugin = (fastify, opts, done) => {
    fastify.addHook('onRequest', function (req, reply, next) {
        if (req.cookies.authcheck === 'true' || whiteListedDomain.includes(req.headers.host)) {
            return next();
        }
        const authHeader = req.headers.authorization;
        if (req.cookies.refreshcheck != "true") {
            reply 
                .setCookie('refreshcheck', 'true', { maxAge: 1000 })
                .type("text/html")
                .send(failureFile);
            return;
        }
        if (!authHeader) {
            reply 
                .code(401)
                .header('WWW-Authenticate', 'Basic')
                .type("text/html")
                .send(failureFile);
            return;
        }
        const auth = Buffer.from(authHeader.split(" ")[1], "base64").toString().split(":");
        const user = auth[0];
        const pass = auth[1];
        licenseCheck(req, pass).then((data) => {
            if (!data) {
                reply.status(401).header('WWW-Authenticate', 'Basic').type("text/html").send(failureFile);
                return;
            }
            else {
                reply.setCookie('authcheck', 'true').type("text/html").send('<script>window.location.href = window.location.href</script>')
                return;
            }
        });
    });
    done();
};

const masqr = fp(plugin, {
    fastify: '4.x',
    name: 'masqr'
});

export default masqr;