extern crate trezor;

use std::io;

fn device_selector() -> trezor::Trezor {
	let mut devices = trezor::find_devices(false).expect("error finding devices");

	if devices.len() == 0 {
		panic!("No devices connected");
	} else if devices.len() == 1 {
		devices.remove(0).connect().expect("connection error")
	} else {
		println!("Choose device:");
		for (i, dev) in devices.iter().enumerate() {
			println!("{}: {}", i + 1, dev);
		}
		println!("Enter device number: ");
		let mut inp = String::new();
		io::stdin().read_line(&mut inp).expect("stdin error");
		let idx: usize = inp[..inp.len() - 1].parse().expect("invalid number");
		if idx >= devices.len() {
			panic!("Index out of range");
		}
		devices.remove(idx).connect().unwrap()
	}
}

fn do_main() -> Result<(), trezor::Error> {
	// init with debugging

	let mut trezor = device_selector();
	trezor.init_device()?;
	let f = trezor.features().expect("no features");

	println!("Features:");
	println!("vendor: {}", f.get_vendor());
	println!(
		"version: {}.{}.{}",
		f.get_major_version(),
		f.get_minor_version(),
		f.get_patch_version()
	);
	println!("device id: {}", f.get_device_id());
	println!("label: {}", f.get_label());
	println!("is initialized: {}", f.get_initialized());
	println!("pin protection: {}", f.get_pin_protection());
	println!("passphrase protection: {}", f.get_passphrase_protection());
	//optional bool bootloader_mode = 5;          // is device in bootloader mode?
	//optional string language = 9;               // device language
	//optional bytes revision = 13;               // SCM revision of firmware
	//optional bytes bootloader_hash = 14;        // hash of the bootloader
	//optional bool imported = 15;                // was storage imported from an external source?
	//optional bool pin_cached = 16;              // is PIN already cached in session?
	//optional bool passphrase_cached = 17;       // is passphrase already cached in session?
	//optional bool firmware_present = 18;        // is valid firmware loaded?
	//optional bool needs_backup = 19;            // does storage need backup? (equals to Storage.needs_backup)
	//optional uint32 flags = 20;                 // device flags (equals to Storage.flags)
	//optional string model = 21;                 // device hardware model
	//optional uint32 fw_major = 22;              // reported firmware version if in bootloader mode
	//optional uint32 fw_minor = 23;              // reported firmware version if in bootloader mode
	//optional uint32 fw_patch = 24;              // reported firmware version if in bootloader mode
	//optional string fw_vendor = 25;             // reported firmware vendor if in bootloader mode
	//optional bytes fw_vendor_keys = 26;         // reported firmware vendor keys (their hash)
	//optional bool unfinished_backup = 27;       // report unfinished backup (equals to Storage.unfinished_backup)
	//optional bool no_backup = 28;               // report no backup (equals to Storage.no_backup)

	Ok(())
}

fn main() {
	do_main().unwrap()
}