standardnotes-server/packages/auth/src/Infra/Redis/RedisTransitionStatusRepository.ts

import * as IORedis from 'ioredis'

import { TransitionStatusRepositoryInterface } from '../../Domain/Transition/TransitionStatusRepositoryInterface'

export class RedisTransitionStatusRepository implements TransitionStatusRepositoryInterface {
  private readonly PREFIX = 'transition'

  constructor(private redisClient: IORedis.Redis) {}

  async updateStatus(
    userUuid: string,
    transitionType: 'items' | 'revisions',
    status: 'STARTED' | 'FAILED',
  ): Promise<void> {
    await this.redisClient.set(`${this.PREFIX}:${transitionType}:${userUuid}`, status)
  }

  async removeStatus(userUuid: string, transitionType: 'items' | 'revisions'): Promise<void> {
    await this.redisClient.del(`${this.PREFIX}:${transitionType}:${userUuid}`)
  }

  async getStatus(userUuid: string, transitionType: 'items' | 'revisions'): Promise<'STARTED' | 'FAILED' | null> {
    const status = (await this.redisClient.get(`${this.PREFIX}:${transitionType}:${userUuid}`)) as
      | 'STARTED'
      | 'FAILED'
      | null

    return status
  }
}