import { db } from '../../db.js'
import { ModelTypes } from '../../cache/dbs/index.generated.js'

import { getId, addMinutes } from 'topkat-utils'
import { UserLockReasonsDefault } from './userAdditionalFields.js'


export async function ensureUserIsNotLocked(ctx: Ctx, userOrId: ModelTypes['user'] | string) {

  const user = typeof userOrId === 'string' ? await db.user.getById(ctx.GM, userOrId, { triggerErrorIfNotSet: true }) : userOrId

  if (user.isLocked) {
    if (user.lockUntil && (new Date(user.lockUntil)).getTime() < Date.now()) {
      // UNLOCK
      user.isLocked = false
      delete user.lockUntil
      await db.user.update(ctx.GM, getId(user), {
        isLocked: false,
        lockUntil: null,
      })
    } else {
      // IS LOCKED ERROR
      throw ctx.error.userLocked({ lockUntil: user.lockUntil, lockedReason: user.lockedReason, userId: getId(user) })
    }
  }

  // May be blacklisted because rateLimiter doesn't always have access to user
  const userInBlacklist = await db.GD_serverBlackList.getOne(ctx.GM, { discriminator: getId(user) })

  if (userInBlacklist?.lockUntil) {
    if (new Date(userInBlacklist.lockUntil) > new Date()) {
      throw ctx.error.userLocked({ lockUntil: userInBlacklist.lockUntil })
    } else {
      await db.GD_serverBlackList.update(ctx.GM, getId(userInBlacklist), {
        lockUntil: null
      })
    }
  }
}


export async function lockUserAndThrow(ctx: Ctx, userId: string, reason: UserLockReasonsDefault | (string & {}), conf: {
  errExtraInfos?: Record<string, any>,
  lockDurationMinutes?: number
} = {}) {
  const lockUntil = addMinutes(new Date(), conf.lockDurationMinutes || 15, 'date')
  await db.user.update(ctx.GM, userId, {
    isLocked: true,
    lockedReason: reason as any,
    lockUntil,
    refreshTokens: [],
  })

  throw ctx.error.userLocked({
    lockedReason: reason,
    lockUntil: addMinutes(new Date(), conf.lockDurationMinutes || 15),
    userId,
    ...(conf.errExtraInfos || {})
  })
}