Rename noSegments to lockCategories

src/routes/postLockCategories.ts

@@ -0,0 +1,75 @@
+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) {
+    // Collect user input data
+    let videoID = req.body.videoID;
+    let userID = req.body.userID;
+    let 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);
+    let 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,
+    });
+}