From 93c7392d6e23733032dbd026b6b0558b0008db1f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tr=C3=A8sAbhi?= <43380238+tresabhi@users.noreply.github.com>
Date: Wed, 5 Jun 2024 18:35:44 -0500
Subject: [PATCH] feat: isolated discovery

---
 package.json               |   3 +-
 scripts/discoverIds.ts     | 105 +++++++++++++++++++++++++++++++++++++
 src/core/blitzkit/asset.ts |   4 +-
 3 files changed, 109 insertions(+), 3 deletions(-)
 create mode 100644 scripts/discoverIds.ts

diff --git a/package.json b/package.json
index b1ab6a183..3f42628a4 100644
--- a/package.json
+++ b/package.json
@@ -30,7 +30,8 @@
     "knip": "knip",
     "update": "yarn upgrade --latest",
     "setup:depotdownloader": "vite-node --script scripts/setupDepotDownloader",
-    "tunnel": "cloudflared tunnel --url http://localhost:3000"
+    "tunnel": "cloudflared tunnel --url http://localhost:3000",
+    "discoverids": "bun run scripts/discoverIds"
   },
   "license": "MIT",
   "dependencies": {
diff --git a/scripts/discoverIds.ts b/scripts/discoverIds.ts
new file mode 100644
index 000000000..418603f5f
--- /dev/null
+++ b/scripts/discoverIds.ts
@@ -0,0 +1,105 @@
+import { times } from 'lodash';
+import { argv } from 'process';
+import { Region, REGIONS } from '../src/constants/regions';
+import { getAccountInfo } from '../src/core/blitz/getAccountInfo';
+import { idToRegion, MIN_IDS } from '../src/core/blitz/idToRegion';
+import { asset } from '../src/core/blitzkit/asset';
+import { DidsReadStream, DidsWriteStream } from '../src/core/streams/dids';
+
+// const RUN_TIME = 1000 * 60 * 60 * 5;
+const RUN_TIME = 1000 * 5;
+const MAX_REQUESTS = 10;
+const ACCOUNTS_PER_CALL = 100;
+const REMOVE_REGION_THRESHOLD = 10;
+
+const production = argv.includes('--production');
+const startTime = Date.now();
+
+async function verify(region: Region, ids: number[]) {
+  const infos = await getAccountInfo(region, ids, undefined, {
+    // cheeky way of requesting nothing
+    fields: 'account_id,-account_id',
+  });
+  const filtered = ids.filter((_, index) => {
+    const info = infos[index];
+    return info !== null;
+  });
+
+  return filtered;
+}
+
+const indexableRegions = [...REGIONS];
+const ids: number[] = [];
+
+const preDiscoveredRaw = await fetch(
+  asset('averages/discovered.dids.lz4', !production),
+).then(async (response) => {
+  if (response.status === 200) {
+    const buffer = await response.arrayBuffer();
+    return new DidsReadStream(buffer).dids();
+  }
+
+  return undefined;
+});
+
+const regionalIdIndex: Record<Region, number> = {
+  asia:
+    preDiscoveredRaw?.find((id) => idToRegion(id) === 'asia') ?? MIN_IDS.asia,
+  com: preDiscoveredRaw?.find((id) => idToRegion(id) === 'com') ?? MIN_IDS.com,
+  eu: preDiscoveredRaw?.find((id) => idToRegion(id) === 'eu') ?? MIN_IDS.eu,
+};
+const zeroStreak: Record<Region, number> = { asia: 0, com: 0, eu: 0 };
+let regionIndex = 0;
+let outOfTimeFlagged = false;
+
+const interval = setInterval(async () => {
+  const region = indexableRegions[regionIndex];
+  const idsToVerify = times(
+    ACCOUNTS_PER_CALL,
+    (index) => regionalIdIndex[region] + index,
+  );
+  const verified = await verify(region, idsToVerify);
+
+  if (verified.length === 0) {
+    zeroStreak[region]++;
+
+    if (zeroStreak[region] >= REMOVE_REGION_THRESHOLD) {
+      const sliceableIndex = indexableRegions.indexOf(region);
+
+      if (sliceableIndex !== -1) {
+        console.log(`Stopped discovery for ${region}`);
+        indexableRegions.splice(sliceableIndex, 1);
+
+        if (indexableRegions.length === 0) {
+          console.log('All players discovered exhaustively!');
+          post();
+        }
+      }
+    }
+  } else {
+    console.log(
+      `${region.padEnd(4)} ${verified.length.toString().padStart(3)}`,
+    );
+
+    zeroStreak[region] = 0;
+    ids.push(...verified);
+  }
+
+  if (Date.now() - startTime > RUN_TIME && !outOfTimeFlagged) {
+    outOfTimeFlagged = true;
+    console.log('Out of time');
+    post();
+  }
+
+  regionIndex = (regionIndex + 1) % indexableRegions.length;
+  regionalIdIndex[region] += ACCOUNTS_PER_CALL;
+}, 1000 / MAX_REQUESTS);
+
+function post() {
+  clearInterval(interval);
+  console.log(`Uploading ${ids.length} ids...`);
+
+  const didsWriteStream = new DidsWriteStream().dids(ids);
+
+  console.log(didsWriteStream.uint8Array.length);
+}
diff --git a/src/core/blitzkit/asset.ts b/src/core/blitzkit/asset.ts
index c9837e63e..db18ee675 100644
--- a/src/core/blitzkit/asset.ts
+++ b/src/core/blitzkit/asset.ts
@@ -1,6 +1,6 @@
 import { ASSETS_REPO } from '../../constants/assets';
 import isDev from './isDev';
 
-export function asset(path: string) {
-  return `https://raw.githubusercontent.com/tresabhi/${ASSETS_REPO}/${isDev() ? 'dev' : 'main'}/${path}`;
+export function asset(path: string, dev = isDev()) {
+  return `https://raw.githubusercontent.com/tresabhi/${ASSETS_REPO}/${dev ? 'dev' : 'main'}/${path}`;
 }