steamguard-cli/src/main.rs

113 lines
2.8 KiB
Rust
Raw Normal View History

extern crate rpassword;
2021-03-27 17:14:34 +01:00
use borrow::BorrowMut;
use io::Write;
2021-03-22 02:21:29 +01:00
use steamguard_cli::*;
use ::std::*;
use text_io::read;
2021-03-26 00:47:44 +01:00
use std::path::Path;
2021-03-27 14:31:38 +01:00
use clap::{App, Arg, crate_version};
2021-03-27 15:35:52 +01:00
use log::*;
mod steamapi;
2021-03-26 00:47:44 +01:00
mod accountmanager;
2021-03-22 02:21:29 +01:00
fn main() {
2021-03-27 14:31:38 +01:00
let matches = App::new("steamguard-cli")
.version(crate_version!())
.bin_name("steamguard")
.author("dyc3 (Carson McManus)")
.about("Generate Steam 2FA codes and confirm Steam trades from the command line.")
.arg(
Arg::with_name("username")
.long("username")
.short("u")
.help("Select the account you want by steam username. By default, the first account in the manifest is selected.")
)
.arg(
Arg::with_name("all")
.long("all")
.short("a")
2021-03-27 17:14:34 +01:00
.takes_value(false)
2021-03-27 14:31:38 +01:00
.help("Select all accounts in the manifest.")
)
.arg(
Arg::with_name("mafiles-path")
.long("mafiles-path")
.short("m")
.default_value("~/maFiles")
.help("Specify which folder your maFiles are in.")
)
.arg(
Arg::with_name("passkey")
.long("passkey")
.short("p")
.help("Specify your encryption passkey.")
)
2021-03-27 15:35:52 +01:00
.arg(
Arg::with_name("verbosity")
.short("v")
.help("Log what is going on verbosely.")
.takes_value(false)
.multiple(true)
)
2021-03-27 14:31:38 +01:00
.subcommand(
App::new("trade")
.about("Interactive interface for trade confirmations")
.arg(
Arg::with_name("accept-all")
.short("a")
.long("accept-all")
2021-03-27 17:14:34 +01:00
.takes_value(false)
2021-03-27 14:31:38 +01:00
.help("Accept all open trade confirmations. Does not open interactive interface.")
)
)
.get_matches();
2021-03-22 02:21:29 +01:00
2021-03-27 15:35:52 +01:00
let verbosity = matches.occurrences_of("verbosity") as usize + 2;
stderrlog::new()
.verbosity(verbosity)
.module(module_path!()).init().unwrap();
2021-03-27 15:35:52 +01:00
let path = Path::new(matches.value_of("mafiles-path").unwrap()).join("manifest.json");
let mut manifest: accountmanager::Manifest;
match accountmanager::Manifest::load(path.as_path()) {
Ok(m) => {
manifest = m;
2021-03-27 13:39:26 +01:00
}
Err(e) => {
2021-03-27 15:35:52 +01:00
error!("Could not load manifest: {}", e);
return;
2021-03-27 13:39:26 +01:00
}
}
2021-03-27 15:35:52 +01:00
manifest.load_accounts();
2021-03-27 17:14:34 +01:00
let mut selected_accounts: Vec<SteamGuardAccount> = vec![];
if matches.is_present("all") {
// manifest.accounts.iter().map(|a| selected_accounts.push(a.b));
for account in manifest.accounts {
selected_accounts.push(account.clone());
}
} else {
for account in manifest.accounts {
if !matches.is_present("username") {
selected_accounts.push(account.clone());
break;
}
if matches.value_of("username").unwrap() == account.account_name {
selected_accounts.push(account.clone());
break;
}
}
}
debug!("selected accounts: {:?}", selected_accounts.iter().map(|a| a.account_name.clone()).collect::<Vec<String>>());
let server_time = steamapi::get_server_time();
for account in selected_accounts {
2021-03-27 15:35:52 +01:00
trace!("{:?}", account);
let code = account.generate_code(server_time);
println!("{}", code);
}
2021-03-22 02:21:29 +01:00
}