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   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 }