#!/bin/sh
set -eu

# ChuSQL 卸载脚本（Linux/macOS）：停服务、删程序与配置、
# 从 .profile 摘掉 PATH 段。数据目录默认一起删。

install_dir=''
data_dir=''
keep_data='no'
keep_config='no'
assume_yes='no'

usage() {
    cat <<'EOF'
Uninstall ChuSQL (Linux/macOS).

Usage: uninstall.sh [options]

  --install-dir DIR   program directory (default: $HOME/.local/share/chusql)
  --data-dir DIR      data directory (default: <install-dir>/data)
  --keep-data         keep the data directory, remove programs only
  --keep-config       keep chusql.toml
  --yes               do not ask (required when there is no terminal)
  -h, --help          show this help
EOF
}

step() { echo "== $1"; }
note() { echo "   $1"; }
fail() {
    echo "!! $1" >&2
    exit 1
}

# 停掉命令行里带这个路径的进程
stop_by_path() {
    target=$1
    [ -n "$target" ] || return 0
    pids=$(ps -eo pid=,args= 2>/dev/null |
        awk -v t="$target" '{ pid = $1; $1 = ""; sub(/^ +/, ""); if (index($0, t) == 1) print pid }' || true)
    [ -n "$pids" ] || return 0
    for pid in $pids; do
        kill "$pid" 2>/dev/null || true
    done
    note "stopped: $pids"
    sleep 1
    for pid in $pids; do
        kill -9 "$pid" 2>/dev/null || true
    done
}

# 问一句 y/n，不是 y 就当没同意
ask_yes_no() {
    printf '%s ' "$1" >&2
    answer=''
    IFS= read -r answer || answer=''
    case "$answer" in
        y | Y | yes | YES | Yes) return 0 ;;
        *) return 1 ;;
    esac
}

while [ $# -gt 0 ]; do
    case "$1" in
        -h | --help) usage; exit 0 ;;
        --install-dir) install_dir="${2:-}"; shift 2 ;;
        --data-dir) data_dir="${2:-}"; shift 2 ;;
        --keep-data) keep_data='yes'; shift ;;
        --keep-config) keep_config='yes'; shift ;;
        --yes) assume_yes='yes'; shift ;;
        *) fail "unknown option: $1 (try --help)" ;;
    esac
done

install_dir=${install_dir:-$HOME/.local/share/chusql}
data_dir=${data_dir:-$install_dir/data}
config_dir="${XDG_CONFIG_HOME:-$HOME/.config}/ChuSQL"
config_file="$config_dir/chusql.toml"
profile="$HOME/.profile"

if [ "$assume_yes" = 'no' ]; then
    echo 'ChuSQL uninstaller'
    echo "  install dir  $install_dir"
    echo "  data dir     $data_dir"
    if [ "$keep_data" = 'yes' ]; then echo '  data         kept'; fi
    if [ -t 0 ]; then
        ask_yes_no 'Remove it? [y/N]' || {
            echo 'nothing was removed'
            exit 0
        }
    else
        fail 'no terminal: rerun with --yes to confirm'
    fi
fi

step 'Stopping the service'
stop_by_path "$install_dir/bin/chusql-server"
stop_by_path "$install_dir/bin/chusql-web"

step 'Removing files'
if [ ! -d "$install_dir" ]; then
    note "not found: $install_dir"
elif [ "$keep_data" = 'yes' ]; then
    for entry in "$install_dir"/*; do
        [ -e "$entry" ] || continue
        if [ "$entry" = "$data_dir" ]; then
            note "kept data: $entry"
            continue
        fi
        rm -rf "$entry"
    done
    note "kept $install_dir (data only)"
else
    rm -rf "$install_dir"
    case "$data_dir" in
        "$install_dir" | "$install_dir"/*) ;;
        *) rm -rf "$data_dir" ;;
    esac
    note "removed $install_dir"
fi

step 'Removing chusql.toml'
if [ "$keep_config" = 'yes' ]; then
    note "kept config: $config_file"
else
    rm -f "$config_file"
    rmdir "$config_dir" 2>/dev/null || true
    note "removed config: $config_file"
fi

step 'Removing the PATH entry'
marker='# ChuSQL'
if [ ! -f "$profile" ]; then
    note "no $profile"
elif grep -q "^$marker" "$profile"; then
    tmp="$profile.chusql-uninstall"
    awk -v m="$marker" '
        skip > 0 { skip--; next }
        index($0, m) == 1 { skip = 2; next }
        { print }
    ' "$profile" >"$tmp" && mv "$tmp" "$profile"
    note "removed the $marker block from $profile"
else
    note "no $marker block in $profile"
fi

step 'Done'
note 'open a new terminal: the old PATH stays in the current one'