From 9515d2c952a0c82f18361cf8615b51272dad1908 Mon Sep 17 00:00:00 2001
From: PedoDeveloper <pedodev>
Date: Sat, 9 Nov 2024 06:59:06 +0000
Subject: [PATCH] Added cleanup function

---
 modules/archive.sh | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/modules/archive.sh b/modules/archive.sh
index 593b2ee..17ed77c 100755
--- a/modules/archive.sh
+++ b/modules/archive.sh
@@ -6,6 +6,11 @@ input_file="${1}"
 archive_pwd="${2}"
 size="$3"
 
+cleanup() {
+	[[ -n "$archive" ]] && rm -f "$archive*"
+	echo "Cleaned up all $archive files" >&2
+}
+
 check_dependencies() {
 	if ! mkdir -p "$archive_dir"; then
 		echo "ERROR: Unable to create archive directory"
@@ -53,6 +58,8 @@ create_archive() {
 	7z a -mhe=on "-p${archive_pwd}" "$volumes" "$archive" "${input_file}" >&2
 }
 
+trap cleanup 1 2 3 6
+
 check_dependencies
 check_input
 set_volumes