@atproto/ozone 0.1.25 → 0.1.27

package/src/team/index.ts

@@ -0,0 +1,213 @@
+import Database from '../db'
+import { Selectable } from 'kysely'
+import { Member } from '../db/schema/member'
+import { Member as TeamMember } from '../lexicon/types/tools/ozone/team/defs'
+import { ProfileViewDetailed } from '../lexicon/types/app/bsky/actor/defs'
+import { InvalidRequestError } from '@atproto/xrpc-server'
+import { chunkArray } from '@atproto/common'
+import AppContext from '../context'
+import { httpLogger } from '../logger'
+
+export type TeamServiceCreator = (db: Database) => TeamService
+
+export class TeamService {
+  constructor(public db: Database) {}
+
+  static creator() {
+    return (db: Database) => new TeamService(db)
+  }
+
+  async list({
+    cursor,
+    limit = 25,
+  }: {
+    cursor?: string
+    limit?: number
+  }): Promise<{ members: Selectable<Member>[]; cursor?: string }> {
+    let builder = this.db.db.selectFrom('member').selectAll()
+    if (cursor) {
+      builder = builder.where('createdAt', '>', new Date(cursor))
+    }
+    const members = await builder
+      .limit(limit)
+      .orderBy('createdAt', 'asc')
+      .execute()
+
+    return { members, cursor: members.at(-1)?.createdAt.toISOString() }
+  }
+
+  async create({
+    role,
+    did,
+    disabled,
+    updatedAt,
+    createdAt,
+    lastUpdatedBy,
+  }: Omit<Selectable<Member>, 'createdAt' | 'updatedAt'> & {
+    createdAt?: Date
+    updatedAt?: Date
+  }): Promise<Selectable<Member>> {
+    const now = new Date()
+    const newMember = await this.db.db
+      .insertInto('member')
+      .values({
+        role,
+        did,
+        disabled,
+        lastUpdatedBy,
+        updatedAt: updatedAt || now,
+        createdAt: createdAt || now,
+      })
+      .returningAll()
+      .executeTakeFirstOrThrow()
+
+    return newMember
+  }
+
+  async upsert({
+    role,
+    did,
+    lastUpdatedBy,
+  }: Pick<
+    Selectable<Member>,
+    'role' | 'did' | 'lastUpdatedBy'
+  >): Promise<void> {
+    const now = new Date()
+    await this.db.db
+      .insertInto('member')
+      .values({
+        role,
+        did,
+        lastUpdatedBy,
+        disabled: false,
+        updatedAt: now,
+        createdAt: now,
+      })
+      .onConflict((oc) =>
+        oc.column('did').doUpdateSet({ role, updatedAt: now, lastUpdatedBy }),
+      )
+      .execute()
+  }
+
+  async update(
+    did: string,
+    updates: Partial<
+      Pick<
+        Selectable<Member>,
+        'role' | 'disabled' | 'lastUpdatedBy' | 'updatedAt'
+      >
+    >,
+  ): Promise<Selectable<Member>> {
+    const { role, disabled, lastUpdatedBy, updatedAt = new Date() } = updates
+    const updatedMember = await this.db.db
+      .updateTable('member')
+      .where('did', '=', did)
+      .set({
+        role,
+        disabled,
+        lastUpdatedBy,
+        updatedAt,
+      })
+      .returningAll()
+      .executeTakeFirstOrThrow()
+
+    return updatedMember
+  }
+
+  async delete(did: string): Promise<void> {
+    await this.db.db.deleteFrom('member').where('did', '=', did).execute()
+  }
+
+  async assertCanDelete(did: string): Promise<void> {
+    const memberExists = await this.doesMemberExist(did)
+
+    if (!memberExists) {
+      throw new InvalidRequestError('member not found', 'MemberNotFound')
+    }
+  }
+
+  async doesMemberExist(did: string): Promise<boolean> {
+    const member = await this.db.db
+      .selectFrom('member')
+      .select('did')
+      .where('did', '=', did)
+      .executeTakeFirst()
+
+    return !!member
+  }
+
+  async getMember(did: string): Promise<Selectable<Member> | undefined> {
+    const member = await this.db.db
+      .selectFrom('member')
+      .selectAll()
+      .where('did', '=', did)
+      .executeTakeFirst()
+
+    return member
+  }
+
+  getMemberRole(member?: Selectable<Member>) {
+    const isAdmin = member?.role === 'tools.ozone.team.defs#roleAdmin'
+    const isModerator =
+      isAdmin || member?.role === 'tools.ozone.team.defs#roleModerator'
+    const isTriage =
+      isModerator || member?.role === 'tools.ozone.team.defs#roleTriage'
+
+    return {
+      isModerator,
+      isAdmin,
+      isTriage,
+    }
+  }
+
+  // getProfiles() only allows 25 DIDs at a time so we need to query in chunks
+  async getProfiles(
+    dids: string[],
+    ctx: AppContext,
+  ): Promise<Map<string, ProfileViewDetailed>> {
+    const profiles = new Map<string, ProfileViewDetailed>()
+
+    try {
+      const headers = await ctx.appviewAuth()
+
+      for (const actors of chunkArray(dids, 25)) {
+        const { data } = await ctx.appviewAgent.api.app.bsky.actor.getProfiles(
+          { actors },
+          headers,
+        )
+
+        data.profiles.forEach((profile) => {
+          profiles.set(profile.did, profile)
+        })
+      }
+    } catch (error) {
+      httpLogger.error(
+        { error, dids },
+        'Failed to get profiles for team members',
+      )
+    }
+
+    return profiles
+  }
+
+  async view(
+    members: Selectable<Member>[],
+    ctx: AppContext,
+  ): Promise<TeamMember[]> {
+    const profiles = await this.getProfiles(
+      members.map(({ did }) => did),
+      ctx,
+    )
+    return members.map((member) => {
+      return {
+        did: member.did,
+        role: member.role,
+        disabled: member.disabled,
+        profile: profiles.get(member.did),
+        createdAt: member.createdAt.toISOString(),
+        updatedAt: member.updatedAt.toISOString(),
+        lastUpdatedBy: member.lastUpdatedBy,
+      }
+    })
+  }
+}

