SponsorBlockServer/src/routes/deleteLockCategories.ts

import { Request, Response } from "express";
import { isUserVIP } from "../utils/isUserVIP";
import { getHash } from "../utils/getHash";
import { db } from "../databases/databases";
import { Category, VideoID } from "../types/segments.model";
import { UserID } from "../types/user.model";

export async function deleteLockCategoriesEndpoint(req: Request, res: Response): Promise<Response> {
    // Collect user input data
    const videoID = req.body.videoID as VideoID;
    const userID = req.body.userID as UserID;
    const categories = req.body.categories as Category[];

    // Check input data is valid
    if (!videoID
        || !userID
        || !categories
        || !Array.isArray(categories)
        || categories.length === 0
    ) {
        return res.status(400).json({
            message: "Bad Format",
        });
    }

    // Check if user is VIP
    const hashedUserID = getHash(userID);
    const userIsVIP = await isUserVIP(hashedUserID);

    if (!userIsVIP) {
        return res.status(403).json({
            message: "Must be a VIP to mark videos.",
        });
    }

    await deleteLockCategories(videoID, categories);

    return res.status(200).json({ message: `Removed lock categories entrys for video ${videoID}` });
}

/**
 *
 * @param videoID
 * @param categories If null, will remove all
 */
export async function deleteLockCategories(videoID: VideoID, categories: Category[]): Promise<void> {
    const entries = (await db.prepare("all", 'SELECT * FROM "lockCategories" WHERE "videoID" = ?', [videoID])).filter((entry: any) => {
        return categories === null || categories.indexOf(entry.category) !== -1;
    });

    for (const entry of entries) {
        await db.prepare("run", 'DELETE FROM "lockCategories" WHERE "videoID" = ? AND "category" = ?', [videoID, entry.category]);
    }
}