feat(wallet): add DRep key support and enhance wallet management

- Introduced `signersDRepKeys` to the NewWallet model and updated related components to handle DRep keys.
- Enhanced wallet creation and management flows to include DRep key handling.
- Updated API endpoints to support DRep key storage and retrieval.
- Improved UI components to display DRep key information and validation messages.
- Refactored error handling in wallet operations to manage account changes effectively.

Author: QSchlegel

package-lock.json

@@ -24,10 +24,10 @@
     "@auth/prisma-adapter": "^1.6.0",
     "@hookform/resolvers": "^3.9.0",
     "@jinglescode/nostr-chat-plugin": "^0.0.11",
-    "@meshsdk/core": "^1.9.0-beta.18",
-    "@meshsdk/core-csl": "^1.9.0-beta.18",
-    "@meshsdk/core-cst": "^1.9.0-beta.19",
-    "@meshsdk/react": "^1.9.0-beta.69",
+    "@meshsdk/core": "^1.9.0-beta.77",
+    "@meshsdk/core-csl": "^1.9.0-beta.77",
+    "@meshsdk/core-cst": "^1.9.0-beta.77",
+    "@meshsdk/react": "^1.9.0-beta.77",
     "@octokit/core": "^6.1.2",
     "@prisma/client": "^6.4.1",
     "@radix-ui/react-accordion": "^1.2.0",
@@ -60,14 +60,14 @@
     "clsx": "^2.1.1",
     "cors": "^2.8.5",
     "dexie": "^4.0.11",
-    "formidable": "^3.5.2",
+    "formidable": "^3.5.4",
     "framer-motion": "^11.11.9",
     "geist": "^1.3.0",
     "idb-keyval": "^6.2.1",
     "jsonld": "^8.3.3",
     "jsonwebtoken": "^9.0.2",
     "lucide-react": "^0.439.0",
-    "next": "^14.2.4",
+    "next": "^15.2.4",
     "next-auth": "^4.24.7",
     "papaparse": "^5.5.3",
     "react": "^18.3.1",

package.json

@@ -0,0 +1,2 @@
+-- AlterTable
+ALTER TABLE "NewWallet" ADD COLUMN     "signersDRepKeys" TEXT[];

prisma/migrations/20250929093740_add_drep_keys_to_new_wallet/migration.sql

