fix: include full treatment info in tournament/game export

* amend export.R preprocessing script to run over all tournament round
  dirs

ref virtualcommons/planning#64

analytics/R/export.R

@@ -29,4 +29,14 @@ run_pipeline <- function(base_dir) {
   readr::write_csv(chat_messages, fs::path_join(c(base_dir, "processed/chatMessages.csv")))
 }
 
-run_pipeline("/dump")
+process_all_rounds <- function(base_path) {
+  # assumes that all numeric directories contain exported tournament round data
+  entries <- fs::dir_ls(base_path)
+  tournament_dirs <- entries[fs::path_ext(entries) == '' & grepl("^[0-9]+$", fs::path_file(entries))]
+
+  for (dir in tournament_dirs) {
+    run_pipeline(dir)
+  }
+}
+
+process_all_rounds("/dump")

server/src/cli.ts

@@ -119,6 +119,7 @@ async function exportTournamentRound(
   const gameQuery = em
     .getRepository(Game)
     .createQueryBuilder("game")
+    .leftJoinAndSelect("game.treatment", "treatment")
     .where("game.tournamentRound.id = :tournamentRoundId", { tournamentRoundId });
   const games = await gameQuery.getMany();
 

server/src/services/replay.ts

@@ -273,13 +273,26 @@ export class AccomplishmentSummarizer {
 export class GameSummarizer {
   constructor(public games: Array<entity.Game>, public path: string) {}
 
+  flattenTreatmentData(game: entity.Game) {
+    const { treatment, ...rest } = game;
+    if (!treatment) return game;
+
+    return {
+      ...rest,
+      treatmentName: treatment.name,
+      treatmentDescription: treatment.description,
+      treatmentMarsEventOverrides: JSON.stringify(treatment.marsEventOverrides),
+    };
+  }
+
   async save() {
-    const header = Object.keys(this.games[0]).map(name => ({
+    const flattenedGames = this.games.map(g => this.flattenTreatmentData(g));
+    const header = Object.keys(flattenedGames[0]).map(name => ({
       id: name,
       title: name,
     }));
     const writer = createObjectCsvWriter({ path: this.path, header });
-    await writer.writeRecords(this.games);
+    await writer.writeRecords(flattenedGames);
   }
 }