Update chroot scripts

root/usr/local/bin/arch-chroot

@@ -5,24 +5,63 @@ if [ "$EUID" -ne 0 ]; then
     exit
 fi
 
+# Take user on the chrooted machine as 1st argument,
+# this will be the logged user after chroot,
+# default to root if none provided
+if [ $# -ge 1 ]; then
+    USERNAME="$1"
+else
+    USERNAME="root"
+fi
+
+MOUNT_POINT="/mnt/arch"
+CLIENT_SCRIPT="/usr/local/bin/chroot-client"
+
 # Make sure the partition is mounted according to fstab
-mount /mnt/arch 2> /dev/null
+mount "$MOUNT_POINT" 2> /dev/null
 
 # Mount necessary directories for chroot to be possible
-mount --types proc /proc /mnt/arch/proc
-mount --rbind /sys /mnt/arch/sys
-mount --make-rslave /mnt/arch/sys
-mount --rbind /dev /mnt/arch/dev
-mount --make-rslave /mnt/arch/dev
+mount --types proc /proc "$MOUNT_POINT/proc"
+mount --rbind /sys "$MOUNT_POINT/sys"
+mount --make-rslave "$MOUNT_POINT/sys"
+mount --rbind /dev "$MOUNT_POINT/dev"
+mount --make-rslave "$MOUNT_POINT/dev"
 
-# Chroot with zsh shell
-chroot /mnt/arch /bin/zsh --login
+# Chroot with custom script, if aviable
+if [ -f "$MOUNT_POINT/$CLIENT_SCRIPT" ]; then
+    chroot "$MOUNT_POINT" "$CLIENT_SCRIPT" "$USERNAME"
+else
+    # If we didn't find the script in the chroot environment
+    # try to find it in this mahcine and copy it over
+    if [ -f "$CLIENT_SCRIPT" ]; then
+        echo "Client script not in chroot environment, copying"
+        mkdir -p "$MOUNT_POINT/$(dirname $CLIENT_SCRIPT)"
+        cp "$CLIENT_SCRIPT" "$MOUNT_POINT/$CLIENT_SCRIPTt"
+        chroot "$MOUNT_POINT" "$CLIENT_SCRIPT" "$USERNAME"
+    else
+        echo "Unable to run chroot client script, proceeding manually"
+        echo "You may need to run 'source /etc/profile' afterwards"
+
+        # Try to obtain chroot user's default shell from /etc/passwd
+        awkstring="BEGIN { FS=\":\" } /$USERNAME/ { print \$7 }"
+        shell="$(cat /etc/passwd | awk "$awkstring")"
+
+        if [ -n "$shell" ]; then
+            echo "Found user's shell, trying to chroot with it ($shell)"
+            chroot "$MOUNT_POINT" "$shell" --login
+        else
+            echo "Unable to find user in chroot's /etc/passwd, using root with bash shell"
+            chroot "$MOUNT_POINT" "/bin/bash" --login
+        fi
+
+    fi
+fi
 
 # Unmount recursively mounted directories
-umount -l /mnt/arch/dev{/shm,/pts,}
-umount -R /mnt/arch
+umount -l "$MOUNT_POINT/dev{/shm,/pts,}"
+umount -R "$MOUNT_POINT"
 
 # Remount partition accordingly to fstab
 # (the above umountings will unmount gentoo completely,
 # which means that remounting is necessary)
-mount /mnt/arch
+mount "$MOUNT_POINT"

root/usr/local/bin/chroot-client

@@ -0,0 +1,17 @@
+#!/bin/sh
+
+if [ $# -ge 1 ]; then
+    if ! id "$1" &>/dev/null; then
+        echo "No such user, give valid username"
+    else
+        USERNAME="$1"
+    fi
+else
+    USERNAME="root"
+fi
+
+source /etc/profile
+exec su "$USERNAME" --login
+exit
+
+

root/usr/local/bin/gentoo-chroot

@@ -1,23 +1,67 @@
 #!/bin/sh
 
+if [ "$EUID" -ne 0 ]; then
+    echo "Must be ran as root"
+    exit
+fi
+
+# Take user on the chrooted machine as 1st argument,
+# this will be the logged user after chroot,
+# default to root if none provided
+if [ $# -ge 1 ]; then
+    USERNAME="$1"
+else
+    USERNAME="root"
+fi
+
+MOUNT_POINT="/mnt/gentoo"
+CLIENT_SCRIPT="/usr/local/bin/chroot-client"
+
 # Make sure the partition is mounted according to fstab
-mount /mnt/gentoo 2>/dev/null
+mount "$MOUNT_POINT" 2> /dev/null
 
 # Mount necessary directories for chroot to be possible
-mount --types proc /proc /mnt/gentoo/proc
-mount --rbind /sys /mnt/gentoo/sys
-mount --make-rslave /mnt/gentoo/sys
-mount --rbind /dev /mnt/gentoo/dev
-mount --make-rslave /mnt/gentoo/dev
+mount --types proc /proc "$MOUNT_POINT/proc"
+mount --rbind /sys "$MOUNT_POINT/sys"
+mount --make-rslave "$MOUNT_POINT/sys"
+mount --rbind /dev "$MOUNT_POINT/dev"
+mount --make-rslave "$MOUNT_POINT/dev"
 
-# Chroot with zsh shell
-chroot /mnt/gentoo /bin/zsh --login
+# Chroot with custom script, if aviable
+if [ -f "$MOUNT_POINT/$CLIENT_SCRIPT" ]; then
+    chroot "$MOUNT_POINT" "$CLIENT_SCRIPT" "$USERNAME"
+else
+    # If we didn't find the script in the chroot environment
+    # try to find it in this mahcine and copy it over
+    if [ -f "$CLIENT_SCRIPT" ]; then
+        echo "Client script not in chroot environment, copying"
+        mkdir -p "$MOUNT_POINT/$(dirname $CLIENT_SCRIPT)"
+        cp "$CLIENT_SCRIPT" "$MOUNT_POINT/$CLIENT_SCRIPTt"
+        chroot "$MOUNT_POINT" "$CLIENT_SCRIPT" "$USERNAME"
+    else
+        echo "Unable to run chroot client script, proceeding manually"
+        echo "You may need to run 'source /etc/profile' afterwards"
+
+        # Try to obtain chroot user's default shell from /etc/passwd
+        awkstring="BEGIN { FS=\":\" } /$USERNAME/ { print \$7 }"
+        shell="$(cat /etc/passwd | awk "$awkstring")"
+
+        if [ -n "$shell" ]; then
+            echo "Found user's shell, trying to chroot with it ($shell)"
+            chroot "$MOUNT_POINT" "$shell" --login
+        else
+            echo "Unable to find user in chroot's /etc/passwd, using root with bash shell"
+            chroot "$MOUNT_POINT" "/bin/bash" --login
+        fi
+
+    fi
+fi
 
 # Unmount recursively mounted directories
-umount -l /mnt/gentoo/dev{/shm,/pts,}
-umount -R /mnt/gentoo
+umount -l "$MOUNT_POINT/dev{/shm,/pts,}"
+umount -R "$MOUNT_POINT"
 
 # Remount partition accordingly to fstab
 # (the above umountings will unmount gentoo completely,
 # which means that remounting is necessary)
-mount /mnt/gentoo
+mount "$MOUNT_POINT"