import { Context } from 'elysia'
import {
  getClientIp,
  getDatabaseConnection,
  jsonResponse,
  validateTurnstile
} from '../lib/util'
import { randomBytes } from 'node:crypto'
import { verifyCodes } from '../lib/tables'
import { and, desc, eq, sql } from 'drizzle-orm'

type Body = {
  token: string
}

export const handler = async (context: Context) => {
  const body = context.body as Body
  const ip = getClientIp(context)
  const code = randomBytes(8).toString('hex')
  const time = Math.floor(Date.now() / 1000)

  if (!ip || !body.token)
    return jsonResponse(
      {
        success: false,
        message: 'Unable to verify captcha key',
        data: 'No token in body or no valid IP'
      },
      400
    )

  const result = await validateTurnstile(body.token, ip)
  if (!result.success)
    return jsonResponse(
      {
        success: false,
        message: 'Unable to verify captcha key',
        data: result
      },
      400
    )

  const dbInfo0 = getDatabaseConnection(0)

  if (!dbInfo0)
    return jsonResponse(
      { success: false, message: 'Failed to connect to database', data: null },
      500
    )
  const { connection: connection0, db: db0 } = dbInfo0

  const codeExists = await db0
    .select({ code: verifyCodes.code })
    .from(verifyCodes)
    .where(
      and(
        eq(verifyCodes.ip, ip),
        eq(verifyCodes.usedTimestamp, 0),
        sql`${verifyCodes.timestamp} >= UNIX_TIMESTAMP() - 600`
      )
    )
    .orderBy(desc(verifyCodes.id))
    .limit(1)
    .execute()
  if (codeExists[0]) {
    connection0.end()
    return jsonResponse(
      {
        success: true,
        message: null,
        data: codeExists[0].code
      },
      200
    )
  }

  await db0.insert(verifyCodes).values({ code, ip, timestamp: time })

  connection0.end()

  return jsonResponse(
    {
      success: true,
      message: null,
      data: code
    },
    200
  )
}