]>
Commit | Line | Data |
---|---|---|
80badf49 C |
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 | } |