atproto/packages/pds/tests/seeds/basic.ts

import { EXAMPLE_LABELER, SeedClient, TestBsky } from '@atproto/dev-env'
import { ids } from '../../src/lexicon/lexicons'
import usersSeed from './users'

export default async (
  sc: SeedClient,
  opts?: { inviteCode?: string; addModLabels?: TestBsky },
) => {
  await usersSeed(sc, opts)

  const alice = sc.dids.alice
  const bob = sc.dids.bob
  const carol = sc.dids.carol
  const dan = sc.dids.dan
  const createdAtMicroseconds = () => ({
    createdAt: new Date().toISOString().replace('Z', '000Z'), // microseconds
  })
  const createdAtTimezone = () => ({
    createdAt: new Date().toISOString().replace('Z', '+00:00'), // iso timezone format
  })

  await sc.follow(alice, bob)
  await sc.follow(alice, carol)
  await sc.follow(alice, dan)
  await sc.follow(carol, alice)
  await sc.follow(bob, alice)
  await sc.follow(bob, carol, createdAtMicroseconds())
  await sc.follow(dan, bob, createdAtTimezone())
  await sc.post(alice, posts.alice[0], undefined, undefined, undefined, {
    labels: {
      $type: 'com.atproto.label.defs#selfLabels',
      values: [{ val: 'self-label' }],
    },
  })
  await sc.post(bob, posts.bob[0], undefined, undefined, undefined, {
    langs: ['en-US', 'i-klingon'],
  })
  const img1 = await sc.uploadFile(
    carol,
    '../dev-env/assets/key-landscape-small.jpg',
    'image/jpeg',
  )
  const img2 = await sc.uploadFile(
    carol,
    '../dev-env/assets/key-alt.jpg',
    'image/jpeg',
  )
  await sc.post(
    carol,
    posts.carol[0],
    undefined,
    [img1, img2], // Contains both images and a quote
    sc.posts[bob][0].ref,
  )
  await sc.post(dan, posts.dan[0])
  await sc.post(
    dan,
    posts.dan[1],
    [
      {
        index: { byteStart: 0, byteEnd: 18 },
        features: [
          {
            $type: `${ids.AppBskyRichtextFacet}#mention`,
            did: alice,
          },
        ],
      },
    ],
    undefined,
    sc.posts[carol][0].ref, // This post contains an images embed
  )
  await sc.post(
    alice,
    posts.alice[1],
    undefined,
    undefined,
    undefined,
    createdAtMicroseconds(),
  )
  await sc.post(
    bob,
    posts.bob[1],
    undefined,
    undefined,
    undefined,
    createdAtTimezone(),
  )
  await sc.post(
    alice,
    posts.alice[2],
    undefined,
    undefined,
    sc.posts[dan][1].ref, // This post contains a record embed which contains an images embed
  )
  await sc.like(bob, sc.posts[alice][1].ref)
  await sc.like(bob, sc.posts[alice][2].ref)
  await sc.like(carol, sc.posts[alice][1].ref)
  await sc.like(carol, sc.posts[alice][2].ref)
  await sc.like(dan, sc.posts[alice][1].ref)
  await sc.like(alice, sc.posts[carol][0].ref, createdAtMicroseconds())
  await sc.like(bob, sc.posts[carol][0].ref, createdAtTimezone())

  const replyImg = await sc.uploadFile(
    bob,
    '../dev-env/assets/key-landscape-small.jpg',
    'image/jpeg',
  )
  // must ensure ordering of replies in indexing
  await sc.network.processAll()
  await sc.reply(
    bob,
    sc.posts[alice][1].ref,
    sc.posts[alice][1].ref,
    replies.bob[0],
    undefined,
    [replyImg],
  )
  await sc.reply(
    carol,
    sc.posts[alice][1].ref,
    sc.posts[alice][1].ref,
    replies.carol[0],
  )
  await sc.network.processAll()
  const alicesReplyToBob = await sc.reply(
    alice,
    sc.posts[alice][1].ref,
    sc.replies[bob][0].ref,
    replies.alice[0],
  )
  await sc.repost(carol, sc.posts[dan][1].ref)
  await sc.repost(dan, sc.posts[alice][1].ref)
  await sc.repost(dan, alicesReplyToBob.ref)

  if (opts?.addModLabels) {
    await createLabel(opts.addModLabels, { did: dan, val: 'repo-action-label' })
  }

  return sc
}

export const posts = {
  alice: ['hey there', 'again', 'yoohoo label_me'],
  bob: ['bob back at it again!', 'bobby boy here', 'yoohoo'],
  carol: ['hi im carol'],
  dan: ['dan here!', '@alice.bluesky.xyz is the best'],
}

export const replies = {
  alice: ['thanks bob'],
  bob: ['hear that label_me label_me_2'],
  carol: ['of course'],
}

const createLabel = async (
  bsky: TestBsky,
  opts: { did: string; val: string },
) => {
  await bsky.db.db
    .insertInto('label')
    .values({
      uri: opts.did,
      cid: '',
      val: opts.val,
      cts: new Date().toISOString(),
      neg: false,
      src: EXAMPLE_LABELER,
    })
    .execute()
}