#!/usr/bin/env bash
# Copyright (C) 2018  Luke Shumaker
# Copyright (C) 2023  Umorpha Systems
# SPDX-License-Identifier: AGPL-3.0-or-later
declare -r NAME=osi-mount
declare -r VERSION=20231023

set -euE
source "$(dirname -- "${BASH_SOURCE[0]}")/lib/osi.sh"

main() {
	local arg_orig=("$@")
	local arg_mode=outside
	local arg_user=
	local arg_rwdirs=()
	local arg_mountopt=
	local args
	if ! args="$(getopt -n "${0##*/}" -o o:hV -l inside,user,root,rwdir:,options:,help,version -- "$@")"; then
		arg_mode=error
	else
		eval "set -- $args"
		while true; do
			case "$1" in
				--inside) shift; arg_mode=inside;;
				--user|--root)
					if [[ -n $arg_user ]]; then
						error 0 "Multiple --user/--root flags given"
						arg_mode=error
					fi
					arg_user=${1#--}
					shift
					;;
				--rwdir) shift; arg_rwdirs+=("$1"); shift;;
				-o|--options) shift; arg_mountopt="$1"; shift;;

				-V|--version) shift; arg_mode=version;;
				-h|--help) shift; arg_mode=usage;;
				--) shift; break;;
				*) error $EXIT_FAILURE 'Internal error.  The programmer writing this tool screwed up.';;
			esac
		done
		case "$arg_mode" in
			outside)
				case "$#" in
					0) error 0 "Must specify an image, mountpoint, and a command"; arg_mode=error;;
					1) error 0 "Must specify a mountpoint and a command"; arg_mode=error;;
					2) error 0 "Must specify a command"; arg_mode=error;;
				esac
				if [[ -z $arg_user ]]; then
					error 0 "Must specify either --root or --user"
					arg_mode=error
				fi
				;;
		esac
		case "$arg_mode" in
			outside|inside)
				arg_device=$(realpath -- "$1")
				arg_mountpoint=$2
				arg_cmd=("${@:3}")
				;;
		esac
	fi

	case "$arg_mode" in
		error)
			print "Try '%q --help' for more information" "${0##*/}" >&2
			return $EXIT_FAILURE
			;;
		version)
			print "%s (osi-tools) %s" "$NAME" "$VERSION"
			return $EXIT_SUCCESS
			;;
		usage)
			print 'Usage: sudo %s [OPTIONS] --<user|root> FILENAME.img MOUNTPOINT COMMAND...' "${0##*/}"
			print 'Operating System Image: Mount'
			echo
			#     000000000011111111112222222222333333333344444444445555555555666666666677777777778
			#     012345678901234567890123456789012345678901234567890123456789012345678901234567890
			print 'Run COMMAND with FILENAME.img mounted at MOUNTPOINT.'
			echo
			print 'OPTIONS:'
			# --inside is internal-only; undocumented
			print '      --user          mount the image RO, run COMMAND as SUDO_USER'
			print '      --root          mount the image RW, run COMMAND as root, protect the rest of the system'
			print '      --rwdir=DIR     keep DIR read+write, even when --root'
			print '  -o, --options LIST  comma-separated list of mount options'
			echo
			print '  -h, --help     display this help'
			print '  -V, --version  output version information'
			return $EXIT_SUCCESS
			;;

		outside)
			needs_root
			needs_sudo
			if out="$(losetup --associated "$arg_device")" && [[ -n $out  ]]; then
				error 0 "Seems to already be mounted somewhere: %s" "$arg_device"
				printf '%s\n' "$out"
				exit $EXIT_FAILURE
			fi

			# I feel guilty pulling in the hugeness of systemd-run
			# just for `--property=ProtectSystem=strict`, but I
			# don't want to implement my own protection system.
			flags=(
				--pty --same-dir --wait --collect --quiet
				--setenv=SUDO_UID="$SUDO_UID"
				--property=PrivateMounts=true
			)
			case "$arg_user" in
				user) :;;
				root)
					flags+=(
						--property=ProtectSystem=strict
						--property=ReadWritePaths="$arg_device"
						"${arg_rwdirs[@]/#/--property=ReadWritePaths=}"
					)
					;;
			esac
			systemd-run "${flags[@]}" -- "${BASH_SOURCE[0]}" --inside "${arg_orig[@]}"

			if out="$(losetup --associated "$arg_device")" && [[ -n $out  ]]; then
				error 0 "umount'ed, but file is still associated with a loop device: %s" "$arg_device"
				printf '%s\n' "$out"
				exit $EXIT_FAILURE
			fi
			;;
		inside)
			needs_root
			needs_sudo

			case "$arg_user" in
				user) arg_mountopt+="${arg_mountopt:+,}ro";;
				root) arg_mountopt+="${arg_mountopt:+,}rw,discard";;
			esac

			cmd=(mount --no-mtab -o "$arg_mountopt" -- "$arg_device" "$arg_mountpoint")
			r=0
			out=$("${cmd[@]}" 2>&1) || r=$?
			if (( r != 0 )); then
				error $r '%s: %s' "${cmd[*]@Q}" "$out"
			fi

			case "$arg_user" in
				user) sudo -u "#${SUDO_UID}" -- "${arg_cmd[@]}";;
				root) command -- "${arg_cmd[@]}";;
			esac
			;;

		*) error $EXIT_FAILURE 'Internal error.  The programmer writing this tool screwed up.';;
	esac
}

main "$@"