poll.horse/src/frontend.ts

"use strict";
import fs from "fs-extra";
import { resolve } from "path";
import { Transform as TransformStream, Stream } from "stream";
import { Router, Request, Response } from "express";
import fetch from 'node-fetch';
import { program } from "commander";
import { FrontendPoll as Poll, PollResult } from "./Poll";
import { MAX_CHARACTER_LENGTH, MAX_POLL_OPTIONS } from "./Config";

const RenderBuffer = new WeakMap();
const RenderReplacements = new WeakMap();

interface NodeError extends Error {
    code?: string;
}

class RenderTransform extends TransformStream {
    constructor(replacements = {}) {
        super();
        RenderReplacements.set(this, replacements);
        RenderBuffer.set(this, "");
    }

    _transform(chunk: any, encoding: string, callback: () => void) {
        const r = RenderReplacements.get(this);
        let c = RenderBuffer.get(this) + chunk.toString();
        Object.entries(r).forEach(([key, value]) => c = c.replace(new RegExp("{{ ?" + key + " ?}}", "ig"), value));

        if (c.match(/\{\{.*?(?<!\}\})$/)) RenderBuffer.set(this, c);
        else {
            RenderBuffer.set(this, "");
            this.push(c.replace(/\\\{/g, "{").replace(/\\\}/g, "}"));
        }
        callback();
    }

    _flush(callback: () => void) {
        const c = RenderBuffer.get(this);
        if (c) this.push(c.replace(/\\\{/g, "{").replace(/\\\}/g, "}"));
        callback();
    }
}

class MinificationTransform extends RenderTransform {
    constructor(replacements = {}) {
        super(replacements);
    }

    _transform(chunk: any, encoding: string, callback: () => void) {
        super._transform(chunk.toString().replace(/\s{2,}/g, ""), encoding, callback);
    }
}

const defaultReplacements = {
    "TITLE": "Poll Horse",
    "DEVELOPER_CONTACT_INFO": "developer@poll.horse",
    "FOOTER_COPYRIGHT": `© ${ (new Date()).getFullYear() } Based Department`,
    "FOOTER_LINKS": `
        <li><a href="https://github.com/Wolvan/poll.horse" target="_blank">Github Repo</a></li>
    `
};
class Defaults2RenderTransform extends MinificationTransform {
    constructor(replacements = {}) {
        super(Object.assign({}, defaultReplacements, replacements));
    }
}
async function displayPage(req: Request, res: Response, htmlFilename: string, replacements = {}, statusCode = 200) {
    const promisifyStream = (fn: Stream) => new Promise(pRes => fn.on("finish", pRes));
    
    try {
        await fs.stat(resolve(__dirname, "../frontend/html", htmlFilename));
        await promisifyStream(
            fs.createReadStream(resolve(__dirname, "../frontend/html", htmlFilename))
            .pipe(new Defaults2RenderTransform(replacements))
            .pipe(res.status(statusCode))
        );
    } catch (error) {
        if (error instanceof Error)
            if ((error as NodeError).code === "ENOENT") {
                await promisifyStream(
                    fs.createReadStream(resolve(__dirname, "../frontend/errors/404.html"))
                    .pipe(new Defaults2RenderTransform())
                    .pipe(res.status(404))
                );
            } else {
                await promisifyStream( 
                    fs.createReadStream(resolve(__dirname, "../frontend/errors/500.html")).pipe(new Defaults2RenderTransform({
                        "JS_ERROR_STACK": (error as NodeError).stack,
                        "HTTP_ERROR_CODE": 500
                    }))
                    .pipe(res.status(500))
                );
            }
    }
}

