@atproto/ozone 0.1.107 → 0.1.109

package/src/verification/service.ts

@@ -0,0 +1,208 @@
+import { Selectable } from 'kysely'
+import {
+  $Typed,
+  AppBskyActorDefs,
+  AtUri,
+  ToolsOzoneModerationDefs,
+  ToolsOzoneVerificationDefs,
+} from '@atproto/api'
+import { Database } from '../db'
+import { CreatedAtUriKeyset, paginate } from '../db/pagination'
+import { Verification } from '../db/schema/verification'
+
+export type VerificationServiceCreator = (db: Database) => VerificationService
+
+export class VerificationService {
+  constructor(public db: Database) {}
+
+  static creator() {
+    return (db: Database) => new VerificationService(db)
+  }
+
+  async create(
+    verifications: Pick<
+      Verification,
+      | 'uri'
+      | 'issuer'
+      | 'subject'
+      | 'handle'
+      | 'displayName'
+      | 'createdAt'
+      | 'cid'
+    >[],
+  ) {
+    return this.db.transaction(async (tx) => {
+      return tx.db
+        .insertInto('verification')
+        .values(verifications)
+        .onConflict((oc) => oc.doNothing())
+        .returningAll()
+        .execute()
+    })
+  }
+
+  async markRevoked({
+    uris,
+    revokedBy,
+    revokedAt,
+    revokeReason,
+  }: {
+    uris: string[]
+    revokedBy?: string
+    revokedAt?: string
+    revokeReason?: string
+  }) {
+    const now = new Date().toISOString()
+    return this.db.transaction(async (tx) => {
+      for (const uri of uris) {
+        return tx.db
+          .updateTable('verification')
+          .set({
+            revokeReason,
+            updatedAt: now,
+            revokedAt: revokedAt || now,
+            // Allow setting revokedBy to a moderator/verifier DID and if it isn't set, default to the author of the verification record
+            revokedBy: revokedBy || new AtUri(uri).host,
+          })
+          .where('uri', '=', uri)
+          .where('revokedAt', 'is', null)
+          .execute()
+      }
+    })
+  }
+
+  async list({
+    sortDirection,
+    cursor,
+    createdAfter,
+    createdBefore,
+    issuers = [],
+    subjects = [],
+    isRevoked,
+    limit = 100,
+  }: {
+    sortDirection?: 'asc' | 'desc'
+    cursor?: string
+    createdAfter?: string
+    createdBefore?: string
+    issuers?: string[]
+    subjects?: string[]
+    isRevoked?: boolean
+    limit?: number
+  }) {
+    const { ref } = this.db.db.dynamic
+
+    let qb = this.db.db.selectFrom('verification').selectAll()
+
+    if (issuers.length) {
+      qb = qb.where('issuer', 'in', issuers)
+    }
+
+    if (isRevoked !== undefined) {
+      qb = qb.where('revokedAt', isRevoked ? 'is not' : 'is', null)
+    }
+
+    if (subjects.length) {
+      qb = qb.where('subject', 'in', subjects)
+    }
+
+    if (createdAfter) {
+      qb = qb.where('createdAt', '>=', createdAfter)
+    }
+
+    if (createdBefore) {
+      qb = qb.where('createdAt', '<=', createdBefore)
+    }
+
+    const keyset = new CreatedAtUriKeyset(ref(`createdAt`), ref('uri'))
+    const paginatedBuilder = paginate(qb, {
+      limit,
+      cursor,
+      keyset,
+      tryIndex: true,
+      direction: sortDirection === 'desc' ? 'desc' : 'asc',
+    })
+
+    const result = await paginatedBuilder.execute()
+    return { verifications: result, cursor: keyset.packFromResult(result) }
+  }
+
+  view(
+    verifications: Selectable<Verification>[],
+    repos: Map<
+      string,
+      | $Typed<ToolsOzoneModerationDefs.RepoViewDetail>
+      | $Typed<ToolsOzoneModerationDefs.RepoViewNotFound>
+    >,
+    profiles: Map<string, AppBskyActorDefs.ProfileViewDetailed>,
+  ): $Typed<ToolsOzoneVerificationDefs.VerificationView>[] {
+    return verifications.map((verification) => {
+      const issuerRepo = repos.get(verification.issuer)
+      const subjectRepo = repos.get(verification.subject)
+      const subjectProfile = profiles.get(verification.subject)
+      const issuerProfile = profiles.get(verification.issuer)
+      return {
+        $type: 'tools.ozone.verification.defs#verificationView',
+        uri: verification.uri,
+        issuer: verification.issuer,
+        subject: verification.subject,
+        createdAt: verification.createdAt,
+        displayName: verification.displayName,
+        handle: verification.handle,
+        updatedAt: verification.updatedAt || undefined,
+        revokedAt: verification.revokedAt || undefined,
+        revokedBy: verification.revokedBy || undefined,
+        revokeReason: verification.revokeReason || undefined,
+        issuerRepo,
+        subjectRepo,
+        subjectProfile: subjectProfile
+          ? {
+              $type: 'app.bsky.actor.defs#profileViewDetailed',
+              ...subjectProfile,
+            }
+          : undefined,
+        issuerProfile: issuerProfile
+          ? {
+              $type: 'app.bsky.actor.defs#profileViewDetailed',
+              ...issuerProfile,
+            }
+          : undefined,
+      }
+    })
+  }
+
+  async getFirehoseCursor() {
+    const entry = await this.db.db
+      .selectFrom('firehose_cursor')
+      .select('cursor')
+      .where('service', '=', 'verification')
+      .executeTakeFirst()
+
+    return entry?.cursor || null
+  }
+
+  createFirehoseCursor() {
+    return this.db.db
+      .insertInto('firehose_cursor')
+      .values({
+        service: 'verification',
+        cursor: null,
+      })
+      .onConflict((oc) => oc.doNothing())
+      .execute()
+  }
+
+  async updateFirehoseCursor(cursor: number) {
+    const updated = await this.db.db
+      .updateTable('firehose_cursor')
+      .set({ cursor })
+      .where('service', '=', 'verification')
+      .where((qb) =>
+        qb.where('cursor', '<', cursor).orWhere('cursor', 'is', null),
+      )
+      .returningAll()
+      .executeTakeFirst()
+
+    return updated?.cursor
+  }
+}

