#!/bin/bash
set -e

BINARY="ostiary"
CONFIG_DIR="$HOME/.config/$BINARY"

RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'
CYAN='\033[0;36m'; NC='\033[0m'

info() { echo -e "${CYAN}[ostiary]${NC} $*"; }
ok()   { echo -e "${GREEN}[ostiary]${NC} $*"; }
warn() { echo -e "${YELLOW}[ostiary]${NC} $*"; }

# ── Find binary ──────────────────────────────────────────────────────────────
BINARY_PATH=$(command -v "$BINARY" 2>/dev/null || true)

removed_binary=false

if [ -n "$BINARY_PATH" ]; then
    info "Found binary: $BINARY_PATH"
    read -rp "  Remove $BINARY_PATH? [y/N] " answer
    if [[ "$answer" =~ ^[Yy]$ ]]; then
        BIN_DIR=$(dirname "$BINARY_PATH")
        if [ -w "$BIN_DIR" ]; then
            rm "$BINARY_PATH"
        elif command -v sudo &>/dev/null; then
            sudo rm "$BINARY_PATH"
        else
            warn "Cannot remove $BINARY_PATH (no write permission and no sudo)"
        fi
        ok "Binary removed"
        removed_binary=true
    fi
else
    # Check common locations manually
    for loc in /usr/local/bin /usr/bin "$HOME/.local/bin"; do
        if [ -f "$loc/$BINARY" ]; then
            info "Found binary: $loc/$BINARY"
            read -rp "  Remove $loc/$BINARY? [y/N] " answer
            if [[ "$answer" =~ ^[Yy]$ ]]; then
                if [ -w "$loc" ]; then
                    rm "$loc/$BINARY"
                elif command -v sudo &>/dev/null; then
                    sudo rm "$loc/$BINARY"
                else
                    warn "Cannot remove $loc/$BINARY"
                fi
                ok "Binary removed"
                removed_binary=true
            fi
            break
        fi
    done
fi

if ! $removed_binary; then
    warn "Binary not found in PATH"
fi

# ── Config directory ─────────────────────────────────────────────────────────
if [ -d "$CONFIG_DIR" ]; then
    echo ""
    info "Config directory: $CONFIG_DIR"
    read -rp "  Remove config and saved settings? [y/N] " answer
    if [[ "$answer" =~ ^[Yy]$ ]]; then
        rm -rf "$CONFIG_DIR"
        ok "Config removed"
    else
        ok "Config kept at $CONFIG_DIR"
    fi
fi

echo ""
ok "Done"