PeerTube/server/models/account/account-blocklist.ts

232 lines
6.3 KiB
TypeScript
Raw Normal View History

import { Op, QueryTypes } from 'sequelize'
2020-07-01 16:05:30 +02:00
import { BelongsTo, Column, CreatedAt, ForeignKey, Model, Scopes, Table, UpdatedAt } from 'sequelize-typescript'
import { handlesToNameAndHost } from '@server/helpers/actors'
2020-06-18 10:45:25 +02:00
import { MAccountBlocklist, MAccountBlocklistAccounts, MAccountBlocklistFormattable } from '@server/types/models'
import { AttributesOnly } from '@shared/typescript-utils'
2020-07-01 16:05:30 +02:00
import { AccountBlock } from '../../../shared/models'
2021-05-11 11:15:29 +02:00
import { ActorModel } from '../actor/actor'
2020-05-29 16:16:24 +02:00
import { ServerModel } from '../server/server'
import { createSafeIn, getSort, searchAttribute } from '../utils'
2020-07-01 16:05:30 +02:00
import { AccountModel } from './account'
enum ScopeNames {
WITH_ACCOUNTS = 'WITH_ACCOUNTS'
}
2019-04-23 09:50:57 +02:00
@Scopes(() => ({
[ScopeNames.WITH_ACCOUNTS]: {
include: [
{
2019-04-23 09:50:57 +02:00
model: AccountModel,
required: true,
as: 'ByAccount'
},
{
2019-04-23 09:50:57 +02:00
model: AccountModel,
required: true,
as: 'BlockedAccount'
}
]
}
2019-04-23 09:50:57 +02:00
}))
@Table({
tableName: 'accountBlocklist',
indexes: [
{
fields: [ 'accountId', 'targetAccountId' ],
unique: true
},
{
fields: [ 'targetAccountId' ]
}
]
})
2021-05-12 14:09:04 +02:00
export class AccountBlocklistModel extends Model<Partial<AttributesOnly<AccountBlocklistModel>>> {
@CreatedAt
createdAt: Date
@UpdatedAt
updatedAt: Date
@ForeignKey(() => AccountModel)
@Column
accountId: number
@BelongsTo(() => AccountModel, {
foreignKey: {
name: 'accountId',
allowNull: false
},
as: 'ByAccount',
onDelete: 'CASCADE'
})
ByAccount: AccountModel
@ForeignKey(() => AccountModel)
@Column
targetAccountId: number
@BelongsTo(() => AccountModel, {
foreignKey: {
name: 'targetAccountId',
allowNull: false
},
as: 'BlockedAccount',
onDelete: 'CASCADE'
})
BlockedAccount: AccountModel
static isAccountMutedByAccounts (accountIds: number[], targetAccountId: number) {
const query = {
attributes: [ 'accountId', 'id' ],
where: {
accountId: {
2020-01-28 14:45:17 +01:00
[Op.in]: accountIds
},
targetAccountId
},
raw: true
}
return AccountBlocklistModel.unscoped()
.findAll(query)
.then(rows => {
const result: { [accountId: number]: boolean } = {}
for (const accountId of accountIds) {
result[accountId] = !!rows.find(r => r.accountId === accountId)
}
return result
})
}
2020-12-08 14:30:29 +01:00
static loadByAccountAndTarget (accountId: number, targetAccountId: number): Promise<MAccountBlocklist> {
const query = {
where: {
accountId,
targetAccountId
}
}
return AccountBlocklistModel.findOne(query)
}
static listForApi (parameters: {
start: number
count: number
sort: string
search?: string
accountId: number
}) {
const { start, count, sort, search, accountId } = parameters
const query = {
offset: start,
limit: count,
order: getSort(sort)
}
const where = {
accountId
}
if (search) {
Object.assign(where, {
[Op.or]: [
searchAttribute(search, '$BlockedAccount.name$'),
searchAttribute(search, '$BlockedAccount.Actor.url$')
]
})
}
Object.assign(query, { where })
return AccountBlocklistModel
.scope([ ScopeNames.WITH_ACCOUNTS ])
2019-08-15 11:53:26 +02:00
.findAndCountAll<MAccountBlocklistAccounts>(query)
.then(({ rows, count }) => {
return { total: count, data: rows }
})
}
2020-12-08 14:30:29 +01:00
static listHandlesBlockedBy (accountIds: number[]): Promise<string[]> {
2020-05-29 16:16:24 +02:00
const query = {
2020-12-08 14:30:29 +01:00
attributes: [ 'id' ],
2020-05-29 16:16:24 +02:00
where: {
accountId: {
[Op.in]: accountIds
}
},
include: [
{
attributes: [ 'id' ],
model: AccountModel.unscoped(),
required: true,
as: 'BlockedAccount',
include: [
{
attributes: [ 'preferredUsername' ],
model: ActorModel.unscoped(),
required: true,
include: [
{
attributes: [ 'host' ],
model: ServerModel.unscoped(),
required: true
}
]
}
]
}
]
}
return AccountBlocklistModel.findAll(query)
.then(entries => entries.map(e => `${e.BlockedAccount.Actor.preferredUsername}@${e.BlockedAccount.Actor.Server.host}`))
}
static getBlockStatus (byAccountIds: number[], handles: string[]): Promise<{ name: string, host: string, accountId: number }[]> {
const sanitizedHandles = handlesToNameAndHost(handles)
const localHandles = sanitizedHandles.filter(h => !h.host)
.map(h => h.name)
const remoteHandles = sanitizedHandles.filter(h => !!h.host)
.map(h => ([ h.name, h.host ]))
const handlesWhere: string[] = []
if (localHandles.length !== 0) {
handlesWhere.push(`("actor"."preferredUsername" IN (:localHandles) AND "server"."id" IS NULL)`)
}
if (remoteHandles.length !== 0) {
handlesWhere.push(`(("actor"."preferredUsername", "server"."host") IN (:remoteHandles))`)
}
const rawQuery = `SELECT "accountBlocklist"."accountId", "actor"."preferredUsername" AS "name", "server"."host" ` +
`FROM "accountBlocklist" ` +
`INNER JOIN "account" ON "account"."id" = "accountBlocklist"."targetAccountId" ` +
`INNER JOIN "actor" ON "actor"."id" = "account"."actorId" ` +
`LEFT JOIN "server" ON "server"."id" = "actor"."serverId" ` +
`WHERE "accountBlocklist"."accountId" IN (${createSafeIn(AccountBlocklistModel.sequelize, byAccountIds)}) ` +
`AND (${handlesWhere.join(' OR ')})`
return AccountBlocklistModel.sequelize.query(rawQuery, {
type: QueryTypes.SELECT as QueryTypes.SELECT,
replacements: { byAccountIds, localHandles, remoteHandles }
})
}
2019-08-20 19:05:31 +02:00
toFormattedJSON (this: MAccountBlocklistFormattable): AccountBlock {
return {
byAccount: this.ByAccount.toFormattedJSON(),
blockedAccount: this.BlockedAccount.toFormattedJSON(),
createdAt: this.createdAt
}
}
}