@atproto/dev-env 0.3.34-rc.0 → 0.3.35

package/src/mock/index.ts

@@ -1,5 +1,5 @@
 import { AtUri } from '@atproto/syntax'
-import { COM_ATPROTO_MODERATION, AtpAgent } from '@atproto/api'
+import AtpAgent, { COM_ATPROTO_MODERATION } from '@atproto/api'
 import { Database } from '@atproto/bsky'
 import { EXAMPLE_LABELER, RecordRef, TestNetwork } from '../index'
 import { postTexts, replyTexts } from './data'
@@ -30,67 +30,90 @@ export async function generateMockSetup(env: TestNetwork) {
     throw new Error('Not found')
   }
 
-  const loggedOut = env.pds.getClient()
-
-  const users = [
+  const clients = {
+    loggedout: env.pds.getClient(),
+    alice: env.pds.getClient(),
+    bob: env.pds.getClient(),
+    carla: env.pds.getClient(),
+  }
+  interface User {
+    email: string
+    did: string
+    handle: string
+    password: string
+    agent: AtpAgent
+  }
+  const users: User[] = [
     {
       email: 'alice@test.com',
+      did: '',
       handle: `alice.test`,
       password: 'hunter2',
+      agent: clients.alice,
     },
     {
       email: 'bob@test.com',
+      did: '',
       handle: `bob.test`,
       password: 'hunter2',
+      agent: clients.bob,
     },
     {
       email: 'carla@test.com',
+      did: '',
       handle: `carla.test`,
       password: 'hunter2',
+      agent: clients.carla,
     },
   ]
+  const alice = users[0]
+  const bob = users[1]
+  const carla = users[2]
 
-  const userAgents = await Promise.all(
-    users.map(async (user, i) => {
-      const client: AtpAgent = env.pds.getClient()
-      await client.createAccount(user)
-      client.assertAuthenticated()
-      await client.app.bsky.actor.profile.create(
-        { repo: client.did },
-        {
-          displayName: ucfirst(user.handle).slice(0, -5),
-          description: `Test user ${i}`,
-        },
-      )
-      return client
-    }),
-  )
-
-  const [alice, bob, carla] = userAgents
+  let _i = 1
+  for (const user of users) {
+    const res = await clients.loggedout.api.com.atproto.server.createAccount({
+      email: user.email,
+      handle: user.handle,
+      password: user.password,
+    })
+    user.agent.api.setHeader('Authorization', `Bearer ${res.data.accessJwt}`)
+    user.did = res.data.did
+    await user.agent.api.app.bsky.actor.profile.create(
+      { repo: user.did },
+      {
+        displayName: ucfirst(user.handle).slice(0, -5),
+        description: `Test user ${_i++}`,
+      },
+    )
+  }
 
   // Create moderator accounts
-  const triageRes = await loggedOut.com.atproto.server.createAccount({
-    email: 'triage@test.com',
-    handle: 'triage.test',
-    password: 'triage-pass',
-  })
+  const triageRes =
+    await clients.loggedout.api.com.atproto.server.createAccount({
+      email: 'triage@test.com',
+      handle: 'triage.test',
+      password: 'triage-pass',
+    })
   await env.ozone.addTriageDid(triageRes.data.did)
-  const modRes = await loggedOut.com.atproto.server.createAccount({
+  const modRes = await clients.loggedout.api.com.atproto.server.createAccount({
     email: 'mod@test.com',
     handle: 'mod.test',
     password: 'mod-pass',
   })
   await env.ozone.addModeratorDid(modRes.data.did)
-  const adminRes = await loggedOut.com.atproto.server.createAccount({
-    email: 'admin-mod@test.com',
-    handle: 'admin-mod.test',
-    password: 'admin-mod-pass',
-  })
+  const adminRes = await clients.loggedout.api.com.atproto.server.createAccount(
+    {
+      email: 'admin-mod@test.com',
+      handle: 'admin-mod.test',
+      password: 'admin-mod-pass',
+    },
+  )
   await env.ozone.addAdminDid(adminRes.data.did)
 
   // Report one user
-  const reporter = picka(userAgents)
-  await reporter.com.atproto.moderation.createReport({
+  const reporter = picka(users)
+  await reporter.agent.api.com.atproto.moderation.createReport({
     reasonType: picka([
       COM_ATPROTO_MODERATION.DefsReasonSpam,
       COM_ATPROTO_MODERATION.DefsReasonOther,
@@ -98,16 +121,16 @@ export async function generateMockSetup(env: TestNetwork) {
     reason: picka(["Didn't look right to me", undefined, undefined]),
     subject: {
       $type: 'com.atproto.admin.defs#repoRef',
-      did: picka(userAgents).did,
+      did: picka(users).did,
     },
   })
 
   // everybody follows everybody
-  const follow = async (author: AtpAgent, subject: AtpAgent) => {
-    await author.app.bsky.graph.follow.create(
-      { repo: author.accountDid },
+  const follow = async (author: User, subject: User) => {
+    await author.agent.api.app.bsky.graph.follow.create(
+      { repo: author.did },
       {
-        subject: subject.accountDid,
+        subject: subject.did,
         createdAt: date.next().value,
       },
     )
@@ -122,8 +145,8 @@ export async function generateMockSetup(env: TestNetwork) {
   // a set of posts and reposts
   const posts: { uri: string; cid: string }[] = []
   for (let i = 0; i < postTexts.length; i++) {
-    const author = picka(userAgents)
-    const post = await author.app.bsky.feed.post.create(
+    const author = picka(users)
+    const post = await author.agent.api.app.bsky.feed.post.create(
       { repo: author.did },
       {
         text: postTexts[i],
@@ -132,8 +155,8 @@ export async function generateMockSetup(env: TestNetwork) {
     )
     posts.push(post)
     if (rand(10) === 0) {
-      const reposter = picka(userAgents)
-      await reposter.app.bsky.feed.repost.create(
+      const reposter = picka(users)
+      await reposter.agent.api.app.bsky.feed.repost.create(
         { repo: reposter.did },
         {
           subject: picka(posts),
@@ -142,8 +165,8 @@ export async function generateMockSetup(env: TestNetwork) {
       )
     }
     if (rand(6) === 0) {
-      const reporter = picka(userAgents)
-      await reporter.com.atproto.moderation.createReport({
+      const reporter = picka(users)
+      await reporter.agent.api.com.atproto.moderation.createReport({
         reasonType: picka([
           COM_ATPROTO_MODERATION.DefsReasonSpam,
           COM_ATPROTO_MODERATION.DefsReasonOther,
@@ -160,11 +183,11 @@ export async function generateMockSetup(env: TestNetwork) {
 
   // make some naughty posts & label them
   const file = Buffer.from(labeledImgB64, 'base64')
-  const uploadedImg = await bob.com.atproto.repo.uploadBlob(file, {
+  const uploadedImg = await bob.agent.api.com.atproto.repo.uploadBlob(file, {
     encoding: 'image/png',
   })
-  const labeledPost = await bob.app.bsky.feed.post.create(
-    { repo: bob.accountDid },
+  const labeledPost = await bob.agent.api.app.bsky.feed.post.create(
+    { repo: bob.did },
     {
       text: 'naughty post',
       embed: {
@@ -180,8 +203,8 @@ export async function generateMockSetup(env: TestNetwork) {
     },
   )
 
-  const filteredPost = await bob.app.bsky.feed.post.create(
-    { repo: bob.accountDid },
+  const filteredPost = await bob.agent.api.app.bsky.feed.post.create(
+    { repo: bob.did },
     {
       text: 'reallly bad post should be deleted',
       createdAt: date.next().value,
@@ -203,13 +226,13 @@ export async function generateMockSetup(env: TestNetwork) {
   for (let i = 0; i < 100; i++) {
     const targetUri = picka(posts).uri
     const urip = new AtUri(targetUri)
-    const target = await alice.app.bsky.feed.post.get({
+    const target = await alice.agent.api.app.bsky.feed.post.get({
       repo: urip.host,
       rkey: urip.rkey,
     })
-    const author = picka(userAgents)
+    const author = picka(users)
     posts.push(
-      await author.app.bsky.feed.post.create(
+      await author.agent.api.app.bsky.feed.post.create(
         { repo: author.did },
         {
           text: picka(replyTexts),
@@ -225,9 +248,9 @@ export async function generateMockSetup(env: TestNetwork) {
 
   // a set of likes
   for (const post of posts) {
-    for (const user of userAgents) {
+    for (const user of users) {
       if (rand(3) === 0) {
-        await user.app.bsky.feed.like.create(
+        await user.agent.api.app.bsky.feed.like.create(
           { repo: user.did },
           {
             subject: post,
@@ -239,11 +262,7 @@ export async function generateMockSetup(env: TestNetwork) {
   }
 
   // a couple feed generators that returns some posts
-  const fg1Uri = AtUri.make(
-    alice.accountDid,
-    'app.bsky.feed.generator',
-    'alice-favs',
-  )
+  const fg1Uri = AtUri.make(alice.did, 'app.bsky.feed.generator', 'alice-favs')
   const fg1 = await env.createFeedGen({
     [fg1Uri.toString()]: async () => {
       const feed = posts
@@ -258,11 +277,14 @@ export async function generateMockSetup(env: TestNetwork) {
     },
   })
   const avatarImg = Buffer.from(blurHashB64, 'base64')
-  const avatarRes = await alice.com.atproto.repo.uploadBlob(avatarImg, {
-    encoding: 'image/png',
-  })
-  const fgAliceRes = await alice.app.bsky.feed.generator.create(
-    { repo: alice.accountDid, rkey: fg1Uri.rkey },
+  const avatarRes = await alice.agent.api.com.atproto.repo.uploadBlob(
+    avatarImg,
+    {
+      encoding: 'image/png',
+    },
+  )
+  const fgAliceRes = await alice.agent.api.app.bsky.feed.generator.create(
+    { repo: alice.did, rkey: fg1Uri.rkey },
     {
       did: fg1.did,
       displayName: 'alices feed',
@@ -272,8 +294,8 @@ export async function generateMockSetup(env: TestNetwork) {
     },
   )
 
-  await alice.app.bsky.feed.post.create(
-    { repo: alice.accountDid },
+  await alice.agent.api.app.bsky.feed.post.create(
+    { repo: alice.did },
     {
       text: 'check out my algorithm!',
       embed: {
@@ -284,7 +306,7 @@ export async function generateMockSetup(env: TestNetwork) {
     },
   )
   for (const user of [alice, bob, carla]) {
-    await user.app.bsky.feed.like.create(
+    await user.agent.api.app.bsky.feed.like.create(
       { repo: user.did },
       {
         subject: fgAliceRes,
@@ -293,11 +315,7 @@ export async function generateMockSetup(env: TestNetwork) {
     )
   }
 
-  const fg2Uri = AtUri.make(
-    bob.accountDid,
-    'app.bsky.feed.generator',
-    'bob-redux',
-  )
+  const fg2Uri = AtUri.make(bob.did, 'app.bsky.feed.generator', 'bob-redux')
   const fg2 = await env.createFeedGen({
     [fg2Uri.toString()]: async () => {
       const feed = posts
@@ -311,8 +329,8 @@ export async function generateMockSetup(env: TestNetwork) {
       }
     },
   })
-  const fgBobRes = await bob.app.bsky.feed.generator.create(
-    { repo: bob.accountDid, rkey: fg2Uri.rkey },
+  const fgBobRes = await bob.agent.api.app.bsky.feed.generator.create(
+    { repo: bob.did, rkey: fg2Uri.rkey },
     {
       did: fg2.did,
       displayName: 'Bobby boy hot new algo',
@@ -320,8 +338,8 @@ export async function generateMockSetup(env: TestNetwork) {
     },
   )
 
-  await alice.app.bsky.feed.post.create(
-    { repo: alice.accountDid },
+  await alice.agent.api.app.bsky.feed.post.create(
+    { repo: alice.did },
     {
       text: `bobs feed is neat too`,
       embed: {
@@ -334,13 +352,14 @@ export async function generateMockSetup(env: TestNetwork) {
 
   // create a labeler account
   {
-    const labeler = env.pds.getClient()
-    const res = await labeler.createAccount({
+    const res = await clients.loggedout.api.com.atproto.server.createAccount({
       email: 'labeler@test.com',
       handle: 'labeler.test',
       password: 'hunter2',
     })
-    await labeler.app.bsky.actor.profile.create(
+    const agent = env.pds.getClient()
+    agent.api.setHeader('Authorization', `Bearer ${res.data.accessJwt}`)
+    await agent.api.app.bsky.actor.profile.create(
       { repo: res.data.did },
       {
         displayName: 'Test Labeler',
@@ -348,7 +367,7 @@ export async function generateMockSetup(env: TestNetwork) {
       },
     )
 
-    await labeler.app.bsky.labeler.service.create(
+    await agent.api.app.bsky.labeler.service.create(
       { repo: res.data.did, rkey: 'self' },
       {
         policies: {
@@ -436,25 +455,25 @@ export async function generateMockSetup(env: TestNetwork) {
       },
     )
     await createLabel(env.bsky.db, {
-      uri: alice.accountDid,
+      uri: alice.did,
       cid: '',
       val: 'rude',
       src: res.data.did,
     })
     await createLabel(env.bsky.db, {
-      uri: `at://${alice.accountDid}/app.bsky.feed.generator/alice-favs`,
+      uri: `at://${alice.did}/app.bsky.feed.generator/alice-favs`,
       cid: '',
       val: 'cool',
       src: res.data.did,
     })
     await createLabel(env.bsky.db, {
-      uri: bob.accountDid,
+      uri: bob.did,
       cid: '',
       val: 'cool',
       src: res.data.did,
     })
     await createLabel(env.bsky.db, {
-      uri: carla.accountDid,
+      uri: carla.did,
       cid: '',
       val: 'spam',
       src: res.data.did,
@@ -463,8 +482,8 @@ export async function generateMockSetup(env: TestNetwork) {
 
   // Create lists and add people to the lists
   {
-    const flowerLovers = await alice.app.bsky.graph.list.create(
-      { repo: alice.accountDid },
+    const flowerLovers = await alice.agent.api.app.bsky.graph.list.create(
+      { repo: alice.did },
       {
         name: 'Flower Lovers',
         purpose: 'app.bsky.graph.defs#curatelist',
@@ -472,8 +491,8 @@ export async function generateMockSetup(env: TestNetwork) {
         description: 'A list of posts about flowers',
       },
     )
-    const labelHaters = await bob.app.bsky.graph.list.create(
-      { repo: bob.accountDid },
+    const labelHaters = await bob.agent.api.app.bsky.graph.list.create(
+      { repo: bob.did },
       {
         name: 'Label Haters',
         purpose: 'app.bsky.graph.defs#modlist',
@@ -481,18 +500,18 @@ export async function generateMockSetup(env: TestNetwork) {
         description: 'A list of people who hate labels',
       },
     )
-    await alice.app.bsky.graph.listitem.create(
-      { repo: alice.accountDid },
+    await alice.agent.api.app.bsky.graph.listitem.create(
+      { repo: alice.did },
       {
-        subject: bob.accountDid,
+        subject: bob.did,
         createdAt: new Date().toISOString(),
         list: new RecordRef(flowerLovers.uri, flowerLovers.cid).uriStr,
       },
     )
-    await bob.app.bsky.graph.listitem.create(
-      { repo: bob.accountDid },
+    await bob.agent.api.app.bsky.graph.listitem.create(
+      { repo: bob.did },
       {
-        subject: alice.accountDid,
+        subject: alice.did,
         createdAt: new Date().toISOString(),
         list: new RecordRef(labelHaters.uri, labelHaters.cid).uriStr,
       },

package/src/moderator-client.ts

@@ -1,5 +1,4 @@
-import {
-  AtpAgent,
+import AtpAgent, {
   ToolsOzoneModerationEmitEvent as EmitModerationEvent,
   ToolsOzoneModerationQueryStatuses as QueryModerationStatuses,
   ToolsOzoneModerationQueryEvents as QueryModerationEvents,
@@ -18,7 +17,7 @@ export class ModeratorClient {
   }
 
   async getEvent(id: number, role?: ModLevel) {
-    const result = await this.agent.tools.ozone.moderation.getEvent(
+    const result = await this.agent.api.tools.ozone.moderation.getEvent(
       { id },
       {
         headers: await this.ozone.modHeaders(role),
@@ -28,7 +27,7 @@ export class ModeratorClient {
   }
 
   async queryStatuses(input: QueryStatusesParams, role?: ModLevel) {
-    const result = await this.agent.tools.ozone.moderation.queryStatuses(
+    const result = await this.agent.api.tools.ozone.moderation.queryStatuses(
       input,
       {
         headers: await this.ozone.modHeaders(role),
@@ -38,9 +37,12 @@ export class ModeratorClient {
   }
 
   async queryEvents(input: QueryEventsParams, role?: ModLevel) {
-    const result = await this.agent.tools.ozone.moderation.queryEvents(input, {
-      headers: await this.ozone.modHeaders(role),
-    })
+    const result = await this.agent.api.tools.ozone.moderation.queryEvents(
+      input,
+      {
+        headers: await this.ozone.modHeaders(role),
+      },
+    )
     return result.data
   }
 
@@ -62,7 +64,7 @@ export class ModeratorClient {
       reason = 'X',
       createdBy = 'did:example:admin',
     } = opts
-    const result = await this.agent.tools.ozone.moderation.emitEvent(
+    const result = await this.agent.api.tools.ozone.moderation.emitEvent(
       { event, subject, subjectBlobCids, createdBy, reason },
       {
         encoding: 'application/json',
@@ -82,7 +84,7 @@ export class ModeratorClient {
     role?: ModLevel,
   ) {
     const { subject, reason = 'X', createdBy = 'did:example:admin' } = opts
-    const result = await this.agent.tools.ozone.moderation.emitEvent(
+    const result = await this.agent.api.tools.ozone.moderation.emitEvent(
       {
         subject,
         event: {

package/src/ozone-service-profile.ts

@@ -18,9 +18,16 @@ export class OzoneServiceProfile {
   }
 
   async createDidAndKey() {
-    await this.thirdPartyPdsClient.createAccount(this.modUserDetails)
+    const modUser =
+      await this.thirdPartyPdsClient.api.com.atproto.server.createAccount(
+        this.modUserDetails,
+      )
+    await this.thirdPartyPdsClient.login({
+      identifier: this.modUserDetails.handle,
+      password: this.modUserDetails.password,
+    })
 
-    this.did = this.thirdPartyPdsClient.accountDid
+    this.did = modUser.data.did
     this.key = await Secp256k1Keypair.create({ exportable: true })
     return { did: this.did, key: this.key }
   }
@@ -34,7 +41,7 @@ export class OzoneServiceProfile {
       throw new Error('No DID/key found!')
     }
     const pdsClient = pds.getClient()
-    const describeRes = await pdsClient.com.atproto.server.describeServer()
+    const describeRes = await pdsClient.api.com.atproto.server.describeServer()
     const newServerDid = describeRes.data.did
 
     const serviceJwtRes =
@@ -43,22 +50,23 @@ export class OzoneServiceProfile {
       })
     const serviceJwt = serviceJwtRes.data.token
 
-    await pdsClient.createAccount(
-      {
-        ...this.modUserDetails,
-        ...userDetails,
-        did: this.did,
-      },
-      {
-        headers: { authorization: `Bearer ${serviceJwt}` },
-        encoding: 'application/json',
-      },
-    )
+    const accountResponse =
+      await pdsClient.api.com.atproto.server.createAccount(
+        {
+          ...this.modUserDetails,
+          ...userDetails,
+          did: this.did,
+        },
+        {
+          headers: { authorization: `Bearer ${serviceJwt}` },
+          encoding: 'application/json',
+        },
+      )
 
-    // For some reason, the tests fail if the client uses the PDS URL to make
-    // its requests. This is a workaround to make the tests pass by simulating
-    // old behavior (that was not relying on the session management).
-    pdsClient.sessionManager.pdsUrl = undefined
+    pdsClient.api.setHeader(
+      'Authorization',
+      `Bearer ${accountResponse.data.accessJwt}`,
+    )
 
     const getDidCredentials =
       await pdsClient.com.atproto.identity.getRecommendedDidCredentials()
@@ -95,9 +103,9 @@ export class OzoneServiceProfile {
       operation: plcOp.data.operation,
     })
 
-    await pdsClient.com.atproto.server.activateAccount()
+    await pdsClient.api.com.atproto.server.activateAccount()
 
-    await pdsClient.app.bsky.actor.profile.create(
+    await pdsClient.api.app.bsky.actor.profile.create(
       { repo: this.did },
       {
         displayName: 'Dev-env Moderation',
@@ -105,7 +113,7 @@ export class OzoneServiceProfile {
       },
     )
 
-    await pdsClient.app.bsky.labeler.service.create(
+    await pdsClient.api.app.bsky.labeler.service.create(
       { repo: this.did, rkey: 'self' },
       {
         policies: {

package/src/ozone.ts

@@ -6,7 +6,7 @@ import { AtpAgent } from '@atproto/api'
 import { createServiceJwt } from '@atproto/xrpc-server'
 import { Keypair, Secp256k1Keypair } from '@atproto/crypto'
 import { DidAndKey, OzoneConfig } from './types'
-import { ADMIN_PASSWORD, EXAMPLE_LABELER } from './const'
+import { ADMIN_PASSWORD } from './const'
 import { createDidAndKey } from './util'
 import { ModeratorClient } from './moderator-client'
 
@@ -103,10 +103,8 @@ export class TestOzone {
     return this.server.ctx
   }
 
-  getClient(): AtpAgent {
-    const agent = new AtpAgent({ service: this.url })
-    agent.configureLabelers([EXAMPLE_LABELER])
-    return agent
+  getClient() {
+    return new AtpAgent({ service: this.url })
   }
 
   getModClient() {

package/src/pds.ts

@@ -73,7 +73,7 @@ export class TestPds {
 
   getClient(): AtpAgent {
     const agent = new AtpAgent({ service: this.url })
-    agent.configureLabelers([EXAMPLE_LABELER])
+    agent.configureLabelersHeader([EXAMPLE_LABELER])
     return agent
   }