server/models/actor/sql/instance-list-followers-query-builder.ts

   1 import { Sequelize } from 'sequelize'
   2 import { ModelBuilder } from '@server/models/shared'
   3 import { parseRowCountResult } from '@server/models/utils'
   4 import { MActorFollowActorsDefault } from '@server/types/models'
   5 import { ActivityPubActorType, FollowState } from '@shared/models'
   6 import { InstanceListFollowsQueryBuilder } from './shared/instance-list-follows-query-builder'
   7
   8 export interface ListFollowersOptions {
   9   actorIds: number[]
  10   start: number
  11   count: number
  12   sort: string
  13   state?: FollowState
  14   actorType?: ActivityPubActorType
  15   search?: string
  16 }
  17
  18 export class InstanceListFollowersQueryBuilder extends InstanceListFollowsQueryBuilder <ListFollowersOptions> {
  19
  20   constructor (
  21     protected readonly sequelize: Sequelize,
  22     protected readonly options: ListFollowersOptions
  23   ) {
  24     super(sequelize, options)
  25   }
  26
  27   async listFollowers () {
  28     this.buildListQuery()
  29
  30     const results = await this.runQuery({ nest: true })
  31     const modelBuilder = new ModelBuilder<MActorFollowActorsDefault>(this.sequelize)
  32
  33     return modelBuilder.createModels(results, 'ActorFollow')
  34   }
  35
  36   async countFollowers () {
  37     this.buildCountQuery()
  38
  39     const result = await this.runQuery()
  40
  41     return parseRowCountResult(result)
  42   }
  43
  44   protected getWhere () {
  45     let where = 'WHERE "ActorFollowing"."id" IN (:actorIds) '
  46     this.replacements.actorIds = this.options.actorIds
  47
  48     if (this.options.state) {
  49       where += 'AND "ActorFollowModel"."state" = :state '
  50       this.replacements.state = this.options.state
  51     }
  52
  53     if (this.options.search) {
  54       const escapedLikeSearch = this.sequelize.escape('%' + this.options.search + '%')
  55
  56       where += `AND (` +
  57         `"ActorFollower->Server"."host" ILIKE ${escapedLikeSearch} ` +
  58         `OR "ActorFollower"."preferredUsername" ILIKE ${escapedLikeSearch} ` +
  59       `)`
  60     }
  61
  62     if (this.options.actorType) {
  63       where += `AND "ActorFollower"."type" = :actorType `
  64       this.replacements.actorType = this.options.actorType
  65     }
  66
  67     return where
  68   }
  69 }