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