DweebUI/controllers/dashboard.js

const { readFileSync, writeFileSync, appendFileSync, readdirSync } = require('fs');
const { execSync } = require("child_process");
const { siteCard } = require('../components/siteCard');
const { containerExec } = require('../functions/system')



exports.Dashboard = async function (req, res) {

    let caddy = 'd-none';

    if (process.env.Proxy_Manager == 'enabled') {
        caddy = '';
    }

    // Render the home page
    res.render("pages/dashboard", {
        name: req.session.user,
        role: req.session.role,
        avatar: req.session.avatar,
        isLoggedIn: true,
        site_list: req.app.locals.site_list,
        caddy: caddy
    });

}



exports.AddSite = async function (req, res) {

    let { domain, type, host, port } = req.body;

    if ( domain && type && host && port) {


        let { domain, type, host, port } = req.body;

        // build caddyfile
        let caddyfile = `${domain} {`
        caddyfile += `\n\t${type} ${host}:${port}`
        caddyfile += `\n\theader {`
        caddyfile += `\n\t\tStrict-Transport-Security "max-age=31536000; includeSubDomains; preload"`
        caddyfile += `\n\t}`
        caddyfile += `\n}`

        
        // save caddyfile
        writeFileSync(`./caddyfiles/sites/${domain}.Caddyfile`, caddyfile, function (err) { console.log(err) });


        // format caddyfile
        let format = {
            container: 'DweebProxy',
            command: `caddy fmt --overwrite /etc/caddy/sites/${domain}.Caddyfile`
        }
        await containerExec(format, function(err, data) {
            if (err) {
                console.error(err);
                return;
            }
            console.log(`Formatted ${domain}.Caddyfile`);
        });
        
        ///////////////// convert caddyfile to json
        let convert = {
            container: 'DweebProxy',
            command: `caddy adapt --config /etc/caddy/sites/${domain}.Caddyfile --pretty >> /etc/caddy/sites/${domain}.json`
        }
        await containerExec(convert, function(err, data) {
            if (err) {
                console.error(err);
                return;
            }
            console.log(`Converted ${domain}.Caddyfile to JSON`);
        });

        ////////////// reload caddy
        let reload = {
            container: 'DweebProxy',
            command: `caddy reload --config /etc/caddy/Caddyfile`
        }
        await containerExec(reload, function(err, data) {
            if (err) {
                console.error(err);
                return;
            }
            console.log(`Reloaded Caddy Config`);
        });

        let site = siteCard(type, domain, host, port, 0);

        req.app.locals.site_list += site;


        res.redirect("/");
    } else {
        // Redirect
        console.log('missing info')
        res.redirect("/");
    }
}


exports.RemoveSite = async function (req, res) {

    for (const [key, value] of Object.entries(req.body)) {

        execSync(`rm ./caddyfiles/sites/${value}.Caddyfile`, (err, stdout, stderr) => {
            if (err) { console.error(`error: ${err.message}`); return; }
            if (stderr) { console.error(`stderr: ${stderr}`); return; }
            console.log(`removed ${value}.Caddyfile`);
        });

    }

    let reload = {
        container: 'DweebProxy',
        command: `caddy reload --config /etc/caddy/Caddyfile`
    }
    await containerExec(reload);

    
    console.log('Removed Site(s)')

    res.redirect("/refreshsites");

    
}


exports.RefreshSites = async function (req, res) {

    let domain, type, host, port;
    let id = 1;

    // Clear site_list.ejs
    req.app.locals.site_list = "";
    

    // check if ./caddyfiles/sites contains any .json files, then delete them
    try {
        let files = readdirSync('./caddyfiles/sites/');
        files.forEach(file => {
            if (file.includes(".json")) {
                execSync(`rm ./caddyfiles/sites/${file}`, (err, stdout, stderr) => {
                    if (err) { console.error(`error: ${err.message}`); return; }
                    if (stderr) { console.error(`stderr: ${stderr}`); return; }
                    console.log(`removed ${file}`);
                });
            }
        });
    } catch (error) { console.log("No .json files to delete") }

    // get list of Caddyfiles
    let sites = readdirSync('./caddyfiles/sites/');


    sites.forEach(site_name => {
        // convert the caddyfile of each site to json
        let convert = {
            container: 'DweebProxy',
            command: `caddy adapt --config ./caddyfiles/sites/${site_name} --pretty >> ./caddyfiles/sites/${site_name}.json`
        }
        containerExec(convert);

        try {
        // read the json file
        let site_file = readFileSync(`./caddyfiles/sites/${site_name}.json`, 'utf8');
        // fix whitespace and parse the json file
        site_file = site_file.replace(/        /g, "  ");
        site_file = JSON.parse(site_file);
        } catch (error) { console.log("No .json file to read") }


        // get the domain, type, host, and port from the json file
        try { domain = site_file.apps.http.servers.srv0.routes[0].match[0].host[0] } catch (error) { console.log("No Domain") }
        try { type = site_file.apps.http.servers.srv0.routes[0].handle[0].routes[0].handle[1].handler } catch (error) { console.log("No Type") }
        try { host = site_file.apps.http.servers.srv0.routes[0].handle[0].routes[0].handle[1].upstreams[0].dial.split(":")[0] } catch (error) { console.log("Not Localhost") }
        try { port = site_file.apps.http.servers.srv0.routes[0].handle[0].routes[0].handle[1].upstreams[0].dial.split(":")[1] } catch (error) { console.log("No Port") }

        // build the site card
        let site = siteCard(type, domain, host, port, id);

        // append the site card to site_list
        req.app.locals.site_list += site;
        
        id++;
    });
    

    res.redirect("/");
 
}



exports.DisableSite = async function (req, res) {
        
    console.log(req.body)
    console.log('Disable Site')

    res.redirect("/");

}


exports.EnableSite = async function (req, res) {
   
    console.log(req.body)
    console.log('Enable Site')

    res.redirect("/");

}