fix: allow failed address checker users to perform unbond

Author: jrwbabylonlab

src/ui/baby/state/StakingState.tsx

@@ -8,8 +8,10 @@ import { useWalletService } from "@/ui/baby/hooks/services/useWalletService";
 import { validateDecimalPoints } from "@/ui/common/components/Staking/Form/validation/validation";
 import { useError } from "@/ui/common/context/Error/ErrorProvider";
 import { usePrice } from "@/ui/common/hooks/client/api/usePrices";
+import { useHealthCheck } from "@/ui/common/hooks/useHealthCheck";
 import { useLogger } from "@/ui/common/hooks/useLogger";
 import { MultistakingFormFields } from "@/ui/common/state/MultistakingState";
+import { GEO_BLOCK_MESSAGE } from "@/ui/common/types/services/healthCheck";
 import {
   createBalanceValidator,
   createMinAmountValidator,
@@ -64,6 +66,10 @@ interface StakingState {
   submitForm(): Promise<void>;
   resetForm(): void;
   calculateFee: (params: Omit<FormData, "feeAmount">) => Promise<number>;
+  disabled?: {
+    title: string;
+    message: string;
+  };
 }
 
 const { StateProvider, useState: useStakingState } =
@@ -79,12 +85,14 @@ const { StateProvider, useState: useStakingState } =
     closePreview: () => {},
     submitForm: async () => {},
     resetForm: () => {},
+    disabled: undefined,
   });
 
 function StakingState({ children }: PropsWithChildren) {
   const [step, setStep] = useState<StakingStep>({ name: "initial" });
 
   const { stake, sendTx, estimateStakingFee } = useDelegationService();
+  const { isGeoBlocked } = useHealthCheck();
   const { validatorMap, loading } = useValidatorService();
   const { balance } = useWalletService();
   const { handleError } = useError();
@@ -104,6 +112,15 @@ function StakingState({ children }: PropsWithChildren) {
     [availableBalance],
   );
 
+  const isDisabled = useMemo(() => {
+    if (isGeoBlocked) {
+      return {
+        title: "Unavailable In Your Region",
+        message: GEO_BLOCK_MESSAGE,
+      };
+    }
+  }, [isGeoBlocked]);
+
   const fieldSchemas = useMemo(
     () =>
       [
@@ -253,6 +270,7 @@ function StakingState({ children }: PropsWithChildren) {
       submitForm,
       resetForm,
       closePreview,
+      disabled: isDisabled,
     };
   }, [
     availableBalance,
@@ -266,6 +284,7 @@ function StakingState({ children }: PropsWithChildren) {
     submitForm,
     resetForm,
     closePreview,
+    isDisabled,
   ]);
 
   return <StateProvider value={context}>{children}</StateProvider>;

src/ui/baby/widgets/StakingForm/index.tsx

@@ -28,6 +28,7 @@ export default function StakingForm({
     babyPrice,
     calculateFee,
     showPreview,
+    disabled,
   } = useStakingState();
 
   const handlePreview = ({
@@ -50,7 +51,7 @@ export default function StakingForm({
 
       <SubmitButton disabled={loading} isGeoBlocked={isGeoBlocked} />
       <StakingModal />
-      <FormAlert isGeoBlocked={isGeoBlocked} />
+      <FormAlert {...disabled} />
     </Form>
   );
 }

src/ui/common/components/Multistaking/MultistakingForm/FormAlert.tsx

@@ -1,41 +1,22 @@
 import { MdErrorOutline } from "react-icons/md";
 
 import { Alert } from "@/ui/common/components/Alerts/Alert";
-import { STAKING_DISABLED } from "@/ui/common/constants";
-import { GEO_BLOCK_MESSAGE } from "@/ui/common/types/services/healthCheck";
 
-interface FormAlertProps {
-  address?: string;
-  isGeoBlocked: boolean;
-}
-
-export const FormAlert = ({ address, isGeoBlocked }: FormAlertProps) => {
-  const stakingDisabledMessage = (
-    <>
-      The Babylon network is under maintenance. New stakes are paused until the
-      network resumes.
-    </>
-  );
-
-  const shouldShowAlert = (address && STAKING_DISABLED) || isGeoBlocked;
-
-  if (!shouldShowAlert) {
+export const FormAlert = ({
+  title,
+  message,
+}: {
+  title?: string;
+  message?: string;
+}) => {
+  if (!title) {
     return null;
   }
 
   return (
     <div className="pt-2">
-      <Alert
-        icon={<MdErrorOutline />}
-        title={
-          isGeoBlocked ? (
-            <strong>Unavailable In Your Region.</strong>
-          ) : (
-            <strong>Staking Currently Unavailable.</strong>
-          )
-        }
-      >
-        {isGeoBlocked ? GEO_BLOCK_MESSAGE : stakingDisabledMessage}
+      <Alert icon={<MdErrorOutline />} title={<strong>{title}</strong>}>
+        {message}
       </Alert>
     </div>
   );

src/ui/common/components/Multistaking/MultistakingForm/MultistakingFormContent.tsx

@@ -2,7 +2,6 @@ import { HiddenField } from "@babylonlabs-io/core-ui";
 
 import { AuthGuard } from "@/ui/common/components/Common/AuthGuard";
 import { MultistakingModal } from "@/ui/common/components/Multistaking/MultistakingModal/MultistakingModal";
-import { useBTCWallet } from "@/ui/common/context/wallet/BTCWalletProvider";
 import { useStakingState } from "@/ui/common/state/StakingState";
 
 import { AmountSection } from "./AmountSection";
@@ -13,8 +12,7 @@ import { StakingFeesSection } from "./StakingFeesSection";
 import { SubmitButton } from "./SubmitButton";
 
 export function MultistakingFormContent() {
-  const { address } = useBTCWallet();
-  const { stakingInfo, blocked: isGeoBlocked } = useStakingState();
+  const { stakingInfo, blocked: isGeoBlocked, disabled } = useStakingState();
 
   return (
     <>
@@ -37,7 +35,7 @@ export function MultistakingFormContent() {
           <SubmitButton />
         </AuthGuard>
 
-        <FormAlert address={address} isGeoBlocked={isGeoBlocked} />
+        <FormAlert {...disabled} />
       </div>
 
       <MultistakingModal />

src/ui/common/components/Multistaking/MultistakingForm/SubmitButton.tsx

@@ -12,7 +12,7 @@ const BUTTON_STYLES: Record<string, string> = {
 
 export function SubmitButton() {
   const { isValid, isValidating, isLoading } = useFormState();
-  const { blocked: isGeoBlocked } = useStakingState();
+  const { blocked: isGeoBlocked, disabled: isDisabled } = useStakingState();
   const error = useFormError();
 
   const renderText = () => {
@@ -44,7 +44,8 @@ export function SubmitButton() {
         isValidating ||
         isLoading ||
         STAKING_DISABLED ||
-        isGeoBlocked
+        isGeoBlocked ||
+        isDisabled !== undefined
       }
     >
       {renderText()}

src/ui/common/context/wallet/BTCWalletProvider.tsx

@@ -35,14 +35,19 @@ import {
   toNetwork,
 } from "@/ui/common/utils/wallet";
 
+import { useAddressScreeningService } from "../../hooks/services/useAddressScreeningService";
+
 const btcConfig = getNetworkConfigBTC();
+const supportedNetworkMessage =
+  "Only Native SegWit and Taproot addresses are supported. Please switch the address type in your wallet and try again.";
 
 interface BTCWalletContextProps {
   loading: boolean;
   network?: networks.Network;
   publicKeyNoCoord: string;
   address: string;
   connected: boolean;
+  failedBtcAddressRiskAssessment: boolean;
   disconnect: () => void;
   open: () => void;
   getAddress: () => Promise<string>;
@@ -70,6 +75,7 @@ const BTCWalletContext = createContext<BTCWalletContextProps>({
   connected: false,
   publicKeyNoCoord: "",
   address: "",
+  failedBtcAddressRiskAssessment: false,
   disconnect: () => {},
   open: () => {},
   getAddress: async () => "",
@@ -87,6 +93,8 @@ const BTCWalletContext = createContext<BTCWalletContextProps>({
 
 export const BTCWalletProvider = ({ children }: PropsWithChildren) => {
   const [loading, setLoading] = useState(true);
+  const [failedBtcAddressRiskAssessment, setFailedBtcAddressRiskAssessment] =
+    useState(false);
   const [btcWalletProvider, setBTCWalletProvider] = useState<IBTCProvider>();
   const [network, setNetwork] = useState<networks.Network>();
   const [publicKeyNoCoord, setPublicKeyNoCoord] = useState("");
@@ -97,24 +105,25 @@ export const BTCWalletProvider = ({ children }: PropsWithChildren) => {
   const { open = () => {}, connected } = useWalletConnect();
   const logger = useLogger();
   const { updateUser } = useSentryUser();
+  const { screenAddress, clearAddressScreeningResult } =
+    useAddressScreeningService();
 
   const btcDisconnect = useCallback(() => {
     setBTCWalletProvider(undefined);
     setNetwork(undefined);
     setPublicKeyNoCoord("");
     setAddress("");
+    setFailedBtcAddressRiskAssessment(false);
 
     updateUser({ btcAddress: null });
-  }, [updateUser]);
+    clearAddressScreeningResult();
+  }, [updateUser, clearAddressScreeningResult]);
 
   const connectBTC = useCallback(
     async (walletProvider: IBTCProvider | null) => {
       if (!walletProvider) return;
       setLoading(true);
 
-      const supportedNetworkMessage =
-        "Only Native SegWit and Taproot addresses are supported. Please switch the address type in your wallet and try again.";
-
       try {
         const network = await walletProvider.getNetwork();
         if (network !== btcConfig.network) {
@@ -164,6 +173,9 @@ export const BTCWalletProvider = ({ children }: PropsWithChildren) => {
           throw emptyProcessedPubKeyError;
         }
 
+        const failedRiskAssessment = await screenAddress(address);
+        setFailedBtcAddressRiskAssessment(failedRiskAssessment);
+
         setBTCWalletProvider(walletProvider);
         setNetwork(toNetwork(network));
         setAddress(address);
@@ -192,7 +204,7 @@ export const BTCWalletProvider = ({ children }: PropsWithChildren) => {
         });
       }
     },
-    [handleError, publicKeyNoCoord, address, logger, updateUser],
+    [handleError, publicKeyNoCoord, address, logger, updateUser, screenAddress],
   );
 
   useEffect(() => {
@@ -288,13 +300,15 @@ export const BTCWalletProvider = ({ children }: PropsWithChildren) => {
       network,
       publicKeyNoCoord,
       address,
+      failedBtcAddressRiskAssessment,
       connected,
       open,
       disconnect: btcDisconnect,
       ...btcWalletMethods,
     }),
     [
       loading,
+      failedBtcAddressRiskAssessment,
       connected,
       network,
       publicKeyNoCoord,

src/ui/common/context/wallet/WalletConnectionProvider.tsx

@@ -7,7 +7,6 @@ import { useTheme } from "next-themes";
 import { useCallback, type PropsWithChildren } from "react";
 
 import { logTermsAcceptance } from "@/ui/common/api/logTermAcceptance";
-import { verifyBTCAddress } from "@/ui/common/api/verifyBTCAddress";
 import { getNetworkConfigBBN } from "@/ui/common/config/network/bbn";
 import { getNetworkConfigBTC } from "@/ui/common/config/network/btc";
 import { ClientError, ERROR_CODES } from "@/ui/common/errors";
@@ -20,7 +19,6 @@ const context = typeof window !== "undefined" ? window : {};
 
 const lifecycleHooks = {
   acceptTermsOfService: logTermsAcceptance,
-  verifyBTCAddress: verifyBTCAddress,
 };
 
 const config: ChainConfigArr = [

src/ui/common/hooks/services/useAddressScreeningService.ts

@@ -0,0 +1,52 @@
+import { useCallback } from "react";
+
+import { verifyBTCAddress } from "../../api/verifyBTCAddress";
+import {
+  clearBtcAddressScreeningResult,
+  getBtcAddressScreeningResult,
+  setBtcAddressScreeningResult,
+} from "../../utils/local_storage/addressScreeningStorage";
+import { useLogger } from "../useLogger";
+
+export function useAddressScreeningService() {
+  const logger = useLogger();
+
+  const screenAddress = useCallback(
+    async (btcAddress: string) => {
+      try {
+        // Check cache first
+        const cachedResult = getBtcAddressScreeningResult();
+        if (cachedResult && cachedResult.btcAddress === btcAddress) {
+          return cachedResult.failedRiskAssessment;
+        }
+
+        // Perform screening
+        const riskAssessment = await verifyBTCAddress(btcAddress);
+        const failedRiskAssessment = !riskAssessment;
+
+        // Cache result
+        setBtcAddressScreeningResult(btcAddress, failedRiskAssessment);
+
+        logger.info("Address screening completed", {
+          btcAddress,
+          failedRiskAssessment,
+        });
+
+        return failedRiskAssessment;
+      } catch (error: any) {
+        logger.error(error, { data: { btcAddress } });
+        return false; // Default to safe
+      }
+    },
+    [logger],
+  );
+
+  const clearAddressScreeningResult = useCallback(() => {
+    clearBtcAddressScreeningResult();
+  }, []);
+
+  return {
+    screenAddress,
+    clearAddressScreeningResult,
+  };
+}