class PrivacyBoostError extends Error {
  code: string;
  message: string;
  retryable: boolean;
}

import { useVault } from '@sunnyside-io/privacy-boost-react';

function ShieldButton() {
  const { shield } = useVault();
  const [error, setError] = useState<string | null>(null);

  const handleShield = async () => {
    setError(null);
    try {
      await shield({ tokenAddress, amount });
    } catch (err: any) {
      if (err.code === 'TRANSACTION_REJECTED') return;
      setError(err.message);
    }
  };

  return (
    <div>
      <button onClick={handleShield}>Deposit</button>
      {error && <p style={{ color: 'red' }}>{error}</p>}
    </div>
  );
}

async function withRetry<T>(
  operation: () => Promise<T>,
  maxRetries = 3
): Promise<T> {
  for (let attempt = 1; attempt <= maxRetries; attempt++) {
    try {
      return await operation();
    } catch (error: any) {
      if (!error.retryable || attempt === maxRetries) throw error;
      await new Promise((r) => setTimeout(r, 1000 * attempt));
    }
  }
  throw new Error('Unreachable');
}

function userMessage(code: string): string {
  const messages: Record<string, string> = {
    NOT_AUTHENTICATED: 'Please connect your wallet first',
    INSUFFICIENT_BALANCE: 'Not enough balance',
    TRANSACTION_REJECTED: 'Transaction cancelled',
    INVALID_AMOUNT: 'Invalid amount',
    INVALID_RECIPIENT: 'Invalid recipient address',
    NETWORK_ERROR: 'Network error. Please try again.',
    RATE_LIMITED: 'Too many requests. Please wait.',
  };
  return messages[code] || 'Something went wrong. Please try again.';
}

try {
  await shield({ tokenAddress, amount });
} catch (err: any) {
  if (err.code === 'TRANSACTION_REJECTED') return;
  setError(userMessage(err.code));
}

catch (err: any) {
  console.error('Operation failed:', err.code, err.message);
  setError(userMessage(err.code));
}