#!/usr/bin/env sh
# version=0.11.0

# Re-exec the script as qemu via sudo (only if needed)
[ "$(whoami)" != qemu ] && exec sudo -E -H -u qemu -- "$0" "$@"

# Environment
PATH="${HOME}/launchers:${HOME}/bin:${PATH}"

# Aliases
alias ls='ls --color=auto'
alias exec='exec '

# Set a restrictive umask to make sure qemu user files are private
umask 7027

# Function to print a colored error
perror() {
    >&2 printf '\033[1;31mKO:\033[0m \033[1m%s\033[0m\n' "$*"
}

# Function to show the usage
public_help() {
    name=$(basename "$0")

    cat << EOF
${name}: usage:
  ${name} running                          - (default behaviour) list running VMs
  ${name} ls                               - list available VMs
  ${name} add <path to script> [<VM name>] - add a launching script
  ${name} edit <VM name>                   - edit VM launching script
  ${name} cat <VM name>                    - print the content of a launching script
  ${name} start <VM name>                  - start a VM
  ${name} attach <VM name>                 - attach to the VM monitor socket
  ${name} rm <VM name>                     - delete launch script
  ${name} diskls                           - list available disk images
  ${name} diskadd <disk name> <size>       - create a disk image
  ${name} diskrm <disk name>               - delete disk image
  ${name} shell                            - start a shell as user qemu
  ${name} do <shell code>                  - run shell input as user qemu
  ${name} spice <running VM> [<TCP port>]  - expose a SPICE socket to TCP
  ${name} help                             - show this help
EOF
}

# Function to throw an invalid usage error (skill issue)
error_usage() {
    perror "invalid usage"
    >&2 public_help

    return 1
}

# Function to start a virtual machine
public_start() {
    if [ "$1" = -f ] || [ "$1" = --foreground ]; then
        daemonize=
    else
        daemonize=-daemonize
    fi

    while echo "$1" | grep -q '^-'; do shift; done;

    export QEMUSH_NAME="$1"

    set -- "$@" \
        -name "$QEMUSH_NAME" \
        -monitor "unix:$(pathof socket),server,nowait" \
        $daemonize

    if ! "$@"; then
        perror "error launching virtual machine \"${QEMUSH_NAME}\""
        return 2
    fi
}

# Attach to a running virtual machine output, the latest opened if no
# argument is provided
public_attach() {
    export QEMUSH_NAME

    if [ -n "$1" ]; then
        QEMUSH_NAME=$1
        socket_path=$(pathof socket)
    else
        socket_path=$(find ~/sockets/monitors -type s -printf '%T@ %p\n' | sort -r -n | head -1 | cut -d \  -f 2-)
        QEMUSH_NAME=$(basename "$socket_path")
    fi

    printf 'Attaching to \033[1m%s\033[0m, escape with C-d (EOF)\n' "$QEMUSH_NAME"
    socat -,rawer,escape=4 "UNIX-CONNECT:${socket_path}"
    echo
}

# List running virtual machines
public_running() {
    cd || return

    echo "Running machines:"
    exec ls -t sockets/monitors
}

# List available virtual machines entrypoints
public_ls() {
    cd || return

    echo "Available machines:"
    exec ls launchers
}

# Create a copy-on-write disk for a virtual machine
public_diskadd() {
    export QEMUSH_NAME="$1"

    exec qemu-img create -f qcow2 "$(pathof disk)" "$2"
}

# Delete a disk
public_diskrm() {
    for disk in "$@"; do
        export QEMUSH_NAME="$disk"

        rm -vi -- "$(pathof disk)"
    done
}

# List available disks
public_diskls() {
    cd || return

    echo "Available disks:"
    exec ls disks
}

# Edit a virtual machine entrypoint with a text editor
public_edit() {
    file="launchers/${1}"
    # I don't even know why shellcheck gives me this warning
    # shellcheck disable=2209
    [ -z "$EDITOR" ] && EDITOR=vi

    set -e
    cd
    touch -- "$file"
    chmod u+x -- "$file"
    exec "$EDITOR" -- "$file"
}

# Delete a virtual machine entrypoint
public_rm() {
    cd ~/launchers || return

    exec rm -vi -- "$@"
}

# Invoke a shell as qemu user in its home directory
public_shell() {
    cd || return

    # Again...
    # shellcheck disable=2209
    [ -z "$SHELL" ] && SHELL=sh

    set -- "$SHELL"
    case "$SHELL" in
        sh|bash|zsh|ksh|mksh|oksh)
            set -- "$@" -i
            ;;
    esac

    exec "$@"
}

# Output the content of an entrypoint, with coloration if on a virtual
# terminal
public_cat() {
    cd ~/launchers || return

    exec cat -- "$@"
}

# Copy a file in entrypoints folder
public_add() {
    if [ -z "$1" ]; then
        perror "specify the path of a launching script you want to add"
        return 1
    fi

    if [ -n "$2" ]; then
        destination="$2"
    else
        destination=$(basename "$1")
    fi
    destination="${HOME}/launchers/${destination}"

    trap return EXIT
    set -e

    cp -vi -- "$1" "$destination"
    chmod 0740 -- "$destination"

    set +e
    trap - EXIT
}

# Run shell commands as qemu
public_do() {
    exec sh -c "$*"
}

# Expose SPICE via TCP
public_spice() {
    export QEMUSH_NAME="$1"

    if [ -n "$2" ]; then
        port=$2
    else
        port=$(first-free-port 5900)
    fi

    exec tmux new -d -s "TCP port ${port} -> ${QEMUSH_NAME}" \
        socat TCP-LISTEN:"${port},reuseaddr,fork" UNIX-CLIENT:"$(pathof spice)"
}

case "$1" in
    "")
        public_running
        ;;
    running|ls|add|edit|cat|start|attach|rm|diskls|diskadd|diskrm|shell|do|spice|help)
        function=$1
        shift

        "public_${function}" "$@"
        ;;
    *)
        error_usage
        ;;
esac