check_uid function

1337kid · Jan 11, 2024 · e06e835 · e06e835
1 parent 91d141d
commit e06e835
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 12 deletions.
diff --git a/scripts/functions.sh b/scripts/functions.sh
@@ -1,5 +1,13 @@
 #!/bin/bash
 
+check_uid() {
+    if [[ $(id -u) != "0" ]]
+    then
+        printf "\033[91mCurrent user does not have root perms\n"
+        exit
+    fi
+}
+
 update_grub_func() {
     if [[ $(which dnf) != "" ]];then
         fedora_version=$(cat /etc/fedora-release | awk '{print $3}')

diff --git a/scripts/install.sh b/scripts/install.sh
@@ -49,12 +49,7 @@ cat <<EOF
 
 EOF
 
-if [[ $(id -u) != "0" ]]
-then
-    printf "\033[91mCurrent user does not have root perms\n"
-    exit
-fi
-
+check_uid
 get_path
 THEME_PATH=$GRUB_PATH"themes/noicegrub"
 printf "\033[1m\033[93m"

diff --git a/scripts/remove.sh b/scripts/remove.sh
@@ -2,15 +2,10 @@
 
 GRUB_PATH=''
 
-if [[ $(id -u) != "0" ]]
-then
-    printf "\033[91mCurrent user does not have root perms\n"
-    exit
-fi
-
 source ./scripts/functions.sh
 
 printf "\033[1m"
+check_uid
 get_path
 THEME_PATH=$GRUB_PATH"themes/noicegrub"
 printf "\033[92m[+] \033[94mRemoving NoiceGRUB theme\n\033[92m"