feat: added ratings leaderboard generation and publication

scripts/build.js

@@ -4,22 +4,22 @@ import { argv } from 'process';
 
 const isProd = !argv.includes('--dev');
 const buildAll = argv.includes('--build=all');
-const buildBot = argv.includes('--build=bot') | buildAll;
-const buildServer = argv.includes('--build=server') | buildAll;
+const buildBot = argv.includes('--build=bot') || buildAll;
+const buildServer = argv.includes('--build=server') || buildAll;
 
 if (isProd) {
   // only remove in production to ensure no time is wasted in dev
-  console.log('🟡 Removing dist...');
+  console.log('Removing dist...');
   await rm('dist', { recursive: true, force: true });
-  console.log('✅ Dist removed');
+  console.log('Dist removed');
 
-  console.log('🟡 Creating dist...');
+  console.log('Creating dist...');
   await mkdir('dist');
-  console.log('✅ Dist created');
+  console.log('Dist created');
 
-  console.log('🟡 Copying package.json...');
+  console.log('Copying package.json...');
   copyFile('package.dist.json', 'dist/package.json').then(() =>
-    console.log('✅ package.json copied'),
+    console.log('package.json copied'),
   );
 }
 
@@ -40,25 +40,25 @@ const commonOptions = {
 };
 
 if (buildBot) {
-  console.log('🟡 Building bot...');
+  console.log('Building bot...');
   build({
     ...commonOptions,
 
     entryPoints: ['src/bot.ts'],
     outfile: 'dist/bot.cjs',
   }).then(() => {
-    console.log('✅ Bot built');
+    console.log('Bot built');
   });
 }
 
 if (buildServer) {
-  console.log('🟡 Building server...');
+  console.log('Building server...');
   build({
     ...commonOptions,
 
     entryPoints: ['src/server.ts'],
     outfile: 'dist/server.cjs',
   }).then(() => {
-    console.log('✅ Server built');
+    console.log('Server built');
   });
 }

scripts/commitMultipleFiles.js

@@ -0,0 +1,67 @@
+export default async function commitMultipleFiles(
+  octokit,
+  owner,
+  repo,
+  branch,
+  message,
+  changes,
+) {
+  // Get the latest commit SHA on the specified branch
+  const { data: refData } = await octokit.git.getRef({
+    owner,
+    repo,
+    ref: `heads/${branch}`,
+  });
+  const latestCommitSha = refData.object.sha;
+
+  // Get the tree SHA for the latest commit
+  const { data: commitData } = await octokit.git.getCommit({
+    owner,
+    repo,
+    commit_sha: latestCommitSha,
+  });
+  const treeSha = commitData.tree.sha;
+
+  // Create blobs for each file
+  const blobs = await Promise.all(
+    changes.map(async (change) => {
+      const { data: blobData } = await octokit.git.createBlob({
+        owner,
+        repo,
+        content: change.content,
+        encoding: 'utf-8',
+      });
+      return {
+        sha: blobData.sha,
+        path: change.path,
+        mode: '100644',
+        type: 'blob',
+      };
+    }),
+  );
+
+  // Create a new tree with the new blobs
+  const { data: treeData } = await octokit.git.createTree({
+    owner,
+    repo,
+    base_tree: treeSha,
+    tree: blobs,
+  });
+
+  // Create a new commit with the new tree
+  const { data: newCommitData } = await octokit.git.createCommit({
+    owner,
+    repo,
+    message,
+    tree: treeData.sha,
+    parents: [latestCommitSha],
+  });
+
+  // Update the branch reference to point to the new commit
+  await octokit.git.updateRef({
+    owner,
+    repo,
+    ref: `heads/${branch}`,
+    sha: newCommitData.sha,
+  });
+}

scripts/generateRatingsLeaderboard.js

@@ -0,0 +1,111 @@
+import { Octokit } from '@octokit/rest';
+import { rm, writeFile } from 'fs/promises';
+import { argv, env } from 'process';
+import commitMultipleFiles from './commitMultipleFiles.js';
+
+const publish = argv.includes('--publish');
+const server = 'na';
+
+const NEIGHBORS = 1000;
+const PLAYERS = await fetch(
+  `https://${server}.wotblitz.com/en/api/rating-leaderboards/season/`,
+)
+  .then((response) => response.json())
+  .then((data) => data.count);
+
+const players = {};
+const leaderboard = [];
+let registered = 0;
+
+async function branchFromPlayer(id) {
+  const { neighbors } = await fetch(
+    `https://${server}.wotblitz.com/en/api/rating-leaderboards/user/${id}/?neighbors=${NEIGHBORS}`,
+  ).then((response) => response.json());
+
+  const firstPlayer = neighbors[0];
+  const lastPlayer = neighbors[neighbors.length - 1];
+
+  const branches = [];
+  if (!players[firstPlayer.number])
+    branches.push(branchFromPlayer(firstPlayer.spa_id));
+  if (!players[lastPlayer.number])
+    branches.push(branchFromPlayer(lastPlayer.spa_id));
+
+  neighbors.forEach((neighbor) => {
+    if (players[neighbor.number]) return;
+
+    players[neighbor.number] = neighbor;
+    registered++;
+  });
+
+  console.log(
+    `Registered ${registered} / ${PLAYERS} (${(
+      (registered / PLAYERS) *
+      100
+    ).toFixed(2)}%)`,
+  );
+
+  await Promise.all(branches);
+}
+
+async function branchFromLeague(id) {
+  const { result } = await fetch(
+    `https://${server}.wotblitz.com/en/api/rating-leaderboards/league/${id}/top/`,
+  ).then((response) => response.json());
+
+  const firstPlayer = result[0];
+  const lastPlayer = result[result.length - 1];
+
+  await Promise.all([
+    branchFromPlayer(firstPlayer.spa_id),
+    branchFromPlayer(lastPlayer.spa_id),
+  ]);
+}
+
+console.log('Branching form leagues...');
+await Promise.all([
+  branchFromLeague(0),
+  branchFromLeague(1),
+  branchFromLeague(2),
+  branchFromLeague(3),
+  branchFromLeague(4),
+]);
+
+console.log('Converting to an array...');
+for (let index = 0; index < PLAYERS; index++) {
+  leaderboard[index] = players[index + 1];
+  if (!leaderboard[index]) console.warn(`${index} had no player`);
+}
+
+const leaderboardJSON = JSON.stringify(leaderboard);
+
+if (publish) {
+  const octokit = new Octokit({ auth: env.GITHUB_TOKEN });
+
+  const info = await fetch(
+    `https://${server}.wotblitz.com/en/api/rating-leaderboards/season/`,
+  ).then((response) => response.json());
+  const time = Math.round(Date.now() / 1000);
+  const normalizedServer = server === 'na' ? 'com' : server;
+
+  const infoPath = `${normalizedServer}/ratings/seasons/${info.current_season}.json`;
+  const leaderboardPath = `${normalizedServer}/ratings/leaderboards/${info.current_season}/${time}.json`;
+  const infoJSON = JSON.stringify(info);
+
+  console.log(`Publishing to season ${info.current_season}...`);
+  commitMultipleFiles(
+    octokit,
+    'tresabhi',
+    'blitzkrieg-db',
+    'main',
+    `${info.current_season}/${time}`,
+    [
+      { path: infoPath, content: infoJSON },
+      { path: leaderboardPath, content: leaderboardJSON },
+    ],
+  );
+} else {
+  console.log('Writing to disk...');
+  await rm('dist.json', { force: true });
+  await writeFile('dist.json', leaderboardJSON);
+}