1
0
Fork 0
peertube/server/models/actor/sql/instance-list-followers-query-builder.ts
Chocobozzz bae616273d
Convert followers/following in raw SQL queries
Prevent weird bug in SQL generation
2022-05-05 10:29:35 +02:00

69 lines
2 KiB
TypeScript

import { Sequelize } from 'sequelize'
import { ModelBuilder } from '@server/models/shared'
import { parseRowCountResult } from '@server/models/utils'
import { MActorFollowActorsDefault } from '@server/types/models'
import { ActivityPubActorType, FollowState } from '@shared/models'
import { InstanceListFollowsQueryBuilder } from './shared/instance-list-follows-query-builder'
export interface ListFollowersOptions {
actorIds: number[]
start: number
count: number
sort: string
state?: FollowState
actorType?: ActivityPubActorType
search?: string
}
export class InstanceListFollowersQueryBuilder extends InstanceListFollowsQueryBuilder <ListFollowersOptions> {
constructor (
protected readonly sequelize: Sequelize,
protected readonly options: ListFollowersOptions
) {
super(sequelize, options)
}
async listFollowers () {
this.buildListQuery()
const results = await this.runQuery({ nest: true })
const modelBuilder = new ModelBuilder<MActorFollowActorsDefault>(this.sequelize)
return modelBuilder.createModels(results, 'ActorFollow')
}
async countFollowers () {
this.buildCountQuery()
const result = await this.runQuery()
return parseRowCountResult(result)
}
protected getWhere () {
let where = 'WHERE "ActorFollowing"."id" IN (:actorIds) '
this.replacements.actorIds = this.options.actorIds
if (this.options.state) {
where += 'AND "ActorFollowModel"."state" = :state '
this.replacements.state = this.options.state
}
if (this.options.search) {
const escapedLikeSearch = this.sequelize.escape('%' + this.options.search + '%')
where += `AND (` +
`"ActorFollower->Server"."host" ILIKE ${escapedLikeSearch} ` +
`OR "ActorFollower"."preferredUsername" ILIKE ${escapedLikeSearch} ` +
`)`
}
if (this.options.actorType) {
where += `AND "ActorFollower"."type" = :actorType `
this.replacements.actorType = this.options.actorType
}
return where
}
}