@berachain/berajs 0.2.8-beta.9 → 0.2.9

package/src/hooks/dex/usePriceImpact.ts

@@ -18,21 +18,28 @@ export function usePriceImpact({
   const { getTokenPrice, isLoading } = useTokenPrices({
     tokens: tokenIn && tokenOut ? [tokenIn, tokenOut] : undefined,
   });
+  const tokenInPrice = getTokenPrice(tokenIn);
+  const tokenOutPrice = getTokenPrice(tokenOut);
+
   const priceImpact = useMemo(() => {
     if (!tokenIn || !tokenOut || !formattedAmountIn || !formattedAmountOut)
       return undefined;
 
-    const tokenInPrice = getTokenPrice(tokenIn);
-    const tokenOutPrice = getTokenPrice(tokenOut);
-
-    if (!tokenInPrice || !tokenOutPrice) return null;
+    if (tokenInPrice == null || tokenOutPrice == null) return null;
     return computePriceImpact(
       tokenInPrice.toString(),
       tokenOutPrice.toString(),
       formattedAmountIn,
       formattedAmountOut,
     );
-  }, [tokenIn, tokenOut, formattedAmountIn, formattedAmountOut]);
+  }, [
+    tokenIn,
+    tokenOut,
+    formattedAmountIn,
+    formattedAmountOut,
+    tokenInPrice,
+    tokenOutPrice,
+  ]);
   return {
     data: priceImpact,
     isLoading,

package/src/hooks/enso/useBendDemultiply.ts

@@ -14,6 +14,7 @@ import {
 
 import { getEnsoClient } from "../../actions/clients/getEnsoClient";
 import { getBendDemultiplyBundle } from "../../actions/enso/getBendDemultiplyBundle";
+import { useBeraFlag } from "../../contexts/BeraFlags/BeraFlags";
 import { POLLING } from "../../enum/polling";
 import { assertPublicClient } from "../../errors/assert";
 import type { Token } from "../../types/dex";
@@ -63,6 +64,9 @@ export function useBendDemultiply(
     primaryAddress,
     ensoWalletV2Address,
   });
+  const proxyAggregatorsThroughVercel = useBeraFlag(
+    "proxyAggregatorsThroughVercel",
+  );
 
   const QUERY_KEY =
     collateralToken &&
@@ -96,6 +100,7 @@ export function useBendDemultiply(
           currentDebt,
           ensoWalletV2Address,
           isAuthorized,
+          proxyAggregatorsThroughVercel ?? false,
         ] as const)
       : null;
 
