monkeytype/backend/worker.ts

import path from "path";
// @ts-ignore
import serviceAccount from "./credentials/serviceAccountKey.json"; // eslint-disable-line require-path-exists/exists
import admin, { ServiceAccount } from "firebase-admin";
import db from "./init/db";
import { config } from "dotenv";

config({ path: path.join(__dirname, ".env") });

async function main(): Promise<void> {
  await db.connect();
  admin.initializeApp({
    credential: admin.credential.cert(
      serviceAccount as unknown as ServiceAccount
    ),
  });
  console.log("Database Connected!!");
  refactor();
}

main();

async function refactor(): Promise<void> {
  console.log("getting all users");

  const usersCollection = db.collection<any>("users");
  const users = await usersCollection.find({}).toArray();
  console.log(users.length);

  for (const user of users) {
    const obj = user.personalBests;

    const lbPb = {
      time: {
        15: {},
        60: {},
      },
    };
    let bestForEveryLanguage = {};
    if (obj?.time?.[15]) {
      obj.time[15].forEach((pb) => {
        if (!bestForEveryLanguage[pb.language]) {
          bestForEveryLanguage[pb.language] = pb;
        } else {
          if (bestForEveryLanguage[pb.language].wpm < pb.wpm) {
            bestForEveryLanguage[pb.language] = pb;
          }
        }
      });
      Object.keys(bestForEveryLanguage).forEach((key) => {
        lbPb.time[15][key] = bestForEveryLanguage[key];
      });
      bestForEveryLanguage = {};
    }
    if (obj?.time?.[60]) {
      obj.time[60].forEach((pb) => {
        if (!bestForEveryLanguage[pb.language]) {
          bestForEveryLanguage[pb.language] = pb;
        } else {
          if (bestForEveryLanguage[pb.language].wpm < pb.wpm) {
            bestForEveryLanguage[pb.language] = pb;
          }
        }
      });
      Object.keys(bestForEveryLanguage).forEach((key) => {
        lbPb.time[60][key] = bestForEveryLanguage[key];
      });
    }

    await usersCollection.updateOne(
      { _id: user._id },
      { $set: { lbPersonalBests: lbPb } }
    );
    console.log(`updated ${user.name}`);
  }
  console.log("done");
}