zkldi_Tachi/database-seeds/scripts/rerunners/import-raw-quests.ts

import { Command } from "commander";
import { CreateGoalID, CreateQuestID, MutateCollection } from "../util";
import fs from "fs";
import { QuestDocument, GoalDocument } from "tachi-common";

const program = new Command();
program.option("-f, --file <quests.json>");

program.parse(process.argv);
const options = program.opts();

// stolen from client/src/types/tachi.ts
type RawQuestDocument = Omit<QuestDocument, "questData" | "questID"> & {
	rawQuestData: Array<RawQuestSection>;
};

type RawQuestSection = {
	title: string;
	desc: string;
	rawGoals: Array<RawQuestGoal>;
};

type RawQuestGoal = {
	goal: Pick<GoalDocument, "criteria" | "charts" | "name">;
	note?: string;
};

const data = JSON.parse(fs.readFileSync(options.file, "utf-8")) as Array<RawQuestDocument>;

const newGoals: Array<GoalDocument> = [];

function HydrateQuest(raw: RawQuestDocument): QuestDocument {
	const questData: QuestDocument["questData"] = [];

	const { game, playtype } = raw;

	for (const rawQuest of raw.rawQuestData) {
		const goals: Array<{ goalID: string; note?: string }> = [];

		for (const rawGoal of rawQuest.rawGoals) {
			const goalID = CreateGoalID(rawGoal.goal.charts, rawGoal.goal.criteria, game, playtype);

			const newGoal: GoalDocument = {
				charts: rawGoal.goal.charts,
				criteria: rawGoal.goal.criteria,
				game,
				playtype,
				goalID,
				name: rawGoal.goal.name,
			} as GoalDocument;

			newGoals.push(newGoal);

			goals.push({
				goalID,
				note: rawGoal.note,
			});
		}

		questData.push({
			title: rawQuest.title,
			desc: rawQuest.desc,
			goals,
		});
	}

	return {
		desc: raw.desc,
		name: raw.name,
		game,
		playtype,
		// just 20 random bytes. can't think of much more creative at the moment.
		questID: CreateQuestID(),
		questData,
	};
}

const newQuests = data.map(HydrateQuest);

MutateCollection("quests.json", (quests) => [...quests, ...newQuests]);

MutateCollection("goals.json", (goals) => {
	// don't duplicate goals
	const goalIDs = new Set(goals.map((e) => e.goalID));

	for (const goal of newGoals) {
		if (goalIDs.has(goal.goalID)) {
			continue;
		}

		// don't allow duplicates in incoming goals, either.
		goalIDs.add(goal.goalID);

		goals.push(goal);
	}

	return goals;
});