Merge pull request #12 from yuki-wtf/add-whitelist-scripts

Add whitelist generation script

indexer/src/index.ts

@@ -4,8 +4,7 @@ import { appType } from "./utils/envs";
 import { indexer } from "./indexer";
 import { viewRefresher } from "./viewRefresher";
 import { indexerApp, viewRefresherApp, generator } from "./utils/const";
-import { whitelistGenerator } from "./whitelistGenerator";
-
+import { generateWhitelistMintGenerations } from "./utils/generateWhitelistMintGenerations";
 logger.info("Starting.");
 
 AppDataSource.initialize().then(async () => {
@@ -17,7 +16,7 @@ AppDataSource.initialize().then(async () => {
         } else if (appType === viewRefresherApp) {
             await viewRefresher()
         } else if(appType === generator) {
-            await whitelistGenerator();
+            await generateWhitelistMintGenerations();
         }
     } catch (e) {
         logger.error(e, "App failed.");

indexer/src/indexer.ts

@@ -18,6 +18,7 @@ import {
 import { IsNull, Not } from "typeorm";
 import assert from "assert";
 import { eventNameMap } from "./utils/const";
+// import { saveWhitelistProofsFromFileToDB } from "./utils/saveWhitelistProofsFromFileToDB";
 
 type ReturnedTransactionStatus = {
   tx_status: string;
@@ -274,6 +275,7 @@ const updatePendingMints = async () => {
 
 const wait = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
 export const indexer = async () => {
+  // saveWhitelistProofsFromFileToDB();
   try {
     while (true) {
       await pullEvents();

indexer/src/utils/db.ts

@@ -12,7 +12,7 @@ import { Mints } from "../entity/mints";
 import { Whitelist } from "../entity/whitelist";
 
 // Uncomment below in local development
-import 'dotenv/config'
+// import 'dotenv/config'
 
 // We need to store bigints in jsonb column, typeorm doesn't support that.
 // Transformers in typeorm run _before_ typeorm's JSON.stringify run, so it is problematic

indexer/src/utils/generateWhitelistMintGenerations.ts

@@ -0,0 +1,33 @@
+import fs from "fs";
+import { AppDataSource } from "./db";
+
+export const generateWhitelistMintGenerations = async () => {
+  const infinite = await AppDataSource.query(`
+    SELECT "transactionOwner", "gameGeneration", "gameState", "createdAt"
+    FROM "public"."infinite"
+    WHERE "transactionType" = 'game_evolved'
+    ORDER BY "gameGeneration" ASC
+`);
+
+  // @ts-ignore
+  const processedData = infinite.reduce((acc, row) => {
+    acc[row.gameGeneration] = {
+      user_id: row.transactionOwner,
+      game_state: row.gameState,
+      timestamp: Math.floor(row.createdAt.getTime() / 1000), // Convert to Unix seconds
+    };
+    return acc;
+  }, {});
+
+  // Convert to JSON string with pretty formatting
+  const jsonData = JSON.stringify(processedData, null, 2);
+
+  // Write to a file
+  fs.writeFile("gol2-whitelist.json", jsonData, (err) => {
+    if (err) {
+      console.error("Error writing file:", err);
+      return;
+    }
+    console.log("Successfully wrote to gol2-whitelist.json");
+  });
+};

indexer/src/whitelistGenerator.ts → indexer/src/utils/saveWhitelistProofsFromFileToDB.ts

@@ -1,10 +1,12 @@
 import fs from 'fs/promises'
-import { Whitelist } from "./entity/whitelist";
-import { AppDataSource } from './utils/db';
+import { Whitelist } from "../entity/whitelist";
+import { AppDataSource } from './db';
 import { exit } from 'process';
 
-export const whitelistGenerator = async () => {
-  console.log("Starting whitelist generator.");
+const filename = "whitelist.json";
+
+export const saveWhitelistProofsFromFileToDB = async () => {
+  console.log(`Retrieving whitelist proofs from "${filename}" file.`);
   const dataRaw = await fs.readFile("whitelist.json", "utf8");
   const data = JSON.parse(dataRaw);