]>
Commit | Line | Data |
---|---|---|
1 | import { createClient, RedisClient } from 'redis' | |
2 | import { logger } from '../helpers/logger' | |
3 | import { generateRandomString } from '../helpers/utils' | |
4 | import { CONFIG, USER_PASSWORD_RESET_LIFETIME, VIDEO_VIEW_LIFETIME } from '../initializers' | |
5 | ||
6 | class Redis { | |
7 | ||
8 | private static instance: Redis | |
9 | private initialized = false | |
10 | private client: RedisClient | |
11 | private prefix: string | |
12 | ||
13 | private constructor () {} | |
14 | ||
15 | init () { | |
16 | // Already initialized | |
17 | if (this.initialized === true) return | |
18 | this.initialized = true | |
19 | ||
20 | this.client = createClient({ | |
21 | host: CONFIG.REDIS.HOSTNAME, | |
22 | port: CONFIG.REDIS.PORT | |
23 | }) | |
24 | ||
25 | this.client.on('error', err => { | |
26 | logger.error('Error in Redis client.', err) | |
27 | process.exit(-1) | |
28 | }) | |
29 | ||
30 | if (CONFIG.REDIS.AUTH) { | |
31 | this.client.auth(CONFIG.REDIS.AUTH) | |
32 | } | |
33 | ||
34 | this.prefix = 'redis-' + CONFIG.WEBSERVER.HOST + '-' | |
35 | } | |
36 | ||
37 | async setResetPasswordVerificationString (userId: number) { | |
38 | const generatedString = await generateRandomString(32) | |
39 | ||
40 | await this.setValue(this.generateResetPasswordKey(userId), generatedString, USER_PASSWORD_RESET_LIFETIME) | |
41 | ||
42 | return generatedString | |
43 | } | |
44 | ||
45 | async getResetPasswordLink (userId: number) { | |
46 | return this.getValue(this.generateResetPasswordKey(userId)) | |
47 | } | |
48 | ||
49 | setView (ip: string, videoUUID: string) { | |
50 | return this.setValue(this.buildViewKey(ip, videoUUID), '1', VIDEO_VIEW_LIFETIME) | |
51 | } | |
52 | ||
53 | async isViewExists (ip: string, videoUUID: string) { | |
54 | return this.exists(this.buildViewKey(ip, videoUUID)) | |
55 | } | |
56 | ||
57 | listJobs (jobsPrefix: string, state: string, mode: 'alpha', order: 'ASC' | 'DESC', offset: number, count: number) { | |
58 | return new Promise<string[]>((res, rej) => { | |
59 | this.client.sort(jobsPrefix + ':jobs:' + state, 'by', mode, order, 'LIMIT', offset.toString(), count.toString(), (err, values) => { | |
60 | if (err) return rej(err) | |
61 | ||
62 | return res(values) | |
63 | }) | |
64 | }) | |
65 | } | |
66 | ||
67 | private getValue (key: string) { | |
68 | return new Promise<string>((res, rej) => { | |
69 | this.client.get(this.prefix + key, (err, value) => { | |
70 | if (err) return rej(err) | |
71 | ||
72 | return res(value) | |
73 | }) | |
74 | }) | |
75 | } | |
76 | ||
77 | private setValue (key: string, value: string, expirationMilliseconds: number) { | |
78 | return new Promise<void>((res, rej) => { | |
79 | this.client.set(this.prefix + key, value, 'PX', expirationMilliseconds, (err, ok) => { | |
80 | if (err) return rej(err) | |
81 | ||
82 | if (ok !== 'OK') return rej(new Error('Redis result is not OK.')) | |
83 | ||
84 | return res() | |
85 | }) | |
86 | }) | |
87 | } | |
88 | ||
89 | private exists (key: string) { | |
90 | return new Promise<boolean>((res, rej) => { | |
91 | this.client.exists(this.prefix + key, (err, existsNumber) => { | |
92 | if (err) return rej(err) | |
93 | ||
94 | return res(existsNumber === 1) | |
95 | }) | |
96 | }) | |
97 | } | |
98 | ||
99 | private generateResetPasswordKey (userId: number) { | |
100 | return 'reset-password-' + userId | |
101 | } | |
102 | ||
103 | private buildViewKey (ip: string, videoUUID: string) { | |
104 | return videoUUID + '-' + ip | |
105 | } | |
106 | ||
107 | static get Instance () { | |
108 | return this.instance || (this.instance = new this()) | |
109 | } | |
110 | } | |
111 | ||
112 | // --------------------------------------------------------------------------- | |
113 | ||
114 | export { | |
115 | Redis | |
116 | } |