From d36b35b7e923e084716c4df0f243d39e0c759048 Mon Sep 17 00:00:00 2001
From: osm0sis <osm0sis@outlook.com>
Date: Thu, 18 Jan 2024 14:27:55 -0400
Subject: [PATCH] Fix ROM injection app replacement on partitions other than
 /system

---
 module/customize.sh    | 11 ++++++++---
 module/post-fs-data.sh | 11 ++++++++---
 2 files changed, 16 insertions(+), 6 deletions(-)

diff --git a/module/customize.sh b/module/customize.sh
index 9495d95..c03ae0d 100644
--- a/module/customize.sh
+++ b/module/customize.sh
@@ -25,9 +25,14 @@ LIST=$MODPATH/example.app_replace.list
 [ -f "$MODPATH/custom.app_replace.list" ] && LIST=$MODPATH/custom.app_replace.list
 for APP in $(grep -v '^#' $LIST); do
     if [ -d "$APP" ]; then
-        HIDEDIR=$MODPATH/$APP
-        mkdir -p $HIDEDIR
-        touch $HIDEDIR/.replace
+        case $APP in
+            /system/*) HIDEDIR=$MODPATH/$APP;;
+            *) HIDEDIR=$MODPATH/system/$APP;;
+        esac
+        if [ ! -f "$HIDEDIR/.replace" ]; then
+            mkdir -p $HIDEDIR
+            touch $HIDEDIR/.replace
+        fi
         ui_print "! $(basename $APP) ROM app disabled"
     fi
 done
diff --git a/module/post-fs-data.sh b/module/post-fs-data.sh
index 4d97981..4e13f2f 100644
--- a/module/post-fs-data.sh
+++ b/module/post-fs-data.sh
@@ -16,9 +16,14 @@ LIST=$MODDIR/example.app_replace.list
 [ -f "$MODDIR/custom.app_replace.list" ] && LIST=$MODDIR/custom.app_replace.list
 for APP in $(grep -v '^#' $LIST); do
     if [ -d "$APP" ]; then
-        HIDEDIR=$MODDIR/$APP
-        mkdir -p $HIDEDIR
-        touch $HIDEDIR/.replace
+        case $APP in
+            /system/*) HIDEDIR=$MODDIR/$APP;;
+            *) HIDEDIR=$MODDIR/system/$APP;;
+        esac
+        if [ ! -f "$HIDEDIR/.replace" ]; then
+            mkdir -p $HIDEDIR
+            touch $HIDEDIR/.replace
+        fi
     fi
 done