seed.ts

import { faker } from '@faker-js/faker'
import { createId } from '@paralleldrive/cuid2'
import { PrismaClient, type User } from '@prisma/client'

import { CATEGORIES } from '../src/lib/constants'
import {
  ARCTIC_GOODS,
  ENBESA_GOODS,
  NEW_WORLD_GOODS,
  OLD_WORLD_GOODS,
  REGIONS_1800,
} from '../src/lib/constants/1800/data'

const prisma = new PrismaClient()

const categories = Object.values(CATEGORIES)
const region = Object.values(REGIONS_1800)
const goods = [OLD_WORLD_GOODS, NEW_WORLD_GOODS, ENBESA_GOODS, ARCTIC_GOODS]

export const generateUserData = (
  length: number,
): Omit<User, 'emailVerified' | 'image'>[] =>
  Array.from({ length }, () => {
    const firstName = faker.person.firstName()
    const lastName = faker.person.lastName()
    const username = faker.internet
      .username({ firstName, lastName })
      .replace(/\./g, '_')

    return {
      biography: faker.person.bio(),
      email: faker.internet.email({
        firstName,
        lastName,
        provider: 'example.fakerjs.dev',
      }),
      id: createId(),
      name: faker.person.fullName({ firstName, lastName }),
      username,
      usernameURL: username.toLowerCase(),
    }
  })

export const generateStampData = (
  length: number,
  users: Omit<User, 'emailVerified' | 'image'>[],
) =>
  Array.from({ length: length }, (_, index) => {
    const categoryIdx = Math.floor(Math.random() * categories.length)
    const rndCategory = categories[categoryIdx]

    const regionIdx = Math.floor(Math.random() * region.length)
    const rndRegion = region[regionIdx]

    const getGood = () => {
      const goodsInRegion = goods[regionIdx]
      const goodIdx = Math.floor(Math.random() * goodsInRegion.length)
      const rndGood = goodsInRegion[goodIdx]
      return rndGood.toLowerCase()
    }
    const getCapital = () => {
      if (rndCategory === 'island' && rndRegion === 'new world') {
        return 'manola'
      }
      return null
    }

    const timestamp = faker.date.past()
    return {
      capital: getCapital(),
      category: rndCategory,
      changedAt: timestamp,
      createdAt: timestamp,
      downloads: faker.number.int({ max: 1000000, min: 0 }),
      game: '1800',
      good: rndCategory === 'production' ? getGood() : null,
      id: createId(),
      markdownDescription: `<h2> Stamp-${index} </h2>`,
      modded: index % 20 === 0 ? true : false,
      region: rndRegion,
      stampFileUrl: 'http://localhost:3000/test-stamp.zip',
      title: `Stamp-${index}-${faker.lorem.words({ max: 12, min: 0 })}`,
      unsafeDescription: `## Stamp-${index}`,
      updatedAt: timestamp,
      userId: users[index % users.length].id,
    }
  })

export const generatePlaceHoldImages = () => {
  return [1, 2].map((image) => {
    const id = createId()
    return {
      id,
      largeUrl: `https://placehold.co/1024x576.png?text=Large${image}\\n${id}`,
      mediumUrl: `https://placehold.co/750x421.png?text=Medium${image}\\n${id}`,
      originalUrl: `https://placehold.co/2000x2000.png?text=Original${image}\\n${id}`,
      smallUrl: `https://placehold.co/500x281.png?text=Small${image}\\n${id}`,
      thumbnailUrl: `https://placehold.co/250x250.png?text=Thumbnail${image}\\n${id}`,
    }
  })
}
async function seed() {
  console.time('Database seed time elapsed')
  const userData = generateUserData(100)
  const stampData = generateStampData(1000, userData)
  await prisma.user.createMany({ data: userData })

  await prisma.stamp.createMany({
    data: stampData,
  })

  for (const stamp of stampData) {
    const selectUser = userData[Math.floor(Math.random() * userData.length)]
    await prisma.stamp.update({
      data: {
        images: {
          createMany: {
            data: generatePlaceHoldImages(),
          },
        },
        likedBy: {
          connect: { id: selectUser.id },
        },
      },
      where: {
        id: stamp.id,
      },
    })
  }
  console.timeEnd('Database seed time elapsed')
  console.log('Seed completed successfully')
}

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