server/lib/redis.ts

   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   private getValue (key: string) {
  58     return new Promise<string>((res, rej) => {
  59       this.client.get(this.prefix + key, (err, value) => {
  60         if (err) return rej(err)
  61
  62         return res(value)
  63       })
  64     })
  65   }
  66
  67   private setValue (key: string, value: string, expirationMilliseconds: number) {
  68     return new Promise<void>((res, rej) => {
  69       this.client.set(this.prefix + key, value, 'PX', expirationMilliseconds, (err, ok) => {
  70         if (err) return rej(err)
  71
  72         if (ok !== 'OK') return rej(new Error('Redis result is not OK.'))
  73
  74         return res()
  75       })
  76     })
  77   }
  78
  79   private exists (key: string) {
  80     return new Promise<boolean>((res, rej) => {
  81       this.client.exists(this.prefix + key, (err, existsNumber) => {
  82         if (err) return rej(err)
  83
  84         return res(existsNumber === 1)
  85       })
  86     })
  87   }
  88
  89   private generateResetPasswordKey (userId: number) {
  90     return 'reset-password-' + userId
  91   }
  92
  93   private buildViewKey (ip: string, videoUUID: string) {
  94     return videoUUID + '-' + ip
  95   }
  96
  97   static get Instance () {
  98     return this.instance || (this.instance = new this())
  99   }
 100 }
 101
 102 // ---------------------------------------------------------------------------
 103
 104 export {
 105   Redis
 106 }