feat: building BoosterConfig using set_booster_* data.

smoke_test.ts

@@ -1294,22 +1294,22 @@ async function main(): Promise<boolean> {
 
 	// openPack — expect error or graceful fail with flat parquet
 	try {
-		const pack = await sdk.booster.openPack("MH3", "draft");
+		const pack = await sdk.booster.openPack("MH3", "play");
 		check("booster.openPack", Array.isArray(pack), `got ${pack.length} cards`);
 	} catch {
 		check("booster.openPack raises Error (no booster data)", true);
 	}
 
 	// openBox — expect error or graceful fail
 	try {
-		const box = await sdk.booster.openBox("MH3", "draft", 1);
+		const box = await sdk.booster.openBox("MH3", "play", 1);
 		check("booster.openBox", Array.isArray(box));
 	} catch {
 		check("booster.openBox raises Error (no booster data)", true);
 	}
 
 	// sheetContents
-	const contents = await sdk.booster.sheetContents("MH3", "draft", "common");
+	const contents = await sdk.booster.sheetContents("MH3", "play", "common");
 	check(
 		"booster.sheetContents",
 		contents === null || typeof contents === "object",

src/booster/simulator.ts

@@ -14,29 +14,136 @@ export class BoosterSimulator {
 	}
 
 	private async _ensure(): Promise<void> {
-		await this._conn.ensureViews("sets", "cards");
+		await this._conn.ensureViews(
+			"sets",
+			"cards",
+			"set_booster_content_weights",
+			"set_booster_contents",
+			"set_booster_sheet_cards",
+			"set_booster_sheets",
+		);
 	}
 
 	private async _getBoosterConfig(
 		setCode: string,
 	): Promise<Record<string, BoosterConfig> | null> {
 		await this._ensure();
 		try {
-			const rows = await this._conn.execute(
-				"SELECT booster FROM sets WHERE code = $1",
+			const setBoosterContentWeightsRows = await this._conn.execute(
+				"SELECT * FROM set_booster_content_weights WHERE setCode = $1",
 				[setCode.toUpperCase()],
 			);
-			if (!rows.length || !rows[0].booster) return null;
-			return rows[0].booster as Record<string, BoosterConfig>;
+			if (!setBoosterContentWeightsRows.length) return null;
+
+			const boosterWeights = new Map<string, Map<number, number>>();
+			for (const row of setBoosterContentWeightsRows) {
+				const name = row.boosterName as string;
+				const index = Number(row.boosterIndex);
+				const weight = Number(row.boosterWeight);
+
+				if (!boosterWeights.has(name)) {
+					boosterWeights.set(name, new Map());
+				}
+				boosterWeights.get(name)!.set(index, weight);
+			}
+
+			const setBoosterContentRows = await this._conn.execute(
+				"SELECT * FROM set_booster_contents WHERE setCode = $1",
+				[setCode.toUpperCase()],
+			);
+
+			if (!setBoosterContentRows.length) return null;
+
+			const boostersByName: Record<string, Map<number, BoosterPack>> = {};
+
+			for (const row of setBoosterContentRows) {
+				const name = row.boosterName as string;
+				const index = Number(row.boosterIndex);
+				const sheetName = row.sheetName as string;
+				const sheetPicks = Number(row.sheetPicks);
+
+				if (!boostersByName[name]) {
+					boostersByName[name] = new Map();
+				}
+
+				const boosterIndexMap = boostersByName[name];
+				if (!boosterIndexMap.has(index)) {
+					const weight = boosterWeights.get(name)?.get(index) ?? 0;
+					boosterIndexMap.set(index, { contents: {}, weight });
+				}
+
+				const booster = boosterIndexMap.get(index)!;
+				booster.contents[sheetName] = sheetPicks;
+			}
+
+			const config: Record<string, BoosterConfig> = {};
+			for (const [name, boosterIndexMap] of Object.entries(boostersByName)) {
+				config[name] = {
+					boosters: Array.from(boosterIndexMap.values()),
+					boostersTotalWeight: Array.from(boosterIndexMap.values()).reduce(
+						(sum, b) => sum + b.weight,
+						0,
+					),
+					sheets: {},
+					sourceSetCodes: [],
+				};
+			}
+
+			const setBoosterSheetsRows = await this._conn.execute(
+				"SELECT * FROM set_booster_sheets WHERE setCode = $1",
+				[setCode.toUpperCase()],
+			);
+
+			const setBoosterSheetCardsRows = await this._conn.execute(
+				"SELECT * FROM set_booster_sheet_cards WHERE setCode = $1",
+				[setCode.toUpperCase()],
+			);
+
+			for (const row of setBoosterSheetsRows) {
+				const boosterName = row.boosterName as string;
+				const sheetName = row.sheetName as string;
+
+				if (config[boosterName]) {
+					const cards: Record<string, number> = {};
+					for (const cardRow of setBoosterSheetCardsRows) {
+						if (
+							cardRow.boosterName === boosterName &&
+							cardRow.sheetName === sheetName
+						) {
+							cards[cardRow.cardUuid as string] = Number(cardRow.cardWeight);
+						}
+					}
+
+					config[boosterName].sheets[sheetName] = {
+						allowDuplicates: true, // Not in data
+						balanceColors: Boolean(row.sheetHasBalanceColors),
+						cards,
+						foil: Boolean(row.sheetIsFoil),
+						totalWeight: Number(row.sheetTotalWeight),
+					};
+				}
+			}
+
+			return config;
 		} catch {
 			return null;
 		}
 	}
 
+	async getBoosterData(
+        setCode: string,
+    ): Promise<Record<string, BoosterConfig> | null> {
+        return this._getBoosterConfig(setCode);
+    }
+
 	async availableTypes(setCode: string): Promise<string[]> {
-		const config = await this._getBoosterConfig(setCode);
-		if (!config) return [];
-		return Object.keys(config);
+		await this._ensure();
+		const rows = await this._conn.execute(
+			"SELECT DISTINCT boosterName FROM set_booster_contents WHERE setCode = $1",
+			[setCode.toUpperCase()],
+		);
+
+		return rows.map((row) => row.boosterName as string);
 	}
 
 	async openPack(setCode: string, boosterType = "draft"): Promise<CardSet[]> {