useSmartRetry

"use client";
 
import { useState } from "react";
import {
  address,
  createSolanaRpc,
} from "@solana/kit";
import { useSmartRetry } from "@/hooks/uxdotsol/use-smart-retry";
 
export function SolBalanceRefresh({
  endpoint,
  owner,
}: {
  endpoint: string;
  owner: string;
}) {
  const rpc = createSolanaRpc(endpoint);
  const [balance, setBalance] = useState<bigint | null>(null);
  const retry = useSmartRetry<bigint>({ client: rpc, maxAttempts: 4 });
 
  async function refreshBalance() {
    setBalance(null);
    try {
      const lamports = await retry.execute(async () => {
        const { value } = await rpc
          .getBalance(address(owner), { commitment: "confirmed" })
          .send();
        return value;
      });
      setBalance(lamports);
    } catch {
      // The hook exposes the final error for the UI.
    }
  }
 
  return (
    <section>
      <p>
        SOL balance:{" "}
        {balance === null
          ? "Not loaded"
          : (Number(balance) / 1_000_000_000).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>
  );
}