import type { Wallet } from '@/model/wallet'

export function getWallets(wallet?: Wallet | Wallet[]): Wallet[] {
  if (!wallet) return []
  return Array.isArray(wallet) ? wallet : [wallet]
}

export function getCashBalance(wallet?: Wallet | Wallet[]): number {
  return getWallets(wallet)[0]?.cash ?? 0
}

export function getGoldBalance(wallet?: Wallet | Wallet[]): string {
  const sum = getWallets(wallet).reduce((total, item) => total + (item.gold ?? 0), 0)
  return sum.toFixed(3)
}