server/controllers/api/users/my-blocklist.ts

   1 import * as express from 'express'
   2 import 'multer'
   3 import { getFormattedObjects } from '../../../helpers/utils'
   4 import {
   5   asyncMiddleware,
   6   asyncRetryTransactionMiddleware,
   7   authenticate,
   8   paginationValidator,
   9   setDefaultPagination,
  10   setDefaultSort,
  11   unblockAccountByAccountValidator
  12 } from '../../../middlewares'
  13 import {
  14   accountsBlocklistSortValidator,
  15   blockAccountValidator,
  16   blockServerValidator,
  17   serversBlocklistSortValidator,
  18   unblockServerByAccountValidator
  19 } from '../../../middlewares/validators'
  20 import { AccountBlocklistModel } from '../../../models/account/account-blocklist'
  21 import { addAccountInBlocklist, addServerInBlocklist, removeAccountFromBlocklist, removeServerFromBlocklist } from '../../../lib/blocklist'
  22 import { ServerBlocklistModel } from '../../../models/server/server-blocklist'
  23 import { UserNotificationModel } from '@server/models/account/user-notification'
  24 import { logger } from '@server/helpers/logger'
  25 import { HttpStatusCode } from '../../../../shared/core-utils/miscs/http-error-codes'
  26
  27 const myBlocklistRouter = express.Router()
  28
  29 myBlocklistRouter.get('/me/blocklist/accounts',
  30   authenticate,
  31   paginationValidator,
  32   accountsBlocklistSortValidator,
  33   setDefaultSort,
  34   setDefaultPagination,
  35   asyncMiddleware(listBlockedAccounts)
  36 )
  37
  38 myBlocklistRouter.post('/me/blocklist/accounts',
  39   authenticate,
  40   asyncMiddleware(blockAccountValidator),
  41   asyncRetryTransactionMiddleware(blockAccount)
  42 )
  43
  44 myBlocklistRouter.delete('/me/blocklist/accounts/:accountName',
  45   authenticate,
  46   asyncMiddleware(unblockAccountByAccountValidator),
  47   asyncRetryTransactionMiddleware(unblockAccount)
  48 )
  49
  50 myBlocklistRouter.get('/me/blocklist/servers',
  51   authenticate,
  52   paginationValidator,
  53   serversBlocklistSortValidator,
  54   setDefaultSort,
  55   setDefaultPagination,
  56   asyncMiddleware(listBlockedServers)
  57 )
  58
  59 myBlocklistRouter.post('/me/blocklist/servers',
  60   authenticate,
  61   asyncMiddleware(blockServerValidator),
  62   asyncRetryTransactionMiddleware(blockServer)
  63 )
  64
  65 myBlocklistRouter.delete('/me/blocklist/servers/:host',
  66   authenticate,
  67   asyncMiddleware(unblockServerByAccountValidator),
  68   asyncRetryTransactionMiddleware(unblockServer)
  69 )
  70
  71 export {
  72   myBlocklistRouter
  73 }
  74
  75 // ---------------------------------------------------------------------------
  76
  77 async function listBlockedAccounts (req: express.Request, res: express.Response) {
  78   const user = res.locals.oauth.token.User
  79
  80   const resultList = await AccountBlocklistModel.listForApi({
  81     start: req.query.start,
  82     count: req.query.count,
  83     sort: req.query.sort,
  84     search: req.query.search,
  85     accountId: user.Account.id
  86   })
  87
  88   return res.json(getFormattedObjects(resultList.data, resultList.total))
  89 }
  90
  91 async function blockAccount (req: express.Request, res: express.Response) {
  92   const user = res.locals.oauth.token.User
  93   const accountToBlock = res.locals.account
  94
  95   await addAccountInBlocklist(user.Account.id, accountToBlock.id)
  96
  97   UserNotificationModel.removeNotificationsOf({
  98     id: accountToBlock.id,
  99     type: 'account',
 100     forUserId: user.id
 101   }).catch(err => logger.error('Cannot remove notifications after an account mute.', { err }))
 102
 103   return res.status(HttpStatusCode.NO_CONTENT_204).end()
 104 }
 105
 106 async function unblockAccount (req: express.Request, res: express.Response) {
 107   const accountBlock = res.locals.accountBlock
 108
 109   await removeAccountFromBlocklist(accountBlock)
 110
 111   return res.status(HttpStatusCode.NO_CONTENT_204).end()
 112 }
 113
 114 async function listBlockedServers (req: express.Request, res: express.Response) {
 115   const user = res.locals.oauth.token.User
 116
 117   const resultList = await ServerBlocklistModel.listForApi({
 118     start: req.query.start,
 119     count: req.query.count,
 120     sort: req.query.sort,
 121     search: req.query.search,
 122     accountId: user.Account.id
 123   })
 124
 125   return res.json(getFormattedObjects(resultList.data, resultList.total))
 126 }
 127
 128 async function blockServer (req: express.Request, res: express.Response) {
 129   const user = res.locals.oauth.token.User
 130   const serverToBlock = res.locals.server
 131
 132   await addServerInBlocklist(user.Account.id, serverToBlock.id)
 133
 134   UserNotificationModel.removeNotificationsOf({
 135     id: serverToBlock.id,
 136     type: 'server',
 137     forUserId: user.id
 138   }).catch(err => logger.error('Cannot remove notifications after a server mute.', { err }))
 139
 140   return res.status(HttpStatusCode.NO_CONTENT_204).end()
 141 }
 142
 143 async function unblockServer (req: express.Request, res: express.Response) {
 144   const serverBlock = res.locals.serverBlock
 145
 146   await removeServerFromBlocklist(serverBlock)
 147
 148   return res.status(HttpStatusCode.NO_CONTENT_204).end()
 149 }