diff --git a/build.rs b/build.rs index 1a807a2b..2a572575 100644 --- a/build.rs +++ b/build.rs @@ -4,31 +4,50 @@ use std::fs; fn main() { let board_dir = "src/board"; - let boards: Vec = fs::read_dir(board_dir).unwrap() + let boards: Vec = fs::read_dir(board_dir) + .expect("failed to read src/board directory") .filter_map(|e| { let p = e.ok()?.path(); if p.is_file() && p.extension()? == "rs" && p.file_stem()? != "mod" { return Some(p.file_stem()?.to_str()?.to_string()); } None - }).collect(); + }) + .collect(); let enabled: Vec<_> = std::env::vars() .filter(|(k, _)| k.starts_with("CARGO_FEATURE_")) .map(|(k, _)| k.replace("CARGO_FEATURE_", "").to_lowercase()) - .filter(|f| boards.contains(f)).collect(); + .filter(|f| boards.contains(f)) + .collect(); match enabled.len() { - 0 => panic!("\n\n{}\n", format!("No board selected! Found: {:?}", boards).red().bold()), + 0 => panic!( + "\n\n{}\n", + format!("No board selected! Found: {:?}", boards) + .red() + .bold() + ), 1 => println!("cargo:rustc-cfg=board_selected"), - _ => panic!("\n\n{}\n", format!("Conflict: Multiple boards enabled {:?}", enabled).yellow().bold()), + _ => panic!( + "\n\n{}\n", + format!("Conflict: Multiple boards enabled {:?}", enabled) + .yellow() + .bold() + ), } let mut env = Environment::new(); - let template = fs::read_to_string("src/board/mod.rs.j2").expect("Missing template .j2"); - env.add_template("mod", &template).unwrap(); - let rendered = env.get_template("mod").unwrap().render(context!(boards => boards)).unwrap(); - fs::write("src/board/mod.rs", rendered).unwrap(); + let template = + fs::read_to_string("src/board/mod.rs.j2").expect("missing src/board/mod.rs.j2"); + env.add_template("mod", &template) + .expect("failed to parse board template"); + let rendered = env + .get_template("mod") + .expect("template not found") + .render(context!(boards => boards)) + .expect("failed to render board module"); + fs::write("src/board/mod.rs", rendered).expect("failed to write src/board/mod.rs"); println!("cargo:rerun-if-changed=src/board/mod.rs.j2"); println!("cargo:rerun-if-changed={}", board_dir); diff --git a/src/main.rs b/src/main.rs index 8bfa90a3..ea91aa83 100644 --- a/src/main.rs +++ b/src/main.rs @@ -60,10 +60,16 @@ async fn run(spawner: Spawner) { let has_display = display.is_some(); - spawner.spawn(radio::radio_task(radio, &COMMANDS, &RESPONSES, &STATUS)).unwrap(); - spawner.spawn(usb::usb_task(usb, &COMMANDS, &RESPONSES, &DISPLAY_COMMANDS, has_display)).unwrap(); + spawner + .spawn(radio::radio_task(radio, &COMMANDS, &RESPONSES, &STATUS)) + .expect("spawn radio_task"); + spawner + .spawn(usb::usb_task(usb, &COMMANDS, &RESPONSES, &DISPLAY_COMMANDS, has_display)) + .expect("spawn usb_task"); if let Some(dp) = display { - spawner.spawn(display::display_task(dp, &STATUS, &DISPLAY_COMMANDS)).unwrap(); + spawner + .spawn(display::display_task(dp, &STATUS, &DISPLAY_COMMANDS)) + .expect("spawn display_task"); } }