46 lines
1 KiB
Rust
46 lines
1 KiB
Rust
|
use std::{
|
||
|
|
sync::{Arc, Mutex},
|
||
|
|
time::{SystemTime, UNIX_EPOCH},
|
||
|
|
};
|
||
|
|
|
||
|
|
use log::*;
|
||
|
|
use steamguard::{steamapi, SteamGuardAccount};
|
||
|
|
|
||
|
|
use crate::AccountManager;
|
||
|
|
|
||
|
|
use super::*;
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Parser)]
|
||
|
|
#[clap(about = "Generate 2FA codes")]
|
||
|
|
pub struct CodeCommand {
|
||
|
|
#[clap(
|
||
|
|
long,
|
||
|
|
help = "Assume the computer's time is correct. Don't ask Steam for the time when generating codes."
|
||
|
|
)]
|
||
|
|
pub offline: bool,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl AccountCommand for CodeCommand {
|
||
|
|
fn execute(
|
||
|
|
&self,
|
||
|
|
_manager: &mut AccountManager,
|
||
|
|
accounts: Vec<Arc<Mutex<SteamGuardAccount>>>,
|
||
|
|
) -> anyhow::Result<()> {
|
||
|
|
let server_time = if self.offline {
|
||
|
|
SystemTime::now().duration_since(UNIX_EPOCH)?.as_secs()
|
||
|
|
} else {
|
||
|
|
steamapi::get_server_time()?.server_time()
|
||
|
|
};
|
||
|
|
debug!("Time used to generate codes: {}", server_time);
|
||
|
|
|
||
|
|
for account in accounts {
|
||
|
|
let account = account.lock().unwrap();
|
||
|
|
info!("Generating code for {}", account.account_name);
|
||
|
|
trace!("{:?}", account);
|
||
|
|
let code = account.generate_code(server_time);
|
||
|
|
println!("{}", code);
|
||
|
|
}
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
}
|