#!/bin/bash

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

# Logging functions
log_info() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

log_warn() {
    echo -e "${YELLOW}[WARN]${NC} $1"
}

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

# Check if running as root
if [ "$EUID" -ne 0 ]; then 
    log_error "Please run as root"
    exit 1
fi

# Stop and disable service
log_info "Stopping service..."
systemctl stop collision-avoidance
systemctl disable collision-avoidance

# Remove service file
log_info "Removing service file..."
rm -f /etc/systemd/system/collision-avoidance.service
systemctl daemon-reload

# Remove files
log_info "Removing files..."
rm -rf /opt/collision_avoidance

# Remove Python packages
log_info "Removing Python packages..."
pip3 uninstall -y flask werkzeug click itsdangerous Jinja2 MarkupSafe

# Remove firewall rules
log_info "Removing firewall rules..."
if command -v firewall-cmd &> /dev/null; then
    firewall-cmd --permanent --remove-port=8010/tcp
    firewall-cmd --permanent --remove-port=8081/tcp
    firewall-cmd --reload
else
    log_warn "firewalld not detected, please remove firewall rules manually"
fi

log_info "Uninstallation completed"