zkldi_Tachi/server/scripts/__KT_DATABASE_MIGRATION/charts-sdvx.ts

/* eslint-disable @typescript-eslint/no-explicit-any */
import { ChartDocument } from "kamaitachi-common";
import db from "../../src/db/db";
import CreateLogCtx from "../../src/logger";
import MigrateRecords from "./migrate";
import { gameOrders } from "kamaitachi-common/js/config";

const logger = CreateLogCtx("charts-sdvx.ts");

async function ConvertFn(c: any): Promise<ChartDocument<"sdvx:Single">> {
    let song = await db.songs.sdvx.findOne({
        id: c.id,
    });

    if (!song) {
        logger.severe(`Cannot find song with ID ${c.id}?`);
        throw new Error(`Cannot find song with ID ${c.id}?`);
    }

    const newChartDoc: ChartDocument<"sdvx:Single"> = {
        rgcID: null,
        chartID: c.chartID,
        difficulty: c.difficulty,
        songID: c.id,
        playtype: c.playtype,
        levelNum: c.levelNum,
        level: c.level.toString(),
        flags: {
            "IN BASE GAME": !!c.flags["IN BASE GAME"],
            OMNIMIX: !!c.flags.OMNIMIX,
            "N-1": !!c.flags["N-1"],
        },
        data: {
            inGameID: c.internals.inGameINTID,
        },
        isPrimary: true,
        versions: [], // sentinel
    };

    let idx = gameOrders.sdvx.indexOf(song.firstVersion!);

    if (idx === -1) {
        logger.warn(`Invalid firstAppearance of ${song.firstVersion!}, running anyway.`);
        newChartDoc.versions = [song.firstVersion!];
    } else {
        newChartDoc.versions = gameOrders.sdvx.slice(idx);
    }

    return newChartDoc;
}

(async () => {
    await MigrateRecords(db.charts.sdvx, "charts-sdvx", ConvertFn);

    process.exit(0);
})();