server/controllers/api/blocklist.ts

   1 import express from 'express'
   2 import { handleToNameAndHost } from '@server/helpers/actors'
   3 import { AccountBlocklistModel } from '@server/models/account/account-blocklist'
   4 import { getServerActor } from '@server/models/application/application'
   5 import { ServerBlocklistModel } from '@server/models/server/server-blocklist'
   6 import { MActorAccountId, MUserAccountId } from '@server/types/models'
   7 import { BlockStatus } from '@shared/models'
   8 import { asyncMiddleware, blocklistStatusValidator, optionalAuthenticate } from '../../middlewares'
   9 import { logger } from '@server/helpers/logger'
  10
  11 const blocklistRouter = express.Router()
  12
  13 blocklistRouter.get('/status',
  14   optionalAuthenticate,
  15   blocklistStatusValidator,
  16   asyncMiddleware(getBlocklistStatus)
  17 )
  18
  19 // ---------------------------------------------------------------------------
  20
  21 export {
  22   blocklistRouter
  23 }
  24
  25 // ---------------------------------------------------------------------------
  26
  27 async function getBlocklistStatus (req: express.Request, res: express.Response) {
  28   const hosts = req.query.hosts as string[]
  29   const accounts = req.query.accounts as string[]
  30   const user = res.locals.oauth?.token.User
  31
  32   const serverActor = await getServerActor()
  33
  34   const byAccountIds = [ serverActor.Account.id ]
  35   if (user) byAccountIds.push(user.Account.id)
  36
  37   const status: BlockStatus = {
  38     accounts: {},
  39     hosts: {}
  40   }
  41
  42   const baseOptions = {
  43     byAccountIds,
  44     user,
  45     serverActor,
  46     status
  47   }
  48
  49   await Promise.all([
  50     populateServerBlocklistStatus({ ...baseOptions, hosts }),
  51     populateAccountBlocklistStatus({ ...baseOptions, accounts })
  52   ])
  53
  54   return res.json(status)
  55 }
  56
  57 async function populateServerBlocklistStatus (options: {
  58   byAccountIds: number[]
  59   user?: MUserAccountId
  60   serverActor: MActorAccountId
  61   hosts: string[]
  62   status: BlockStatus
  63 }) {
  64   const { byAccountIds, user, serverActor, hosts, status } = options
  65
  66   if (!hosts || hosts.length === 0) return
  67
  68   const serverBlocklistStatus = await ServerBlocklistModel.getBlockStatus(byAccountIds, hosts)
  69
  70   logger.debug('Got server blocklist status.', { serverBlocklistStatus, byAccountIds, hosts })
  71
  72   for (const host of hosts) {
  73     const block = serverBlocklistStatus.find(b => b.host === host)
  74
  75     status.hosts[host] = getStatus(block, serverActor, user)
  76   }
  77 }
  78
  79 async function populateAccountBlocklistStatus (options: {
  80   byAccountIds: number[]
  81   user?: MUserAccountId
  82   serverActor: MActorAccountId
  83   accounts: string[]
  84   status: BlockStatus
  85 }) {
  86   const { byAccountIds, user, serverActor, accounts, status } = options
  87
  88   if (!accounts || accounts.length === 0) return
  89
  90   const accountBlocklistStatus = await AccountBlocklistModel.getBlockStatus(byAccountIds, accounts)
  91
  92   logger.debug('Got account blocklist status.', { accountBlocklistStatus, byAccountIds, accounts })
  93
  94   for (const account of accounts) {
  95     const sanitizedHandle = handleToNameAndHost(account)
  96
  97     const block = accountBlocklistStatus.find(b => b.name === sanitizedHandle.name && b.host === sanitizedHandle.host)
  98
  99     status.accounts[sanitizedHandle.handle] = getStatus(block, serverActor, user)
 100   }
 101 }
 102
 103 function getStatus (block: { accountId: number }, serverActor: MActorAccountId, user?: MUserAccountId) {
 104   return {
 105     blockedByServer: !!(block && block.accountId === serverActor.Account.id),
 106     blockedByUser: !!(block && user && block.accountId === user.Account.id)
 107   }
 108 }