@b3dotfun/sdk 0.0.65-test.5 → 0.0.66

package/src/global-account/react/components/LinkAccount/LinkAccount.tsx

@@ -1,69 +1,138 @@
 import app from "@b3dotfun/sdk/global-account/app";
-import { Button, ManageAccountModalProps, useB3, useModalStore, useQueryB3 } from "@b3dotfun/sdk/global-account/react";
+import { ecosystemWalletId } from "@b3dotfun/sdk/shared/constants";
+import { thirdwebB3Mainnet } from "@b3dotfun/sdk/shared/constants/chains/b3Chain";
 import { client } from "@b3dotfun/sdk/shared/utils/thirdweb";
-import { truncateAddress } from "@b3dotfun/sdk/shared/utils/truncateAddress";
-import { Copy, Loader2, Pencil, UnlinkIcon } from "lucide-react";
-import { useRef, useState } from "react";
+import { Loader2, Mail, Phone, WalletIcon } from "lucide-react";
+import { useCallback, useEffect, useState } from "react";
 import { toast } from "sonner";
-import { useProfiles, useUnlinkProfile } from "thirdweb/react";
-
-import { Chain } from "thirdweb";
+import { useLinkProfile, useProfiles } from "thirdweb/react";
+import { createWallet, preAuthenticate, WalletId } from "thirdweb/wallets";
+import { WalletRow } from "../..";
+import { LinkAccountModalProps, useModalStore } from "../../stores/useModalStore";
 import { getProfileDisplayInfo } from "../../utils/profileDisplay";
+import { useB3 } from "../B3Provider/useB3";
+import { AppleIcon } from "../icons/AppleIcon";
+import { DiscordIcon } from "../icons/DiscordIcon";
+import { FarcasterIcon } from "../icons/FarcasterIcon";
+import { GoogleIcon } from "../icons/GoogleIcon";
+import { XIcon } from "../icons/XIcon";
+import { Button } from "../ui/button";
+type OTPStrategy = "email" | "phone";
+type SocialStrategy = "google" | "x" | "discord" | "apple" | "farcaster";
+type Strategy = OTPStrategy | SocialStrategy | "wallet";
 
-// Helper function to check if a string is a wallet address and format it
-const formatProfileTitle = (title: string): { displayTitle: string; isAddress: boolean } => {
-  // Check if title looks like an Ethereum address (0x followed by 40 hex characters)
-  const isEthereumAddress = /^0x[a-fA-F0-9]{40}$/.test(title);
+interface AuthMethod {
+  id: Strategy;
+  label: string;
+  enabled: boolean;
+  icon: React.ReactNode;
+  walletType?: WalletId;
+}
 
-  if (isEthereumAddress) {
-    return {
-      displayTitle: truncateAddress(title),
-      isAddress: true,
-    };
-  }
+const AUTH_METHODS: AuthMethod[] = [
+  { id: "email", label: "Email", enabled: true, icon: <Mail className="text-b3-primary-blue size-6" /> },
+  { id: "phone", label: "Phone", enabled: true, icon: <Phone className="text-b3-primary-blue size-6" /> },
+  { id: "google", label: "Google", enabled: true, icon: <GoogleIcon className="size-6" /> },
+  { id: "x", label: "X (Twitter)", enabled: true, icon: <XIcon className="size-6" /> },
+  { id: "discord", label: "Discord", enabled: true, icon: <DiscordIcon className="size-6" /> },
+  { id: "apple", label: "Apple", enabled: true, icon: <AppleIcon className="size-6" /> },
+  {
+    id: "farcaster",
+    label: "Farcaster",
+    enabled: true,
+    icon: <FarcasterIcon className="size-6" />,
+  },
+];
 
-  return {
-    displayTitle: title,
-    isAddress: false,
-  };
-};
+const WALLET_METHODS: AuthMethod[] = [
+  {
+    id: "wallet",
+    label: "Wallet",
+    enabled: true,
+    icon: <WalletIcon className="size-6" />,
+    walletType: "com.coinbase.wallet",
+  },
+  { id: "wallet", label: "Wallet", enabled: true, icon: <WalletIcon className="size-6" />, walletType: "io.metamask" },
+  {
+    id: "wallet",
+    label: "Wallet",
+    enabled: true,
+    icon: <WalletIcon className="size-6" />,
+    walletType: "me.rainbow",
+  },
+  {
+    id: "wallet",
+    label: "Wallet",
+    enabled: true,
+    icon: <WalletIcon className="size-6" />,
+    walletType: "app.phantom",
+  },
+  { id: "wallet", label: "Wallet", enabled: true, icon: <WalletIcon className="size-6" />, walletType: "io.rabby" },
+  {
+    id: "wallet",
+    label: "Wallet",
+    enabled: true,
+    icon: <WalletIcon className="size-6" />,
+    walletType: "walletConnect",
+  },
+];
 
