diff options
Diffstat (limited to 'server/middlewares')
-rw-r--r-- | server/middlewares/cache/cache.ts | 14 | ||||
-rw-r--r-- | server/middlewares/cache/shared/api-cache.ts | 45 | ||||
-rw-r--r-- | server/middlewares/validators/feeds.ts | 46 | ||||
-rw-r--r-- | server/middlewares/validators/users.ts | 4 |
4 files changed, 103 insertions, 6 deletions
diff --git a/server/middlewares/cache/cache.ts b/server/middlewares/cache/cache.ts index e14160ba8..6041c76c3 100644 --- a/server/middlewares/cache/cache.ts +++ b/server/middlewares/cache/cache.ts | |||
@@ -17,12 +17,22 @@ function cacheRoute (duration: string) { | |||
17 | function cacheRouteFactory (options: APICacheOptions) { | 17 | function cacheRouteFactory (options: APICacheOptions) { |
18 | const instance = new ApiCache({ ...defaultOptions, ...options }) | 18 | const instance = new ApiCache({ ...defaultOptions, ...options }) |
19 | 19 | ||
20 | return instance.buildMiddleware.bind(instance) | 20 | return { instance, middleware: instance.buildMiddleware.bind(instance) } |
21 | } | ||
22 | |||
23 | // --------------------------------------------------------------------------- | ||
24 | |||
25 | function buildPodcastGroupsCache (options: { | ||
26 | channelId: number | ||
27 | }) { | ||
28 | return 'podcast-feed-' + options.channelId | ||
21 | } | 29 | } |
22 | 30 | ||
23 | // --------------------------------------------------------------------------- | 31 | // --------------------------------------------------------------------------- |
24 | 32 | ||
25 | export { | 33 | export { |
26 | cacheRoute, | 34 | cacheRoute, |
27 | cacheRouteFactory | 35 | cacheRouteFactory, |
36 | |||
37 | buildPodcastGroupsCache | ||
28 | } | 38 | } |
diff --git a/server/middlewares/cache/shared/api-cache.ts b/server/middlewares/cache/shared/api-cache.ts index 7c366db00..c6197b972 100644 --- a/server/middlewares/cache/shared/api-cache.ts +++ b/server/middlewares/cache/shared/api-cache.ts | |||
@@ -27,7 +27,13 @@ export class ApiCache { | |||
27 | private readonly options: APICacheOptions | 27 | private readonly options: APICacheOptions |
28 | private readonly timers: { [ id: string ]: NodeJS.Timeout } = {} | 28 | private readonly timers: { [ id: string ]: NodeJS.Timeout } = {} |
29 | 29 | ||
30 | private readonly index: { all: string[] } = { all: [] } | 30 | private readonly index = { |
31 | groups: [] as string[], | ||
32 | all: [] as string[] | ||
33 | } | ||
34 | |||
35 | // Cache keys per group | ||
36 | private groups: { [groupIndex: string]: string[] } = {} | ||
31 | 37 | ||
32 | constructor (options: APICacheOptions) { | 38 | constructor (options: APICacheOptions) { |
33 | this.options = { | 39 | this.options = { |
@@ -43,7 +49,7 @@ export class ApiCache { | |||
43 | 49 | ||
44 | return asyncMiddleware( | 50 | return asyncMiddleware( |
45 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { | 51 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { |
46 | const key = Redis.Instance.getPrefix() + 'api-cache-' + req.originalUrl | 52 | const key = this.getCacheKey(req) |
47 | const redis = Redis.Instance.getClient() | 53 | const redis = Redis.Instance.getClient() |
48 | 54 | ||
49 | if (!Redis.Instance.isConnected()) return this.makeResponseCacheable(res, next, key, duration) | 55 | if (!Redis.Instance.isConnected()) return this.makeResponseCacheable(res, next, key, duration) |
@@ -62,6 +68,29 @@ export class ApiCache { | |||
62 | ) | 68 | ) |
63 | } | 69 | } |
64 | 70 | ||
71 | clearGroupSafe (group: string) { | ||
72 | const run = async () => { | ||
73 | const cacheKeys = this.groups[group] | ||
74 | if (!cacheKeys) return | ||
75 | |||
76 | for (const key of cacheKeys) { | ||
77 | try { | ||
78 | await this.clear(key) | ||
79 | } catch (err) { | ||
80 | logger.error('Cannot clear ' + key, { err }) | ||
81 | } | ||
82 | } | ||
83 | |||
84 | delete this.groups[group] | ||
85 | } | ||
86 | |||
87 | void run() | ||
88 | } | ||
89 | |||
90 | private getCacheKey (req: express.Request) { | ||
91 | return Redis.Instance.getPrefix() + 'api-cache-' + req.originalUrl | ||
92 | } | ||
93 | |||
65 | private shouldCacheResponse (response: express.Response) { | 94 | private shouldCacheResponse (response: express.Response) { |
66 | if (!response) return false | 95 | if (!response) return false |
67 | if (this.options.excludeStatus.includes(response.statusCode)) return false | 96 | if (this.options.excludeStatus.includes(response.statusCode)) return false |
@@ -69,8 +98,16 @@ export class ApiCache { | |||
69 | return true | 98 | return true |
70 | } | 99 | } |
71 | 100 | ||
72 | private addIndexEntries (key: string) { | 101 | private addIndexEntries (key: string, res: express.Response) { |
73 | this.index.all.unshift(key) | 102 | this.index.all.unshift(key) |
103 | |||
104 | const groups = res.locals.apicacheGroups || [] | ||
105 | |||
106 | for (const group of groups) { | ||
107 | if (!this.groups[group]) this.groups[group] = [] | ||
108 | |||
109 | this.groups[group].push(key) | ||
110 | } | ||
74 | } | 111 | } |
75 | 112 | ||
76 | private filterBlacklistedHeaders (headers: OutgoingHttpHeaders) { | 113 | private filterBlacklistedHeaders (headers: OutgoingHttpHeaders) { |
@@ -177,7 +214,7 @@ export class ApiCache { | |||
177 | self.accumulateContent(res, content) | 214 | self.accumulateContent(res, content) |
178 | 215 | ||
179 | if (res.locals.apicache.cacheable && res.locals.apicache.content) { | 216 | if (res.locals.apicache.cacheable && res.locals.apicache.content) { |
180 | self.addIndexEntries(key) | 217 | self.addIndexEntries(key, res) |
181 | 218 | ||
182 | const headers = res.locals.apicache.headers || res.getHeaders() | 219 | const headers = res.locals.apicache.headers || res.getHeaders() |
183 | const cacheObject = self.createCacheObject( | 220 | const cacheObject = self.createCacheObject( |
diff --git a/server/middlewares/validators/feeds.ts b/server/middlewares/validators/feeds.ts index 0bfe89e6f..ee8615cae 100644 --- a/server/middlewares/validators/feeds.ts +++ b/server/middlewares/validators/feeds.ts | |||
@@ -3,6 +3,7 @@ import { param, query } from 'express-validator' | |||
3 | import { HttpStatusCode } from '../../../shared/models/http/http-error-codes' | 3 | import { HttpStatusCode } from '../../../shared/models/http/http-error-codes' |
4 | import { isValidRSSFeed } from '../../helpers/custom-validators/feeds' | 4 | import { isValidRSSFeed } from '../../helpers/custom-validators/feeds' |
5 | import { exists, isIdOrUUIDValid, isIdValid, toCompleteUUID } from '../../helpers/custom-validators/misc' | 5 | import { exists, isIdOrUUIDValid, isIdValid, toCompleteUUID } from '../../helpers/custom-validators/misc' |
6 | import { buildPodcastGroupsCache } from '../cache' | ||
6 | import { | 7 | import { |
7 | areValidationErrors, | 8 | areValidationErrors, |
8 | checkCanSeeVideo, | 9 | checkCanSeeVideo, |
@@ -43,6 +44,21 @@ function setFeedFormatContentType (req: express.Request, res: express.Response, | |||
43 | acceptableContentTypes = [ 'application/xml', 'text/xml' ] | 44 | acceptableContentTypes = [ 'application/xml', 'text/xml' ] |
44 | } | 45 | } |
45 | 46 | ||
47 | return feedContentTypeResponse(req, res, next, acceptableContentTypes) | ||
48 | } | ||
49 | |||
50 | function setFeedPodcastContentType (req: express.Request, res: express.Response, next: express.NextFunction) { | ||
51 | const acceptableContentTypes = [ 'application/rss+xml', 'application/xml', 'text/xml' ] | ||
52 | |||
53 | return feedContentTypeResponse(req, res, next, acceptableContentTypes) | ||
54 | } | ||
55 | |||
56 | function feedContentTypeResponse ( | ||
57 | req: express.Request, | ||
58 | res: express.Response, | ||
59 | next: express.NextFunction, | ||
60 | acceptableContentTypes: string[] | ||
61 | ) { | ||
46 | if (req.accepts(acceptableContentTypes)) { | 62 | if (req.accepts(acceptableContentTypes)) { |
47 | res.set('Content-Type', req.accepts(acceptableContentTypes) as string) | 63 | res.set('Content-Type', req.accepts(acceptableContentTypes) as string) |
48 | } else { | 64 | } else { |
@@ -55,6 +71,8 @@ function setFeedFormatContentType (req: express.Request, res: express.Response, | |||
55 | return next() | 71 | return next() |
56 | } | 72 | } |
57 | 73 | ||
74 | // --------------------------------------------------------------------------- | ||
75 | |||
58 | const videoFeedsValidator = [ | 76 | const videoFeedsValidator = [ |
59 | query('accountId') | 77 | query('accountId') |
60 | .optional() | 78 | .optional() |
@@ -82,6 +100,31 @@ const videoFeedsValidator = [ | |||
82 | } | 100 | } |
83 | ] | 101 | ] |
84 | 102 | ||
103 | // --------------------------------------------------------------------------- | ||
104 | |||
105 | const videoFeedsPodcastValidator = [ | ||
106 | query('videoChannelId') | ||
107 | .custom(isIdValid), | ||
108 | |||
109 | async (req: express.Request, res: express.Response, next: express.NextFunction) => { | ||
110 | if (areValidationErrors(req, res)) return | ||
111 | if (!await doesVideoChannelIdExist(req.query.videoChannelId, res)) return | ||
112 | |||
113 | return next() | ||
114 | } | ||
115 | ] | ||
116 | |||
117 | const videoFeedsPodcastSetCacheKey = [ | ||
118 | (req: express.Request, res: express.Response, next: express.NextFunction) => { | ||
119 | if (req.query.videoChannelId) { | ||
120 | res.locals.apicacheGroups = [ buildPodcastGroupsCache({ channelId: req.query.videoChannelId }) ] | ||
121 | } | ||
122 | |||
123 | return next() | ||
124 | } | ||
125 | ] | ||
126 | // --------------------------------------------------------------------------- | ||
127 | |||
85 | const videoSubscriptionFeedsValidator = [ | 128 | const videoSubscriptionFeedsValidator = [ |
86 | query('accountId') | 129 | query('accountId') |
87 | .custom(isIdValid), | 130 | .custom(isIdValid), |
@@ -126,7 +169,10 @@ const videoCommentsFeedsValidator = [ | |||
126 | export { | 169 | export { |
127 | feedsFormatValidator, | 170 | feedsFormatValidator, |
128 | setFeedFormatContentType, | 171 | setFeedFormatContentType, |
172 | setFeedPodcastContentType, | ||
129 | videoFeedsValidator, | 173 | videoFeedsValidator, |
174 | videoFeedsPodcastValidator, | ||
130 | videoSubscriptionFeedsValidator, | 175 | videoSubscriptionFeedsValidator, |
176 | videoFeedsPodcastSetCacheKey, | ||
131 | videoCommentsFeedsValidator | 177 | videoCommentsFeedsValidator |
132 | } | 178 | } |
diff --git a/server/middlewares/validators/users.ts b/server/middlewares/validators/users.ts index 7ebea048d..3d311b15b 100644 --- a/server/middlewares/validators/users.ts +++ b/server/middlewares/validators/users.ts | |||
@@ -11,6 +11,7 @@ import { | |||
11 | isUserBlockedReasonValid, | 11 | isUserBlockedReasonValid, |
12 | isUserDescriptionValid, | 12 | isUserDescriptionValid, |
13 | isUserDisplayNameValid, | 13 | isUserDisplayNameValid, |
14 | isUserEmailPublicValid, | ||
14 | isUserNoModal, | 15 | isUserNoModal, |
15 | isUserNSFWPolicyValid, | 16 | isUserNSFWPolicyValid, |
16 | isUserP2PEnabledValid, | 17 | isUserP2PEnabledValid, |
@@ -213,6 +214,9 @@ const usersUpdateMeValidator = [ | |||
213 | body('password') | 214 | body('password') |
214 | .optional() | 215 | .optional() |
215 | .custom(isUserPasswordValid), | 216 | .custom(isUserPasswordValid), |
217 | body('emailPublic') | ||
218 | .optional() | ||
219 | .custom(isUserEmailPublicValid), | ||
216 | body('email') | 220 | body('email') |
217 | .optional() | 221 | .optional() |
218 | .isEmail(), | 222 | .isEmail(), |