#!/bin/sh
service_template()
{
    cat <<EOF
[Unit]
Description=${service_name}

[Service]
# Type=forking

Environment=DISPLAY=$DISPLAY
Environment=XAUTHORITY=$XAUTHORITY
Environment=SSH_AUTH_SOCK=%t/keyring/ssh

ExecStart=${exe} ${args}
Restart=always

[Install]
WantedBy=default.target
EOF
}

set -e
exe=$1
shift
args=$*

case "$exe" in
    sudo) exit 1;;
    /*) ;;
    */*) exe=$(realpath -e "$exe") ;;
    *) exe=$(which "$exe") ;;
esac

if [ "$(id -u)" = 0 ]; then
    service_dir=/etc/systemd/system
    user=
else
    service_dir=$HOME/.config/systemd/user
    user=y
fi

service_name=${exe##*/}
service_file=${service_dir}/${service_name}.service

[ -d "$service_dir" ] || mkdir -p "$service_dir"

systemctl="systemctl ${user:+--user}"

if [ -e "$service_file" ]; then
    "${EDITOR:-/usr/bin/vi}" "$service_file"
    $systemctl daemon-reload
    $systemctl status "$service_name"
else
    service_template > "$service_file"
    $systemctl daemon-reload
    $systemctl enable "$service_name"
    $systemctl start "$service_name"
fi