export default function init(router: Router): void {
    router.get("/:id/r", async (req, res) => {
        const id = req.params.id;
        try {
            const poll: PollResult = await fetch(
                (program.opts().backendBaseUrl || "http://localhost:" + program.opts().port) + "/_backend/poll-result/" + id
            ).then(r => r.json()) as PollResult;
            if (!poll || poll.error) return res.redirect("/");
            const totalVotes = Object.values(poll.votes).reduce((acc, cur) => acc + cur, 0);
            const pollOptionsDivs = Object.entries(poll.votes).map(([option, votes]) => `
                <div class="poll-option" option="${ option }">
                    <div class="poll-option-info">
                        <div class="poll-option-text">${ option }</div><div class="poll-option-votes">${ votes }</div>
                    </div>
                    <div class="progress">
                        <div class="poll-bar">
                            <div class="poll-bar-fill" style="width: ${ (votes / totalVotes) * 100 }%"></div>
                        </div><div class="poll-bar-text">${ Math.round((votes / totalVotes) * 100) }</div>
                    </div>
                </div>
            `).join("");

            await displayPage(req, res, "result.html", {
                "POLL_ID": id,
                "POLL_TITLE": poll.title,
                "POLL_OPTION_DIVS": pollOptionsDivs,
                "POLL_VOTES_TOTAL": totalVotes,
                "BACKEND_BASE_PATH": (program.opts().backendBaseUrl || ""),
                "POLL_OPTION_VOTES": Buffer.from(JSON.stringify(Object.entries(poll.votes))).toString("base64")
            });
        } catch (error) {
            console.log(error);
            res.redirect(`/`);
        }
    });
    router.get("/:id", async (req, res) => {
        const id = req.params.id;
        const options = (typeof req.query.options === "string" ? req.query.options.split("\uFFFE") : []).filter(i => i);
        try {
            const poll: Poll = await fetch(
                (program.opts().backendBaseUrl || "http://localhost:" + program.opts().port) + "/_backend/poll/" + id
            ).then(r => r.json()) as Poll;
            if (!poll || poll.error) return res.redirect("/");

            const pollOptions = poll.options.map(option =>
                `<div class="poll-option">
                    <div class="input-container">
                        <input type="${poll.multiSelect ? "checkbox" : "radio"}" name="poll-option" value="${option}"" ${options.includes(option) ? "checked" : ""}/><div class="checkmark"><svg class="checkmark" xmlns="http://www.w3.org/2000/svg" viewBox="0 0 40 40"><path class="checkmark__check" fill="none" d="M14.1 27.2l7.1 7.2 16.7-16.8"/></svg></div>
                    </div><div class="text">${option}</div>
                </div>`
            ).join("");

            await displayPage(req, res, "poll.html", {
                "POLL_ID": poll.id,
                "POLL_TITLE": poll.title,
                "POLL_OPTION_DIVS": pollOptions,
                "BACKEND_BASE_PATH": (program.opts().backendBaseUrl || ""),
                "FORM_SUBMISSION_ERROR": req.query.error,
                "FORM_SUBMISSION_ERROR_SHOWN_CLASS": req.query.error ? "error-visible" : "",
            });
        } catch (error) {
            console.log(error);
            res.redirect(`/`);
        }
    });

    router.get("/", (req, res) => {
        const options = (typeof req.query.options === "string" ? req.query.options.split("\uFFFE") : [])
            .filter(i => i)
            .concat(Array(3).fill(""))
            .slice(0, 3);
        const pollOptionDivs = options.map(option => `
            <div class="poll-option">
                <input type="text" name="poll-option" maxlength="${MAX_CHARACTER_LENGTH}" placeholder="Enter your option here" value="${option}">
            </div>
        `).join("");

        displayPage(req, res, "index.html", {
            "BACKEND_BASE_PATH": (program.opts().backendBaseUrl || ""),
            "FORM_SUBMISSION_ERROR": req.query.error,
            "FORM_SUBMISSION_ERROR_SHOWN_CLASS": req.query.error ? "error-visible" : "",
            "FORM_TITLE": req.query.title || "",
            "FORM_DUPECHECK_IP": req.query.dupecheck === "ip" ? "selected" : "",
            "FORM_DUPECHECK_COOKIE": req.query.dupecheck === "cookie" ? "selected" : "",
            "FORM_DUPECHECK_NONE": req.query.dupecheck === "none" ? "selected" : "",
            "FORM_MULTI_SELECT": req.query.multiselect === "true" ? "checked" : "",  
            "FORM_CAPTCHA": req.query.captcha === "true" ? "checked" : "",
            "FORM_OPTION_DIVS": pollOptionDivs,
            "MAX_POLL_OPTIONS": MAX_POLL_OPTIONS,
            "MAX_CHARACTER_LENGTH": MAX_CHARACTER_LENGTH
        });
    });
}