package/src/verification/util.ts

@@ -0,0 +1,50 @@
+import { $Typed, ToolsOzoneModerationDefs } from '@atproto/api'
+import { addAccountInfoToRepoViewDetail, getPdsAccountInfos } from '../api/util'
+import { AppContext } from '../context'
+import { ModerationService } from '../mod-service'
+import { ParsedLabelers } from '../util'
+
+export const getReposForVerifications = async (
+  ctx: AppContext,
+  labelers: ParsedLabelers,
+  modService: ModerationService,
+  dids: string[],
+  isModerator: boolean,
+) => {
+  const [partialRepos, accountInfo] = await Promise.all([
+    modService.views.repoDetails(dids, labelers),
+    getPdsAccountInfos(ctx, dids),
+  ])
+
+  const repos = new Map<
+    string,
+    | $Typed<ToolsOzoneModerationDefs.RepoViewDetail>
+    | $Typed<ToolsOzoneModerationDefs.RepoViewNotFound>
+  >(
+    dids.map((did) => {
+      const partialRepo = partialRepos.get(did)
+      if (!partialRepo) {
+        return [
+          did,
+          {
+            did,
+            $type: 'tools.ozone.moderation.defs#repoViewNotFound',
+          },
+        ]
+      }
+      return [
+        did,
+        {
+          ...addAccountInfoToRepoViewDetail(
+            partialRepo,
+            accountInfo.get(did) || null,
+            isModerator,
+          ),
+          $type: 'tools.ozone.moderation.defs#repoViewDetail',
+        },
+      ]
+    }),
+  )
+
+  return repos
+}

package/tests/__snapshots__/verification-listener.test.ts.snap

