add getSegmentInfo with tests

src/app.ts

@@ -28,7 +28,7 @@ import {corsMiddleware} from './middleware/cors';
 import {apiCspMiddleware} from './middleware/apiCsp';
 import {rateLimitMiddleware} from './middleware/requestRateLimit';
 import dumpDatabase, {redirectLink} from './routes/dumpDatabase';
-
+import {endpoint as getSegmentInfo} from './routes/getSegmentInfo';
 
 export function createServer(callback: () => void) {
     // Create a service (the app object is just a callback).
@@ -133,6 +133,9 @@ function setupRoutes(app: Express) {
     //get if user is a vip
     app.post('/api/segmentShift', postSegmentShift);
 
+    //get segment info
+    app.get('/api/segmentInfo', getSegmentInfo);
+
     if (config.postgres) {
         app.get('/database', (req, res) => dumpDatabase(req, res, true));
         app.get('/database.json', (req, res) => dumpDatabase(req, res, false));

src/routes/getSegmentInfo.ts

@@ -0,0 +1,75 @@
+import { Request, Response } from 'express';
+import { db } from '../databases/databases';
+import { DBSegment, SegmentUUID } from "../types/segments.model";
+
+const isValidSegmentUUID = (str: string): Boolean => /^([a-f0-9]{64}|[a-f0-9]{8}\-[a-f0-9]{4}\-[a-f0-9]{4}\-[a-f0-9]{4}\-[a-f0-9]{12})/.test(str)
+
+async function getSegmentFromDBByUUID(UUID: SegmentUUID): Promise<DBSegment> {
+    try {
+        return await db.prepare('get', `SELECT * FROM "sponsorTimes" WHERE "UUID" = ?`, [UUID])
+    } catch (err) {
+        return null
+    }
+}
+
+async function getSegmentsByUUID(UUIDs: SegmentUUID[]): Promise<DBSegment[]> {
+    const DBSegments: DBSegment[] = [];
+    for (let UUID of UUIDs) {
+        // if UUID is invalid, skip
+        if (!isValidSegmentUUID(UUID)) {
+            continue;
+        }
+        DBSegments.push(await getSegmentFromDBByUUID(UUID as SegmentUUID));
+    }
+    return DBSegments
+}
+
+async function handleGetSegmentInfo(req: Request, res: Response) {
+    // If using params instead of JSON, only one UUID can be pulled
+    let UUIDs = req.query.UUIDs
+    ? JSON.parse(req.query.UUIDs as string)
+    : req.query.UUID
+        ? [req.query.UUID]
+        : null
+    // deduplicate with set
+    UUIDs = [ ...new Set(UUIDs)];
+    // if more than 10 entries, slice
+    if (UUIDs.length > 10) UUIDs = UUIDs.slice(0, 10);
+    if (!Array.isArray(UUIDs) || !UUIDs) {
+        res.status(400).send("UUIDs parameter does not match format requirements.");
+        return false;
+    }
+    const DBSegments = await getSegmentsByUUID(UUIDs);
+    if (DBSegments === null || DBSegments === undefined) {
+        res.sendStatus(400);
+        return false;
+    }
+    if (DBSegments.length === 0) {
+        res.sendStatus(404);
+        return false;
+    }
+    return DBSegments;
+}
+
+async function endpoint(req: Request, res: Response): Promise<void> {
+    try {
+        const DBSegments = await handleGetSegmentInfo(req, res);
+
+        // If false, res.send has already been called
+        if (DBSegments) {
+            //send result
+            res.send(DBSegments);
+        }
+    } catch (err) {
+        if (err instanceof SyntaxError) { // catch JSON.parse error
+            res.status(400).send("UUIDs parameter does not match format requirements.");
+        } else res.status(500).send();
+    }
+}
+
+export {
+    getSegmentFromDBByUUID,
+    getSegmentsByUUID,
+    handleGetSegmentInfo,
+    endpoint
+};