top500jsons script ok version

prisma/scripts/top500jsons.ts

@@ -1,6 +1,116 @@
+import { Prisma } from "@prisma/client";
+import { getWeaponNormalized } from "../../lib/lists/weapons";
+import prisma from "../client";
+import cb from "./data/october_clam_blitz_2020.json";
+import rm from "./data/october_rainmaker_2020.json";
 import sz from "./data/october_splat_zones_2020.json";
-// import tc from "./data/october_tower_control_2020.json";
-// import rm from "./data/october_rainmaker_2020.json";
-// import cb from "./data/october_clam_blitz_2020.json"
+import tc from "./data/october_tower_control_2020.json";
 
-console.log(sz[0]);
+const MONTH = 10;
+const YEAR = 2020;
+
+const filterJson = (result: any) => !result.cheater;
+
+const jsonToInput = (
+  result: any,
+  mode: "SZ" | "TC" | "RM" | "CB",
+  month: number,
+  year: number
+): Prisma.XRankPlacementCreateInput => ({
+  playerName: result.name,
+  mode,
+  month,
+  year,
+  ranking: result.rank,
+  xPower: result.x_power,
+  weapon: getWeaponNormalized(result.weapon.name),
+  player: {
+    connectOrCreate: {
+      create: {
+        name: result.name,
+        switchAccountId: result.unique_id,
+      },
+      where: {
+        switchAccountId: result.unique_id,
+      },
+    },
+  },
+});
+
+const jsonToWeaponArrMap = (acc: Map<string, string[]>, result: any) => {
+  const weaponArr = acc.get(result.unique_id) ?? [];
+  acc.set(
+    result.unique_id,
+    weaponArr.concat(getWeaponNormalized(result.weapon.name))
+  );
+
+  return acc;
+};
+
+const main = async () => {
+  const builds = await prisma.build.findMany({
+    where: {
+      top500: false,
+    },
+    include: { user: { include: { player: true } } },
+  });
+
+  const inputSZ = sz
+    .filter(filterJson)
+    .map((json) => jsonToInput(json, "SZ", MONTH, YEAR));
+
+  const inputTC = tc
+    .filter(filterJson)
+    .map((json) => jsonToInput(json, "TC", MONTH, YEAR));
+
+  const inputRM = rm
+    .filter(filterJson)
+    .map((json) => jsonToInput(json, "RM", MONTH, YEAR));
+
+  const inputCB = cb
+    .filter(filterJson)
+    .map((json) => jsonToInput(json, "CB", MONTH, YEAR));
+
+  const idsToWeapons = [...sz, ...tc, ...rm, ...cb].reduce(
+    jsonToWeaponArrMap,
+    new Map()
+  );
+
+  const idsToUpdate: number[] = [];
+
+  builds.forEach((build) => {
+    if (!build.user.player?.switchAccountId) return;
+    const weaponArr = idsToWeapons.get(build.user.player.switchAccountId);
+    if (!weaponArr) return;
+    if (!weaponArr.includes(getWeaponNormalized(build.weapon))) return;
+
+    idsToUpdate.push(build.id);
+  });
+
+  const newPlacements = [
+    ...inputSZ,
+    ...inputTC,
+    ...inputRM,
+    ...inputCB,
+  ].map((input) => prisma.xRankPlacement.create({ data: input }));
+
+  await prisma.$transaction([...newPlacements]);
+  console.log("new placements submitted");
+  await prisma.build.updateMany({
+    where: {
+      id: {
+        in: idsToUpdate,
+      },
+    },
+    data: {
+      top500: true,
+    },
+  });
+  console.log("builds updated:", idsToUpdate.length);
+};
+
+main()
+  .catch((e) => console.error(e))
+  .finally(async () => {
+    await prisma.$disconnect();
+  });