Files
RedBear-OS/usbctl/src/main.rs
T
2020-02-15 09:56:32 +01:00

57 lines
2.1 KiB
Rust

use clap::{App, Arg};
use xhcid_interface::XhciClientHandle;
fn main() {
let matches = App::new("usbctl")
.arg(
Arg::with_name("SCHEME")
.takes_value(true)
.required(true)
.long("scheme")
.short("s"),
)
.subcommand(
App::new("port")
.arg(Arg::with_name("PORT").takes_value(true).required(true))
.subcommand(App::new("status"))
.subcommand(
App::new("endpoint")
.arg(
Arg::with_name("ENDPOINT_NUM")
.takes_value(true)
.required(true),
)
.subcommand(App::new("status")),
),
)
.get_matches();
let scheme = matches.value_of("SCHEME").expect("no scheme");
if let Some(port_scmd_matches) = matches.subcommand_matches("port") {
let port = port_scmd_matches
.value_of("PORT")
.expect("invalid utf-8 for PORT argument")
.parse::<usize>()
.expect("expected PORT to be an integer");
let handle = XhciClientHandle::new(scheme.to_owned(), port);
if let Some(_status_scmd_matches) = port_scmd_matches.subcommand_matches("status") {
let state = handle.port_state().expect("Failed to get port state");
println!("{}", state.as_str());
} else if let Some(endp_scmd_matches) = port_scmd_matches.subcommand_matches("endpoint") {
let endp_num = endp_scmd_matches
.value_of("ENDPOINT_NUM")
.expect("no valid ENDPOINT_NUM")
.parse::<u8>()
.expect("expected ENDPOINT_NUM to be an 8-bit integer");
let mut endp_handle = handle
.open_endpoint(endp_num)
.expect("Failed to open endpoint");
let state = endp_handle.status().expect("Failed to get endpoint state");
println!("{}", state.as_str());
}
}
}