Implement wallet connection checks before creating new wallets

- Added user address validation in the new wallet creation process across multiple components to ensure users are connected before proceeding.
- Enhanced error messaging to inform users when their wallet is not connected, improving user experience and preventing unnecessary actions.

Author: QSchlegel

src/components/pages/homepage/wallets/new-wallet-flow/shared/useWalletFlowState.tsx

@@ -449,6 +449,14 @@ export function useWalletFlowState(): WalletFlowState {
 
   async function handleCreateNewWallet() {
     if (router.pathname == "/wallets/new-wallet-flow/save") {
+      if (!userAddress) {
+        toast({
+          title: "Error",
+          description: "Please connect your wallet before creating a wallet.",
+          variant: "destructive",
+        });
+        return;
+      }
       setLoading(true);
       createNewWallet({
         name: name,
@@ -457,7 +465,7 @@ export function useWalletFlowState(): WalletFlowState {
         signersDescriptions: signersDescriptions,
         signersStakeKeys: signersStakeKeys,
         signersDRepKeys: signersDRepKeys,
-        ownerAddress: userAddress!,
+        ownerAddress: userAddress,
         numRequiredSigners: numRequiredSigners,
         stakeCredentialHash: stakeKey || undefined,
         scriptType: nativeScriptType,

src/components/pages/homepage/wallets/new-wallet/index.tsx

@@ -239,6 +239,14 @@ export default function PageNewWallet() {
 
   async function handleCreateNewWallet() {
     if (router.pathname == "/wallets/new-wallet") {
+      if (!userAddress) {
+        toast({
+          title: "Error",
+          description: "Please connect your wallet before creating a wallet.",
+          variant: "destructive",
+        });
+        return;
+      }
       setLoading(true);
       createNewWallet({
         name: name,
@@ -247,7 +255,7 @@ export default function PageNewWallet() {
         signersDescriptions: signersDescriptions,
         signersStakeKeys: signersStakeKeys,
         signersDRepKeys: signersDRepKeys,
-        ownerAddress: userAddress!,
+        ownerAddress: userAddress,
         numRequiredSigners: numRequiredSigners,
         stakeCredentialHash: stakeKey || undefined,
         scriptType: nativeScriptType,

src/components/pages/wallet/info/migration/useMigrationWalletFlowState.tsx

@@ -335,6 +335,15 @@ export function useMigrationWalletFlowState(appWallet: Wallet): MigrationWalletF
       return;
     }
 
+    if (!userAddress) {
+      toast({
+        title: "Error",
+        description: "Please connect your wallet before creating a wallet invite link.",
+        variant: "destructive",
+      });
+      return;
+    }
+
     setLoading(true);
     try {
       const walletData = {
@@ -345,7 +354,7 @@ export function useMigrationWalletFlowState(appWallet: Wallet): MigrationWalletF
         signersStakeKeys: signersStakeKeys,
         signersDRepKeys: signersDRepKeys,
         numRequiredSigners: numRequiredSigners,
-        ownerAddress: userAddress ?? "",
+        ownerAddress: userAddress,
         stakeCredentialHash: stakeKey || undefined,
         scriptType: nativeScriptType || undefined,
       };

src/server/api/routers/wallets.ts

@@ -342,7 +342,14 @@ export const walletRouter = createTRPCRouter({
     )
     .mutation(async ({ ctx, input }) => {
       const sessionAddress = requireSessionAddress(ctx);
-      if (sessionAddress !== input.ownerAddress) {
+      const sessionWallets: string[] = (ctx as any).sessionWallets ?? [];
+      
+      // Allow ownerAddress to be either the sessionAddress or any address in sessionWallets
+      const isAuthorized = 
+        sessionAddress === input.ownerAddress || 
+        sessionWallets.includes(input.ownerAddress);
+      
+      if (!isAuthorized) {
         throw new TRPCError({ code: "FORBIDDEN", message: "Owner address mismatch" });
       }
       const numRequired = (input.scriptType === "all" || input.scriptType === "any") ? null : input.numRequiredSigners;