server-manager.sh

#!/bin/bash

# DUT Reservation Server Manager
# This script provides easy commands to manage your Flask web server

SERVICE_NAME="dut-reservation.service"
SERVER_DIR="/home/ubuntu/mkim/testbed-reservation"
LOG_FILE="$SERVER_DIR/server.log"

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

# Function to print colored output
print_status() {
    echo -e "${GREEN}[INFO]${NC} $1"
}

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

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

print_header() {
    echo -e "${BLUE}================================${NC}"
    echo -e "${BLUE}  DUT Reservation Server Manager${NC}"
    echo -e "${BLUE}================================${NC}"
}

# Function to check if service exists
check_service() {
    if ! systemctl list-unit-files | grep -q "$SERVICE_NAME"; then
        print_error "Service $SERVICE_NAME not found. Please run: ./server-manager.sh install"
        exit 1
    fi
}

# Function to check if running as root
check_root() {
    if [[ $EUID -eq 0 ]]; then
        print_error "This script should not be run as root"
        exit 1
    fi
}

# Function to show usage
show_usage() {
    print_header
    echo "Usage: $0 [COMMAND]"
    echo ""
    echo "Commands:"
    echo "  start     - Start the web server"
    echo "  stop      - Stop the web server"
    echo "  restart   - Restart the web server"
    echo "  status    - Show server status"
    echo "  logs      - Show recent logs"
    echo "  logs-follow - Follow logs in real-time"
    echo "  install   - Install the systemd service"
    echo "  uninstall - Remove the systemd service"
    echo "  manual    - Start server manually (for debugging)"
    echo "  health    - Run health check"
    echo "  config    - Show configuration"
    echo "  help      - Show this help message"
    echo ""
    echo "Examples:"
    echo "  $0 start"
    echo "  $0 status"
    echo "  $0 logs"
}

# Function to kill any existing manual Flask processes
kill_manual_processes() {
    print_status "Checking for existing Flask processes..."
    
    # Find and kill any manual Flask server processes
    local flask_pids=$(pgrep -f "python3.*server.py" | grep -v $$)
    if [ ! -z "$flask_pids" ]; then
        print_warning "Found existing Flask processes: $flask_pids"
        print_status "Stopping manual Flask processes..."
        for pid in $flask_pids; do
            kill -TERM $pid 2>/dev/null
            sleep 1
            if kill -0 $pid 2>/dev/null; then
                print_warning "Process $pid still running, force killing..."
                kill -KILL $pid 2>/dev/null
            fi
        done
        sleep 2
    else
        print_status "No existing Flask processes found"
    fi
}

# Function to start the server
start_server() {
    print_status "Starting DUT reservation server..."
    
    # Kill any existing manual processes first
    kill_manual_processes
    
    # Try to start via systemd
    if systemctl list-unit-files | grep -q "$SERVICE_NAME"; then
        print_status "Starting via systemd service..."
        sudo systemctl start $SERVICE_NAME
        if [ $? -eq 0 ]; then
            print_status "Server started successfully via systemd!"
            sleep 2
            show_status
        else
            print_warning "Systemd service failed to start, trying manual start..."
            start_manual_background
        fi
    else
        print_warning "Systemd service not found, starting manually..."
        start_manual_background
    fi
}

# Function to start server manually in background
start_manual_background() {
    print_status "Starting server manually in background..."
    
    cd "$SERVER_DIR"
    export FLASK_APP=server.py
    export FLASK_ENV=production
    
    # Start in background and redirect output
    nohup python3 server.py > server.log 2>&1 &
    local pid=$!
    
    # Wait a moment and check if it started successfully
    sleep 3
    if kill -0 $pid 2>/dev/null; then
        print_status "Server started successfully! PID: $pid"
        print_status "Logs are being written to: $LOG_FILE"
        sleep 2
        show_status
    else
        print_error "Failed to start server manually"
        exit 1
    fi
}

# Function to stop the server
stop_server() {
    print_status "Stopping DUT reservation server..."
    
    # First try to stop via systemd
    if systemctl list-unit-files | grep -q "$SERVICE_NAME"; then
        print_status "Stopping systemd service..."
        sudo systemctl stop $SERVICE_NAME 2>/dev/null
    fi
    
    # Then kill any manual processes
    kill_manual_processes
    
    print_status "Server stopped successfully!"
}

