import {Logger} from '../utils/logger'; import {getHash} from '../utils/getHash'; import {isUserVIP} from '../utils/isUserVIP'; import {db} from '../databases/databases'; import {Request, Response} from 'express'; export async function postLockCategories(req: Request, res: Response): Promise { // Collect user input data const videoID = req.body.videoID; let userID = req.body.userID; const categories = req.body.categories; // Check input data is valid if (!videoID || !userID || !categories || !Array.isArray(categories) || categories.length === 0 ) { res.status(400).json({ message: 'Bad Format', }); return; } // Check if user is VIP userID = getHash(userID); const userIsVIP = await isUserVIP(userID); if (!userIsVIP) { res.status(403).json({ message: 'Must be a VIP to mark videos.', }); return; } // Get existing lock categories markers let noCategoryList = await db.prepare('all', 'SELECT "category" from "lockCategories" where "videoID" = ?', [videoID]); if (!noCategoryList || noCategoryList.length === 0) { noCategoryList = []; } else { noCategoryList = noCategoryList.map((obj: any) => { return obj.category; }); } // get user categories not already submitted that match accepted format let categoriesToMark = categories.filter((category) => { return !!category.match(/^[_a-zA-Z]+$/); }).filter((category) => { return noCategoryList.indexOf(category) === -1; }); // remove any duplicates categoriesToMark = categoriesToMark.filter((category, index) => { return categoriesToMark.indexOf(category) === index; }); // create database entry for (const category of categoriesToMark) { try { await db.prepare('run', `INSERT INTO "lockCategories" ("videoID", "userID", "category") VALUES(?, ?, ?)`, [videoID, userID, category]); } catch (err) { Logger.error("Error submitting 'lockCategories' marker for category '" + category + "' for video '" + videoID + "'"); Logger.error(err); res.status(500).json({ message: "Internal Server Error: Could not write marker to the database.", }); } } res.status(200).json({ submitted: categoriesToMark, }); }