useSmartRetry

$npx shadcn@latest add https://uxdotsol.xyz/r/use-smart-retry.json

"use client";
 
import { useState } from "react";
import {
  LAMPORTS_PER_SOL,
  type Connection,
  type PublicKey,
} from "@solana/web3.js";
import { useSmartRetry } from "@/hooks/use-smart-retry";
 
export function SolBalanceRefresh({
  connection,
  owner,
}: {
  connection: Connection;
  owner: PublicKey;
}) {
  const [balance, setBalance] = useState<number | null>(null);
  const retry = useSmartRetry<number>({ client: connection, maxAttempts: 4 });
 
  async function refreshBalance() {
    setBalance(null);
    try {
      const lamports = await retry.execute(() =>
        connection.getBalance(owner, "confirmed"),
      );
      setBalance(lamports / LAMPORTS_PER_SOL);
    } catch {
      // The hook exposes the final error for the UI.
    }
  }
 
  return (
    <section>
      <p>SOL balance: {balance === null ? "Not loaded" : balance.toFixed(4)}</p>
      <button disabled={retry.isRetrying} onClick={refreshBalance}>
        {retry.isRetrying
          ? `RPC busy, retrying (${retry.attempt}/4)...`
          : "Refresh balance"}
      </button>
      {!retry.isRetrying && retry.error && balance === null ? (
        <p role="alert">Could not reach the Solana RPC.</p>
      ) : null}
    </section>
  );
}