@@ -0,0 +1,146 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`verification-listener indexes new and revoked verifications 1`] = `
+Array [
+  Object {
+    "$type": "tools.ozone.verification.defs#verificationView",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "displayName": "Alice",
+    "handle": "alice.test",
+    "issuer": "user(0)",
+    "issuerProfile": Object {
+      "$type": "app.bsky.actor.defs#profileViewDetailed",
+      "associated": Object {
+        "feedgens": 0,
+        "labeler": false,
+        "lists": 0,
+        "starterPacks": 0,
+      },
+      "avatar": "https://bsky.public.url/img/avatar/plain/user(3)/cids(0)@jpeg",
+      "createdAt": "1970-01-01T00:00:00.000Z",
+      "description": "hi im bob label_me",
+      "did": "user(0)",
+      "displayName": "bobby",
+      "followersCount": 2,
+      "followsCount": 2,
+      "handle": "bob.test",
+      "indexedAt": "1970-01-01T00:00:00.000Z",
+      "labels": Array [],
+      "postsCount": 3,
+      "viewer": Object {
+        "blockedBy": false,
+        "muted": false,
+      },
+    },
+    "issuerRepo": Object {
+      "$type": "tools.ozone.moderation.defs#repoViewDetail",
+      "did": "user(0)",
+      "email": "bob@test.com",
+      "handle": "bob.test",
+      "indexedAt": "1970-01-01T00:00:00.000Z",
+      "invites": Array [],
+      "invitesDisabled": false,
+      "labels": Array [],
+      "moderation": Object {},
+      "relatedRecords": Array [
+        Object {
+          "$type": "app.bsky.actor.profile",
+          "avatar": Object {
+            "$type": "blob",
+            "mimeType": "image/jpeg",
+            "ref": Object {
+              "$link": "cids(0)",
+            },
+            "size": 3976,
+          },
+          "createdAt": "1970-01-01T00:00:00.000Z",
+          "description": "hi im bob label_me",
+          "displayName": "bobby",
+        },
+      ],
+    },
+    "revokedAt": "1970-01-01T00:00:00.000Z",
+    "revokedBy": "user(0)",
+    "subject": "user(1)",
+    "subjectProfile": Object {
+      "$type": "app.bsky.actor.defs#profileViewDetailed",
+      "associated": Object {
+        "feedgens": 0,
+        "labeler": false,
+        "lists": 0,
+        "starterPacks": 0,
+      },
+      "avatar": "https://bsky.public.url/img/avatar/plain/user(2)/cids(0)@jpeg",
+      "createdAt": "1970-01-01T00:00:00.000Z",
+      "description": "its me!",
+      "did": "user(1)",
+      "displayName": "ali",
+      "followersCount": 2,
+      "followsCount": 3,
+      "handle": "alice.test",
+      "indexedAt": "1970-01-01T00:00:00.000Z",
+      "labels": Array [
+        Object {
+          "cid": "cids(1)",
+          "cts": "1970-01-01T00:00:00.000Z",
+          "src": "user(1)",
+          "uri": "record(1)",
+          "val": "self-label-a",
+        },
+        Object {
+          "cid": "cids(1)",
+          "cts": "1970-01-01T00:00:00.000Z",
+          "src": "user(1)",
+          "uri": "record(1)",
+          "val": "self-label-b",
+        },
+      ],
+      "postsCount": 4,
+      "viewer": Object {
+        "blockedBy": false,
+        "muted": false,
+      },
+    },
+    "subjectRepo": Object {
+      "$type": "tools.ozone.moderation.defs#repoViewDetail",
+      "did": "user(1)",
+      "email": "alice@test.com",
+      "handle": "alice.test",
+      "indexedAt": "1970-01-01T00:00:00.000Z",
+      "invites": Array [],
+      "invitesDisabled": false,
+      "labels": Array [],
+      "moderation": Object {},
+      "relatedRecords": Array [
+        Object {
+          "$type": "app.bsky.actor.profile",
+          "avatar": Object {
+            "$type": "blob",
+            "mimeType": "image/jpeg",
+            "ref": Object {
+              "$link": "cids(0)",
+            },
+            "size": 3976,
+          },
+          "createdAt": "1970-01-01T00:00:00.000Z",
+          "description": "its me!",
+          "displayName": "ali",
+          "labels": Object {
+            "$type": "com.atproto.label.defs#selfLabels",
+            "values": Array [
+              Object {
+                "val": "self-label-a",
+              },
+              Object {
+                "val": "self-label-b",
+              },
+            ],
+          },
+        },
+      ],
+    },
+    "updatedAt": "1970-01-01T00:00:00.000Z",
+    "uri": "record(0)",
+  },
+]
+`;

