steamguard-cli/src/accountmanager.rs

use std::fs::File;
use std::io::BufReader;
use std::path::Path;
use serde::{Serialize, Deserialize};
use std::error::Error;
use steamguard::SteamGuardAccount;
use log::*;

#[derive(Debug, Serialize, Deserialize)]
pub struct Manifest {
	pub encrypted: bool,
	pub entries: Vec<ManifestEntry>,
	pub first_run: bool,
	pub periodic_checking: bool,
	pub periodic_checking_interval: i32,
	pub periodic_checking_checkall: bool,
	pub auto_confirm_market_transactions: bool,
	pub auto_confirm_trades: bool,

	#[serde(skip)]
	pub accounts: Vec<SteamGuardAccount>,
	#[serde(skip)]
	folder: String, // I wanted to use a Path here, but it was too hard to make it work...
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ManifestEntry {
	pub encryption_iv: Option<String>,
	pub encryption_salt: Option<String>,
	pub filename: String,
	#[serde(rename = "steamid")]
	pub steam_id: u64,
}

impl Manifest {
	pub fn load(path: &Path) -> Result<Manifest, Box<dyn Error>> {
		debug!("loading manifest: {:?}", &path);
		match File::open(path) {
			Ok(file) => {
				let reader = BufReader::new(file);
				match serde_json::from_reader(reader) {
					Ok(m) => {
						let mut manifest: Manifest = m;
						manifest.folder = String::from(path.parent().unwrap().to_str().unwrap());
						return Ok(manifest);
					}
					Err(e) => {
						return Err(Box::new(e));
					}
				}
			}
			Err(e) => {
				return Err(Box::new(e));
			}
		}
	}

	pub fn load_accounts(&mut self) {
		for entry in &self.entries {
			let path = Path::new(&self.folder).join(&entry.filename);
			debug!("loading account: {:?}", path);
			match File::open(path) {
				Ok(f) => {
					let reader = BufReader::new(f);
					match serde_json::from_reader(reader) {
						Ok(a) => {
							let account: SteamGuardAccount = a;
							self.accounts.push(account);
						}
						Err(e) => {
							error!("invalid json: {}", e)
						}
					}
				}
				Err(e) => {
					error!("unable to open account: {}", e)
				}
			}
		}
	}
}