@atproto/ozone/src/db/migrations/20231219T205730722Z-init.ts

Backend service for moderating the Bluesky network.

import { Kysely } from 'kysely'

export async function up(db: Kysely<unknown>): Promise<void> {
  // Moderation event
  await db.schema
    .createTable('moderation_event')
    .addColumn('id', 'serial', (col) => col.primaryKey())
    .addColumn('action', 'varchar', (col) => col.notNull())
    .addColumn('subjectType', 'varchar', (col) => col.notNull())
    .addColumn('subjectDid', 'varchar', (col) => col.notNull())
    .addColumn('subjectUri', 'varchar')
    .addColumn('subjectCid', 'varchar')
    .addColumn('comment', 'text')
    .addColumn('meta', 'jsonb')
    .addColumn('createdAt', 'varchar', (col) => col.notNull())
    .addColumn('createdBy', 'varchar', (col) => col.notNull())
    .addColumn('reversedAt', 'varchar')
    .addColumn('reversedBy', 'varchar')
    .addColumn('durationInHours', 'integer')
    .addColumn('expiresAt', 'varchar')
    .addColumn('reversedReason', 'text')
    .addColumn('createLabelVals', 'varchar')
    .addColumn('negateLabelVals', 'varchar')
    .addColumn('legacyRefId', 'integer')
    .execute()

  // Moderation subject status
  await db.schema
    .createTable('moderation_subject_status')
    .addColumn('id', 'serial', (col) => col.primaryKey())

    // Identifiers
    .addColumn('did', 'varchar', (col) => col.notNull())
    // Default to '' so that we can apply unique constraints on did and recordPath columns
    .addColumn('recordPath', 'varchar', (col) => col.notNull().defaultTo(''))
    .addColumn('blobCids', 'jsonb')
    .addColumn('recordCid', 'varchar')

    // human review team state
    .addColumn('reviewState', 'varchar', (col) => col.notNull())
    .addColumn('comment', 'varchar')
    .addColumn('muteUntil', 'varchar')
    .addColumn('lastReviewedAt', 'varchar')
    .addColumn('lastReviewedBy', 'varchar')

    // report state
    .addColumn('lastReportedAt', 'varchar')
    .addColumn('lastAppealedAt', 'varchar')

    // visibility/intervention state
    .addColumn('takendown', 'boolean', (col) => col.defaultTo(false).notNull())
    .addColumn('suspendUntil', 'varchar')
    .addColumn('appealed', 'boolean')

    // timestamps
    .addColumn('createdAt', 'varchar', (col) => col.notNull())
    .addColumn('updatedAt', 'varchar', (col) => col.notNull())
    .addUniqueConstraint('moderation_status_unique_idx', ['did', 'recordPath'])
    .execute()

  await db.schema
    .createIndex('moderation_subject_status_blob_cids_idx')
    .on('moderation_subject_status')
    .using('gin')
    .column('blobCids')
    .execute()

  // Label
  await db.schema
    .createTable('label')
    .addColumn('id', 'bigserial', (col) => col.primaryKey())
    .addColumn('src', 'varchar', (col) => col.notNull())
    .addColumn('uri', 'varchar', (col) => col.notNull())
    .addColumn('cid', 'varchar', (col) => col.notNull())
    .addColumn('val', 'varchar', (col) => col.notNull())
    .addColumn('neg', 'boolean', (col) => col.notNull())
    .addColumn('cts', 'varchar', (col) => col.notNull())
    .execute()
  await db.schema
    .createIndex('unique_label_idx')
    .unique()
    .on('label')
    .columns(['src', 'uri', 'cid', 'val'])
    .execute()
  await db.schema
    .createIndex('label_uri_index')
    .on('label')
    .column('uri')
    .execute()

  // Push Events
  await db.schema
    .createTable('repo_push_event')
    .addColumn('id', 'serial', (col) => col.primaryKey())
    .addColumn('eventType', 'varchar', (col) => col.notNull())
    .addColumn('subjectDid', 'varchar', (col) => col.notNull())
    .addColumn('takedownRef', 'varchar')
    .addColumn('confirmedAt', 'timestamptz')
    .addColumn('lastAttempted', 'timestamptz')
    .addColumn('attempts', 'integer', (col) => col.notNull().defaultTo(0))
    .addUniqueConstraint('repo_push_event_unique_evt', [
      'subjectDid',
      'eventType',
    ])
    .execute()
  await db.schema
    .createIndex('repo_push_confirmation_idx')
    .on('repo_push_event')
    .columns(['confirmedAt', 'attempts'])
    .execute()

  await db.schema
    .createTable('record_push_event')
    .addColumn('id', 'serial', (col) => col.primaryKey())
    .addColumn('eventType', 'varchar', (col) => col.notNull())
    .addColumn('subjectDid', 'varchar', (col) => col.notNull())
    .addColumn('subjectUri', 'varchar', (col) => col.notNull())
    .addColumn('subjectCid', 'varchar')
    .addColumn('takedownRef', 'varchar')
    .addColumn('confirmedAt', 'timestamptz')
    .addColumn('lastAttempted', 'timestamptz')
    .addColumn('attempts', 'integer', (col) => col.notNull().defaultTo(0))
    .addUniqueConstraint('record_push_event_unique_evt', [
      'subjectUri',
      'eventType',
    ])
    .execute()
  await db.schema
    .createIndex('record_push_event_did_type_idx')
    .on('record_push_event')
    .columns(['subjectDid', 'eventType'])
    .execute()
  await db.schema
    .createIndex('record_push_confirmation_idx')
    .on('record_push_event')
    .columns(['confirmedAt', 'attempts'])
    .execute()

  await db.schema
    .createTable('blob_push_event')
    .addColumn('id', 'serial', (col) => col.primaryKey())
    .addColumn('eventType', 'varchar', (col) => col.notNull())
    .addColumn('subjectDid', 'varchar', (col) => col.notNull())
    .addColumn('subjectBlobCid', 'varchar', (col) => col.notNull())
    .addColumn('subjectUri', 'varchar')
    .addColumn('takedownRef', 'varchar')
    .addColumn('confirmedAt', 'timestamptz')
    .addColumn('lastAttempted', 'timestamptz')
    .addColumn('attempts', 'integer', (col) => col.notNull().defaultTo(0))
    .addUniqueConstraint('blob_push_event_unique_evt', [
      'subjectDid',
      'subjectBlobCid',
      'eventType',
    ])
    .execute()
  await db.schema
    .createIndex('blob_push_confirmation_idx')
    .on('blob_push_event')
    .columns(['confirmedAt', 'attempts'])
    .execute()
}

export async function down(db: Kysely<unknown>): Promise<void> {
  await db.schema.dropTable('moderation_event').execute()
  await db.schema.dropTable('moderation_subject_status').execute()
  await db.schema.dropTable('label').execute()
  await db.schema.dropTable('repo_push_event').execute()
  await db.schema.dropTable('record_push_event').execute()
  await db.schema.dropTable('blob_push_event').execute()
}