database_editor/app/routes/private_routes.js

var db = require("../database/sqlite.database.js");
const logger = require("../utils/logger.util.js");
const Database = require("better-sqlite3");
const path = require("path");
const fs = require("fs");
const app_conf = require("../config/app.conf.js");
const db_conf = require("../config/db.conf.js");
const messages = require("../config/messages.conf.js");

const express = require('express');
const router = express.Router();

function errorPageRenderer(res, emoji, message) {
    res.render("error", {
        page_title: app_conf.name,
        css_path: path.join(app_conf.paths.css.web, "stylesheet.css"),
        js_path: path.join(app_conf.paths.js.web, "error.js"),
        error_emoji: emoji,
        error_message: message
    })
}

router.delete("/remove", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    if (req.query["element"]) {
        try {
            const remove = db.getDB().prepare(`DELETE FROM ${db_conf.table_name} WHERE Name='${req.query["element"]}';`);
            if (remove.run()["changes"]) {
                res.status(200).send();
            } else {
                res.status(404).send();
            }
        } catch (error) {
            res.status(500).send();
            logger.error(error.message);
        }
    } else {
        res.status(400).send();
    }
});

router.post("/upload", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    if (!req.files || !req.files.file) {
        res.status(422).send();
    } else {
        const uploadedFile = req.files.file;

        if (db.getDB()) {
            db.getDB().close();
        }
        fs.copyFile(uploadedFile.tempFilePath, db_conf.path, (error) => {
            if (error) {
                logger.error(error);
            } else {
                try {
                    db.setDB(new Database(db_conf.path, { fileMustExist: true }));
                } catch (error) {
                    logger.error(error);
                }
            }
        });
    }
    res.status(201).send();
});

router.get("/download", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    res.download(db_conf.path);
});

router.post("/add", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    if (req.query["element"] && req.body) {
        try {
            let keys = Object.keys(req.body);
            let values = `'${req.query["element"]}',`;

            for (const k of keys) {
                if (req.body[k] === "null") {
                    values += `NULL,`
                } else {
                    values += `'${req.body[k]}',`;
                }
            }

            if (values) {
                values = values.slice(0, -1);
                const update = db.getDB().prepare(`INSERT INTO ${db_conf.table_name} VALUES (${values});`);
                update.run();
                res.status(200).send();
            } else {
                res.status(400).send();
            }
        } catch (error) {
            res.status(500).send();
            logger.error(error.message);
        }
    } else {
        res.status(400).send();
    }
});

router.post("/edit", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    if (req.query["element"] && req.body) {
        try {
            let keys = Object.keys(req.body);
            let values = "";

            for (const k of keys) {
                if (req.body[k] === "null") {
                    values += `${k}=NULL,`
                } else {
                    values += `${k}='${req.body[k]}',`;
                }
            }

            if (values) {
                values = values.slice(0, -1);
                const update = db.getDB().prepare(`UPDATE ${db_conf.table_name} SET ${values} WHERE Name='${req.query["element"]}'`);
                update.run();
                res.status(200).send();
            } else {
                res.status(400).send();
            }
        } catch (error) {
            res.status(500).send();
            logger.error(error.message);
        }
    } else {
        res.status(400).send();
    }
});

router.get("/add", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    try {
        const select_get = db.getDB().prepare(`SELECT * FROM ${db_conf.table_name};`);
        const row = select_get.get();

        const sql_header = Object.keys(row);
        var table_header = `<th>${sql_header[0]}</th>\n`;
        sql_header.splice(0, 1);

        if (row) {
            var column_options = {};
            for (const key of sql_header) {
                table_header += `<th>${key}</th>\n`;

                const select_distinct = db.getDB().prepare(`
                    SELECT DISTINCT ${key}
                    FROM ${db_conf.table_name};
                    `);
                values = select_distinct.all();

                for (v of values) {
                    if (column_options[key] == null) {
                        column_options[key] = Object.values(v);
                    } else {
                        column_options[key] = column_options[key].concat(Object.values(v));
                    }
                }
            }

            var table_data = "";
            const sql_row_values = Object.values(row);

            table_data += `<tr><td><input type="text" placeholder="Sensor name"></td>\n`;
            for (let i = 1; i < sql_row_values.length; i++) {
                table_data += `<td><select></select></td>\n`;
            }

            table_data += "</tr>";
            res.render("add", {
                page_title: app_conf.name,
                css_path: path.join(app_conf.paths.css.web, "stylesheet.css"),
                js_path: path.join(app_conf.paths.js.web, "add.js"),
                column_options: JSON.stringify(column_options),
                table_header: table_header,
                table_data: table_data
            });
        } else {
            errorPageRenderer(res, "&#10067;", messages.empty_db);
        }
    } catch (error) {
        errorPageRenderer(res, "&#127755;&#128165;", messages.internal_server_error);
        logger.error(error.message);
    }
});

