pkgb

#!/bin/bash
set -Euo pipefail
_DEBUGPIPE=''

CLEAN=0
REFRESH=1
ASSUMEYES="-y"
EXACT="--exact"
FUZZER=""

if [[ -f "$HOME/.pkgbdebug" ]]; then
    _DEBUGPIPE=$(mktemp)
    echo "Started in Debug mode!"
    echo "Debug pipe: $_DEBUGPIPE"
    _trace_handler() {
        if [[ ! -p $_DEBUGPIPE ]]; then
            mkfifo "$_DEBUGPIPE"
        fi
        echo -e "\n[DBG]➜\t$BASH_COMMAND" &>"$_DEBUGPIPE"
    }
    set -o functrace
    trap '_trace_handler' DEBUG
fi

_fail() {
    printf '\e[0m'
    printf '\n\e[38;2;200;0;0m%s\e[0m\n' "FAIL: $1"
    exit 1
}

_info() {
    printf '\e[0m'
    printf '\e[38;2;0;200;0m%s\e[0m\n' "$1"
}

_terminate_handler() {
    _fail "Caught terminate signal"
}

_error_handler() {
    ORIGIN=$1
    ERRNO=$2
    printf '\e[0m'
    printf '\n\e[1m\e[48;2;236;240;241m\e[38;2;200;0;0m %s \e[0m' "!! > Internal failure. Command '$ORIGIN' failed with $ERRNO"
    printf '\n'
    exit 1
}

trap '_terminate_handler' SIGINT SIGTERM
trap '_error_handler $BASH_COMMAND $?' ERR

ACTIONS=('add' 'del' 'export' 'list' 'purge' 'upgrade' 'checkout', 'repair')
INFO_HEADER="<ESC> Exit, <PgUp> Preview up, <PgDown> Preview down <F5> Apply"

SUDO_CMD=$(command -v sudo || echo '')
if [[ -z "$SUDO_CMD" ]]; then
    if [[ $EUID -ne 0 ]]; then
        _fail 'sudo not found and current user is not root'
    fi
fi

_installed() {
    local result
    command -v "$1" &> /dev/null
    result=$?
    return $result
}

_get_installed_pkgs() {
    dpkg-query --show --showformat='${Package} ${db:Status-Status}\n' | grep installed | awk '{print $1}'
}

_pkgb_clean() {
    $SUDO_CMD apt-get autoremove "$ASSUMEYES"
}

_pkgb_refresh() {
    if [[ $REFRESH = 1 ]]; then
        $SUDO_CMD apt-get update
    fi
}

pkgb_repair() {
    echo "Trying to repair broken system state"
    apt install --fix-broken
}

pkgb_add() {
    local installed_pkg
    local all_pkg
    local selection

    _pkgb_refresh

    installed_pkg=$(_get_installed_pkgs)
    all_pkg=$(apt-cache pkgnames)

    selection=$(echo "$installed_pkg" "$all_pkg" | tr ' ' '\n' | sort | uniq -u | "$FUZZER" "${FUZZER_OPTS[@]}" --query="$1" --preview='apt-cache show {}' --prompt='Install packages > ' --bind "F5:accept" || echo '')
    echo "-- Selected packages --"
    echo "$selection"

    if [[ -n "$selection" ]]; then
        cmd=$(echo "$selection" | tr '\n' ' ')
        "$SUDO_CMD" apt-get install $cmd "$ASSUMEYES"
    fi
}

pkgb_del() {
    local selection
    selection=$(_get_installed_pkgs | "$FUZZER" "${FUZZER_OPTS[@]}" --preview='dpkg-query -s {}' --query="$1" --prompt='Uninstall packages > ' || echo '')
    echo "-- Selected packages --"
    echo "$selection"

    if [[ -n "$selection" ]]; then
        cmd=$(echo "$selection" | tr '\n' ' ')
        "$SUDO_CMD" apt-get remove $cmd "$ASSUMEYES"
    fi
}

pkgb_purge() {
    local selection
    selection=$(_get_installed_pkgs | "$FUZZER" "${FUZZER_OPTS[@]}" --preview='dpkg-query -s {}' --query="$1" --prompt='Purge packages > ' || echo '')
    echo "-- Selected packages --"
    echo "$selection"

    if [[ -n "$selection" ]]; then
        cmd=$(echo "$selection" | tr '\n' ' ')
        "$SUDO_CMD" apt-get purge $cmd "$ASSUMEYES"
    fi
}

pkgb_list() {
    _get_installed_pkgs | "$FUZZER" --preview='dpkg-query -s {}; printf "\n -- Installed Files --\n"; dpkg-query -L {}' --prompt='Installed packages > ' --query="$1" || true
}

pkgb_checkout() {
    local selection
    local goto
    selection=$(_get_installed_pkgs | "$FUZZER" --preview='printf "\n -- Installed Files --\n"; dpkg-query -L {}' --prompt='Installed packages > ' --query="$1" || true)
    if [[ -n "$selection" ]]; then
        goto=$(dpkg-query -L "$selection" | "$FUZZER" --prompt='Goto >' || true)
        if [[ -n "$goto" ]]; then
            sh -c "cd $goto; exec $SHELL"
        fi
    fi
}

pkgb_export() {
    _get_installed_pkgs
}

pkgb_upgrade() {
    _pkgb_refresh
    $SUDO_CMD apt-get dist-upgrade
}

pkgb_help() {
    echo "Usage: $(basename "$0") [-c] [-s] [-f] [-h] (${ACTIONS[*]}) [QUERY]" >&2
    echo -e '\nOptions:'
    echo -e '\t -c: Will run autoremove after action'
    echo -e '\t -s: Safe execute. Disables "always assume yes" for apt'
    echo -e '\t -f: Fast execute. Skip "apt-get update" before action'
    echo -e '\t -h: Show help and exit'
}

OPTIND=1 # Reset in case getopts has been used previously in the shell.

while getopts 'csfhu' OPTION; do
    case "$OPTION" in
    c)
        _info "-> Requested autoclean"
        CLEAN=1
        ;;
    u)
        _info "-> Disabled exact mode"
        EXACT=""
        ;;
    s)
        _info "-> Not assume always assume yes"
        ASSUMEYES=""
        ;;
    f)
        _info "-> Skipping source update"
        REFRESH=0
        ;;
    h)
        pkgb_help
        exit 1
        ;;
    *)
        echo "Invalid option"
        pkgb_help
        exit 1
        ;;

    esac

done
shift "$((OPTIND - 1))"

if [[ $# -lt 1 ]]; then
    echo "No action specified"
    pkgb_help
    exit 1
fi

if _installed "sk"; then
    FUZZER="sk"
    echo "Using sk"
elif _installed "fzf"; then
    FUZZER="fzf"
    echo "Using fzf"
else
    _fail "No fuzzy matcher installed. Please install fzf or sk!"
fi

FUZZER_OPTS=("--no-height" "--multi" "--bind" "pgdn:preview-down" $EXACT "--bind" "pgup:preview-up" "--header" "$INFO_HEADER")

for a in "${ACTIONS[@]}"; do
    if [ "$a" == "$1" ]; then
        shift
        "pkgb_${a}" "$*"
        if [[ $CLEAN = 1 ]]; then
            _pkgb_clean
        fi
        exit 0
    fi
done
echo "Unknown Action: $1"
pkgb_help
exit 1