package/tests/__snapshots__/get-record.test.ts.snap

@@ -66,6 +66,7 @@ Object {
           },
           "size": 3976,
         },
+        "createdAt": "1970-01-01T00:00:00.000Z",
         "description": "its me!",
         "displayName": "ali",
         "labels": Object {
@@ -165,6 +166,7 @@ Object {
           },
           "size": 3976,
         },
+        "createdAt": "1970-01-01T00:00:00.000Z",
         "description": "its me!",
         "displayName": "ali",
         "labels": Object {

package/tests/__snapshots__/get-repo.test.ts.snap

@@ -52,6 +52,7 @@ Object {
         },
         "size": 3976,
       },
+      "createdAt": "1970-01-01T00:00:00.000Z",
       "description": "its me!",
       "displayName": "ali",
       "labels": Object {

package/tests/__snapshots__/get-starter-pack.test.ts.snap

@@ -0,0 +1,466 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`admin get starter pack view getStarterPack() gets a starterpack by uri 1`] = `
+Object {
+  "starterPack": Object {
+    "cid": "cids(0)",
+    "creator": Object {
+      "avatar": "https://bsky.public.url/img/avatar/plain/user(1)/cids(1)@jpeg",
+      "createdAt": "1970-01-01T00:00:00.000Z",
+      "did": "user(0)",
+      "displayName": "ali",
+      "handle": "alice.test",
+      "labels": Array [
+        Object {
+          "cid": "cids(2)",
+          "cts": "1970-01-01T00:00:00.000Z",
+          "src": "user(0)",
+          "uri": "record(3)",
+          "val": "self-label-a",
+        },
+        Object {
+          "cid": "cids(2)",
+          "cts": "1970-01-01T00:00:00.000Z",
+          "src": "user(0)",
+          "uri": "record(3)",
+          "val": "self-label-b",
+        },
+      ],
+      "viewer": Object {
+        "blockedBy": false,
+        "muted": false,
+      },
+    },
+    "feeds": Array [
+      Object {
+        "cid": "cids(3)",
+        "creator": Object {
+          "avatar": "https://bsky.public.url/img/avatar/plain/user(1)/cids(1)@jpeg",
+          "createdAt": "1970-01-01T00:00:00.000Z",
+          "description": "its me!",
+          "did": "user(0)",
+          "displayName": "ali",
+          "handle": "alice.test",
+          "indexedAt": "1970-01-01T00:00:00.000Z",
+          "labels": Array [
+            Object {
+              "cid": "cids(2)",
+              "cts": "1970-01-01T00:00:00.000Z",
+              "src": "user(0)",
+              "uri": "record(3)",
+              "val": "self-label-a",
+            },
+            Object {
+              "cid": "cids(2)",
+              "cts": "1970-01-01T00:00:00.000Z",
+              "src": "user(0)",
+              "uri": "record(3)",
+              "val": "self-label-b",
+            },
+          ],
+          "viewer": Object {
+            "blockedBy": false,
+            "muted": false,
+          },
+        },
+        "did": "did:web:example.com",
+        "displayName": "alice's feedgen",
+        "indexedAt": "1970-01-01T00:00:00.000Z",
+        "labels": Array [],
+        "likeCount": 0,
+        "uri": "record(2)",
+        "viewer": Object {},
+      },
+    ],
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "joinedAllTimeCount": 0,
+    "joinedWeekCount": 0,
+    "labels": Array [],
+    "list": Object {
+      "cid": "cids(4)",
+      "indexedAt": "1970-01-01T00:00:00.000Z",
+      "labels": Array [],
+      "listItemCount": 3,
+      "name": "n/a",
+      "purpose": "app.bsky.graph.defs#referencelist",
+      "uri": "record(1)",
+      "viewer": Object {
+        "muted": false,
+      },
+    },
+    "listItemsSample": Array [
+      Object {
+        "subject": Object {
+          "did": "user(2)",
+          "handle": "carol.test",
+          "labels": Array [],
+          "viewer": Object {
+            "blockedBy": false,
+            "muted": false,
+          },
+        },
+        "uri": "record(4)",
+      },
+      Object {
+        "subject": Object {
+          "avatar": "https://bsky.public.url/img/avatar/plain/user(4)/cids(1)@jpeg",
+          "createdAt": "1970-01-01T00:00:00.000Z",
+          "description": "hi im bob label_me",
+          "did": "user(3)",
+          "displayName": "bobby",
+          "handle": "bob.test",
+          "indexedAt": "1970-01-01T00:00:00.000Z",
+          "labels": Array [],
+          "viewer": Object {
+            "blockedBy": false,
+            "muted": false,
+          },
+        },
+        "uri": "record(5)",
+      },
+      Object {
+        "subject": Object {
+          "associated": Object {
+            "chat": Object {
+              "allowIncoming": "none",
+            },
+          },
+          "did": "user(5)",
+          "handle": "dan.test",
+          "labels": Array [],
+          "viewer": Object {
+            "blockedBy": false,
+            "muted": false,
+          },
+        },
+        "uri": "record(6)",
+      },
+    ],
+    "record": Object {
+      "$type": "app.bsky.graph.starterpack",
+      "createdAt": "1970-01-01T00:00:00.000Z",
+      "feeds": Array [
+        Object {
+          "uri": "record(2)",
+        },
+      ],
+      "list": "record(1)",
+      "name": "alice's starter pack",
+    },
+    "uri": "record(0)",
+  },
+}
+`;
+
+exports[`admin get starter pack view getStarterPack() gets a starterpack while taken down 1`] = `
+Object {
+  "cid": "cids(0)",
+  "creator": Object {
+    "avatar": "https://bsky.public.url/img/avatar/plain/user(1)/cids(1)@jpeg",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "did": "user(0)",
+    "displayName": "ali",
+    "handle": "alice.test",
+    "labels": Array [
+      Object {
+        "cid": "cids(2)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(3)",
+        "val": "self-label-a",
+      },
+      Object {
+        "cid": "cids(2)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(3)",
+        "val": "self-label-b",
+      },
+    ],
+    "viewer": Object {
+      "blockedBy": false,
+      "muted": false,
+    },
+  },
+  "feeds": Array [
+    Object {
+      "cid": "cids(3)",
+      "creator": Object {
+        "avatar": "https://bsky.public.url/img/avatar/plain/user(1)/cids(1)@jpeg",
+        "createdAt": "1970-01-01T00:00:00.000Z",
+        "description": "its me!",
+        "did": "user(0)",
+        "displayName": "ali",
+        "handle": "alice.test",
+        "indexedAt": "1970-01-01T00:00:00.000Z",
+        "labels": Array [
+          Object {
+            "cid": "cids(2)",
+            "cts": "1970-01-01T00:00:00.000Z",
+            "src": "user(0)",
+            "uri": "record(3)",
+            "val": "self-label-a",
+          },
+          Object {
+            "cid": "cids(2)",
+            "cts": "1970-01-01T00:00:00.000Z",
+            "src": "user(0)",
+            "uri": "record(3)",
+            "val": "self-label-b",
+          },
+        ],
+        "viewer": Object {
+          "blockedBy": false,
+          "muted": false,
+        },
+      },
+      "did": "did:web:example.com",
+      "displayName": "alice's feedgen",
+      "indexedAt": "1970-01-01T00:00:00.000Z",
+      "labels": Array [],
+      "likeCount": 0,
+      "uri": "record(2)",
+      "viewer": Object {},
+    },
+  ],
+  "indexedAt": "1970-01-01T00:00:00.000Z",
+  "joinedAllTimeCount": 0,
+  "joinedWeekCount": 0,
+  "labels": Array [],
+  "list": Object {
+    "cid": "cids(4)",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "labels": Array [],
+    "listItemCount": 3,
+    "name": "n/a",
+    "purpose": "app.bsky.graph.defs#referencelist",
+    "uri": "record(1)",
+    "viewer": Object {
+      "muted": false,
+    },
+  },
+  "listItemsSample": Array [
+    Object {
+      "subject": Object {
+        "did": "user(2)",
+        "handle": "carol.test",
+        "labels": Array [],
+        "viewer": Object {
+          "blockedBy": false,
+          "followedBy": "record(6)",
+          "following": "record(5)",
+          "muted": false,
+        },
+      },
+      "uri": "record(4)",
+    },
+    Object {
+      "subject": Object {
+        "avatar": "https://bsky.public.url/img/avatar/plain/user(4)/cids(1)@jpeg",
+        "createdAt": "1970-01-01T00:00:00.000Z",
+        "description": "hi im bob label_me",
+        "did": "user(3)",
+        "displayName": "bobby",
+        "handle": "bob.test",
+        "indexedAt": "1970-01-01T00:00:00.000Z",
+        "labels": Array [],
+        "viewer": Object {
+          "blockedBy": false,
+          "followedBy": "record(9)",
+          "following": "record(8)",
+          "muted": false,
+        },
+      },
+      "uri": "record(7)",
+    },
+    Object {
+      "subject": Object {
+        "associated": Object {
+          "chat": Object {
+            "allowIncoming": "none",
+          },
+        },
+        "did": "user(5)",
+        "handle": "dan.test",
+        "labels": Array [],
+        "viewer": Object {
+          "blockedBy": false,
+          "following": "record(11)",
+          "muted": false,
+        },
+      },
+      "uri": "record(10)",
+    },
+  ],
+  "record": Object {
+    "$type": "app.bsky.graph.starterpack",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "feeds": Array [
+      Object {
+        "uri": "record(2)",
+      },
+    ],
+    "list": "record(1)",
+    "name": "alice's starter pack",
+  },
+  "uri": "record(0)",
+}
+`;
+
+exports[`admin get starter pack view getStarterPack() gets a starterpack while taken down 2`] = `
+Object {
+  "cid": "cids(0)",
+  "creator": Object {
+    "avatar": "https://bsky.public.url/img/avatar/plain/user(1)/cids(1)@jpeg",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "did": "user(0)",
+    "displayName": "ali",
+    "handle": "alice.test",
+    "labels": Array [
+      Object {
+        "cid": "cids(2)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(3)",
+        "val": "self-label-a",
+      },
+      Object {
+        "cid": "cids(2)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(3)",
+        "val": "self-label-b",
+      },
+    ],
+    "viewer": Object {
+      "blockedBy": false,
+      "muted": false,
+    },
+  },
+  "feeds": Array [
+    Object {
+      "cid": "cids(3)",
+      "creator": Object {
+        "avatar": "https://bsky.public.url/img/avatar/plain/user(1)/cids(1)@jpeg",
+        "createdAt": "1970-01-01T00:00:00.000Z",
+        "description": "its me!",
+        "did": "user(0)",
+        "displayName": "ali",
+        "handle": "alice.test",
+        "indexedAt": "1970-01-01T00:00:00.000Z",
+        "labels": Array [
+          Object {
+            "cid": "cids(2)",
+            "cts": "1970-01-01T00:00:00.000Z",
+            "src": "user(0)",
+            "uri": "record(3)",
+            "val": "self-label-a",
+          },
+          Object {
+            "cid": "cids(2)",
+            "cts": "1970-01-01T00:00:00.000Z",
+            "src": "user(0)",
+            "uri": "record(3)",
+            "val": "self-label-b",
+          },
+        ],
+        "viewer": Object {
+          "blockedBy": false,
+          "muted": false,
+        },
+      },
+      "did": "did:web:example.com",
+      "displayName": "alice's feedgen",
+      "indexedAt": "1970-01-01T00:00:00.000Z",
+      "labels": Array [],
+      "likeCount": 0,
+      "uri": "record(2)",
+      "viewer": Object {},
+    },
+  ],
+  "indexedAt": "1970-01-01T00:00:00.000Z",
+  "joinedAllTimeCount": 0,
+  "joinedWeekCount": 0,
+  "labels": Array [
+    Object {
+      "cid": "cids(0)",
+      "cts": "1970-01-01T00:00:00.000Z",
+      "src": "user(2)",
+      "uri": "record(0)",
+      "val": "!takedown",
+    },
+  ],
+  "list": Object {
+    "cid": "cids(4)",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "labels": Array [],
+    "listItemCount": 3,
+    "name": "n/a",
+    "purpose": "app.bsky.graph.defs#referencelist",
+    "uri": "record(1)",
+    "viewer": Object {
+      "muted": false,
+    },
+  },
+  "listItemsSample": Array [
+    Object {
+      "subject": Object {
+        "did": "user(3)",
+        "handle": "carol.test",
+        "labels": Array [],
+        "viewer": Object {
+          "blockedBy": false,
+          "muted": false,
+        },
+      },
+      "uri": "record(4)",
+    },
+    Object {
+      "subject": Object {
+        "avatar": "https://bsky.public.url/img/avatar/plain/user(5)/cids(1)@jpeg",
+        "createdAt": "1970-01-01T00:00:00.000Z",
+        "description": "hi im bob label_me",
+        "did": "user(4)",
+        "displayName": "bobby",
+        "handle": "bob.test",
+        "indexedAt": "1970-01-01T00:00:00.000Z",
+        "labels": Array [],
+        "viewer": Object {
+          "blockedBy": false,
+          "muted": false,
+        },
+      },
+      "uri": "record(5)",
+    },
+    Object {
+      "subject": Object {
+        "associated": Object {
+          "chat": Object {
+            "allowIncoming": "none",
+          },
+        },
+        "did": "user(6)",
+        "handle": "dan.test",
+        "labels": Array [],
+        "viewer": Object {
+          "blockedBy": false,
+          "muted": false,
+        },
+      },
+      "uri": "record(6)",
+    },
+  ],
+  "record": Object {
+    "$type": "app.bsky.graph.starterpack",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "feeds": Array [
+      Object {
+        "uri": "record(2)",
+      },
+    ],
+    "list": "record(1)",
+    "name": "alice's starter pack",
+  },
+  "uri": "record(0)",
+}
+`;