From 42c249fb4df05dbb6fee75005d813e1a4becd37c Mon Sep 17 00:00:00 2001
From: nnyyxxxx <ethanm112806@gmail.com>
Date: Thu, 29 Aug 2024 19:55:25 -0400
Subject: [PATCH] Change remaining sudos

---
 src/commands/utils/numlock.sh | 18 +++++++++++-------
 1 file changed, 11 insertions(+), 7 deletions(-)

diff --git a/src/commands/utils/numlock.sh b/src/commands/utils/numlock.sh
index b1ac782a..2b23c3f1 100755
--- a/src/commands/utils/numlock.sh
+++ b/src/commands/utils/numlock.sh
@@ -1,12 +1,15 @@
 #!/bin/sh -e
 
+. ../common-script.sh
+
 # setleds can be used in all distros
 # This method works by calling a script using systemd service
 
 # Create a script to toggle numlock
+
 create_file() {
   echo "Creating script..."
-  sudo tee "/usr/local/bin/numlock" >/dev/null <<'EOF'
+  $ESCALATION_TOOL tee "/usr/local/bin/numlock" >/dev/null <<'EOF'
 #!/bin/bash
 
 for tty in /dev/tty{1..6}
@@ -15,13 +18,13 @@ do
 done
 EOF
 
-  sudo chmod +x /usr/local/bin/numlock
+  $ESCALATION_TOOL chmod +x /usr/local/bin/numlock
 }
 
 # Create a systemd service to run the script on boot
 create_service() {
   echo "Creating service..."
-  sudo tee "/etc/systemd/system/numlock.service" >/dev/null <<'EOF'
+  $ESCALATION_TOOL tee "/etc/systemd/system/numlock.service" >/dev/null <<'EOF'
 [Unit]
 Description=numlock
         
@@ -35,7 +38,7 @@ WantedBy=multi-user.target
 EOF
 }
 
-main() {
+numlockSetup() {
   # Check if the script and service files exists
   if [ ! -f "/usr/local/bin/numlock" ]; then
     create_file
@@ -48,13 +51,14 @@ main() {
   printf "Do you want to enable Numlock on boot? (y/n): "
   read -r confirm
   if [ "$confirm" = "y" ] || [ "$confirm" = "Y" ]; then
-    sudo systemctl enable numlock.service --quiet
+    $ESCALATION_TOOL systemctl enable numlock.service --quiet
     echo "Numlock will be enabled on boot"
   else
-    sudo systemctl disable numlock.service --quiet
+    $ESCALATION_TOOL systemctl disable numlock.service --quiet
     echo "Numlock will not be enabled on boot"
 
   fi
 }
 
-main
+checkEscalationTool
+numlockSetup