Add some more tests and fix existing broken ones

server/scripts/set-indexes.ts

@@ -1,11 +1,6 @@
-import monk from "monk";
+/* eslint-disable no-await-in-loop */
 import { Command } from "commander";
-import CreateLogCtx from "../src/lib/logger/logger";
-import { IndexOptions } from "mongodb";
-import { ValidDatabases } from "tachi-common";
-import { CONF_INFO } from "../src/lib/setup/config";
-
-const logger = CreateLogCtx(__filename);
+import { SetIndexes } from "../src/external/mongo/indexes";
 
 const program = new Command();
 
@@ -18,131 +13,4 @@ program.option(
 program.parse(process.argv);
 const options = program.opts();
 
-interface Index {
-    fields: Record<string, unknown>;
-    options?: IndexOptions;
-}
-
-function index(fields: Record<string, unknown>, options?: IndexOptions) {
-    return { fields, options };
-}
-
-const UNIQUE = { unique: true };
-
-const staticIndexes: Partial<Record<ValidDatabases, Index[]>> = {
-    scores: [index({ scoreID: 1 }, UNIQUE)],
-    "personal-bests": [
-        index({ chartID: 1, userID: 1 }, UNIQUE),
-        index({ chartID: 1, "scoreData.percent": 1 }),
-    ],
-    sessions: [
-        // lol
-        index({ timeStarted: 1, timeEnded: 1, userID: 1, game: 1, playtype: 1 }),
-        index({ name: "text" }),
-    ],
-    "game-stats": [index({ userID: 1, game: 1, playtype: 1 }, UNIQUE)],
-    "folder-chart-lookup": [index({ chartID: 1, folderID: 1 }, UNIQUE)],
-    "tierlist-data": [
-        index({ tierlistDataID: 1 }, UNIQUE),
-        index({ chartID: 1, tierlistID: 1 }),
-        index({ chartID: 1, tierlistID: 1, type: 1 }),
-        index({ chartID: 1, tierlistID: 1, type: 1, key: 1 }, UNIQUE),
-    ],
-    goals: [index({ goalID: 1 }, UNIQUE)],
-    "user-goals": [index({ goalID: 1, userID: 1 }, UNIQUE), index({ goalID: 1 })],
-    milestones: [index({ milestoneID: 1 }, UNIQUE), index({ group: 1, game: 1, playtype: 1 })],
-    "user-milestones": [
-        index({ milestoneID: 1, userID: 1 }, UNIQUE),
-        index({ userID: 1, game: 1, playtype: 1 }),
-    ],
-    imports: [index({ importID: 1 }, UNIQUE)],
-    "import-timings": [
-        index({ importID: 1 }, UNIQUE),
-        index({ timestamp: 1 }),
-        index({ total: 1 }),
-    ],
-    users: [index({ id: 1 }, UNIQUE)],
-    tierlists: [
-        index({ tierlistID: 1 }, UNIQUE),
-        index({ game: 1, playtype: 1, isDefault: 1 }, UNIQUE),
-    ],
-    folders: [
-        index({ folderID: 1 }, UNIQUE),
-        index({ game: 1, playtype: 1 }),
-        index({ game: 1, playtype: 1, table: 1 }),
-        index({ game: 1, playtype: 1, table: 1, tableIndex: 1 }),
-    ],
-    "kai-auth-tokens": [index({ userID: 1, service: 1 }, UNIQUE)],
-    "charts-iidx": [
-        index(
-            { "data.arcChartID": 1 },
-            { unique: true, partialFilterExpression: { "data.arcChartID": { $type: "string" } } }
-        ),
-        index({ "data.hashSHA256": 1 }),
-    ],
-    "bms-course-lookup": [index({ md5sums: 1 }, UNIQUE)],
-    "api-tokens": [index({ token: 1 }, UNIQUE)],
-};
-
-const indexes: Partial<Record<ValidDatabases, Index[]>> = staticIndexes;
-
-for (const game of CONF_INFO.SUPPORTED_GAMES) {
-    if (indexes[`charts-${game}` as ValidDatabases]) {
-        indexes[`charts-${game}` as ValidDatabases]!.push(
-            index({ chartID: 1 }, UNIQUE),
-            index(
-                { songID: 1, difficulty: 1, playtype: 1, isPrimary: 1 },
-                { unique: true, partialFilterExpression: { isPrimary: { $eq: true } } }
-            )
-        );
-    } else {
-        indexes[`charts-${game}` as ValidDatabases] = [
-            index({ chartID: 1 }, UNIQUE),
-            index({ songID: 1, difficulty: 1, playtype: 1, isPrimary: 1 }, UNIQUE),
-        ];
-    }
-
-    if (indexes[`songs-${game}` as ValidDatabases]) {
-        indexes[`songs-${game}` as ValidDatabases]!.push(
-            index({ id: 1 }, UNIQUE),
-            index({ title: "text", artist: "text", "alt-titles": "text", "search-titles": "text" })
-        );
-    } else {
-        indexes[`songs-${game}` as ValidDatabases] = [
-            index({ id: 1 }, UNIQUE),
-            index({ title: 1 }),
-            index({ title: "text", artist: "text", "alt-titles": "text", "search-titles": "text" }),
-        ];
-    }
-}
-
-export async function SetIndexes(dbst: string) {
-    const db = monk(`${process.env.MONGO_BASE_URL ?? "127.0.0.1"}/${dbst}`);
-
-    logger.info(`Starting indexing for ${options.db ?? "ktblackdb"}...`);
-
-    for (const collection in indexes) {
-        if (options.reset) {
-            // eslint-disable-next-line no-await-in-loop
-            await db.get(collection).dropIndexes();
-            logger.info(`Reset ${collection}.`);
-        }
-
-        // @ts-expect-error dru(n)kts
-        for (const index of indexes[collection]) {
-            // eslint-disable-next-line no-await-in-loop
-            const r = await db.get(collection).createIndex(index.fields, index.options);
-
-            logger.info(r);
-        }
-    }
-
-    db.close();
-
-    logger.info("Done.");
-}
-
-// if calling this as a script -- similar to pythons if __name__ == "__main__"
-if (require.main === module) {
-    SetIndexes(options.db ?? "ktblackdb").then(() => process.exit(0));
-}
+SetIndexes(options.db, options.reset).then(() => process.exit(0));