diff options
Diffstat (limited to 'server/middlewares/validators/blocklist.ts')
-rw-r--r-- | server/middlewares/validators/blocklist.ts | 94 |
1 files changed, 94 insertions, 0 deletions
diff --git a/server/middlewares/validators/blocklist.ts b/server/middlewares/validators/blocklist.ts new file mode 100644 index 000000000..9dbd5e512 --- /dev/null +++ b/server/middlewares/validators/blocklist.ts | |||
@@ -0,0 +1,94 @@ | |||
1 | import { param, body } from 'express-validator/check' | ||
2 | import * as express from 'express' | ||
3 | import { logger } from '../../helpers/logger' | ||
4 | import { areValidationErrors } from './utils' | ||
5 | import { isAccountNameWithHostExist } from '../../helpers/custom-validators/accounts' | ||
6 | import { UserModel } from '../../models/account/user' | ||
7 | import { AccountBlocklistModel } from '../../models/account/account-blocklist' | ||
8 | import { isHostValid } from '../../helpers/custom-validators/servers' | ||
9 | import { ServerBlocklistModel } from '../../models/server/server-blocklist' | ||
10 | |||
11 | const blockAccountByAccountValidator = [ | ||
12 | body('accountName').exists().withMessage('Should have an account name with host'), | ||
13 | |||
14 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { | ||
15 | logger.debug('Checking blockAccountByAccountValidator parameters', { parameters: req.body }) | ||
16 | |||
17 | if (areValidationErrors(req, res)) return | ||
18 | if (!await isAccountNameWithHostExist(req.body.accountName, res)) return | ||
19 | |||
20 | return next() | ||
21 | } | ||
22 | ] | ||
23 | |||
24 | const unblockAccountByAccountValidator = [ | ||
25 | param('accountName').exists().withMessage('Should have an account name with host'), | ||
26 | |||
27 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { | ||
28 | logger.debug('Checking unblockAccountByAccountValidator parameters', { parameters: req.params }) | ||
29 | |||
30 | if (areValidationErrors(req, res)) return | ||
31 | if (!await isAccountNameWithHostExist(req.params.accountName, res)) return | ||
32 | |||
33 | const user = res.locals.oauth.token.User as UserModel | ||
34 | const targetAccount = res.locals.account | ||
35 | if (!await isUnblockAccountExists(user.Account.id, targetAccount.id, res)) return | ||
36 | |||
37 | return next() | ||
38 | } | ||
39 | ] | ||
40 | |||
41 | const unblockServerByAccountValidator = [ | ||
42 | param('host').custom(isHostValid).withMessage('Should have an account name with host'), | ||
43 | |||
44 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { | ||
45 | logger.debug('Checking unblockServerByAccountValidator parameters', { parameters: req.params }) | ||
46 | |||
47 | if (areValidationErrors(req, res)) return | ||
48 | |||
49 | const user = res.locals.oauth.token.User as UserModel | ||
50 | if (!await isUnblockServerExists(user.Account.id, req.params.host, res)) return | ||
51 | |||
52 | return next() | ||
53 | } | ||
54 | ] | ||
55 | |||
56 | // --------------------------------------------------------------------------- | ||
57 | |||
58 | export { | ||
59 | blockAccountByAccountValidator, | ||
60 | unblockAccountByAccountValidator, | ||
61 | unblockServerByAccountValidator | ||
62 | } | ||
63 | |||
64 | // --------------------------------------------------------------------------- | ||
65 | |||
66 | async function isUnblockAccountExists (accountId: number, targetAccountId: number, res: express.Response) { | ||
67 | const accountBlock = await AccountBlocklistModel.loadByAccountAndTarget(accountId, targetAccountId) | ||
68 | if (!accountBlock) { | ||
69 | res.status(404) | ||
70 | .send({ error: 'Account block entry not found.' }) | ||
71 | .end() | ||
72 | |||
73 | return false | ||
74 | } | ||
75 | |||
76 | res.locals.accountBlock = accountBlock | ||
77 | |||
78 | return true | ||
79 | } | ||
80 | |||
81 | async function isUnblockServerExists (accountId: number, host: string, res: express.Response) { | ||
82 | const serverBlock = await ServerBlocklistModel.loadByAccountAndHost(accountId, host) | ||
83 | if (!serverBlock) { | ||
84 | res.status(404) | ||
85 | .send({ error: 'Server block entry not found.' }) | ||
86 | .end() | ||
87 | |||
88 | return false | ||
89 | } | ||
90 | |||
91 | res.locals.serverBlock = serverBlock | ||
92 | |||
93 | return true | ||
94 | } | ||