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