From 25e9c523d395db77dcaa2d32ffbf9f6edf12caf4 Mon Sep 17 00:00:00 2001
From: lubeilin <1791778603@qq.com>
Date: Fri, 12 May 2023 16:54:44 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=B7=B2=E7=9F=A5=E9=97=AE?=
 =?UTF-8?q?=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 switch-desktop/src/main.rs | 29 ++++++++++++-----------------
 1 file changed, 12 insertions(+), 17 deletions(-)

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);
 }