@@ -116,6 +121,7 @@ export function useBendDemultiply(
       currentDebt,
       ensoWalletV2Address,
       isAuthorized,
+      proxyAggregatorsThroughVercel,
     ]) => {
       assertPublicClient(publicClient);
 
@@ -140,7 +146,9 @@ export function useBendDemultiply(
         marketId,
         primaryAddress,
         chainId,
-        ensoClient: getEnsoClient(),
+        ensoClient: getEnsoClient({
+          baseURL: proxyAggregatorsThroughVercel ? "/api/enso" : undefined,
+        }),
         slippage,
         publicClient,
         collateralPriceInLoanAsset,

package/src/hooks/enso/useBendMultiply.ts

@@ -14,6 +14,7 @@ import {
 
 import { getEnsoClient } from "../../actions/clients/getEnsoClient";
 import { getBendMultiplyBundle } from "../../actions/enso/getBendMultiplyBundle";
+import { useBeraFlag } from "../../contexts/BeraFlags/BeraFlags";
 import { POLLING } from "../../enum/polling";
 import { assertPublicClient } from "../../errors/assert";
 import type { Token } from "../../types/dex";
@@ -65,6 +66,9 @@ export function useBendMultiply(
     primaryAddress,
     ensoWalletV2Address,
   });
+  const proxyAggregatorsThroughVercel = useBeraFlag(
+    "proxyAggregatorsThroughVercel",
+  );
 
   const QUERY_KEY =
     collateralToken &&
@@ -99,6 +103,7 @@ export function useBendMultiply(
           formattedCollateralPriceInLoanAsset,
           ensoWalletV2Address,
           isAuthorized,
+          proxyAggregatorsThroughVercel ?? false,
         ] as const)
       : null;
 
@@ -120,6 +125,7 @@ export function useBendMultiply(
       formattedCollateralPriceInLoanAsset,
       ensoWalletV2Address,
       isAuthorized,
+      proxyAggregatorsThroughVercel,
     ]) => {
       assertPublicClient(publicClient);
 
@@ -146,7 +152,9 @@ export function useBendMultiply(
         marketId,
         primaryAddress,
         chainId,
-        ensoClient: getEnsoClient(),
+        ensoClient: getEnsoClient({
+          baseURL: proxyAggregatorsThroughVercel ? "/api/enso" : undefined,
+        }),
         slippage,
         leverageMultiplier,
         publicClient,

package/src/hooks/enso/useBendZapSupply.ts

@@ -8,6 +8,7 @@ import { defaultChainId as berachainChainId } from "@berachain/config/internal";
 import { useBeraWallet, usePublicClient } from "@berachain/wagmi/hooks";
 
 import { ensoErc4626 } from "../../actions/dex/aggregators/enso/ensoErc4626";
+import { useBeraFlag } from "../../contexts/BeraFlags/BeraFlags";
 import { POLLING } from "../../enum/polling";
 import {
   assertAddress,
@@ -86,17 +87,24 @@ export function useBendZapSupply(
     options,
   );
 
-  const { data: zapSupplyAmount } = useGetConvertToAssets({
+  const {
+    data: zapSupplyAmount,
+    isLoading: isZapSupplyAmountLoading,
+    error: zapSupplyAmountError,
+  } = useGetConvertToAssets({
     sharesAmount: BigInt(swrResponse.data?.amountOut ?? 0),
     vaultAddress: bendVault.address,
+    vaultChainId: bendVault.chainId,
   });
 
-  const { data: priceImpact } = usePriceImpact({
-    tokenIn: tokenIn,
-    tokenOut: loanToken,
-    formattedAmountIn: amount,
-    formattedAmountOut: zapSupplyAmount?.formatted ?? "0",
-  });
+  const { data: priceImpact, isLoading: isPriceImpactLoading } = usePriceImpact(
+    {
+      tokenIn: tokenIn,
+      tokenOut: loanToken,
+      formattedAmountIn: amount,
+      formattedAmountOut: zapSupplyAmount?.formatted ?? "0",
+    },
+  );
 
   const isCrossChain = tokenIn?.chainId && tokenIn.chainId !== berachainChainId;
 
@@ -135,7 +143,10 @@ export function useBendZapSupply(
   return {
     ...swrResponse,
     zapSupplyAmount,
+    zapSupplyAmountError,
+    isZapSupplyAmountLoading,
     priceImpact,
+    isPriceImpactLoading,
     hasInsufficientNativeBalance,
     isCrossChain,
     nativeFee,
@@ -157,8 +168,12 @@ export function useBendZapSupplyHelper(
   options?: DefaultHookOptions,
 ) {
   const { address: fromAddress } = useBeraWallet();
+  const proxyAggregatorsThroughVercel = useBeraFlag(
+    "proxyAggregatorsThroughVercel",
+  );
 
   const isEnabled =
+    (options?.opts?.isEnabled ?? true) &&
     !!tokenIn &&
     !!loanToken &&
     !!bendVault &&
@@ -173,12 +188,14 @@ export function useBendZapSupplyHelper(
   const QUERY_KEY = isEnabled
     ? ([
         "useBendZapSupply",
+        fromAddress,
         tokenIn,
         loanToken,
         bendVault,
         rewardVault,
         amount,
         slippage,
+        proxyAggregatorsThroughVercel ?? false,
       ] as const)
     : null;
 
@@ -186,6 +203,7 @@ export function useBendZapSupplyHelper(
     QUERY_KEY,
     async ([
       ,
+      fromAddress,
       tokenIn,
       loanToken,
       bendVault,
@@ -206,6 +224,7 @@ export function useBendZapSupplyHelper(
         slippage,
         account: fromAddress,
         rewardVault,
+        proxyAggregatorsThroughVercel,
       });
     },
     {

package/src/hooks/enso/useEnsoSwapBundle.ts

@@ -10,6 +10,7 @@ import { useBeraWallet, useConfig } from "@berachain/wagmi/hooks";
 
 import { getEnsoClient } from "../../actions/clients/getEnsoClient";
 import { getEnsoSwapBundle } from "../../actions/enso/getEnsoSwapBundle";
+import { useBeraFlag } from "../../contexts/BeraFlags/BeraFlags";
 import { POLLING } from "../../enum/polling";
 import {
   assertAddress,
@@ -45,7 +46,12 @@ export function useEnsoSwapBundle({
 }) {
   const { address: fromAddress } = useBeraWallet();
   const wagmiConfig = useConfig();
-  const ensoClient = getEnsoClient();
+  const proxyAggregatorsThroughVercel = useBeraFlag(
+    "proxyAggregatorsThroughVercel",
+  );
+  const ensoClient = getEnsoClient({
+    baseURL: proxyAggregatorsThroughVercel ? "/api/enso" : undefined,
+  });
 
   const { refresh: refreshBalances } = usePollWalletBalances();
 
@@ -59,12 +65,20 @@ export function useEnsoSwapBundle({
     !!tokenIn.chainId;
 
   const QUERY_KEY = isEnabled
-    ? (["useEnsoSwapBundle", tokenIn, tokenOut, amount, slippage] as const)
+    ? ([
+        "useEnsoSwapBundle",
+        fromAddress,
+        tokenIn,
+        tokenOut,
+        amount,
+        slippage,
+        proxyAggregatorsThroughVercel ?? false,
+      ] as const)
     : null;
 
   const swrResponse = useSWR(
     QUERY_KEY,
-    async ([, tokenIn, tokenOut, amount, slippage]) => {
+    async ([, fromAddress, tokenIn, tokenOut, amount, slippage]) => {
       assertAddress(fromAddress, "fromAddress");
       assertAddress(tokenIn?.address, "tokenInAddress");
       assertDefined(tokenIn?.chainId, "tokenInChainId");

package/src/hooks/enso/useEnsoUserTokensWithBalances.ts

@@ -4,6 +4,7 @@ import { useBeraWallet, useConfig } from "@berachain/wagmi/hooks";
 
 import { getEnsoClient } from "../../actions/clients/getEnsoClient";
 import { getEnsoUserTokensWithBalances } from "../../actions/enso/getEnsoUserTokensWithBalances";
+import { useBeraFlag } from "../../contexts/BeraFlags/BeraFlags";
 import { POLLING } from "../../enum/polling";
 import type { BalanceToken } from "../../types/dex";
 import type { DefaultHookOptions } from "../../types/global";
@@ -14,11 +15,19 @@ export function useEnsoUserTokensWithBalances(
 ) {
   const { address: fromAddress } = useBeraWallet();
   const wagmiConfig = useConfig();
+  const proxyAggregatorsThroughVercel = useBeraFlag(
+    "proxyAggregatorsThroughVercel",
+  );
   const isEnabled = !!fromAddress && (options?.opts?.isEnabled ?? true);
 
   const QUERY_KEY =
     isEnabled && wagmiConfig
-      ? (["useEnsoUserTokensWithBalances", fromAddress, chainId] as const)
+      ? ([
+          "useEnsoUserTokensWithBalances",
+          fromAddress,
+          chainId,
+          proxyAggregatorsThroughVercel ?? false,
+        ] as const)
       : null;
 
   const swrResponse = useSWR(
@@ -28,7 +37,9 @@ export function useEnsoUserTokensWithBalances(
         account: fromAddress,
         wagmiConfig,
         chainId: chainId ?? "all",
-        ensoClient: getEnsoClient(),
+        ensoClient: getEnsoClient({
+          baseURL: proxyAggregatorsThroughVercel ? "/api/enso" : undefined,
+        }),
       }),
     {
       refreshInterval: POLLING.NORMAL,

package/src/hooks/exports.ts

@@ -13,7 +13,6 @@ export * from "./dex/useAllUserPools";
 export * from "./dex/useApiPool";
 export * from "./dex/useBexStatus";
 export * from "./dex/useCreatePool";
-export * from "./dex/useExploitedTokens";
 export { useGlobalLiquidityAndSwapVolume } from "./dex/useGlobalLiquidityAndSwapVolume";
 export * from "./dex/useLiquidityMismatch";
 export * from "./dex/useLpPosition";
@@ -36,20 +35,6 @@ export * from "./enso/useBendZapSupply";
 export * from "./enso/useEnsoSwapBundle";
 export * from "./enso/useEnsoUserTokensWithBalances";
 export * from "./enso/useZapStakeBera";
-export { useCreateProposal } from "./governance/useCreateProposal";
-export * from "./governance/useGetPastVotes";
-export * from "./governance/useHasVoted";
-export * from "./governance/useIsCanceller";
-export * from "./governance/usePollAllProposals";
-export * from "./governance/usePollProposal";
-export * from "./governance/usePollProposalThreshold";
-export * from "./governance/usePollProposalVotes";
-export * from "./governance/usePollUserDelegates";
-export * from "./governance/useProposalFromTx";
-export * from "./governance/useProposalSnapshot";
-export * from "./governance/useProposalState";
-export * from "./governance/useProposalTimelockState";
-export * from "./governance/useQuorum";
 export * from "./honey/useCappedGlobally";
 export * from "./honey/useCappedRelatively";
 export * from "./honey/useCollateralWeights";
@@ -63,6 +48,7 @@ export * from "./honey/useIsBadCollateralAsset";
 export * from "./honey/useIsBasketModeEnabled";
 export * from "./honey/usePythLatestPrices";
 export * from "./perps/usePythUpdateFee";
+export { useAutoclaimedIncentives } from "./pol/useAutoclaimedIncentives";
 export * from "./pol/useBgtAprSimulation";
 export { useBgtUnstakedBalance } from "./pol/useBgtUnstakedBalance";
 export { useClaimableFees } from "./pol/useClaimableFees";
@@ -126,6 +112,7 @@ export * from "./validators/useValidator";
 export { useValidatorAnalytics } from "./validators/useValidatorAnalytics";
 export { useValidatorCommission } from "./validators/useValidatorCommission";
 export { useValidatorEstimatedBgtPerYear } from "./validators/useValidatorEstimatedBgtPerYear";
+export { useValidatorIncentiveDistribution } from "./validators/useValidatorIncentiveDistribution";
 export { useValidatorQueuedCommission } from "./validators/useValidatorQueuedCommission";
 export { useValidatorQueuedOperatorAddress } from "./validators/useValidatorQueuedOperatorAddress";
 export { useValidatorQueuedRewardAllocation } from "./validators/useValidatorQueuedRewardAllocation";

package/src/hooks/governance/exports.ts

@@ -0,0 +1,14 @@
+export { useCreateProposal } from "./useCreateProposal";
+export * from "./useGetPastVotes";
+export * from "./useHasVoted";
+export * from "./useIsCanceller";
+export * from "./usePollAllProposals";
+export * from "./usePollProposal";
+export * from "./usePollProposalThreshold";
+export * from "./usePollProposalVotes";
+export * from "./usePollUserDelegates";
+export * from "./useProposalFromTx";
+export * from "./useProposalSnapshot";
+export * from "./useProposalState";
+export * from "./useProposalTimelockState";
+export * from "./useQuorum";

package/src/hooks/pol/useAutoclaimedIncentives.ts

@@ -0,0 +1,49 @@
+import useSWR from "swr";
+
+import { useBeraWallet } from "@berachain/wagmi/hooks";
+
+import { getAutoclaimedIncentives } from "../../actions/pol/getAutoclaimedIncentives";
+import { BeraError } from "../../errors/BeraError";
+import type { AutoclaimedIncentivesResponse } from "../../types/bribe-boost";
+import type { DefaultHookReturnType } from "../../types/global";
+
+interface UseAutoclaimedIncentivesArgs {
+  // Hard gate: when false the SWR key is null, so the API is never called.
+  // The banner consumer flips this to true only after both `!isBgtPolRewardToken`
+  // and `bgtIncentiveDistributor.paused() === true` have been verified.
+  enabled: boolean;
+}
+
+export function useAutoclaimedIncentives({
+  enabled,
+}: UseAutoclaimedIncentivesArgs): DefaultHookReturnType<AutoclaimedIncentivesResponse | null> {
+  const { address: account } = useBeraWallet();
+
+  const QUERY_KEY =
+    enabled && account
+      ? (["useAutoclaimedIncentives", account] as const)
+      : null;
+
+  const swrResponse = useSWR(
+    QUERY_KEY,
+    async ([, account]) => {
+      if (!account) {
+        throw new BeraError({
+          message: "useAutoclaimedIncentives needs a logged in account",
+          level: "error",
+        });
+      }
+      return await getAutoclaimedIncentives({ account });
+    },
+    {
+      revalidateOnFocus: false,
+      revalidateOnReconnect: false,
+      revalidateIfStale: false,
+    },
+  );
+
+  return {
+    ...swrResponse,
+    refresh: () => swrResponse?.mutate?.(),
+  };
+}

package/src/hooks/pol/useStakedAPR.ts

@@ -7,7 +7,7 @@ import type { DefaultHookReturnType } from "../../types/global";
 import { dailyResolution } from "../../utils/filter";
 
 type UseStakedAPR = {
-  _timestamp: number;
+  timestamp: number;
   apr: string;
 }[];
 export function useStakedAPR({
@@ -26,7 +26,7 @@ export function useStakedAPR({
     );
 
     return sortedStakedBeraSnapshots.map((item) => ({
-      _timestamp: item.timestamp,
+      timestamp: item.timestamp,
       apr: item.apr,
     }));
   });

package/src/hooks/pol/useStakedSnapshots.ts

@@ -1,17 +1,13 @@
 import useSWR from "swr";
 import type { Address } from "viem";
 
+import type { VaultStatsByDayResponse } from "@berachain/graphql/api";
+
 import { getStakingDailyAssets } from "../../actions/pol/getStakingDailyAssets";
 import type { DefaultHookReturnType } from "../../types/global";
 import { dailyResolution } from "../../utils/filter";
 
-type UseStakedBeraSnapshot = Record<
-  Address,
-  {
-    _timestamp: number;
-    total_assets: string;
-  }[]
->;
+type UseStakedBeraSnapshot = Record<Address, VaultStatsByDayResponse>;
 export function useStakedSnapshots({
   range,
   addresses,
@@ -30,7 +26,7 @@ export function useStakedSnapshots({
           range,
         });
         const sortedStakingDailyAssets = dailyResolution(
-          unstakingDailyAssets.sort((a, b) => a._timestamp - b._timestamp),
+          unstakingDailyAssets.sort((a, b) => a.timestamp - b.timestamp),
         );
         return {
           [address]: sortedStakingDailyAssets,
@@ -41,7 +37,7 @@ export function useStakedSnapshots({
     // Merge all address data into a single object
     return sortedStakingDailyAssetsArray.reduce(
       (acc, addressData) => ({ ...acc, ...addressData }),
-      {} as Record<Address, { _timestamp: number; total_assets: string }[]>,
+      {} as UseStakedBeraSnapshot,
     );
   });
 

package/src/hooks/validators/useValidator.ts

@@ -56,6 +56,8 @@ export function useValidator(
                 lastDayDistributedBGTAmount:
                   indexerValidator?.dynamicData?.lastDayDistributedBGTAmount ??
                   "",
+                lastDayProposedBlockCount:
+                  indexerValidator?.dynamicData?.lastDayProposedBlockCount ?? 0,
                 activeBoostAmountRank:
                   indexerValidator?.dynamicData?.activeBoostAmountRank ?? 0,
 

package/src/hooks/validators/useValidatorIncentiveDistribution.ts

@@ -0,0 +1,39 @@
+import useSWR from "swr";
+import type { Address } from "viem";
+
+import type { GetValidatorIncentiveDistributionQuery } from "@berachain/graphql/pol/subgraph";
+
+import { getValidatorIncentiveDistribution } from "../../actions/validators/getValidatorIncentiveDistribution";
+import { POLLING } from "../../enum/polling";
+import type { DefaultHookReturnType } from "../../types/global";
+
+/**
+ * Historical per-validator incentive distribution. Split from
+ * `useValidatorAnalytics` so the IncentivesEarned chart can switch data
+ * sources independently of the other charts that still consume the umbrella
+ * `GetValidatorAnalytics` query.
+ */
+export function useValidatorIncentiveDistribution({
+  pubkey,
+  dayRange,
+}: {
+  pubkey: Address | undefined;
+  dayRange: number;
+}): DefaultHookReturnType<GetValidatorIncentiveDistributionQuery> {
+  const QUERY_KEY =
+    pubkey && dayRange
+      ? (["useValidatorIncentiveDistribution", pubkey, dayRange] as const)
+      : null;
+
+  const swrResponse = useSWR(
+    QUERY_KEY,
+    async ([, pubkey, dayRange]) => {
+      return await getValidatorIncentiveDistribution({ pubkey, dayRange });
+    },
+    {
+      refreshInterval: POLLING.SLOW,
+    },
+  );
+
+  return { ...swrResponse, refresh: () => swrResponse.mutate() };
+}

package/src/types/bribe-boost.d.ts

@@ -32,6 +32,14 @@ export interface BribeBoostRewardProof {
     | null;
 }
 
+// Response shape of GET /api/v1/wallets/:wallet/autoclaimed:
+//   { [wallet]: { [validatorPubkey]: { [tokenAddress]: rawAmount (stringified bigint) } } }
+// Top-level wallet key is the lowercase wallet address that was queried.
+export type AutoclaimedIncentivesResponse = Record<
+  Address,
+  Record<Address, Record<Address, string>>
+>;
+
 export interface BribeBoostValidatorApr {
   id: string;
   boost_total_bgt: string;

package/src/types/exports.ts

@@ -2,7 +2,6 @@ export * from "./bribe-boost.d";
 export * from "./dex.d";
 export * from "./enso.d";
 export * from "./global.d";
-export * from "./governance.d";
 export * from "./honeySwap.d";
 export * from "./pol.d";
 export * from "./staking.d";

package/src/types/governance/exports.ts

@@ -0,0 +1 @@
+export * from "../governance.d";

package/src/utils/beraFetch.ts

@@ -1,6 +1,7 @@
 import { BeraTracing, type SpanContext } from "../errors/BeraTracing";
 import { RequestError } from "../errors/RequestError";
 import { getErrorResponse } from "./getErrorResponse";
+import { sanitizeRpcUrl } from "./sanitizeRpcUrl";
 
 type RequestOptions = RequestInit;
 
@@ -18,10 +19,10 @@ export async function beraFetch(
   },
 ) {
   try {
-    let spanUrl = (
+    let spanUrl = sanitizeRpcUrl(
       typeof endpoint === "string" || endpoint instanceof URL
         ? endpoint.toString()
-        : endpoint.url.toString()
+        : endpoint.url.toString(),
     ).replace(/0x[a-fA-F0-9]{40}/g, ":address");
 
     if (spanContext?.keepQueryParams !== true && URL.canParse(spanUrl)) {
@@ -43,10 +44,15 @@ export async function beraFetch(
       }
     }
 
+    const endpointName =
+      typeof endpoint === "string" || endpoint instanceof URL
+        ? undefined
+        : endpoint.name;
+
     return await BeraTracing.startSpan(
       {
         op: "beraFetch",
-        name: "beraFetch",
+        name: endpointName ? `beraFetch ${endpointName}` : "beraFetch",
         attributes:
           typeof endpoint === "string" || endpoint instanceof URL
             ? {

package/src/utils/exports.ts

@@ -24,5 +24,6 @@ export {
   wBeraToken as internal_wBeraToken,
 } from "./tokens";
 export * from "./tokenWrapping";
+export { tracedTransport } from "./tracedTransport";
 export * from "./truncateHash";
 export * from "./validatorUptimeStatus";

package/src/utils/getServerSideClient.ts

@@ -7,6 +7,8 @@ import {
   getRpcUrls,
 } from "@berachain/config/internal";
 
+import { tracedTransport } from "./tracedTransport";
+
 /**
  * This is a singleton to avoid creating a new client for each request.
  *
@@ -31,10 +33,10 @@ export const getServerSidePublicClient = (
     rpcs.publicJsonRpcUrl,
   ]
     .filter((i) => i)
-    .map((url) => http(url));
+    .map((url) => tracedTransport(http(url)));
 
   if (fallbackTransports.length === 0) {
-    fallbackTransports.push(http());
+    fallbackTransports.push(tracedTransport(http()));
   }
   const chain = chainId === ChainId.MAINNET ? berachain : berachainBepolia;
   const client = createPublicClient({

package/src/utils/sanitizeRpcUrl.ts

@@ -0,0 +1,22 @@
+/**
+ * Strips secret tokens from RPC URLs for safe use in span attributes and logs.
+ *
+ * QuickNode URLs embed API keys as path segments, e.g.:
+ *   https://polished-morning-sun.bera-mainnet.quiknode.pro/abc123def/
+ * This function trims the subpath, producing:
+ *   https://polished-morning-sun.bera-mainnet.quiknode.pro/
+ */
+export function sanitizeRpcUrl(url: string | undefined): string {
+  if (!url) return "";
+  try {
+    const parsed = new URL(url);
+    if (parsed.hostname.endsWith(".quiknode.pro")) {
+      parsed.pathname = "/";
+      parsed.search = "";
+      return parsed.toString();
+    }
+    return url;
+  } catch {
+    return url;
+  }
+}

package/src/utils/tracedTransport.ts

@@ -0,0 +1,35 @@
+import type { Transport } from "viem";
+
+import { BeraTracing } from "../errors/BeraTracing";
+import { sanitizeRpcUrl } from "./sanitizeRpcUrl";
+
+/**
+ * Wraps a viem Transport to trace each RPC request via BeraTracing.
+ *
+ * Each `request()` call is wrapped in a span with the JSON-RPC method name,
+ * transport type, and URL as attributes. Works with `http()`, `fallback()`,
+ * and any other viem transport.
+ */
+export function tracedTransport<T extends Transport>(transport: T): T {
+  return ((opts: Parameters<Transport>[0]) => {
+    const result = transport(opts);
+    const originalRequest = result.request;
+
+    result.request = async (args, options) => {
+      return BeraTracing.startSpan(
+        {
+          name: `RPC ${args.method}`,
+          op: "rpc.request",
+          attributes: {
+            "rpc.method": args.method,
+            "rpc.transport": result.config.type,
+            "rpc.url": sanitizeRpcUrl(result.value?.url),
+          },
+        },
+        () => originalRequest(args, options),
+      );
+    };
+
+    return result;
+  }) as T;
+}