router.get("/edit", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    if (req.query["element"]) {
        try {
            const select_get = db.getDB().prepare(`SELECT * FROM ${db_conf.table_name} WHERE Name='${req.query["element"]}';`);
            const row = select_get.get();

            if (row) {
                const sql_header = Object.keys(row);
                var table_header = `<th>${sql_header[0]}</th>\n`;
                sql_header.splice(0, 1);

                var column_options = {};
                for (const key of sql_header) {
                    table_header += `<th>${key}</th>\n`;

                    const select_distinct = db.getDB().prepare(`
                    SELECT DISTINCT ${key}
                    FROM ${db_conf.table_name}
                    WHERE ${key} IS NOT NULL
                    AND ${key} NOT IN (
                    SELECT ${key}
                    FROM ${db_conf.table_name}
                    WHERE Name='${req.query["element"]}'
                    AND ${key} IS NOT NULL);
                    `);
                    values = select_distinct.all();

                    for (v of values) {
                        if (column_options[key] == null) {
                            column_options[key] = Object.values(v);
                        } else {
                            column_options[key] = column_options[key].concat(Object.values(v));
                        }
                    }
                }

                var table_data = "";
                const sql_row_values = Object.values(row);
                table_data += `<tr><td>${sql_row_values[0]}</td>\n`;
                sql_row_values.splice(0, 1);

                for (const v of sql_row_values) {
                    table_data += `<td><select><option selected="selected">${v}</option></select></td>\n`;
                }

                table_data += "</tr>";

                res.render("edit", {
                    page_title: app_conf.name,
                    css_path: path.join(app_conf.paths.css.web, "stylesheet.css"),
                    js_path: path.join(app_conf.paths.js.web, "edit.js"),
                    column_options: JSON.stringify(column_options),
                    table_header: table_header,
                    table_data: table_data
                });
            } else {
                errorPageRenderer(res, "&#128269;&#10060;", messages.not_found);
            }
        } catch (error) {
            errorPageRenderer(res, "&#127755;&#128165;", messages.internal_server_error);
            logger.error(error.message);
        }
    } else {
        errorPageRenderer(res, "&#9995;&#9940;", messages.bad_request);
    }
});

router.get("/", async (req, res) => {
    logger.info(`${req.method}: "${req.url}" => ${req.get("User-Agent")}`);

    try {
        const select = db.getDB().prepare(`SELECT * FROM ${db_conf.table_name}`);
        const rows = select.all();

        if (rows[0]) {
            const sql_header = Object.keys(rows[0]);

            var table_header = "";
            for (const key of sql_header) {
                table_header += `<th>${key}</th>\n`;
            }

            var table_data = "";
            for (const row of rows) {
                const sql_row_values = Object.values(row);
                table_data += "<tr>";

                for (const v of sql_row_values) {
                    table_data += `<td>${v}</td>\n`;
                }

                table_data += "</tr>";
            }

            res.render("index", {
                page_title: app_conf.name,
                css_path: path.join(app_conf.paths.css.web, "stylesheet.css"),
                js_path: path.join(app_conf.paths.js.web, "index.js"),
                table_header: table_header,
                table_data: table_data
            });
        } else {
            errorPageRenderer(res, "&#10067;", messages.empty_db);
        }
    } catch (error) {
        errorPageRenderer(res, "&#127755;&#128165;", messages.internal_server_error);
        logger.error(error.message);
    }
});

module.exports = router;