From 161afbc28529a0a45797010610626c5128ac89d6 Mon Sep 17 00:00:00 2001
From: oddlama <oddlama@oddlama.org>
Date: Thu, 2 Jan 2020 23:56:36 +0100
Subject: [PATCH] Added chroot indicator

---
 scripts/functions.sh   | 10 ++++++++--
 scripts/main_chroot.sh |  3 ---
 2 files changed, 8 insertions(+), 5 deletions(-)

diff --git a/scripts/functions.sh b/scripts/functions.sh
index 7985f32..3353fac 100644
--- a/scripts/functions.sh
+++ b/scripts/functions.sh
@@ -271,7 +271,13 @@ gentoo_umount() {
 }
 
 gentoo_chroot() {
-	[[ $# -gt 0 ]] || die "Missing command argument"
+	if [[ $# -eq 0 ]]; then
+		cat > "$TMP_DIR/.bashrc" <<EOF
+# Set the PS1 to a recognizable value
+export PS1="(chroot) \$PS1"
+EOF
+		gentoo_chroot /bin/bash --init-file "$TMP_DIR/.bashrc"
+	fi
 
 	[[ $EXECUTED_IN_CHROOT != true ]] \
 		|| die "Already in chroot"
@@ -301,6 +307,6 @@ gentoo_chroot() {
 	disable_logging
 	EXECUTED_IN_CHROOT=true \
 		TMP_DIR=$TMP_DIR \
-		exec chroot "$ROOT_MOUNTPOINT" "$GENTOO_BOOTSTRAP_DIR/scripts/main_chroot.sh" "$@" \
+		exec chroot -- "$ROOT_MOUNTPOINT" "$GENTOO_BOOTSTRAP_DIR/scripts/main_chroot.sh" "$@" \
 		|| die "Failed to chroot into '$ROOT_MOUNTPOINT'"
 }
diff --git a/scripts/main_chroot.sh b/scripts/main_chroot.sh
index a4deb4c..2a5c733 100755
--- a/scripts/main_chroot.sh
+++ b/scripts/main_chroot.sh
@@ -14,8 +14,5 @@ export NPROC_ONE="$(($NPROC + 1))"
 export MAKEFLAGS="-j$NPROC"
 export EMERGE_DEFAULT_OPTS="--jobs=$NPROC_ONE --load-average=$NPROC"
 
-# Set the PS1 to a recognizable value
-export PS1="(chroot) $PS1"
-
 # Execute the requested command
 exec "$@"