From 5779bf74bda2aeb4c794487248d5a84ae2c14153 Mon Sep 17 00:00:00 2001
From: Einliterflasche <einliterflasche@pm.me>
Date: Mon, 8 Jul 2024 15:43:33 +0200
Subject: [PATCH] add --refund-to-internal-wallet flag for buy-xmr.

---
 swap/src/api.rs         |  7 +++++++
 swap/src/cli/command.rs | 45 ++++++++++++++++++++++++++++++-----------
 2 files changed, 40 insertions(+), 12 deletions(-)

diff --git a/swap/src/api.rs b/swap/src/api.rs
index 340c2e39..142f6ebc 100644
--- a/swap/src/api.rs
+++ b/swap/src/api.rs
@@ -267,6 +267,13 @@ impl Context {
             tasks: Arc::new(PendingTaskList::default()),
         }
     }
+
+    pub fn bitcoint_wallet(&self) -> Option<Arc<bitcoin::Wallet>> {
+        match &self.bitcoin_wallet {
+            None => None,
+            Some(wallet) => Some(wallet.clone()),
+        }
+    }
 }
 
 impl fmt::Debug for Context {
diff --git a/swap/src/cli/command.rs b/swap/src/cli/command.rs
index af72df1b..416be1f9 100644
--- a/swap/src/cli/command.rs
+++ b/swap/src/cli/command.rs
@@ -65,10 +65,34 @@ where
             seller: Seller { seller },
             bitcoin,
             bitcoin_change_address,
+            refund_to_internal_wallet,
             monero,
             monero_receive_address,
             tor,
         } => {
+            let context = Context::build(
+                Some(bitcoin),
+                Some(monero),
+                Some(tor),
+                data,
+                is_testnet,
+                debug,
+                json,
+                None,
+            )
+            .await?;
+
+            let bitcoin_change_address = match bitcoin_change_address {
+                Some(addr) => addr,
+                None => {
+                    context
+                        .bitcoint_wallet()
+                        .expect("bitcoin wallet should exist")
+                        .new_address()
+                        .await?
+                }
+            };
+
             let monero_receive_address =
                 monero_address::validate_is_testnet(monero_receive_address, is_testnet)?;
             let bitcoin_change_address =
@@ -81,17 +105,6 @@ where
                 swap_id: Uuid::new_v4(),
             });
 
-            let context = Context::build(
-                Some(bitcoin),
-                Some(monero),
-                Some(tor),
-                data,
-                is_testnet,
-                debug,
-                json,
-                None,
-            )
-            .await?;
             (context, request)
         }
         CliCommand::History => {
@@ -303,7 +316,15 @@ enum CliCommand {
             help = "The bitcoin address where any form of change or excess funds should be sent to",
             parse(try_from_str = bitcoin_address::parse)
         )]
-        bitcoin_change_address: bitcoin::Address,
+        bitcoin_change_address: Option<bitcoin::Address>,
+
+        #[structopt(
+            conflicts_with = "change-address",
+            long = "refund-to-internal-wallet",
+            help = "Instead of providing a bitcoin address to deposit refunded Bitcoin to, \
+                this option will deposit them in the internal wallet for use in future swaps."
+        )]
+        refund_to_internal_wallet: bool,
 
         #[structopt(flatten)]
         monero: Monero,