import { BelongsTo, Column, CreatedAt, ForeignKey, Model, Scopes, Table, UpdatedAt } from 'sequelize-typescript' import { AccountModel } from './account' import { getSort } from '../utils' import { AccountBlock } from '../../../shared/models/blocklist' import { Op } from 'sequelize' import * as Bluebird from 'bluebird' import { MAccountBlocklist, MAccountBlocklistAccounts, MAccountBlocklistFormattable } from '@server/typings/models' enum ScopeNames { WITH_ACCOUNTS = 'WITH_ACCOUNTS' } @Scopes(() => ({ [ScopeNames.WITH_ACCOUNTS]: { include: [ { model: AccountModel, required: true, as: 'ByAccount' }, { model: AccountModel, required: true, as: 'BlockedAccount' } ] } })) @Table({ tableName: 'accountBlocklist', indexes: [ { fields: [ 'accountId', 'targetAccountId' ], unique: true }, { fields: [ 'targetAccountId' ] } ] }) export class AccountBlocklistModel extends Model { @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 isAccountMutedBy (accountId: number, targetAccountId: number) { return AccountBlocklistModel.isAccountMutedByMulti([ accountId ], targetAccountId) .then(result => result[accountId]) } static isAccountMutedByMulti (accountIds: number[], targetAccountId: number) { const query = { attributes: [ 'accountId', 'id' ], where: { accountId: { [Op.in]: accountIds // FIXME: sequelize ANY seems broken }, 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 }) } static loadByAccountAndTarget (accountId: number, targetAccountId: number): Bluebird { const query = { where: { accountId, targetAccountId } } return AccountBlocklistModel.findOne(query) } static listForApi (accountId: number, start: number, count: number, sort: string) { const query = { offset: start, limit: count, order: getSort(sort), where: { accountId } } return AccountBlocklistModel .scope([ ScopeNames.WITH_ACCOUNTS ]) .findAndCountAll(query) .then(({ rows, count }) => { return { total: count, data: rows } }) } toFormattedJSON (this: MAccountBlocklistFormattable): AccountBlock { return { byAccount: this.ByAccount.toFormattedJSON(), blockedAccount: this.BlockedAccount.toFormattedJSON(), createdAt: this.createdAt } } }