1 import { FindOptions, Op, QueryTypes } from 'sequelize'
2 import { BelongsTo, Column, CreatedAt, ForeignKey, Model, Table, UpdatedAt } from 'sequelize-typescript'
3 import { handlesToNameAndHost } from '@server/helpers/actors'
4 import { MAccountBlocklist, MAccountBlocklistFormattable } from '@server/types/models'
5 import { AttributesOnly } from '@shared/typescript-utils'
6 import { AccountBlock } from '../../../shared/models'
7 import { ActorModel } from '../actor/actor'
8 import { ServerModel } from '../server/server'
9 import { createSafeIn, getSort, searchAttribute } from '../utils'
10 import { AccountModel } from './account'
13 tableName: 'accountBlocklist',
16 fields: [ 'accountId', 'targetAccountId' ],
20 fields: [ 'targetAccountId' ]
24 export class AccountBlocklistModel extends Model<Partial<AttributesOnly<AccountBlocklistModel>>> {
32 @ForeignKey(() => AccountModel)
36 @BelongsTo(() => AccountModel, {
44 ByAccount: AccountModel
46 @ForeignKey(() => AccountModel)
48 targetAccountId: number
50 @BelongsTo(() => AccountModel, {
52 name: 'targetAccountId',
58 BlockedAccount: AccountModel
60 static isAccountMutedByAccounts (accountIds: number[], targetAccountId: number) {
62 attributes: [ 'accountId', 'id' ],
72 return AccountBlocklistModel.unscoped()
75 const result: { [accountId: number]: boolean } = {}
77 for (const accountId of accountIds) {
78 result[accountId] = !!rows.find(r => r.accountId === accountId)
85 static loadByAccountAndTarget (accountId: number, targetAccountId: number): Promise<MAccountBlocklist> {
93 return AccountBlocklistModel.findOne(query)
96 static listForApi (parameters: {
103 const { start, count, sort, search, accountId } = parameters
105 const getQuery = (forCount: boolean) => {
106 const query: FindOptions = {
109 order: getSort(sort),
114 Object.assign(query.where, {
116 searchAttribute(search, '$BlockedAccount.name$'),
117 searchAttribute(search, '$BlockedAccount.Actor.url$')
122 if (forCount !== true) {
141 AccountBlocklistModel.count(getQuery(true)),
142 AccountBlocklistModel.findAll(getQuery(false))
143 ]).then(([ total, data ]) => ({ total, data }))
146 static listHandlesBlockedBy (accountIds: number[]): Promise<string[]> {
148 attributes: [ 'id' ],
156 attributes: [ 'id' ],
157 model: AccountModel.unscoped(),
159 as: 'BlockedAccount',
162 attributes: [ 'preferredUsername' ],
163 model: ActorModel.unscoped(),
167 attributes: [ 'host' ],
168 model: ServerModel.unscoped(),
178 return AccountBlocklistModel.findAll(query)
179 .then(entries => entries.map(e => `${e.BlockedAccount.Actor.preferredUsername}@${e.BlockedAccount.Actor.Server.host}`))
182 static getBlockStatus (byAccountIds: number[], handles: string[]): Promise<{ name: string, host: string, accountId: number }[]> {
183 const sanitizedHandles = handlesToNameAndHost(handles)
185 const localHandles = sanitizedHandles.filter(h => !h.host)
188 const remoteHandles = sanitizedHandles.filter(h => !!h.host)
189 .map(h => ([ h.name, h.host ]))
191 const handlesWhere: string[] = []
193 if (localHandles.length !== 0) {
194 handlesWhere.push(`("actor"."preferredUsername" IN (:localHandles) AND "server"."id" IS NULL)`)
197 if (remoteHandles.length !== 0) {
198 handlesWhere.push(`(("actor"."preferredUsername", "server"."host") IN (:remoteHandles))`)
201 const rawQuery = `SELECT "accountBlocklist"."accountId", "actor"."preferredUsername" AS "name", "server"."host" ` +
202 `FROM "accountBlocklist" ` +
203 `INNER JOIN "account" ON "account"."id" = "accountBlocklist"."targetAccountId" ` +
204 `INNER JOIN "actor" ON "actor"."id" = "account"."actorId" ` +
205 `LEFT JOIN "server" ON "server"."id" = "actor"."serverId" ` +
206 `WHERE "accountBlocklist"."accountId" IN (${createSafeIn(AccountBlocklistModel.sequelize, byAccountIds)}) ` +
207 `AND (${handlesWhere.join(' OR ')})`
209 return AccountBlocklistModel.sequelize.query(rawQuery, {
210 type: QueryTypes.SELECT as QueryTypes.SELECT,
211 replacements: { byAccountIds, localHandles, remoteHandles }
215 toFormattedJSON (this: MAccountBlocklistFormattable): AccountBlock {
217 byAccount: this.ByAccount.toFormattedJSON(),
218 blockedAccount: this.BlockedAccount.toFormattedJSON(),
219 createdAt: this.createdAt