package/tests/__snapshots__/verification.test.ts.snap

@@ -0,0 +1,288 @@
+// Jest Snapshot v1, https://goo.gl/fbAQLP
+
+exports[`verification list returns paginated list of verifications 1`] = `
+Object {
+  "$type": "tools.ozone.verification.defs#verificationView",
+  "createdAt": "1970-01-01T00:00:00.000Z",
+  "displayName": "bobby",
+  "handle": "bob.test",
+  "issuer": "user(0)",
+  "issuerProfile": Object {
+    "$type": "app.bsky.actor.defs#profileViewDetailed",
+    "associated": Object {
+      "feedgens": 0,
+      "labeler": false,
+      "lists": 0,
+      "starterPacks": 0,
+    },
+    "avatar": "https://bsky.public.url/img/avatar/plain/user(3)/cids(0)@jpeg",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "description": "its me!",
+    "did": "user(0)",
+    "displayName": "ali",
+    "followersCount": 2,
+    "followsCount": 3,
+    "handle": "alice.test",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "labels": Array [
+      Object {
+        "cid": "cids(1)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(1)",
+        "val": "self-label-a",
+      },
+      Object {
+        "cid": "cids(1)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(1)",
+        "val": "self-label-b",
+      },
+    ],
+    "postsCount": 4,
+    "verification": Object {
+      "trustedVerifierStatus": "valid",
+      "verifications": Array [],
+      "verifiedStatus": "none",
+    },
+    "viewer": Object {
+      "blockedBy": false,
+      "muted": false,
+    },
+  },
+  "issuerRepo": Object {
+    "$type": "tools.ozone.moderation.defs#repoViewDetail",
+    "did": "user(0)",
+    "email": "alice@test.com",
+    "handle": "alice.test",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "invites": Array [],
+    "invitesDisabled": false,
+    "labels": Array [],
+    "moderation": Object {},
+    "relatedRecords": Array [
+      Object {
+        "$type": "app.bsky.actor.profile",
+        "avatar": Object {
+          "$type": "blob",
+          "mimeType": "image/jpeg",
+          "ref": Object {
+            "$link": "cids(0)",
+          },
+          "size": 3976,
+        },
+        "createdAt": "1970-01-01T00:00:00.000Z",
+        "description": "its me!",
+        "displayName": "ali",
+        "labels": Object {
+          "$type": "com.atproto.label.defs#selfLabels",
+          "values": Array [
+            Object {
+              "val": "self-label-a",
+            },
+            Object {
+              "val": "self-label-b",
+            },
+          ],
+        },
+      },
+    ],
+  },
+  "subject": "user(1)",
+  "subjectProfile": Object {
+    "$type": "app.bsky.actor.defs#profileViewDetailed",
+    "associated": Object {
+      "feedgens": 0,
+      "labeler": false,
+      "lists": 0,
+      "starterPacks": 0,
+    },
+    "avatar": "https://bsky.public.url/img/avatar/plain/user(2)/cids(0)@jpeg",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "description": "hi im bob label_me",
+    "did": "user(1)",
+    "displayName": "bobby",
+    "followersCount": 2,
+    "followsCount": 2,
+    "handle": "bob.test",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "labels": Array [],
+    "postsCount": 3,
+    "verification": Object {
+      "trustedVerifierStatus": "none",
+      "verifications": Array [
+        Object {
+          "createdAt": "1970-01-01T00:00:00.000Z",
+          "isValid": true,
+          "issuer": "user(0)",
+          "uri": "record(0)",
+        },
+      ],
+      "verifiedStatus": "valid",
+    },
+    "viewer": Object {
+      "blockedBy": false,
+      "muted": false,
+    },
+  },
+  "subjectRepo": Object {
+    "$type": "tools.ozone.moderation.defs#repoViewDetail",
+    "did": "user(1)",
+    "email": "bob@test.com",
+    "handle": "bob.test",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "invites": Array [],
+    "invitesDisabled": false,
+    "labels": Array [],
+    "moderation": Object {},
+    "relatedRecords": Array [
+      Object {
+        "$type": "app.bsky.actor.profile",
+        "avatar": Object {
+          "$type": "blob",
+          "mimeType": "image/jpeg",
+          "ref": Object {
+            "$link": "cids(0)",
+          },
+          "size": 3976,
+        },
+        "createdAt": "1970-01-01T00:00:00.000Z",
+        "description": "hi im bob label_me",
+        "displayName": "bobby",
+      },
+    ],
+  },
+  "updatedAt": "1970-01-01T00:00:00.000Z",
+  "uri": "record(0)",
+}
+`;
+
+exports[`verification list returns paginated list of verifications 2`] = `
+Object {
+  "$type": "tools.ozone.verification.defs#verificationView",
+  "createdAt": "1970-01-01T00:00:00.000Z",
+  "displayName": "",
+  "handle": "carol.test",
+  "issuer": "user(0)",
+  "issuerProfile": Object {
+    "$type": "app.bsky.actor.defs#profileViewDetailed",
+    "associated": Object {
+      "feedgens": 0,
+      "labeler": false,
+      "lists": 0,
+      "starterPacks": 0,
+    },
+    "avatar": "https://bsky.public.url/img/avatar/plain/user(2)/cids(0)@jpeg",
+    "createdAt": "1970-01-01T00:00:00.000Z",
+    "description": "its me!",
+    "did": "user(0)",
+    "displayName": "ali",
+    "followersCount": 2,
+    "followsCount": 3,
+    "handle": "alice.test",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "labels": Array [
+      Object {
+        "cid": "cids(1)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(1)",
+        "val": "self-label-a",
+      },
+      Object {
+        "cid": "cids(1)",
+        "cts": "1970-01-01T00:00:00.000Z",
+        "src": "user(0)",
+        "uri": "record(1)",
+        "val": "self-label-b",
+      },
+    ],
+    "postsCount": 4,
+    "verification": Object {
+      "trustedVerifierStatus": "valid",
+      "verifications": Array [],
+      "verifiedStatus": "none",
+    },
+    "viewer": Object {
+      "blockedBy": false,
+      "muted": false,
+    },
+  },
+  "issuerRepo": Object {
+    "$type": "tools.ozone.moderation.defs#repoViewDetail",
+    "did": "user(0)",
+    "email": "alice@test.com",
+    "handle": "alice.test",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "invites": Array [],
+    "invitesDisabled": false,
+    "labels": Array [],
+    "moderation": Object {},
+    "relatedRecords": Array [
+      Object {
+        "$type": "app.bsky.actor.profile",
+        "avatar": Object {
+          "$type": "blob",
+          "mimeType": "image/jpeg",
+          "ref": Object {
+            "$link": "cids(0)",
+          },
+          "size": 3976,
+        },
+        "createdAt": "1970-01-01T00:00:00.000Z",
+        "description": "its me!",
+        "displayName": "ali",
+        "labels": Object {
+          "$type": "com.atproto.label.defs#selfLabels",
+          "values": Array [
+            Object {
+              "val": "self-label-a",
+            },
+            Object {
+              "val": "self-label-b",
+            },
+          ],
+        },
+      },
+    ],
+  },
+  "revokeReason": "Testing",
+  "revokedAt": "1970-01-01T00:00:00.000Z",
+  "revokedBy": "user(0)",
+  "subject": "user(1)",
+  "subjectProfile": Object {
+    "$type": "app.bsky.actor.defs#profileViewDetailed",
+    "associated": Object {
+      "feedgens": 0,
+      "labeler": false,
+      "lists": 0,
+      "starterPacks": 0,
+    },
+    "did": "user(1)",
+    "followersCount": 2,
+    "followsCount": 1,
+    "handle": "carol.test",
+    "labels": Array [],
+    "postsCount": 2,
+    "viewer": Object {
+      "blockedBy": false,
+      "muted": false,
+    },
+  },
+  "subjectRepo": Object {
+    "$type": "tools.ozone.moderation.defs#repoViewDetail",
+    "did": "user(1)",
+    "email": "carol@test.com",
+    "handle": "carol.test",
+    "indexedAt": "1970-01-01T00:00:00.000Z",
+    "invites": Array [],
+    "invitesDisabled": false,
+    "labels": Array [],
+    "moderation": Object {},
+    "relatedRecords": Array [],
+  },
+  "updatedAt": "1970-01-01T00:00:00.000Z",
+  "uri": "record(0)",
+}
+`;