From d4c1758f1b5426c7ad3abe01c55a0f799ed80e1e Mon Sep 17 00:00:00 2001
From: Carson McManus <carson.mcmanus1@gmail.com>
Date: Wed, 28 Jul 2021 15:48:06 -0400
Subject: [PATCH] implement --accept-all

---
 src/lib.rs  | 10 +++++++++-
 src/main.rs | 11 +++++++++--
 2 files changed, 18 insertions(+), 3 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index 955f40f..2bab7ae 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -211,7 +211,7 @@ impl SteamGuardAccount {
 	///
 	/// Host: https://steamcommunity.com
 	/// Steam Endpoint: `POST /mobileconf/ajaxop`
-	fn send_confirmation_ajax(&self, conf: Confirmation, operation: String) -> anyhow::Result<()> {
+	fn send_confirmation_ajax(&self, conf: &Confirmation, operation: String) -> anyhow::Result<()> {
 		ensure!(operation == "allow" || operation == "cancel");
 
 		let url = "https://steamcommunity.com".parse::<Url>().unwrap();
@@ -241,6 +241,14 @@ impl SteamGuardAccount {
 		ensure!(resp.success);
 		Ok(())
 	}
+
+	pub fn accept_confirmation(&self, conf: &Confirmation) -> anyhow::Result<()> {
+		self.send_confirmation_ajax(conf, "allow".into())
+	}
+
+	pub fn deny_confirmation(&self, conf: &Confirmation) -> anyhow::Result<()> {
+		self.send_confirmation_ajax(conf, "cancel".into())
+	}
 }
 
 #[cfg(test)]
diff --git a/src/main.rs b/src/main.rs
index abb7efe..fb1129c 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -124,7 +124,7 @@ fn main() {
 
 	debug!("selected accounts: {:?}", selected_accounts.iter().map(|a| a.account_name.clone()).collect::<Vec<String>>());
 
-	if matches.is_present("trade") {
+	if let Some(trade_matches) = matches.subcommand_matches("trade") {
 		info!("trade");
 		for a in selected_accounts.iter_mut() {
 			let mut account = a; // why is this necessary?
@@ -133,9 +133,16 @@ fn main() {
 			loop {
 				match account.get_trade_confirmations() {
 					Ok(confs) => {
-						for conf in confs {
+						for conf in &confs {
 							println!("{}", conf.description());
 						}
+						if trade_matches.is_present("accept-all") {
+							info!("accepting all confirmations");
+							for conf in &confs {
+								let result = account.accept_confirmation(conf);
+								debug!("accept confirmation result: {:?}", result);
+							}
+						}
 						break;
 					}
 					Err(_) => {