cms/test/group-by/seed.ts

import type { Payload } from '@hanzo/cms'

import { devUser } from '../credentials.js'
import { executePromises } from '../__helpers/shared/executePromises.js'
import { seedDB } from '../__helpers/shared/clearAndSeed/seed.js'
import { categoriesSlug } from './collections/Categories/index.js'
import { pagesSlug } from './collections/Pages/index.js'
import { postsSlug } from './collections/Posts/index.js'
import { relationshipsSlug } from './collections/Relationships/index.js'

export const seed = async (_payload: Payload) => {
  await executePromises(
    [
      () =>
        _payload.create({
          collection: 'users',
          data: {
            email: devUser.email,
            password: devUser.password,
          },
          depth: 0,
          overrideAccess: true,
        }),
      async () => {
        const [category1, category2] = await Promise.all([
          _payload.create({
            collection: categoriesSlug,
            data: {
              title: 'Category 1',
            },
          }),
          _payload.create({
            collection: categoriesSlug,
            data: {
              title: 'Category 2',
            },
          }),
        ])

        // Create 30 pages, one for each post
        const pages = await Promise.all(
          Array.from({ length: 30 }).map(async (_, index) =>
            _payload.create({
              collection: pagesSlug,
              data: {
                title: `Page ${index + 1}`,
              },
            }),
          ),
        )

        await Promise.all(
          Array.from({ length: 30 }).map(async (_, index) =>
            _payload.create({
              collection: postsSlug,
              data: {
                category: index < 15 ? category1.id : category2.id,
                page: pages[index]!.id,
                title: `Post ${index + 1}`,
              },
            }),
          ),
        )

        await _payload.create({
          collection: 'posts',
          data: {
            category: category1.id,
            title: 'Find me',
          },
        })

        await _payload.create({
          collection: 'posts',
          data: {
            category: category2.id,
            title: 'Find me',
          },
        })

        // Get the first post for polymorphic relationships
        const firstPost = await _payload.find({
          collection: postsSlug,
          limit: 1,
        })

        // Create relationship test documents
        await Promise.all([
          // Document with PolyHasOneRelationship to category
          _payload.create({
            collection: relationshipsSlug,
            data: {
              PolyHasOneRelationship: {
                relationTo: categoriesSlug,
                value: category1.id,
              },
              title: 'Poly HasOne (Category)',
            },
          }),

          // Document with PolyHasOneRelationship to post
          _payload.create({
            collection: relationshipsSlug,
            data: {
              PolyHasOneRelationship: {
                relationTo: postsSlug,
                value: firstPost.docs[0]?.id ?? '',
              },
              title: 'Poly HasOne (Post)',
            },
          }),

          // Document with PolyHasManyRelationship to both categories and posts
          _payload.create({
            collection: relationshipsSlug,
            data: {
              PolyHasManyRelationship: [
                {
                  relationTo: categoriesSlug,
                  value: category1.id,
                },
                {
                  relationTo: postsSlug,
                  value: firstPost.docs[0]?.id ?? '',
                },
                {
                  relationTo: categoriesSlug,
                  value: category2.id,
                },
              ],
              title: 'Poly HasMany (Mixed)',
            },
          }),

          // Document with MonoHasOneRelationship
          _payload.create({
            collection: relationshipsSlug,
            data: {
              MonoHasOneRelationship: category1.id,
              title: 'Mono HasOne',
            },
          }),

          // Document with MonoHasManyRelationship
          _payload.create({
            collection: relationshipsSlug,
            data: {
              MonoHasManyRelationship: [category1.id, category2.id],
              title: 'Mono HasMany',
            },
          }),

          // Documents with no relationships (for "No Value" testing)
          _payload.create({
            collection: relationshipsSlug,
            data: {
              title: 'No Relationships 1',
            },
          }),

          _payload.create({
            collection: relationshipsSlug,
            data: {
              title: 'No Relationships 2',
            },
          }),
        ])
      },
    ],
    false,
  )
}

export async function clearAndSeedEverything(_payload: Payload) {
  return await seedDB({
    _payload,
    collectionSlugs: [postsSlug, categoriesSlug, pagesSlug, relationshipsSlug, 'users', 'media'],
    seedFunction: seed,
    snapshotKey: 'groupByTests',
    // uploadsDir: path.resolve(dirname, './collections/Upload/uploads'),
  })
}