#!/bin/bash

_sudo(){
  command -v sudo > /dev/null && echo sudo || true
}

$(_sudo) mkdir -p /etc/sudoers.d

if [ "$1" != "--force" ]; then
    if [ -f /usr/sbin/start-systemd-namespace ]; then
        echo "It appears you have already installed the systemd hack."
        echo "To forcibly reinstall, run this script with the \`--force\` parameter."
        exit
    fi
    if [ -z "$WSL_DISTRO_NAME" ]; then
        echo "It appears that you are not running on WSL."
        echo "To forcibly install anyway, run this script with the \`--force\` parameter."
        exit
    fi
fi

self_dir="$(dirname $0)"

function interop_prefix {

	win_location="/mnt/"
	if [ -f /etc/wsl.conf ]; then
		tmp="$(awk -F '=' '/root/ {print $2}' /etc/wsl.conf | awk '{$1=$1;print}')"
		[ "$tmp" == "" ] || win_location="$tmp"
		unset tmp
	fi
	echo "$win_location"

	unset win_location
}

function sysdrive_prefix {
	win_location="$(interop_prefix)"
	hard_reset=0
	for pt in $(ls "$win_location"); do
		if [ $(echo "$pt" | wc -l) -eq 1 ]; then
			if [ -d "$win_location$pt/Windows/System32" ]; then
				hard_reset=1
				win_location="$pt"
				break
			fi
		fi
	done

	if [ $hard_reset -eq 0 ]; then
		win_location="c"
	fi

	echo "$win_location"

	unset win_location
	unset hard_reset
}

$(_sudo) hwclock -s
$(_sudo) apt-get update && $(_sudo) apt-get install -yqq daemonize dbus-user-session fontconfig

$(_sudo) cp "$self_dir/start-systemd-namespace" /usr/sbin/start-systemd-namespace
$(_sudo) cp "$self_dir/enter-systemd-namespace" /usr/sbin/enter-systemd-namespace
$(_sudo) chmod +x /usr/sbin/enter-systemd-namespace

$(_sudo) tee /etc/sudoers.d/systemd-namespace >/dev/null <<EOF
Defaults        env_keep += WSLPATH
Defaults        env_keep += WSLENV
Defaults        env_keep += WSL_INTEROP
Defaults        env_keep += WSL_DISTRO_NAME
Defaults        env_keep += PRE_NAMESPACE_PATH
Defaults        env_keep += PRE_NAMESPACE_PWD

Defaults        env_keep += WSL2_IP

%sudo ALL=(ALL) NOPASSWD: /usr/sbin/enter-systemd-namespace
EOF

if ! grep 'start-systemd-namespace' /etc/bash.bashrc >/dev/null; then
  $(_sudo) sed -i 2a"# Start or enter a PID namespace in WSL2\nsource /usr/sbin/start-systemd-namespace\n" /etc/bash.bashrc
fi

$(_sudo) ln -sf /dev/null /etc/systemd/user/sockets.target.wants/dirmngr.socket || true
$(_sudo) ln -sf /dev/null /etc/systemd/user/sockets.target.wants/gpg-agent*.socket || true
$(_sudo) ln -sf /dev/null /lib/systemd/system/sysinit.target.wants/proc-sys-fs-binfmt_misc.automount || true
$(_sudo) ln -sf /dev/null /lib/systemd/system/sysinit.target.wants/proc-sys-fs-binfmt_misc.mount || true
$(_sudo) ln -sf /dev/null /lib/systemd/system/sysinit.target.wants/systemd-binfmt.service || true
