import {
  loadConfig,
  saveConfig,
  type EthagentConfig,
  type EthagentIdentity,
} from './config.js'
import {
  rmSecret,
  hasSecret,
  whichBackend,
  type KeyBackend,
} from './secrets.js'

const IDENTITY_ACCOUNT = 'ethereum:default'

export type IdentityStatus = {
  address: string
  createdAt: string
  backend: KeyBackend | 'browser-wallet'
  backup?: EthagentIdentity['backup']
  agentCard?: EthagentIdentity['agentCard']
  source?: EthagentIdentity['source']
  agentId?: string
  chainId?: number
} | null

export async function getIdentityStatus(config?: EthagentConfig): Promise<IdentityStatus> {
  const resolved = config ?? (await loadConfig())
  if (!resolved?.identity) return null
  if (resolved.identity.source === 'erc8004' || resolved.identity.agentId) {
    return {
      address: resolved.identity.address,
      createdAt: resolved.identity.createdAt,
      backend: 'browser-wallet',
      backup: resolved.identity.backup,
      agentCard: resolved.identity.agentCard,
      source: resolved.identity.source,
      agentId: resolved.identity.agentId,
      chainId: resolved.identity.chainId,
    }
  }
  const present = await hasSecret(IDENTITY_ACCOUNT)
  if (!present) return null
  const backend = await whichBackend()
  return {
    address: resolved.identity.address,
    createdAt: resolved.identity.createdAt,
    backend,
    backup: resolved.identity.backup,
    agentCard: resolved.identity.agentCard,
    source: resolved.identity.source,
    chainId: resolved.identity.chainId,
  }
}

export async function setTokenIdentity(
  config: EthagentConfig,
  identity: EthagentIdentity,
): Promise<EthagentConfig> {
  if (!identity.address || !identity.agentId || !identity.agentUri || !identity.ownerAddress) {
    throw new Error('Token identity is missing ERC-8004 metadata')
  }
  const next: EthagentConfig = {
    ...config,
    identity: {
      ...identity,
      source: 'erc8004',
    },
  }
  if (!next.erc8004 && identity.chainId && identity.rpcUrl && identity.identityRegistryAddress) {
    next.erc8004 = {
      chainId: identity.chainId,
      rpcUrl: identity.rpcUrl,
      identityRegistryAddress: identity.identityRegistryAddress,
    }
  }
  await saveConfig(next)
  return next
}

export async function updateIdentityBackup(
  config: EthagentConfig,
  backup: NonNullable<EthagentIdentity['backup']>,
): Promise<EthagentConfig> {
  if (!config.identity) throw new Error('No identity set')
  const next: EthagentConfig = {
    ...config,
    identity: {
      ...config.identity,
      backup,
    },
  }
  await saveConfig(next)
  return next
}

export async function clearIdentity(config: EthagentConfig): Promise<EthagentConfig> {
  await rmSecret(IDENTITY_ACCOUNT)
  if (!config.identity) return config
  const { identity: _drop, ...rest } = config
  void _drop
  const next = { ...rest } as EthagentConfig
  await saveConfig(next)
  return next
}