# Function to restart the server
restart_server() {
    print_status "Restarting DUT reservation server..."
    
    # Stop the server (handles both systemd and manual processes)
    stop_server
    
    # Wait a moment
    sleep 2
    
    # Start the server (will try systemd first, then manual)
    start_server
}

# Function to show server status
show_status() {
    print_status "Server Status:"
    echo ""
    
    # Check systemd service status
    if systemctl list-unit-files | grep -q "$SERVICE_NAME"; then
        print_status "Systemd Service Status:"
        sudo systemctl status $SERVICE_NAME --no-pager -l
        echo ""
    else
        print_warning "Systemd service not installed"
    fi
    
    # Check for manual Flask processes
    local flask_pids=$(pgrep -f "python3.*server.py" | grep -v $$)
    if [ ! -z "$flask_pids" ]; then
        print_status "Manual Flask Processes:"
        for pid in $flask_pids; do
            echo "  PID: $pid - $(ps -p $pid -o cmd= --no-headers)"
        done
        echo ""
    else
        print_warning "No manual Flask processes found"
    fi
    
    # Show if server is accessible
    if curl -s http://localhost:8080 > /dev/null 2>&1; then
        print_status "Server is accessible at http://localhost:8080"
    else
        print_warning "Server may not be accessible at http://localhost:8080"
    fi
    
    # Show health dashboard
    if curl -s http://localhost:8080/health > /dev/null 2>&1; then
        print_status "Health dashboard available at http://localhost:8080/health"
    fi
}

# Function to show logs
show_logs() {
    print_status "Recent server logs:"
    echo ""
    sudo journalctl -u $SERVICE_NAME --no-pager -l -n 50
}

# Function to follow logs
follow_logs() {
    print_status "Following server logs (Ctrl+C to stop):"
    echo ""
    sudo journalctl -u $SERVICE_NAME --no-pager -l -f
}

# Function to install the service
install_service() {
    print_status "Installing systemd service..."
    
    if [ ! -f "$SERVER_DIR/dut-reservation.service" ]; then
        print_error "Service file not found at $SERVER_DIR/dut-reservation.service"
        exit 1
    fi
    
    sudo cp "$SERVER_DIR/dut-reservation.service" /etc/systemd/system/
    sudo systemctl daemon-reload
    sudo systemctl enable $SERVICE_NAME
    
    print_status "Service installed and enabled!"
    print_status "You can now use: $0 start"
}

# Function to uninstall the service
uninstall_service() {
    print_status "Uninstalling systemd service..."
    
    sudo systemctl stop $SERVICE_NAME 2>/dev/null
    sudo systemctl disable $SERVICE_NAME 2>/dev/null
    sudo rm -f /etc/systemd/system/$SERVICE_NAME
    sudo systemctl daemon-reload
    
    print_status "Service uninstalled!"
}

# Function to start server manually
start_manual() {
    print_status "Starting server manually (for debugging)..."
    print_warning "This will run in the foreground. Press Ctrl+C to stop."
    echo ""
    
    cd "$SERVER_DIR"
    export FLASK_APP=server.py
    export FLASK_ENV=production
    python3 server.py
}

# Function to run health check
run_health_check() {
    print_status "Running health check..."
    cd "$SERVER_DIR"
    
    if [ -f "test_health_setup.py" ]; then
        python3 test_health_setup.py
    else
        print_warning "Health check script not found"
    fi
}

# Function to show configuration
show_config() {
    print_status "Server Configuration:"
    echo ""
    echo "Service Name: $SERVICE_NAME"
    echo "Server Directory: $SERVER_DIR"
    echo "Log File: $LOG_FILE"
    echo ""
    
    if [ -f "$SERVER_DIR/health_config.yaml" ]; then
        echo "Health Configuration:"
        cat "$SERVER_DIR/health_config.yaml"
    fi
}

# Main script logic
check_root

case "${1:-help}" in
    start)
        start_server
        ;;
    stop)
        stop_server
        ;;
    restart)
        restart_server
        ;;
    status)
        show_status
        ;;
    logs)
        show_logs
        ;;
    logs-follow)
        follow_logs
        ;;
    install)
        install_service
        ;;
    uninstall)
        uninstall_service
        ;;
    manual)
        start_manual
        ;;
    health)
        run_health_check
        ;;
    config)
        show_config
        ;;
    help|--help|-h)
        show_usage
        ;;
    *)
        print_error "Unknown command: $1"
        echo ""
        show_usage
        exit 1
        ;;
esac