sendou.ink/prisma/seed.ts

//const { PrismaClient } = require("@prisma/client");
import { PrismaClient } from "@prisma/client";
const fs = require("fs");
const path = require("path");

const prisma = new PrismaClient();

const main = async () => {
  fs.readFile(path.join(__dirname, "/.env"), function (err: any, data: any) {
    if (!err) {
      for (const line of data.toString().split("\n")) {
        if (!line.startsWith("DATABASE_URL=")) {
          continue;
        }

        if (!line.includes("localhost:")) {
          console.error("trying to seed a database not in localhost");
          process.exit(1);
        }
      }
    } else {
      console.error(err);
      process.exit(1);
    }
  });

  await prisma.build.deleteMany({});
  await prisma.profile.deleteMany({});
  await prisma.user.deleteMany({});
  const testUser = await prisma.user.create({
    data: {
      username: "N-ZAP",
      discriminator: "6227",
      discordId: "455039198672453645",
      discordAvatar: "f809176af93132c3db5f0a5019e96339",
      profile: {
        create: {
          bio: "My cool bio! Supports markdown too: **bolded**",
          country: "US",
          customUrlPath: "tester",
          sensMotion: 4.5,
          sensStick: -2.0,
          weaponPool: [
            "Tenta Brella",
            "Range Blaster",
            "Luna Blaster",
            "N-ZAP '89",
          ],
          twitterName: "nintendovs",
          twitchName: "nintendo",
          youtubeId: "UCAtobAxsQcACwDZCSH9uJfA",
        },
      },
    },
  });

  console.log("User created");

  await prisma.xRankPlacement.deleteMany({});
  await prisma.player.deleteMany({});

  const modes = ["SZ", "TC", "RM", "CB"];
  let ranking = 0;

  await Promise.all(
    Array(100)
      .fill(null)
      .map((_, i) => {
        const playerName = i % 2 === 0 ? `Player${i}` : `選手${i}`;
        const mode = modes.shift()!;
        modes.push(mode);

        if (mode === "SZ") ranking++;

        return prisma.xRankPlacement.create({
          data: {
            playerName,
            mode: mode as "SZ" | "TC" | "RM" | "CB",
            month: 12,
            year: 2020,
            ranking,
            xPower: 3000 - i * 0.5,
            weapon: "Splattershot Jr.",
            player:
              i === 0 || i > 3
                ? {
                    create: {
                      switchAccountId: "" + i,
                      name: playerName,
                      user:
                        i === 0
                          ? {
                              connect: {
                                id: testUser.id,
                              },
                            }
                          : undefined,
                    },
                  }
                : {
                    connect: {
                      switchAccountId: "0",
                    },
                  },
          },
        });
      })
  );

  console.log("X Rank placements created");

  await Promise.all([
    prisma.build.create({
      data: {
        user: {
          connect: {
            id: testUser.id,
          },
        },
        weapon: "Splattershot Jr.",
        headAbilities: ["SS", "SS", "SS", "SS"],
        clothingAbilities: ["SS", "SS", "SS", "SS"],
        shoesAbilities: ["SS", "SS", "SS", "SS"],
        title: "Amazing test build",
        description: "Just testing.",
        top500: true,
        jpn: false,
        abilityPoints: {
          SS: 57,
        },
      },
    }),
    prisma.build.create({
      data: {
        user: {
          connect: {
            id: testUser.id,
          },
        },
        weapon: "Splattershot Jr.",
        headAbilities: ["QR", "QR", "QR", "QR"],
        clothingAbilities: ["QR", "QR", "QR", "QR"],
        shoesAbilities: ["QR", "QR", "QR", "QR"],
        title: "SECOND test build",
        description: "Just testing again.",
        top500: true,
        jpn: false,
        abilityPoints: {
          SS: 57,
        },
      },
    }),
  ]);

  console.log("Builds created");
};

main()
  .catch((e) => console.error(e))
  .finally(async () => {
    await prisma.$disconnect();
  });

export {};