-import { Referrals, Users } from "@b3dotfun/b3-api";
-import ModalHeader from "../ModalHeader/ModalHeader";
-
-export const LinkAccount = ({
-  partnerId,
-  onLogout,
+export function LinkAccount({
+  onSuccess: onSuccessCallback,
+  onError,
+  onClose,
   chain,
-}: {
-  partnerId: string;
-  onLogout?: () => void;
-  chain: Chain;
-}) => {
-  const [unlinkingAccountId, setUnlinkingAccountId] = useState<string | null>(null);
-  const { data: profilesRaw = [], isLoading: isLoadingProfiles } = useProfiles({ client });
-  const { mutate: unlinkProfile, isPending: isUnlinking } = useUnlinkProfile();
-  const setB3ModalContentType = useModalStore(state => state.setB3ModalContentType);
-  const isLinking = useModalStore(state => state.isLinking);
-  const setB3ModalOpen = useModalStore(state => state.setB3ModalOpen);
-  const contentType = useModalStore(state => state.contentType);
-  const { user, setUser } = useB3();
-  const [isUpdatingCode, setIsUpdatingCode] = useState(false);
-  const [newReferralCode, setNewReferralCode] = useState("");
-  const [isEditingCode, setIsEditingCode] = useState(false);
-  const referallCodeRef = useRef<HTMLInputElement>(null);
-  const { data: referrals, isLoading: isLoadingReferrals } = useQueryB3(
-    "referrals",
-    "find",
-    { query: { referrerId: user?.userId } },
-    !!user?.userId,
+  partnerId,
+  className,
+}: LinkAccountModalProps & { className?: string }) {
+  const { isLinking, linkingMethod, setLinkingState, navigateBack, setB3ModalContentType } = useModalStore();
+  const [selectedMethod, setSelectedMethod] = useState<Strategy | null>(null);
+  const [email, setEmail] = useState("");
+  const [phone, setPhone] = useState("");
+  const [otp, setOtp] = useState("");
+  const [otpSent, setOtpSent] = useState(false);
+  const [error, setError] = useState<string | null>(null);
+  const { data: profilesRaw = [] } = useProfiles({ client });
+
+  // Get connected auth methods
+  const connectedAuthMethods = profilesRaw
+    .filter((profile: any) => !["custom_auth_endpoint"].includes(profile.type))
+    .map((profile: any) => profile.type as Strategy);
+
+  // Filter available auth methods
+  const availableAuthMethods = AUTH_METHODS.filter(
+    method => !connectedAuthMethods.includes(method.id) && method.enabled,
   );
-  const showReferralInfo = (contentType as ManageAccountModalProps)?.showReferralInfo ?? false;
+
+  const profiles = profilesRaw
+    .filter((profile: any) => !["custom_auth_endpoint"].includes(profile.type))
+    .map((profile: any) => ({
+      ...getProfileDisplayInfo(profile),
+      originalProfile: profile,
+    }));
+
+  const { account } = useB3();
+  const { mutate: linkProfile } = useLinkProfile();
+
+  const onSuccess = useCallback(async () => {
+    await onSuccessCallback?.();
+  }, [onSuccessCallback]);
+
+  // Reset linking state when component unmounts
+  useEffect(() => {
+    return () => {
+      if (isLinking) {
+        setLinkingState(false);
+      }
+    };
+  }, [isLinking, setLinkingState]);
 
   const mutationOptions = {
     onError: (error: Error) => {
-      console.error("Error Unlinking account:", error);
+      console.error("Error linking account:", error);
       toast.error(error.message);
+      setLinkingState(false);
+      onError?.(error);
     },
     onSuccess: async (data: any) => {
       console.log("Raw Link Account Data:", data);
@@ -76,316 +145,348 @@ export const LinkAccount = ({
     },
   };
 
-  // Fetch referred users
-  const currentReferralCode = user?.referralCode || user?.userId || "";
+  const validateInput = () => {
+    if (selectedMethod === "email") {
+      if (!email) {
+        setError("Please enter your email address");
+        return false;
+      }
+      if (!/^[^\s@]+@[^\s@]+\.[^\s@]+$/.test(email)) {
+        setError("Please enter a valid email address");
+        return false;
+      }
+    } else if (selectedMethod === "phone") {
+      if (!phone) {
+        setError("Please enter your phone number");
+        return false;
+      }
+      if (!/^\+?[\d\s-]{10,}$/.test(phone)) {
+        setError("Please enter a valid phone number");
+        return false;
+      }
+    }
+    setError(null);
+    return true;
+  };
+
+  const handleSendOTP = async () => {
+    if (!validateInput()) return;
 
-  const handleCopyCode = async () => {
     try {
-      await navigator.clipboard.writeText(currentReferralCode);
-      toast.success("Referral code copied to clipboard!");
+      setLinkingState(true, selectedMethod);
+      setError(null);
+
+      if (selectedMethod === "email") {
+        await preAuthenticate({
+          client,
+          strategy: "email",
+          email,
+          ecosystem: {
+            id: ecosystemWalletId,
+            partnerId: partnerId,
+          },
+        });
+      } else if (selectedMethod === "phone") {
+        await preAuthenticate({
+          client,
+          strategy: "phone",
+          phoneNumber: phone,
+          ecosystem: {
+            id: ecosystemWalletId,
+            partnerId: partnerId,
+          },
+        });
+      }
+
+      setOtpSent(true);
     } catch (error) {
-      toast.error("Failed to copy referral code");
+      console.error("Error sending OTP:", error);
+      setError(error instanceof Error ? error.message : "Failed to send OTP");
+      onError?.(error as Error);
+      setLinkingState(false);
     }
   };
 
-  const handleUpdateReferralCode = async () => {
-    if (!newReferralCode) return;
+  const handleLinkAccount = async () => {
+    if (!otp) {
+      console.error("No OTP entered");
+      setError("Please enter the verification code");
+      return;
+    }
 
-    setIsUpdatingCode(true);
     try {
-      // @ts-expect-error - setReferralCode is not typed for some reason
-      const newUser = await app.service("users").setReferralCode({
-        userId: user?.userId,
-        referralCode: newReferralCode,
-      });
-      setUser(newUser as unknown as Users);
-      toast.success("Referral code updated successfully!");
-      setIsEditingCode(false);
-      setNewReferralCode("");
+      setOtpSent(false);
+      setLinkingState(true, selectedMethod);
+      setError(null);
+
+      if (selectedMethod === "email") {
+        await linkProfile(
+          {
+            client,
+            strategy: "email",
+            email,
+            verificationCode: otp,
+          },
+          mutationOptions,
+        );
+      } else if (selectedMethod === "phone") {
+        await linkProfile(
+          {
+            client,
+            strategy: "phone",
+            phoneNumber: phone,
+            verificationCode: otp,
+          },
+          mutationOptions,
+        );
+      }
     } catch (error) {
-      toast.error("Failed to update referral code");
-    } finally {
-      setIsUpdatingCode(false);
+      console.error("Error linking account:", error);
+      setError(error instanceof Error ? error.message : "Failed to link account");
+      onError?.(error as Error);
     }
   };
 
-  const profiles = profilesRaw
-    .filter((profile: any) => !["custom_auth_endpoint"].includes(profile.type))
-    .map((profile: any) => ({
-      ...getProfileDisplayInfo(profile),
-      originalProfile: profile,
-    }));
-
-  const handleUnlink = async (profile: any) => {
-    setUnlinkingAccountId(profile.title);
+  const handleLinkWallet = async (walletType: WalletId) => {
+    setLinkingState(true, "wallet");
+    console.log("selectedMethod", walletType);
     try {
-      unlinkProfile({ client, profileToUnlink: profile.originalProfile }, mutationOptions);
+      if (!walletType) {
+        throw new Error("Wallet type not found");
+      }
+      await linkProfile(
+        {
+          client,
+          strategy: "wallet",
+          wallet: createWallet(walletType),
+          chain: thirdwebB3Mainnet,
+        },
+        mutationOptions,
+      );
     } catch (error) {
-      console.error("Error unlinking account:", error);
-    } finally {
-      setUnlinkingAccountId(null);
+      console.error("Error linking account:", error);
+      setError(error instanceof Error ? error.message : "Failed to link account");
+      onError?.(error as Error);
     }
   };
 
-  const handleOpenLinkModal = () => {
-    setB3ModalOpen(true);
-    setB3ModalContentType({
-      type: "linkNewAccount",
-      partnerId,
-      chain,
-      onSuccess: async () => {
-        // Let the LinkAccount component handle modal closing
-      },
-      onError: () => {
-        // Let the LinkAccount component handle errors
-      },
-      onClose: () => {
-        // Let the LinkAccount component handle closing
-      },
-    });
+  const handleSocialLink = async (strategy: SocialStrategy) => {
+    try {
+      console.log("handleSocialLink", strategy);
+      setLinkingState(true, strategy);
+      setError(null);
+
+      const result = await linkProfile(
+        {
+          client,
+          strategy,
+        },
+        mutationOptions,
+      );
+
+      console.log("result", result);
+
+      // Don't close the modal yet, wait for auth to complete
+      onSuccess?.();
+    } catch (error) {
+      console.error("Error linking with social:", error);
+      setError(error instanceof Error ? error.message : "Failed to link social account");
+      onError?.(error as Error);
+      setLinkingState(false);
+    }
   };
 
+  // Add effect to handle social auth completion
+  useEffect(() => {
+    if (isLinking && linkingMethod && !selectedMethod) {
+      // This means we're in a social auth flow
+      const checkAuthStatus = async () => {
+        try {
+          // Wait a bit to ensure auth is complete
+          await new Promise(resolve => setTimeout(resolve, 1000));
+          onClose?.();
+        } catch (error) {
+          console.error("Error checking auth status:", error);
+          setLinkingState(false);
+        }
+      };
+
+      checkAuthStatus();
+    }
+  }, [isLinking, linkingMethod, selectedMethod, onClose, setLinkingState]);
+
+  const handleBack = useCallback(() => {
+    if (isLinking) return;
+    setSelectedMethod(null);
+    setEmail("");
+    setPhone("");
+    setOtp("");
+    setOtpSent(false);
+    setError(null);
+    setLinkingState(false);
+  }, [isLinking, setSelectedMethod, setEmail, setPhone, setOtp, setOtpSent, setError, setLinkingState]);
+
+  const handleFinishedLinking = useCallback(
+    (success: boolean) => {
+      if (success) {
+        onSuccess?.();
+        onClose?.();
+      }
+
+      setLinkingState(false);
+      navigateBack();
+      setB3ModalContentType({
+        type: "manageAccount",
+        activeTab: "settings",
+        setActiveTab: () => {},
+        chain,
+        partnerId,
+      });
+    },
+    [chain, navigateBack, partnerId, setB3ModalContentType, setLinkingState, onSuccess, onClose],
+  );
+
+  useEffect(() => {
+    if (isLinking) {
+      handleFinishedLinking(true);
+    }
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+  }, [profiles.length]);
+
+  if (!account) {
+    return <div className="text-b3-foreground-muted py-8 text-center">Please connect your account first</div>;
+  }
+
   return (
-    <div className="linked-accounts-settings">
-      <ModalHeader title="Linked Accounts" />
-      {/* Linked Accounts Section */}
-      <div className="linked-accounts-section space-y-4 p-5">
-        {isLoadingProfiles ? (
-          <div className="linked-accounts-loading flex justify-center py-8">
-            <Loader2 className="text-b3-grey animate-spin" />
-          </div>
-        ) : profiles.length > 0 ? (
-          <div className="linked-accounts-list space-y-4">
-            {profiles.map(profile => (
-              <div
-                key={profile.title}
-                className="linked-account-item hover:bg-b3-line group flex cursor-pointer items-center justify-between rounded-xl p-4 transition-colors"
-              >
-                <div className="linked-account-info flex items-center gap-3">
-                  {profile.imageUrl ? (
-                    <img
-                      src={profile.imageUrl}
-                      alt={profile.title}
-                      className="linked-account-avatar linked-account-avatar-image size-10 rounded-full"
-                    />
-                  ) : (
-                    <div className="linked-account-avatar linked-account-avatar-placeholder bg-b3-primary-wash flex h-10 w-10 items-center justify-center rounded-full">
-                      <span className="linked-account-initial text-b3-grey font-neue-montreal-semibold text-sm uppercase">
-                        {profile.initial}
-                      </span>
-                    </div>
-                  )}
-                  <div className="linked-account-details">
-                    <div className="linked-account-title-row flex items-center gap-2">
-                      {(() => {
-                        const { displayTitle, isAddress } = formatProfileTitle(profile.title);
-
-                        const handleCopyAddress = async (e: React.MouseEvent) => {
-                          e.stopPropagation();
-                          try {
-                            await navigator.clipboard.writeText(profile.title);
-                            toast.success("Address copied to clipboard!");
-                          } catch (error) {
-                            toast.error("Failed to copy address");
-                          }
-                        };
-
-                        return (
-                          <div className="flex items-center gap-1">
-                            <span
-                              className={`linked-account-title text-b3-grey font-neue-montreal-semibold ${
-                                isAddress
-                                  ? "font-mono text-sm" // Use monospace font for addresses
-                                  : "break-words" // Use break-words for emails/names (better than break-all)
-                              }`}
-                              title={isAddress ? profile.title : undefined} // Show full address on hover
-                            >
-                              {displayTitle}
-                            </span>
-                            {isAddress && (
-                              <button
-                                onClick={handleCopyAddress}
-                                className="linked-account-copy-button ml-1 rounded p-1 opacity-0 transition-opacity hover:bg-gray-100 group-hover:opacity-100"
-                                title="Copy full address"
-                              >
-                                <Copy size={12} className="text-gray-500 hover:text-gray-700" />
-                              </button>
-                            )}
-                          </div>
-                        );
-                      })()}
-                      <span className="linked-account-type text-b3-foreground-muted font-neue-montreal-medium bg-b3-primary-wash rounded px-2 py-0.5 text-xs">
-                        {profile.type.toUpperCase()}
-                      </span>
-                    </div>
-                    <div className="linked-account-subtitle text-b3-foreground-muted font-neue-montreal-medium text-sm">
-                      {profile.subtitle}
-                    </div>
-                  </div>
-                </div>
-                <Button
-                  variant="ghost"
-                  size="icon"
-                  className="linked-account-unlink-button text-b3-grey hover:text-b3-negative"
-                  onClick={() => handleUnlink(profile)}
-                  disabled={unlinkingAccountId === profile.title || isUnlinking}
-                >
-                  {unlinkingAccountId === profile.title || isUnlinking ? (
-                    <Loader2 className="linked-account-unlink-loading animate-spin" />
-                  ) : (
-                    <UnlinkIcon size={16} className="linked-account-unlink-icon" />
-                  )}
-                </Button>
-              </div>
-            ))}
-          </div>
-        ) : (
-          <div className="linked-accounts-empty text-b3-foreground-muted py-8 text-center">
-            No linked accounts found
-          </div>
+    <div className={`b3-link-account space-y-6 p-6 ${className || ""}`} data-testid="link-account">
+      <div className="b3-link-account-header flex items-center justify-between">
+        <h2 className="b3-link-account-title text-b3-grey font-neue-montreal-semibold text-2xl">Link New Account</h2>
+        {selectedMethod && (
+          <Button variant="ghost" className="text-b3-grey hover:text-b3-grey/80" onClick={handleBack}>
+            Backs
+          </Button>
         )}
-
-        {/* Link New Account Button */}
-        <Button
-          onClick={handleOpenLinkModal}
-          disabled={isLinking}
-          className="b3-modal-link-new-account-button bg-b3-primary-blue hover:bg-b3-primary-blue/90 border-white/12 group relative h-12 w-full rounded-xl border-2 px-[18px] text-base font-semibold text-white shadow-[inset_0px_0px_0px_1px_rgba(10,13,18,0.18),inset_0px_-2px_0px_0px_rgba(10,13,18,0.05)] transition-all"
-        >
-          <div className="flex items-center justify-center gap-1.5">
-            <svg
-              width="20"
-              height="20"
-              viewBox="0 0 20 20"
-              fill="none"
-              xmlns="http://www.w3.org/2000/svg"
-              className="shrink-0"
-            >
-              <path
-                d="M10.0001 13.3333V6.66667M6.66675 10H13.3334M18.3334 10C18.3334 14.6024 14.6025 18.3333 10.0001 18.3333C5.39771 18.3333 1.66675 14.6024 1.66675 10C1.66675 5.39763 5.39771 1.66667 10.0001 1.66667C14.6025 1.66667 18.3334 5.39763 18.3334 10Z"
-                stroke="currentColor"
-                strokeWidth="1.67"
-                strokeLinecap="round"
-                strokeLinejoin="round"
-              />
-            </svg>
-            <span>Link new account</span>
-          </div>
-        </Button>
       </div>
 
-      {showReferralInfo && (
-        /* Referral Section */
-        <div className="referrals-section space-y-4">
-          <h3 className="referrals-title text-b3-grey font-neue-montreal-semibold text-xl">Referrals</h3>
-
-          {/* Referral Code */}
-          <div className="referral-code-container bg-b3-line rounded-xl p-4">
-            {isEditingCode && (
-              <div className="referral-code-header-editing">
-                <div className="referral-code-title text-b3-grey font-neue-montreal-semibold">Your Referral Code</div>
-                <div className="referral-code-description text-b3-foreground-muted font-neue-montreal-medium text-sm">
-                  Share this code with friends to earn rewards
-                </div>
-              </div>
-            )}
-            <div className="referral-code-content flex items-center justify-between">
-              {!isEditingCode && (
-                <div className="referral-code-header">
-                  <div className="referral-code-title text-b3-grey font-neue-montreal-semibold">Your Referral Code</div>
-                  <div className="referral-code-description text-b3-foreground-muted font-neue-montreal-medium text-sm">
-                    Share this code with friends to earn rewards
+      {!selectedMethod ? (
+        <div className="b3-link-account-methods grid gap-3">
+          {availableAuthMethods.map(method => (
+            <Button
+              key={method.id}
+              variant="outline"
+              className="b3-link-account-method-button border-b3-line hover:border-b3-primary-blue/30 hover:bg-b3-primary-blue/5 text-b3-grey font-neue-montreal-medium h-14 justify-start bg-transparent px-6 text-base transition-all duration-200"
+              data-method={method.id}
+              onClick={() => {
+                if (method.id === "email" || method.id === "phone") {
+                  setSelectedMethod(method.id);
+                } else {
+                  handleSocialLink(method.id as SocialStrategy);
+                }
+              }}
+              disabled={linkingMethod === method.id}
+            >
+              {isLinking && linkingMethod === method.id ? (
+                <Loader2 className="h-5 w-5 animate-spin" />
+              ) : (
+                <div className="b3-link-account-method-content flex items-center gap-4">
+                  <div className="b3-link-account-method-icon flex items-center justify-center rounded-full">
+                    {method.icon}
                   </div>
+                  <span className="b3-link-account-method-label font-medium">{method.label}</span>
                 </div>
               )}
-              <div className="referral-code-actions flex items-center gap-2">
-                {isEditingCode ? (
-                  <div className="referral-code-edit-form flex items-center gap-2">
-                    <input
-                      type="text"
-                      value={newReferralCode}
-                      onChange={e => setNewReferralCode(e.target.value)}
-                      className="referral-code-input rounded-lg border border-gray-200 bg-white px-3 py-1.5 text-sm"
-                      placeholder="Enter new code"
-                      ref={referallCodeRef}
-                    />
-                    <Button
-                      size="sm"
-                      className="referral-code-save-button"
-                      onClick={handleUpdateReferralCode}
-                      disabled={isUpdatingCode || !newReferralCode}
-                    >
-                      {isUpdatingCode ? (
-                        <Loader2 className="referral-code-save-loading h-4 w-4 animate-spin" />
-                      ) : (
-                        "Save"
-                      )}
-                    </Button>
-                    <Button
-                      size="sm"
-                      variant="ghost"
-                      className="referral-code-cancel-button"
-                      onClick={() => {
-                        setIsEditingCode(false);
-                        setNewReferralCode("");
-                      }}
-                    >
-                      Cancel
-                    </Button>
-                  </div>
-                ) : (
-                  <>
-                    <div className="referral-code-display rounded-lg border border-gray-200 bg-white px-3 py-1.5 text-sm">
-                      {currentReferralCode}
-                    </div>
-                    <Button size="icon" variant="ghost" className="referral-code-copy-button" onClick={handleCopyCode}>
-                      <Copy className="referral-code-copy-icon h-4 w-4" />
-                    </Button>
-                    <Button
-                      size="icon"
-                      variant="ghost"
-                      className="referral-code-edit-button"
-                      onClick={() => {
-                        setIsEditingCode(true);
-                        setTimeout(() => {
-                          referallCodeRef.current?.focus();
-                        }, 100);
-                      }}
-                    >
-                      <Pencil className="referral-code-edit-icon h-4 w-4" />
-                    </Button>
-                  </>
-                )}
-              </div>
+            </Button>
+          ))}
+          {WALLET_METHODS.map(method => {
+            if (!method.walletType) {
+              return null;
+            }
+
+            return (
+              <WalletRow
+                key={method.walletType}
+                walletId={method.walletType as WalletId}
+                onClick={() => handleLinkWallet(method.walletType as WalletId)}
+                isLoading={isLinking}
+              />
+            );
+          })}
+          {availableAuthMethods.length === 0 && (
+            <div className="text-b3-foreground-muted py-8 text-center">
+              All available authentication methods have been connected
             </div>
-          </div>
-
-          {/* Referred Users */}
-          <div className="referred-users-container bg-b3-line rounded-xl p-4">
-            <div className="referred-users-title text-b3-grey font-neue-montreal-semibold mb-4">Referred Users</div>
-            {isLoadingReferrals ? (
-              <div className="referred-users-loading flex justify-center py-4">
-                <Loader2 className="h-6 w-6 animate-spin text-gray-400" />
-              </div>
-            ) : referrals?.data?.length ? (
-              <div className="referred-users-list space-y-3">
-                {referrals.data.map((referral: Referrals) => (
-                  <div
-                    key={String(referral._id)}
-                    className="referred-user-item flex items-center justify-between rounded-lg bg-white p-3"
-                  >
-                    <div className="referred-user-id text-sm font-medium">{referral.referreeId}</div>
-                    <div className="referred-user-date text-sm text-gray-500">
-                      {new Date(referral.createdAt).toLocaleDateString()}
-                    </div>
-                  </div>
-                ))}
+          )}
+        </div>
+      ) : (
+        <div className="b3-link-account-form space-y-4">
+          {selectedMethod === "email" && (
+            <div className="space-y-2">
+              <label className="text-b3-grey font-neue-montreal-medium text-sm">Email Address</label>
+              <input
+                type="email"
+                placeholder="Enter your email"
+                className="bg-b3-line text-b3-grey font-neue-montreal-medium focus:ring-b3-primary-blue/20 w-full rounded-xl p-4 focus:outline-none focus:ring-2"
+                value={email}
+                onChange={e => setEmail(e.target.value)}
+                disabled={otpSent || (isLinking && linkingMethod === "email")}
+              />
+            </div>
+          )}
+
+          {selectedMethod === "phone" && (
+            <div className="space-y-2">
+              <label className="text-b3-grey font-neue-montreal-medium text-sm">Phone Number</label>
+              <input
+                type="tel"
+                placeholder="Enter your phone number"
+                className="bg-b3-line text-b3-grey font-neue-montreal-medium focus:ring-b3-primary-blue/20 w-full rounded-xl p-4 focus:outline-none focus:ring-2"
+                value={phone}
+                onChange={e => setPhone(e.target.value)}
+                disabled={otpSent || (isLinking && linkingMethod === "phone")}
+              />
+              <p className="text-b3-foreground-muted font-neue-montreal-medium text-sm">
+                Include country code (e.g., +1 for US)
+              </p>
+            </div>
+          )}
+
+          {error && <div className="text-b3-negative font-neue-montreal-medium py-2 text-sm">{error}</div>}
+
+          {(selectedMethod === "email" || selectedMethod === "phone") &&
+            (otpSent ? (
+              <div className="space-y-4">
+                <div className="space-y-2">
+                  <label className="text-b3-grey font-neue-montreal-medium text-sm">Verification Code</label>
+                  <input
+                    type="text"
+                    placeholder="Enter verification code"
+                    className="bg-b3-line text-b3-grey font-neue-montreal-medium focus:ring-b3-primary-blue/20 w-full rounded-xl p-4 focus:outline-none focus:ring-2"
+                    value={otp}
+                    onChange={e => setOtp(e.target.value)}
+                  />
+                </div>
+                <Button
+                  className="bg-b3-primary-blue hover:bg-b3-primary-blue/90 font-neue-montreal-semibold h-12 w-full text-white"
+                  onClick={handleLinkAccount}
+                >
+                  Link Account
+                </Button>
               </div>
             ) : (
-              <div className="referred-users-empty py-4 text-center text-gray-500">No referred users yet</div>
-            )}
-          </div>
+              <Button
+                className="bg-b3-primary-blue hover:bg-b3-primary-blue/90 font-neue-montreal-semibold h-12 w-full text-white"
+                onClick={handleSendOTP}
+                disabled={(!email && !phone) || (isLinking && linkingMethod === selectedMethod)}
+              >
+                {isLinking && linkingMethod === selectedMethod ? (
+                  <Loader2 className="animate-spin" />
+                ) : (
+                  "Send Verification Code"
+                )}
+              </Button>
+            ))}
         </div>
       )}
     </div>
   );
-};
+}