X-Git-Url: https://git.immae.eu/?a=blobdiff_plain;f=server%2Flib%2Fredis.ts;h=b4cd6f8e7e0afe75189f775d7433519ee931a242;hb=fa11e79d4ad9211848938af490493078437bf500;hp=f77d0b62c20ba22d3a92c01da5e097b7b35cf8c1;hpb=97567dd81f508dd6295ac4d73d849aa2ce0a6549;p=github%2FChocobozzz%2FPeerTube.git diff --git a/server/lib/redis.ts b/server/lib/redis.ts index f77d0b62c..b4cd6f8e7 100644 --- a/server/lib/redis.ts +++ b/server/lib/redis.ts @@ -6,13 +6,14 @@ import { CONTACT_FORM_LIFETIME, USER_EMAIL_VERIFY_LIFETIME, USER_PASSWORD_RESET_LIFETIME, + USER_PASSWORD_CREATE_LIFETIME, VIDEO_VIEW_LIFETIME, WEBSERVER } from '../initializers/constants' import { CONFIG } from '../initializers/config' type CachedRoute = { - body: string, + body: string contentType?: string statusCode?: string } @@ -24,7 +25,8 @@ class Redis { private client: RedisClient private prefix: string - private constructor () {} + private constructor () { + } init () { // Already initialized @@ -49,9 +51,9 @@ class Redis { return Object.assign({}, (CONFIG.REDIS.AUTH && CONFIG.REDIS.AUTH != null) ? { password: CONFIG.REDIS.AUTH } : {}, (CONFIG.REDIS.DB) ? { db: CONFIG.REDIS.DB } : {}, - (CONFIG.REDIS.HOSTNAME && CONFIG.REDIS.PORT) ? - { host: CONFIG.REDIS.HOSTNAME, port: CONFIG.REDIS.PORT } : - { path: CONFIG.REDIS.SOCKET } + (CONFIG.REDIS.HOSTNAME && CONFIG.REDIS.PORT) + ? { host: CONFIG.REDIS.HOSTNAME, port: CONFIG.REDIS.PORT } + : { path: CONFIG.REDIS.SOCKET } ) } @@ -63,7 +65,7 @@ class Redis { return this.prefix } - /************* Forgot password *************/ + /* ************ Forgot password ************ */ async setResetPasswordVerificationString (userId: number) { const generatedString = await generateRandomString(32) @@ -73,11 +75,19 @@ class Redis { return generatedString } + async setCreatePasswordVerificationString (userId: number) { + const generatedString = await generateRandomString(32) + + await this.setValue(this.generateResetPasswordKey(userId), generatedString, USER_PASSWORD_CREATE_LIFETIME) + + return generatedString + } + async getResetPasswordLink (userId: number) { return this.getValue(this.generateResetPasswordKey(userId)) } - /************* Email verification *************/ + /* ************ Email verification ************ */ async setVerifyEmailVerificationString (userId: number) { const generatedString = await generateRandomString(32) @@ -91,7 +101,7 @@ class Redis { return this.getValue(this.generateVerifyEmailKey(userId)) } - /************* Contact form per IP *************/ + /* ************ Contact form per IP ************ */ async setContactFormIp (ip: string) { return this.setValue(this.generateContactFormKey(ip), '1', CONTACT_FORM_LIFETIME) @@ -101,7 +111,7 @@ class Redis { return this.exists(this.generateContactFormKey(ip)) } - /************* Views per IP *************/ + /* ************ Views per IP ************ */ setIPVideoView (ip: string, videoUUID: string) { return this.setValue(this.generateViewKey(ip, videoUUID), '1', VIDEO_VIEW_LIFETIME) @@ -111,7 +121,7 @@ class Redis { return this.exists(this.generateViewKey(ip, videoUUID)) } - /************* API cache *************/ + /* ************ API cache ************ */ async getCachedRoute (req: express.Request) { const cached = await this.getObject(this.generateCachedRouteKey(req)) @@ -120,17 +130,17 @@ class Redis { } setCachedRoute (req: express.Request, body: any, lifetime: number, contentType?: string, statusCode?: number) { - const cached: CachedRoute = Object.assign({}, { - body: body.toString() - }, - (contentType) ? { contentType } : null, - (statusCode) ? { statusCode: statusCode.toString() } : null + const cached: CachedRoute = Object.assign( + {}, + { body: body.toString() }, + (contentType) ? { contentType } : null, + (statusCode) ? { statusCode: statusCode.toString() } : null ) return this.setObject(this.generateCachedRouteKey(req), cached, lifetime) } - /************* Video views *************/ + /* ************ Video views ************ */ addVideoView (videoId: number) { const keyIncr = this.generateVideoViewKey(videoId) @@ -173,7 +183,7 @@ class Redis { ]) } - /************* Keys generation *************/ + /* ************ Keys generation ************ */ generateCachedRouteKey (req: express.Request) { return req.method + '-' + req.originalUrl @@ -207,7 +217,7 @@ class Redis { return 'contact-form-' + ip } - /************* Redis helpers *************/ + /* ************ Redis helpers ************ */ private getValue (key: string) { return new Promise((res, rej) => { @@ -265,7 +275,7 @@ class Redis { }) } - private setObject (key: string, obj: { [ id: string ]: string }, expirationMilliseconds: number) { + private setObject (key: string, obj: { [id: string]: string }, expirationMilliseconds: number) { return new Promise((res, rej) => { this.client.hmset(this.prefix + key, obj, (err, ok) => { if (err) return rej(err) @@ -282,7 +292,7 @@ class Redis { } private getObject (key: string) { - return new Promise<{ [ id: string ]: string }>((res, rej) => { + return new Promise<{ [id: string]: string }>((res, rej) => { this.client.hgetall(this.prefix + key, (err, value) => { if (err) return rej(err)