From 6c4d6008e39504282eaff5f6494ff32208928ce5 Mon Sep 17 00:00:00 2001
From: timoxa0 <tima012007@gmail.com>
Date: Sat, 2 Nov 2024 23:12:32 +0500
Subject: [PATCH] mklni: Add cleanup args

---
 mklni | 39 +++++++++++++++++++++++++++++++++++++++
 1 file changed, 39 insertions(+)

diff --git a/mklni b/mklni
index 2ce550e..a0553c9 100755
--- a/mklni
+++ b/mklni
@@ -70,6 +70,10 @@ build)
                 chroot_into
                 break 2 ;;
 
+            -r | -c | --cleanraw | --cleanup)
+                run_makelni "$opt"
+                break ;;
+
             -h | -help | --help)
                 usage ;;
         
@@ -96,6 +100,12 @@ usage() {
     # Make image
     $0 build path/to/lnibuild
 
+    # Clean raw images
+    $0 -r or --cleanraw
+
+    # Clean build chroots
+    $0 -c or --cleanup
+
     # Chroot into
     $0 chroot
 EOF
@@ -328,6 +338,35 @@ build()
     exit 0
 }
 
+run_makelni()
+{
+    [ ! -d "$CHROOTDIR" ] && {
+        log "Chroot not installed" error
+        exit 1
+    }
+
+    log "Running makelni $1"
+
+    prepare_chroot || {
+        log "Failed to setup chroot" error
+        exit 1
+    }
+    
+    chroot "$CHROOTDIR" /bin/bash -c "cd /makelni && ./makelni $1"
+    local exitcode=$?
+
+    detach_chroot || {
+        log "Failed to detach chroot" error
+        exit 1
+    }
+
+    [ "$exitcode" -eq "0" ] || {
+        log "Failed to build image" error
+    }
+
+    return "$exitcode"
+}
+
 if [ "$(id -u)" != "0" ]; then
     log "$0 must be run as root"
     exit 3