import { Command } from 'commander'

import { handleError } from '@/shared/utils/error-handler'
import { formatOutput } from '@/shared/utils/output'

import { DiscordClient } from '../client'
import { DiscordCredentialManager } from '../credential-manager'

async function getAction(userId: string, options: { pretty?: boolean }): Promise<void> {
  try {
    const credManager = new DiscordCredentialManager()
    const config = await credManager.load()

    if (!config.token) {
      console.log(formatOutput({ error: 'Not authenticated. Run "auth extract" first.' }, options.pretty))
      process.exit(1)
    }

    const client = await new DiscordClient().login({ token: config.token })
    const note = await client.getUserNote(userId)

    if (note === null) {
      console.log(formatOutput({ note: null }, options.pretty))
    } else {
      console.log(formatOutput(note, options.pretty))
    }
  } catch (error) {
    handleError(error as Error)
  }
}

async function setAction(userId: string, note: string, options: { pretty?: boolean }): Promise<void> {
  try {
    const credManager = new DiscordCredentialManager()
    const config = await credManager.load()

    if (!config.token) {
      console.log(formatOutput({ error: 'Not authenticated. Run "auth extract" first.' }, options.pretty))
      process.exit(1)
    }

    const client = await new DiscordClient().login({ token: config.token })
    const result = await client.setUserNote(userId, note)

    console.log(formatOutput(result, options.pretty))
  } catch (error) {
    handleError(error as Error)
  }
}

export const noteCommand = new Command('note')
  .description('User note commands')
  .addCommand(
    new Command('get')
      .description('Get note for a user')
      .argument('<user-id>', 'User ID')
      .option('--pretty', 'Pretty print JSON output')
      .action(getAction),
  )
  .addCommand(
    new Command('set')
      .description('Set note for a user')
      .argument('<user-id>', 'User ID')
      .argument('<note>', 'Note content')
      .option('--pretty', 'Pretty print JSON output')
      .action(setAction),
  )