chore: refactor into new task system

This commit is contained in:
DecDuck
2025-06-06 10:00:16 +10:00
parent 951a741f3e
commit 816355be0c
3 changed files with 22 additions and 7 deletions

View File

@ -14,6 +14,9 @@ export const taskGroups = {
"import:game": {
concurrency: true,
},
"ludusavi:import": {
concurrency: false,
},
} as const;
export type TaskGroup = keyof typeof taskGroups;

View File

@ -10,6 +10,7 @@ import cleanupObjects from "./registry/objects";
import { taskGroups, type TaskGroup } from "./group";
import prisma from "../db/database";
import { type } from "arktype";
import ludusavi from "./registry/ludusavi";
// a task that has been run
type FinishedTask = {
@ -59,6 +60,7 @@ class TaskHandler {
this.saveScheduledTask(cleanupSessions);
this.saveScheduledTask(checkUpdate);
this.saveScheduledTask(cleanupObjects);
this.saveScheduledTask(ludusavi)
}
/**

View File

@ -0,0 +1,143 @@
import { defineDropTask } from "..";
import yaml from "js-yaml";
import { Platform } from "~/prisma/client";
import type { LudusaviPlatformEntryCreateOrConnectWithoutLudusaviEntryInput } from "~/prisma/client/models";
import prisma from "../../db/database";
type ConnectOrCreateShorthand =
LudusaviPlatformEntryCreateOrConnectWithoutLudusaviEntryInput;
type LudusaviModel = {
[key: string]: {
files?: {
[key: string]: {
tags?: Array<string>;
when?: Array<{ os?: string }>;
};
};
registry?: { [key: string]: { tags?: Array<string> } };
steam?: { id: number };
};
};
export default defineDropTask({
buildId: () => `ludusavi:import:${new Date().toISOString()}`,
name: "Import Ludusavi",
acls: [],
taskGroup: "ludusavi:import",
async run({ log, progress }) {
const manifest = yaml.load(
await $fetch<string>(
"https://raw.githubusercontent.com/mtkennerly/ludusavi-manifest/refs/heads/master/data/manifest.yaml",
),
) as LudusaviModel;
let currentProgress = 10;
progress(currentProgress);
const entries = Object.entries(manifest);
const increment = (1 / entries.length) * 0.9;
for (const [name, data] of entries) {
if (!data.files && !data.registry) continue;
const iterableFiles = data.files ? Object.entries(data.files) : undefined;
function findFilesForOperatingSystem(os: string) {
return iterableFiles?.filter((e) =>
e[1].when?.find((v) => v.os === os),
);
}
const connectOrCreate: ConnectOrCreateShorthand[] = [];
const windowsData = {
registry: data.registry,
files: findFilesForOperatingSystem("windows"),
};
if (windowsData.registry || windowsData.files) {
const create: ConnectOrCreateShorthand = {
where: {
ludusaviEntryName_platform: {
ludusaviEntryName: name,
platform: Platform.Windows,
},
},
create: {
platform: Platform.Windows,
files: windowsData.files?.map((e) => e[0]) ?? [],
registry: Object.entries(windowsData.registry ?? {}).map(
(e) => e[0],
),
},
};
connectOrCreate.push(create);
}
const linuxData = {
files: findFilesForOperatingSystem("linux"),
};
if (linuxData.files) {
const create: ConnectOrCreateShorthand = {
where: {
ludusaviEntryName_platform: {
ludusaviEntryName: name,
platform: Platform.Linux,
},
},
create: {
platform: Platform.Linux,
files: linuxData.files?.map((e) => e[0]) ?? [],
registry: [],
},
};
connectOrCreate.push(create);
}
const macData = {
files: findFilesForOperatingSystem("mac"),
};
if (macData.files) {
const create: ConnectOrCreateShorthand = {
where: {
ludusaviEntryName_platform: {
ludusaviEntryName: name,
platform: Platform.macOS,
},
},
create: {
platform: Platform.macOS,
files: macData.files?.map((e) => e[0]) ?? [],
registry: [],
},
};
connectOrCreate.push(create);
}
const steamId = data.steam?.id.toString() ?? null;
await prisma.ludusaviEntry.upsert({
where: {
name,
},
create: {
name,
steamId,
entries: { connectOrCreate },
},
update: {
steamId,
entries: { connectOrCreate },
},
});
currentProgress += increment;
log(`Imported game "${name}"`);
}
},
});