Don't add tx to wallet when there's error

ziesha-network · Jul 13, 2023 · 031809b · 031809b
1 parent ffc0c37
commit 031809b
Showing 1 changed file with 23 additions and 10 deletions.
diff --git a/src/cli/wallet/send.rs b/src/cli/wallet/send.rs
@@ -84,7 +84,7 @@ pub async fn send(
                             } else {
                                 wallet.user(0).add_tx(tx.clone().into());
                                 wallet.save(wallet_path).unwrap();
-                                println!("Sent");
+                                println!("Sent!");
                             }
                         }
                         GeneralAddress::MpnAddress(to) => {
@@ -110,9 +110,13 @@ pub async fn send(
                                     token_id: TokenId::Ziesha,
                                 },
                             );
-                            wallet.user(0).add_tx(pay.clone().into());
-                            wallet.save(wallet_path).unwrap();
-                            println!("{:#?}", client.transact(pay.clone().into()).await?);
+                            if let Some(err) = client.transact(pay.clone().into()).await?.error {
+                                println!("Error: {}", err);
+                            } else {
+                                wallet.user(0).add_tx(pay.clone().into());
+                                wallet.save(wallet_path).unwrap();
+                                println!("Sent!");
+                            }
                         }
                     }
                 }
@@ -142,9 +146,14 @@ pub async fn send(
                                 },
                                 to.to_string().parse().unwrap(), // TODO: WTH :D
                             );
-                            wallet.user(0).add_tx(pay.clone().into());
-                            wallet.save(wallet_path).unwrap();
-                            println!("{:#?}", client.transact(pay.clone().into()).await?);
+
+                            if let Some(err) = client.transact(pay.clone().into()).await?.error {
+                                println!("Error: {}", err);
+                            } else {
+                                wallet.user(0).add_tx(pay.clone().into());
+                                wallet.save(wallet_path).unwrap();
+                                println!("Sent!");
+                            }
                         }
 
                         GeneralAddress::MpnAddress(to) => {
@@ -168,9 +177,13 @@ pub async fn send(
                                 },
                                 new_nonce,
                             );
-                            wallet.user(0).add_tx(tx.clone().into());
-                            wallet.save(wallet_path).unwrap();
-                            println!("{:#?}", client.transact(tx.clone().into()).await?);
+                            if let Some(err) = client.transact(tx.clone().into()).await?.error {
+                                println!("Error: {}", err);
+                            } else {
+                                wallet.user(0).add_tx(tx.clone().into());
+                                wallet.save(wallet_path).unwrap();
+                                println!("Sent!");
+                            }
                         }
                     }
                 }