From 0cfc9d0f4acac43ce20b445ac883b7cf0ca1dc79 Mon Sep 17 00:00:00 2001 From: littleblack111 Date: Sun, 21 Sep 2025 23:08:48 +0800 Subject: format --- src/tasks.rs | 124 ++++++++++++++++++++++++++++++++++++++++------------------- 1 file changed, 85 insertions(+), 39 deletions(-) (limited to 'src/tasks.rs') diff --git a/src/tasks.rs b/src/tasks.rs index 0e196b2..a4cab8b 100644 --- a/src/tasks.rs +++ b/src/tasks.rs @@ -14,13 +14,30 @@ pub enum Name { impl Display for Name { fn fmt(&self, f: &mut core::fmt::Formatter<'_>) -> core::fmt::Result { - write!(f, "{:?}", self) + write!( + f, + "{:?}", + self + ) } } -pub const TASKS: &[Task] = &[Task { name: Name::Nop, exec: no_op }, Task { name: Name::Exec, exec: exec }, NETWORK_T]; +pub const TASKS: &[Task] = &[ + Task { + name: Name::Nop, + exec: no_op, + }, + Task { + name: Name::Exec, + exec: exec, + }, + NETWORK_T, +]; -pub const NETWORK_T: Task = Task { name: Name::Network, exec: network }; +pub const NETWORK_T: Task = Task { + name: Name::Network, + exec: network, +}; fn no_op(_: &[&str]) -> Result { Ok("".to_string()) @@ -30,9 +47,14 @@ fn exec(cmds: &[&str]) -> Result { use std::process::Command; // TODO: fmt and add exit code, and add support for both stdout and stderr, and // support for no output - match Command::new(cmds[0]).args(&cmds[1..]).output() { + match Command::new(cmds[0]) + .args(&cmds[1..]) + .output() + { Ok(o) => { - if o.status.success() { + if o.status + .success() + { Ok(String::from_utf8_lossy(&o.stdout).to_string()) } else { Err(String::from_utf8_lossy(&o.stderr).to_string()) @@ -44,7 +66,9 @@ fn exec(cmds: &[&str]) -> Result { fn network(switch: &[&str]) -> Result { use std::process::{Command, Stdio}; - let switch: bool = switch[0].parse::().map_err(|e| e.to_string())?; + let switch: bool = switch[0] + .parse::() + .map_err(|e| e.to_string())?; // TODO: windows and linux @@ -52,52 +76,70 @@ fn network(switch: &[&str]) -> Result { #[cfg(target_os = "macos")] { use std::io::Read; - match Command::new("networksetup").arg("-listallhardwareports").stdout(Stdio::piped()).spawn() { + match Command::new("networksetup") + .arg("-listallhardwareports") + .stdout(Stdio::piped()) + .spawn() + { Ok(devs) => { match Command::new("awk") .arg(r#"/Wi-Fi|AirPort/{getline; print $2}"#) - .stdin(Stdio::from(match devs.stdout { - Some(s) => s, - None => match devs.stderr { - Some(mut e) => { - let mut reason = String::new(); - if let Err(e) = e.read_to_string(&mut reason) { - return Err(format!("networksetup failed to list hardware: {e}")); - } - return Err(format!("networksetup failed to list hardware: {reason}")); - } - None => Err("networksetup failed to list hardware".to_string())?, - }, - })) + .stdin( + Stdio::from( + match devs.stdout { + Some(s) => s, + None => match devs.stderr { + Some(mut e) => { + let mut reason = String::new(); + if let Err(e) = e.read_to_string(&mut reason) { + return Err(format!("networksetup failed to list hardware: {e}")); + } + return Err(format!("networksetup failed to list hardware: {reason}")); + } + None => Err("networksetup failed to list hardware".to_string())?, + }, + }, + ), + ) .stdout(Stdio::piped()) .spawn() { Ok(dev) => match Command::new("networksetup") .arg("-setairportpower") - .arg(match dev.stdout { - Some(mut o) => { - let mut output = String::new(); - if let Err(e) = o.read_to_string(&mut output) { - return Err(format!("awk failed to get device from networksetup list hardware: {e}")); - }; - output - } - None => match dev.stderr { - Some(mut e) => { - let mut reason = String::new(); - if let Err(e) = e.read_to_string(&mut reason) { + .arg( + match dev.stdout { + Some(mut o) => { + let mut output = String::new(); + if let Err(e) = o.read_to_string(&mut output) { return Err(format!("awk failed to get device from networksetup list hardware: {e}")); - } - return Err(format!("awk failed to get device from networksetup list hardware: {reason}")); + }; + output } - None => Err("awk failed to get device from networksetup list hardware".to_string())?, + None => match dev.stderr { + Some(mut e) => { + let mut reason = String::new(); + if let Err(e) = e.read_to_string(&mut reason) { + return Err(format!("awk failed to get device from networksetup list hardware: {e}")); + } + return Err(format!("awk failed to get device from networksetup list hardware: {reason}")); + } + None => Err("awk failed to get device from networksetup list hardware".to_string())?, + }, + }, + ) + .arg( + if switch { + "on" + } else { + "off" }, - }) - .arg(if switch { "on" } else { "off" }) + ) .output() { Ok(o) => { - if o.status.success() { + if o.status + .success() + { Ok(String::from_utf8_lossy(&o.stdout).to_string()) } else { Err(String::from_utf8_lossy(&o.stderr).to_string()) @@ -126,7 +168,11 @@ impl TryFrom<&str> for &'static Task { fn try_from(value: &str) -> Result { for task in TASKS { - if task.name.to_string() == value { + if task + .name + .to_string() + == value + { return Ok(task); } } -- cgit v1.2.3