@@ -72,6 +72,7 @@ model NewWallet {
   description         String?
   signersAddresses    String[]
   signersStakeKeys    String[]
+  signersDRepKeys     String[]
   signersDescriptions String[]
   numRequiredSigners  Int?
   ownerAddress        String

prisma/schema.prisma

@@ -17,7 +17,7 @@ export default function CardUI({
   headerDom?: ReactNode;
 }) {
   return (
-    <Card className={`self-start ${cardClassName}`}>
+    <Card className={`w-full max-w-4xl ${cardClassName}`}>
       <CardHeader className="flex flex-row items-center justify-between space-y-0 pb-2">
         <CardTitle className="text-xl font-medium">{title}</CardTitle>
         {headerDom && headerDom}
@@ -33,7 +33,7 @@ export default function CardUI({
           </>
         )}
       </CardHeader>
-      <CardContent>
+      <CardContent className="overflow-y-auto max-h-[calc(100vh-200px)]">
         <div className="mt-1 flex flex-col gap-2">
           {description && (
             <p className="text-sm text-muted-foreground">{description}</p>

src/components/common/card-content.tsx

@@ -35,26 +35,33 @@ import {
   BreadcrumbSeparator,
 } from "@/components/ui/breadcrumb";
 
-// Simple error boundary component
-class SimpleErrorBoundary extends Component<
+// Enhanced error boundary component for wallet errors
+class WalletErrorBoundary extends Component<
   { children: ReactNode; fallback: ReactNode },
-  { hasError: boolean }
+  { hasError: boolean; error: Error | null }
 > {
   constructor(props: { children: ReactNode; fallback: ReactNode }) {
     super(props);
-    this.state = { hasError: false };
+    this.state = { hasError: false, error: null };
   }
 
-  static getDerivedStateFromError() {
-    return { hasError: true };
+  static getDerivedStateFromError(error: Error) {
+    return { hasError: true, error };
   }
 
   componentDidCatch(error: Error, errorInfo: any) {
-    console.error('Error caught by boundary:', error, errorInfo);
+    console.error('Error caught by wallet boundary:', error, errorInfo);
+    
+    // Handle specific wallet errors
+    if (error.message.includes("account changed")) {
+      console.log("Wallet account changed error caught by boundary, reloading page...");
+      window.location.reload();
+      return;
+    }
   }
 
   render() {
-    if (this.state.hasError) {
+    if (this.state.hasError && this.state.error && !this.state.error.message.includes("account changed")) {
       return this.props.fallback;
     }
     return this.props.children;
@@ -75,6 +82,30 @@ export default function RootLayout({
   const userAddress = useUserStore((state) => state.userAddress);
   const setUserAddress = useUserStore((state) => state.setUserAddress);
 
+  // Global error handler for unhandled promise rejections
+  useEffect(() => {
+    const handleUnhandledRejection = (event: PromiseRejectionEvent) => {
+      console.error('Unhandled promise rejection:', event.reason);
+      
+      // Handle wallet-related errors specifically
+      if (event.reason && typeof event.reason === 'object') {
+        const error = event.reason as Error;
+        if (error.message && error.message.includes("account changed")) {
+          console.log("Account changed error caught by global handler, reloading page...");
+          event.preventDefault(); // Prevent the error from being logged to console
+          window.location.reload();
+          return;
+        }
+      }
+    };
+
+    window.addEventListener('unhandledrejection', handleUnhandledRejection);
+    
+    return () => {
+      window.removeEventListener('unhandledrejection', handleUnhandledRejection);
+    };
+  }, []);
+
   const { mutate: createUser } = api.user.createUser.useMutation({
     onError: (e) => console.error(e),
   });
@@ -87,56 +118,57 @@ export default function RootLayout({
     (async () => {
       if (!connected || !wallet) return;
 
-      // 1) Set user address in store
-      let address = (await wallet.getUsedAddresses())[0];
-      if (!address) address = (await wallet.getUnusedAddresses())[0];
-      if (address) setUserAddress(address);
+      try {
+        // 1) Set user address in store
+        let address = (await wallet.getUsedAddresses())[0];
+        if (!address) address = (await wallet.getUnusedAddresses())[0];
+        if (address) setUserAddress(address);
 
-      // 2) Get stake address
-      const stakeAddress = (await wallet.getRewardAddresses())[0];
-      if (!stakeAddress || !address) {
-        console.error("No stake address or payment address found");
-        return;
-      }
+        // 2) Get stake address
+        const stakeAddress = (await wallet.getRewardAddresses())[0];
+        if (!stakeAddress || !address) {
+          console.error("No stake address or payment address found");
+          return;
+        }
 
-      // 3) Get DRep key hash
-      const dRepKey = await wallet.getDRep();
-      if (!dRepKey) {
-        console.error("No DRep key found");
-        return;
-      }
-      const drepKeyHash = dRepKey.publicKeyHash;
-      if (!drepKeyHash) {
-        console.error("No DRep key hash found:", drepKeyHash);
-        return;
-      }
-
-      // 4) If user doesn't exist create it
-      if (!isLoading && user === null) {
-        const nostrKey = generateNsec();
-        createUser({
-          address,
-          stakeAddress,
-          drepKeyHash,
-          nostrKey: JSON.stringify(nostrKey),
-        });
-      }
+        // 3) Get DRep key hash (optional)
+        let drepKeyHash = "";
+        try {
+          const dRepKey = await wallet.getDRep();
+          console.log("DRep key:", dRepKey);
+          if (dRepKey && dRepKey.publicKeyHash) {
+            drepKeyHash = dRepKey.publicKeyHash;
+          }
+        } catch (error) {
+          // DRep key is optional, so we continue without it
+          console.log("No DRep key available for this wallet");
+        }
 
-      // 5) If user exists but missing fields, update it
-      if (
-        !isLoading &&
-        user &&
-        user !== null &&
-        (user.stakeAddress !== stakeAddress || user.drepKeyHash !== drepKeyHash)
-      ) {
-        updateUser({
-          address,
-          stakeAddress,
-          drepKeyHash,
-        });
+        // 4) Create or update user (upsert pattern handles both cases)
+        if (!isLoading) {
+          const nostrKey = generateNsec();
+          createUser({
+            address,
+            stakeAddress,
+            drepKeyHash,
+            nostrKey: JSON.stringify(nostrKey),
+          });
+        }
+      } catch (error) {
+        console.error("Error in wallet initialization effect:", error);
+        
+        // If we get an "account changed" error, reload the page
+        if (error instanceof Error && error.message.includes("account changed")) {
+          console.log("Account changed detected, reloading page...");
+          window.location.reload();
+          return;
+        }
+        
+        // For other errors, don't throw to prevent app crash
+        // The user can retry by reconnecting their wallet
       }
     })();
-  }, [connected, wallet, user, isLoading, generateNsec, setUserAddress]);
+  }, [connected, wallet, user, isLoading, createUser, generateNsec, setUserAddress]);
 
   const isWalletPath = router.pathname.includes("/wallets/[wallet]");
   const walletPageRoute = router.pathname.split("/wallets/[wallet]/")[1];
@@ -268,7 +300,22 @@ export default function RootLayout({
         </header>
 
         <main className="relative flex flex-1 flex-col gap-4 overflow-y-auto overflow-x-hidden p-4 md:p-8">
-          {pageIsPublic || userAddress ? children : <PageHomepage />}
+          <WalletErrorBoundary 
+            fallback={
+              <div className="flex flex-col items-center justify-center h-full">
+                <h2 className="text-xl font-semibold mb-2">Something went wrong</h2>
+                <p className="text-gray-600 mb-4">Please try refreshing the page or reconnecting your wallet.</p>
+                <button 
+                  onClick={() => window.location.reload()} 
+                  className="px-4 py-2 bg-blue-500 text-white rounded hover:bg-blue-600"
+                >
+                  Refresh Page
+                </button>
+              </div>
+            }
+          >
+            {pageIsPublic || userAddress ? children : <PageHomepage />}
+          </WalletErrorBoundary>
         </main>
       </div>
     </div>

src/components/common/overall-layout/layout.tsx

@@ -51,28 +51,77 @@ export default function UserDropDownWrapper({
   });
 
   async function unlinkDiscord(): Promise<void> {
-    const address = (await wallet.getUsedAddresses())[0];
-    unlinkDiscordMutation.mutate({ address: address ?? "" });
-    setOpen(false);
+    try {
+      const usedAddresses = await wallet.getUsedAddresses();
+      const address = usedAddresses[0];
+      unlinkDiscordMutation.mutate({ address: address ?? "" });
+      setOpen(false);
+    } catch (error) {
+      console.error("Error getting wallet address for Discord unlink:", error);
+      if (error instanceof Error && error.message.includes("account changed")) {
+        console.log("Account changed during Discord unlink, aborting");
+        return;
+      }
+      // Show error to user
+      toast({
+        title: "Error",
+        description: "Failed to get wallet address",
+        variant: "destructive",
+        duration: 3000,
+      });
+    }
   }
 
   const { data: discordData } = api.user.getUserDiscordId.useQuery({
     address: userAddress ?? "",
   });
 
   async function handleCopyAddress() {
-    let userAddress = (await wallet.getUsedAddresses())[0];
-    if (userAddress === undefined) {
-      userAddress = (await wallet.getUnusedAddresses())[0];
+    try {
+      let userAddress: string | undefined;
+      try {
+        const usedAddresses = await wallet.getUsedAddresses();
+        userAddress = usedAddresses[0];
+      } catch (error) {
+        if (error instanceof Error && error.message.includes("account changed")) {
+          console.log("Account changed during address copy, aborting");
+          return;
+        }
+        throw error;
+      }
+      
+      if (userAddress === undefined) {
+        try {
+          const unusedAddresses = await wallet.getUnusedAddresses();
+          userAddress = unusedAddresses[0];
+        } catch (error) {
+          if (error instanceof Error && error.message.includes("account changed")) {
+            console.log("Account changed during unused address retrieval, aborting");
+            return;
+          }
+          throw error;
+        }
+      }
+      
+      if (userAddress) {
+        navigator.clipboard.writeText(userAddress);
+        toast({
+          title: "Copied",
+          description: "Address copied to clipboard",
+          duration: 5000,
+        });
+        setOpen(false);
+        if (onAction) onAction();
+      }
+    } catch (error) {
+      console.error("Error copying wallet address:", error);
+      toast({
+        title: "Error",
+        description: "Failed to copy address",
+        variant: "destructive",
+        duration: 3000,
+      });
     }
-    navigator.clipboard.writeText(userAddress!);
-    toast({
-      title: "Copied",
-      description: "Address copied to clipboard",
-      duration: 5000,
-    });
-    setOpen(false);
-    if (onAction) onAction();
   }
 
   function handleLogout() {