import { InvalidRequestError } from '@atproto/xrpc-server'
import { AppContext } from '../../../../context'
import { Server } from '../../../../lexicon'
import { AccountCodes } from '../../../../lexicon/types/com/atproto/server/createInviteCodes'
import { genInvCodes } from './util'

export default function (server: Server, ctx: AppContext) {
  server.com.atproto.server.createInviteCodes({
    auth: ctx.authVerifier.adminToken,
    handler: async ({ input }) => {
      if (ctx.cfg.entryway) {
        throw new InvalidRequestError(
          'Account invites are managed by the entryway service',
        )
      }

      const { codeCount, useCount } = input.body

      const forAccounts = input.body.forAccounts ?? ['admin']

      const accountCodes: AccountCodes[] = []
      for (const account of forAccounts) {
        const codes = genInvCodes(ctx.cfg, codeCount)
        accountCodes.push({ account, codes })
      }
      await ctx.accountManager.createInviteCodes(accountCodes, useCount)

      return {
        encoding: 'application/json',
        body: { codes: accountCodes },
      }
    },
  })
}