mipilin/routes/util.ts

import { NodePgDatabase } from "drizzle-orm/node-postgres";
import type { Request, Response } from "express";
import { inviteCodes, updates } from "../db/schema.js";
import { count, desc, eq } from "drizzle-orm";
import fs from "node:fs/promises";

export enum UserStatus {
    MODERATOR = 0b001,
    BANNED = 0b010,
    TRUSTED = 0b100
}

const nonceChars =
    "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz1234567890-_";
let nonce: string;
export function setNonce() {
    nonce = "";
    for (let i = 0; i < 32; i++)
        nonce += nonceChars[Math.floor(Math.random() * nonceChars.length)];
    return nonce;
}
export function getNonce() {
    if (!nonce) throw new Error("Nonce doesn't exist");
    return nonce;
}

let moods: string[], moodsSorted: string[];
export async function getMoods() {
    if (!moods)
        moods = (await fs.readFile("./static/moods.txt"))
            .toString("utf-8")
            .split(";");
    if (!moodsSorted) moodsSorted = Array.from(moods).sort();
    return { moods, moodsSorted };
}

export async function render(
    db: NodePgDatabase,
    page: string,
    title: string,
    res: Response,
    req: Request,
    stuff?: Object
) {
    //? maybe you should cache this and save the current mood to the session until it's changed
    const { moods } = await getMoods();
    let currentMood: string;
    if (req.session["loggedIn"]) {
        const update = (
            await db
                .select({ mood: updates.mood })
                .from(updates)
                .where(eq(updates.user, req.session["uid"]))
                .orderBy(desc(updates.date))
                .limit(1)
        )[0];
        currentMood = moods[update?.mood];
    }

    const o = {
        title,
        session: req.session,
        flashes: req.flash(),
        moods,
        currentMood,
        nonce
    };
    res.render(page, { ...o, ...stuff });
}
export async function render404(
    db: NodePgDatabase,
    res: Response,
    req: Request
) {
    res.statusCode = 404;
    render(db, "404", "not found", res, req);
}

const inviteCodeChars = "abcdefghijklmnopqrstuvwxyz0123456789";
export async function createInviteCode(
    db: NodePgDatabase,
    user: number,
    expires: Date,
    confers = 0
) {
    let existingToken = 1,
        token: string;
    while (existingToken) {
        token = user.toString().padStart(4, "0") + "-";
        for (let i = 0; i < 17; i++) {
            if ((i + 1) % 6 === 0) {
                token += "-";
                continue;
            }
            token +=
                inviteCodeChars[
                    Math.floor(Math.random() * inviteCodeChars.length)
                ];
        }
        existingToken = (
            await db
                .select({ value: count() })
                .from(inviteCodes)
                .where(eq(inviteCodes.token, token))
        )[0].value;
    }

    //@ts-expect-error
    await db.insert(inviteCodes).values({
        token,
        user: user || undefined,
        granted: new Date(Date.now()),
        expires,
        confers
    });
    return token;
}