diff --git a/switch-desktop/src/main.rs b/switch-desktop/src/main.rs index d024343..3c625cf 100644 --- a/switch-desktop/src/main.rs +++ b/switch-desktop/src/main.rs @@ -146,25 +146,20 @@ fn main() { #[cfg(any(target_os = "linux", target_os = "macos"))] fn main() { + if sudo::RunningAs::Root != sudo::check() { + println!( + "{}", + style("需要使用root权限执行(Need to execute with root permission)...").red() + ); + sudo::escalate_if_needed().unwrap(); + } let args = BaseArgs::parse(); - match args.command { - Commands::Start(_) | Commands::Stop => { - if sudo::RunningAs::Root != sudo::check() { - println!( - "{}", - style("需要使用root权限执行(Need to execute with root permission)...").red() - ); - sudo::escalate_if_needed().unwrap(); - } - let home = dirs::home_dir().unwrap().join(".switch_desktop"); - config::set_home(home); - if let Commands::Start(start_args) = &args.command { - if start_args.log { - let _ = log_init(); - } - } + let home = dirs::home_dir().unwrap().join(".switch_desktop"); + config::set_home(home); + if let Commands::Start(start_args) = &args.command { + if start_args.log { + let _ = log_init(); } - _ => {} } unix::main0(args); }