validate_install.sh

#!/bin/bash

# MeshC2 Installation Validation Script
# Run this after installation to verify everything is working

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

print_status() {
    echo -e "${BLUE}[CHECK]${NC} $1"
}

print_success() {
    echo -e "${GREEN}[PASS]${NC} $1"
}

print_error() {
    echo -e "${RED}[FAIL]${NC} $1"
}

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

echo "=== MeshC2 Installation Validation ==="
echo

# Check if service user exists
print_status "Checking service user..."
if id meshc2 &>/dev/null; then
    print_success "Service user 'meshc2' exists"
else
    print_error "Service user 'meshc2' not found"
fi

# Check if directories exist
print_status "Checking directories..."
directories=("/opt/meshc2" "/etc/meshc2" "/var/log/meshc2" "/var/lib/meshc2")
for dir in "${directories[@]}"; do
    if [[ -d "$dir" ]]; then
        print_success "Directory exists: $dir"
    else
        print_error "Directory missing: $dir"
    fi
done

# Check if files exist
print_status "Checking critical files..."
files=(
    "/opt/meshc2/src/meshtastic_troubleshoot/main.py"
    "/opt/meshc2/venv/bin/python"
    "/etc/systemd/system/meshc2.service"
    "/usr/local/bin/meshc2"
)
for file in "${files[@]}"; do
    if [[ -f "$file" ]]; then
        print_success "File exists: $file"
    else
        print_error "File missing: $file"
    fi
done

# Check if virtual environment works
print_status "Testing virtual environment..."
if /opt/meshc2/venv/bin/python -c "import meshtastic; print('Meshtastic library OK')" 2>/dev/null; then
    print_success "Virtual environment and dependencies OK"
else
    print_error "Virtual environment or dependencies issue"
fi

# Check if service is configured
print_status "Checking systemd service..."
if systemctl list-unit-files | grep -q meshc2.service; then
    print_success "Systemd service is configured"
    
    if systemctl is-enabled meshc2.service &>/dev/null; then
        print_success "Service is enabled"
    else
        print_warning "Service is not enabled (run: sudo systemctl enable meshc2)"
    fi
    
    if systemctl is-active meshc2.service &>/dev/null; then
        print_success "Service is running"
    else
        print_warning "Service is not running (run: sudo systemctl start meshc2)"
    fi
else
    print_error "Systemd service not found"
fi

# Check if command wrapper works
print_status "Testing command wrapper..."
if command -v meshc2 &>/dev/null; then
    print_success "meshc2 command is available"
    
    if meshc2 --help &>/dev/null; then
        print_success "meshc2 command executes successfully"
    else
        print_error "meshc2 command execution failed"
    fi
else
    print_error "meshc2 command not found in PATH"
fi

# Check for Meshtastic devices
print_status "Checking for Meshtastic devices..."
if ls /dev/ttyUSB* &>/dev/null || ls /dev/ttyACM* &>/dev/null; then
    print_success "Serial devices found"
    
    # Try to list devices
    if meshc2 --list-devices &>/dev/null; then
        print_success "Device detection working"
    else
        print_error "Device detection failed"
    fi
else
    print_warning "No serial devices found (connect Meshtastic device)"
fi

echo
echo "=== Validation Complete ==="
echo
echo "If any checks failed, try:"
echo "1. Re-run the installer: sudo ./install.sh"
echo "2. Check logs: sudo journalctl -u meshc2"
echo "3. Manual service start: sudo systemctl start meshc2"
echo