simono41/steamguard-cli
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

move import subcommand impl

Browse source
This commit is contained in:
Carson McManus 2022-06-19 11:43:37 -04:00
parent 37539f7679
commit 73281ef069
2 changed files with 22 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
src/accountmanager.rs
View file

@ -180,8 +180,8 @@ impl Manifest {
.insert(account.account_name.clone(), Arc::new(Mutex::new(account))); .insert(account.account_name.clone(), Arc::new(Mutex::new(account)));
} }
pub fn import_account(&mut self, import_path: String) -> anyhow::Result<()> { pub fn import_account(&mut self, import_path: &String) -> anyhow::Result<()> {
let path = Path::new(&import_path); let path = Path::new(import_path);
ensure!(path.exists(), "{} does not exist.", import_path); ensure!(path.exists(), "{} does not exist.", import_path);
ensure!(path.is_file(), "{} is not a file.", import_path); ensure!(path.is_file(), "{} is not a file.", import_path);
@ -576,7 +576,7 @@ mod tests {
let mut loaded_manifest = Manifest::new(manifest_path.as_path()); let mut loaded_manifest = Manifest::new(manifest_path.as_path());
assert!(matches!( assert!(matches!(
loaded_manifest.import_account( loaded_manifest.import_account(
tmp_dir &tmp_dir
.path() .path()
.join("asdf1234.maFile") .join("asdf1234.maFile")
.into_os_string() .into_os_string()

36
src/main.rs
View file

@ -229,7 +229,9 @@ fn run() -> anyhow::Result<()> {
Some(cli::Subcommands::Setup(args)) => { Some(cli::Subcommands::Setup(args)) => {
return do_subcmd_setup(args, &mut manifest); return do_subcmd_setup(args, &mut manifest);
}, },
Some(cli::Subcommands::Import(args)) => {todo!()}, Some(cli::Subcommands::Import(args)) => {
return do_subcmd_import(args, &mut manifest);
},
Some(cli::Subcommands::Encrypt(args)) => { Some(cli::Subcommands::Encrypt(args)) => {
return do_subcmd_encrypt(args, &mut manifest); return do_subcmd_encrypt(args, &mut manifest);
}, },
@ -239,22 +241,6 @@ fn run() -> anyhow::Result<()> {
_ => {}, _ => {},
} }
if let Some(import_matches) = matches.subcommand_matches("import") {
for file_path in import_matches.values_of("files").unwrap() {
match manifest.import_account(file_path.into()) {
Ok(_) => {
info!("Imported account: {}", file_path);
}
Err(err) => {
bail!("Failed to import account: {} {}", file_path, err);
}
}
}
manifest.save()?;
return Ok(());
}
let mut selected_accounts: Vec<Arc<Mutex<SteamGuardAccount>>>; let mut selected_accounts: Vec<Arc<Mutex<SteamGuardAccount>>>;
loop { loop {
match get_selected_accounts(&matches, &mut manifest) { match get_selected_accounts(&matches, &mut manifest) {
@ -682,6 +668,22 @@ fn do_subcmd_setup(args: cli::ArgsSetup, manifest: &mut accountmanager::Manifest
return Ok(()); return Ok(());
} }
fn do_subcmd_import(args: cli::ArgsImport, manifest: &mut accountmanager::Manifest) -> anyhow::Result<()> {
for file_path in args.files {
match manifest.import_account(&file_path) {
Ok(_) => {
info!("Imported account: {}", &file_path);
}
Err(err) => {
bail!("Failed to import account: {} {}", &file_path, err);
}
}
}
manifest.save()?;
return Ok(());
}
fn do_subcmd_encrypt(args: cli::ArgsEncrypt, manifest: &mut accountmanager::Manifest) -> anyhow::Result<()> { fn do_subcmd_encrypt(args: cli::ArgsEncrypt, manifest: &mut accountmanager::Manifest) -> anyhow::Result<()> {
if !manifest.has_passkey() { if !manifest.has_passkey() {
let mut passkey; let mut passkey;