mirror of https://github.com/Chocobozzz/PeerTube
91 lines
3.0 KiB
TypeScript
91 lines
3.0 KiB
TypeScript
import { sequelizeTypescript } from '@server/initializers/database.js'
|
|
import { getServerActor } from '@server/models/application/application.js'
|
|
import { MAccountBlocklist, MAccountId, MAccountHost, MServerBlocklist } from '@server/types/models/index.js'
|
|
import { AccountBlocklistModel } from '../models/account/account-blocklist.js'
|
|
import { ServerBlocklistModel } from '../models/server/server-blocklist.js'
|
|
import { UserNotificationModel } from '@server/models/user/user-notification.js'
|
|
import { logger } from '@server/helpers/logger.js'
|
|
|
|
async function addAccountInBlocklist (options: {
|
|
byAccountId: number
|
|
targetAccountId: number
|
|
|
|
removeNotificationOfUserId: number | null // If blocked by a user
|
|
}) {
|
|
const { byAccountId, targetAccountId, removeNotificationOfUserId } = options
|
|
|
|
await sequelizeTypescript.transaction(async t => {
|
|
return AccountBlocklistModel.upsert({
|
|
accountId: byAccountId,
|
|
targetAccountId
|
|
}, { transaction: t })
|
|
})
|
|
|
|
UserNotificationModel.removeNotificationsOf({
|
|
id: targetAccountId,
|
|
type: 'account',
|
|
forUserId: removeNotificationOfUserId
|
|
}).catch(err => logger.error('Cannot remove notifications after an account mute.', { err }))
|
|
}
|
|
|
|
async function addServerInBlocklist (options: {
|
|
byAccountId: number
|
|
targetServerId: number
|
|
|
|
removeNotificationOfUserId: number | null
|
|
}) {
|
|
const { byAccountId, targetServerId, removeNotificationOfUserId } = options
|
|
|
|
await sequelizeTypescript.transaction(async t => {
|
|
return ServerBlocklistModel.upsert({
|
|
accountId: byAccountId,
|
|
targetServerId
|
|
}, { transaction: t })
|
|
})
|
|
|
|
UserNotificationModel.removeNotificationsOf({
|
|
id: targetServerId,
|
|
type: 'server',
|
|
forUserId: removeNotificationOfUserId
|
|
}).catch(err => logger.error('Cannot remove notifications after a server mute.', { err }))
|
|
}
|
|
|
|
function removeAccountFromBlocklist (accountBlock: MAccountBlocklist) {
|
|
return sequelizeTypescript.transaction(async t => {
|
|
return accountBlock.destroy({ transaction: t })
|
|
})
|
|
}
|
|
|
|
function removeServerFromBlocklist (serverBlock: MServerBlocklist) {
|
|
return sequelizeTypescript.transaction(async t => {
|
|
return serverBlock.destroy({ transaction: t })
|
|
})
|
|
}
|
|
|
|
async function isBlockedByServerOrAccount (targetAccount: MAccountHost, userAccount?: MAccountId) {
|
|
const serverAccountId = (await getServerActor()).Account.id
|
|
const sourceAccounts = [ serverAccountId ]
|
|
|
|
if (userAccount) sourceAccounts.push(userAccount.id)
|
|
|
|
const accountMutedHash = await AccountBlocklistModel.isAccountMutedByAccounts(sourceAccounts, targetAccount.id)
|
|
if (accountMutedHash[serverAccountId] || (userAccount && accountMutedHash[userAccount.id])) {
|
|
return true
|
|
}
|
|
|
|
const instanceMutedHash = await ServerBlocklistModel.isServerMutedByAccounts(sourceAccounts, targetAccount.Actor.serverId)
|
|
if (instanceMutedHash[serverAccountId] || (userAccount && instanceMutedHash[userAccount.id])) {
|
|
return true
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
export {
|
|
addAccountInBlocklist,
|
|
addServerInBlocklist,
|
|
removeAccountFromBlocklist,
|
|
removeServerFromBlocklist,
|
|
isBlockedByServerOrAccount
|
|
}
|