#!/bin/bash

# autoSMART Uninstaller
# Version: 1.0
# Description: Complete removal of autoSMART system to prevent orphaned files

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
INSTALL_DIR="/opt/autoSMART"
CONFIG_DIR="/etc/autosmart"
PVE_CONFIG_DIR="/etc/pve/autoSMART"
SERVICE_NAME="autosmart"
LOG_DIR="/var/log/autosmart"
SYSTEMD_SERVICE="/etc/systemd/system/${SERVICE_NAME}.service"

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

log_info() {
    echo -e "${BLUE}[INFO]${NC} $1"
}

log_success() {
    echo -e "${GREEN}[SUCCESS]${NC} $1"
}

log_warning() {
    echo -e "${YELLOW}[WARNING]${NC} $1"
}

log_error() {
    echo -e "${RED}[ERROR]${NC} $1"
}

log_info "🗑️  autoSMART Uninstaller v1.0"
log_info "==============================="

# Check if running as root
if [[ $EUID -ne 0 ]]; then
   log_error "This script must be run as root (use sudo)"
   exit 1
fi

# Stop and disable systemd service
if systemctl is-active --quiet "$SERVICE_NAME" 2>/dev/null; then
    log_info "Stopping autoSMART service..."
    systemctl stop "$SERVICE_NAME"
    log_success "Service stopped"
else
    log_info "Service is not running"
fi

if systemctl is-enabled --quiet "$SERVICE_NAME" 2>/dev/null; then
    log_info "Disabling autoSMART service..."
    systemctl disable "$SERVICE_NAME"
    log_success "Service disabled"
fi

# Remove systemd service file
if [[ -f "$SYSTEMD_SERVICE" ]]; then
    log_info "Removing systemd service file..."
    rm -f "$SYSTEMD_SERVICE"
    systemctl daemon-reload
    log_success "Service file removed"
fi

# Remove installation directory
if [[ -d "$INSTALL_DIR" ]]; then
    log_info "Removing installation directory: $INSTALL_DIR"
    rm -rf "$INSTALL_DIR"
    log_success "Installation directory removed"
else
    log_info "Installation directory does not exist"
fi

# Remove configuration directory
if [[ -d "$CONFIG_DIR" ]]; then
    log_info "Removing configuration directory: $CONFIG_DIR"
    rm -rf "$CONFIG_DIR"
    log_success "Configuration directory removed"
else
    log_info "Configuration directory does not exist"
fi

# Remove PVE configuration directory (if exists)
if [[ -d "$PVE_CONFIG_DIR" ]]; then
    log_info "Removing PVE configuration directory: $PVE_CONFIG_DIR"
    rm -rf "$PVE_CONFIG_DIR"
    log_success "PVE configuration directory removed"
fi

# Remove log directory
if [[ -d "$LOG_DIR" ]]; then
    log_info "Removing log directory: $LOG_DIR"
    rm -rf "$LOG_DIR"
    log_success "Log directory removed"
fi

# Remove cron jobs (if any)
if crontab -l 2>/dev/null | grep -q "autosmart"; then
    log_info "Removing autoSMART cron jobs..."
    (crontab -l 2>/dev/null | grep -v "autosmart") | crontab -
    log_success "Cron jobs removed"
fi

# Remove temporary files
TEMP_FILES=(
    "/tmp/autosmart*"
    "/tmp/smart-*"
    "/var/tmp/autosmart*"
)

for pattern in "${TEMP_FILES[@]}"; do
    if ls $pattern 1> /dev/null 2>&1; then
        log_info "Removing temporary files: $pattern"
        rm -rf $pattern
    fi
done

# Remove user and group (if created specifically for autoSMART)
if id "autosmart" &>/dev/null; then
    log_warning "Found autosmart user - leaving intact (may be used by database)"
    log_info "To remove user manually: userdel autosmart"
fi

# Clean up any remaining processes
PROCESSES=$(pgrep -f "autosmart|smart-collector" || true)
if [[ -n "$PROCESSES" ]]; then
    log_warning "Found running autoSMART processes: $PROCESSES"
    log_info "Terminating processes..."
    pkill -f "autosmart|smart-collector" || true
    sleep 2
    pkill -9 -f "autosmart|smart-collector" || true
    log_success "Processes terminated"
fi

# Remove from PATH modifications (if any)
PROFILE_FILES=(
    "/etc/profile.d/autosmart.sh"
    "/etc/bash.bashrc.d/autosmart.sh"
)

for file in "${PROFILE_FILES[@]}"; do
    if [[ -f "$file" ]]; then
        log_info "Removing PATH modification: $file"
        rm -f "$file"
    fi
done

# Clean package manager cache related to autoSMART dependencies
log_info "Cleaning package cache..."
if command -v apt-get &> /dev/null; then
    apt-get clean >/dev/null 2>&1 || true
elif command -v yum &> /dev/null; then
    yum clean all >/dev/null 2>&1 || true
fi

# Final verification
REMAINING_FILES=$(find /etc /opt /var -name "*autosmart*" -o -name "*autoSMART*" 2>/dev/null | head -10)
if [[ -n "$REMAINING_FILES" ]]; then
    log_warning "Some autoSMART files may still exist:"
    echo "$REMAINING_FILES"
    log_info "These may be database files or manually created configurations"
fi

log_success "✅ autoSMART uninstallation complete!"
log_info ""
log_info "📋 Summary of removed components:"
log_info "  • Systemd service: $SERVICE_NAME"
log_info "  • Installation directory: $INSTALL_DIR"
log_info "  • Configuration directory: $CONFIG_DIR"
log_info "  • Log directory: $LOG_DIR"
log_info "  • Temporary files and processes"
log_info ""
log_info "💡 Notes:"
log_info "  • Database data is preserved (not removed)"
log_info "  • System packages (Perl, PostgreSQL client) are preserved"
log_info "  • User 'autosmart' is preserved if it exists"
log_info ""
log_info "🔄 System is now clean and ready for fresh installation"

exit 0