From 924bd4262f715764cc97437217f057023713757c Mon Sep 17 00:00:00 2001
From: AngeD <ange.duhayon@epitech.eu>
Date: Sun, 26 Mar 2023 12:58:30 +0200
Subject: [PATCH] feat: dkprune -r option

---
 bin/dkprune | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/bin/dkprune b/bin/dkprune
index 0b358a2..e9a78c0 100755
--- a/bin/dkprune
+++ b/bin/dkprune
@@ -14,6 +14,7 @@ Options:
     -I  Remove unused images
     -n  Remove unused networks
     -a  All of the above, equivalent of -CvIn
+    -r  Run as root
 EOF
     exit 0
 }
@@ -30,11 +31,7 @@ function add_opt() {
     echo "$RESULT"
 }
 
-RUNTIME="$(basename "$(command -v podman || command -v docker)")"
-
-if ! $RUNTIME container ls > /dev/null 2>&1 && [ "$EUID" != 0 ]; then
-    exec sudo -- "$0" "$@"
-fi
+RUNTIME="$(command -v podman || command -v docker)"
 
 c=1
 # C=2
@@ -45,7 +42,7 @@ n=32
 # a="$((c + C + v + I + n))"
 
 CLEAN=0
-while getopts hcCviIna o; do
+while getopts hcCviInar o; do
     case "$o" in
         h)
             usage
@@ -75,6 +72,9 @@ while getopts hcCviIna o; do
             CON_OPT='-f'
             IMG_OPT='-a'
             ;;
+        r)
+            exec sudo -- "$0" "$(sed 's/r//' <<< "${@:1}")"
+            ;;
         *)
             exit 1
             ;;