sendou.ink/scripts/transfer-weapon-pools.ts

/* eslint-disable no-console */
import "dotenv/config";
import { db } from "~/db/sql";

async function main() {
  const weaponPools = await db
    .selectFrom("UserWeapon")
    .select([
      "UserWeapon.userId",
      "UserWeapon.weaponSplId",
      "UserWeapon.userId",
    ])
    .where("UserWeapon.order", "!=", 5)
    .orderBy("UserWeapon.order asc")
    .execute();

  // group by userId
  const weaponPoolsByUserId = weaponPools.reduce(
    (acc, weaponPool) => {
      if (!acc[weaponPool.userId]) {
        acc[weaponPool.userId] = [];
      }

      acc[weaponPool.userId].push(weaponPool);

      return acc;
    },
    {} as Record<string, typeof weaponPools>,
  );

  for (const [userId, weaponPools] of Object.entries(weaponPoolsByUserId)) {
    const weaponPoolIds = weaponPools.map(
      (weaponPool) => weaponPool.weaponSplId,
    );

    await db
      .updateTable("User")
      .set({
        qWeaponPool: JSON.stringify(weaponPoolIds),
      })
      .where("User.id", "=", Number(userId))
      .execute();
  }

  console.log("done with the transfer");
}

void main();