Add root check

ChrisTitusTech · Oct 13, 2024 · c997a3d · c997a3d
1 parent 79eb752
commit c997a3d
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 0 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/tui/Cargo.toml b/tui/Cargo.toml
@@ -30,6 +30,7 @@ tree-sitter-highlight = "0.24.2"
 tree-sitter-bash = "0.23.1"
 anstyle = "1.0.8"
 ansi-to-tui = "6.0.0"
+sudo = "0.6.0"
 zips = "0.1.7"
 
 [build-dependencies]

diff --git a/tui/src/main.rs b/tui/src/main.rs
@@ -36,6 +36,12 @@ struct Args {
 }
 
 fn main() -> io::Result<()> {
+    if sudo::check() != sudo::RunningAs::User {
+        eprintln!("Error: This program is not intended to be run with elevated privileges.");
+        eprintln!("Please run this program as a regular user to ensure proper security and functionality.");
+        std::process::exit(1);
+    }
+
     let args = Args::parse();
 
     let mut state = AppState::new(args.theme, args.override_validation);