stop-server.sh

#!/bin/bash

# Minecraft Bedrock Server Stop Script
# This script gracefully stops the Minecraft Bedrock server

set -euo pipefail

# Source configuration
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "$SCRIPT_DIR/common.sh"
load_config
validate_config || { printf "%s\n" "Unsafe or invalid configuration" >&2; exit 1; }

SAVE_HELD=false
release_save_hold() {
    local status=$?
    trap - EXIT INT TERM
    if [[ "$SAVE_HELD" == true ]]; then
        sudo -u "$SERVER_USER" screen -S "$SCREEN_SESSION_NAME" -X stuff "save resume"$'\r' || true
    fi
    exit "$status"
}
trap release_save_hold EXIT
trap 'exit 130' INT
trap 'exit 143' TERM

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

# Logging function
log() {
    local level="$1"
    shift
    local message="$*"
    local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
    
    case "$level" in
        INFO)  echo -e "${GREEN}[INFO]${NC} $message" ;;
        WARN)  echo -e "${YELLOW}[WARN]${NC} $message" ;;
        ERROR) echo -e "${RED}[ERROR]${NC} $message" ;;
        DEBUG) echo -e "${BLUE}[DEBUG]${NC} $message" ;;
    esac
    
    # Also log to file if log directory exists
    if [[ -d "$LOG_DIR" ]]; then
        echo "[$timestamp] [$level] $message" >> "$LOG_FILE"
    fi
}

# Check if server is running
is_server_running() {
    has_server_process || has_screen_session
}

# Send a message to all players before shutdown
send_shutdown_warning() {
    local countdown="$1"
    
    log INFO "Sending shutdown warning to players ($countdown seconds)..."
    
    local message="Server will shut down in $countdown seconds. Please save your progress!"
    
    # Send the message to the server console
    sudo -u "$SERVER_USER" screen -S "$SCREEN_SESSION_NAME" -X stuff "say $message"$'\r' || {
        log WARN "Failed to send shutdown warning"
        return 1
    }
}

# Send a command to the server console
send_server_command() {
    local command="$1"
    
    if is_server_running; then
        sudo -u "$SERVER_USER" screen -S "$SCREEN_SESSION_NAME" -X stuff "$command"$'\r' || {
            log ERROR "Failed to send command: $command"
            return 1
        }
        return 0
    else
        log ERROR "Server is not running"
        return 1
    fi
}

# Gracefully stop the server
graceful_stop() {
    local force="$1"
    
    if ! is_server_running; then
        log INFO "Server is not running"
        return 0
    fi
    
    log INFO "Stopping Minecraft Bedrock Server gracefully..."
    
    # If not forcing, send warnings to players
    if [[ "$force" != "force" ]]; then
        # Send warnings at 60, 30, 15, 5 seconds
        send_shutdown_warning 60 || log WARN "Failed to send first shutdown warning"
        sleep 45
        
        if is_server_running; then
            send_shutdown_warning 15 || log WARN "Failed to send second shutdown warning"
            sleep 10
            
            if is_server_running; then
                send_shutdown_warning 5 || log WARN "Failed to send final shutdown warning"
                sleep 5
            fi
        fi
    fi
    
    if is_server_running; then
        # Save the world before stopping
        log INFO "Saving world data..."
        SAVE_HELD=true
        send_server_command "save hold" || log WARN "Failed to send 'save hold' command"
        sleep 2
        send_server_command "save query" || log WARN "Failed to send 'save query' command"
        sleep 3
        if send_server_command "save resume"; then SAVE_HELD=false; fi
        sleep 2
        
        # Send stop command
        log INFO "Sending stop command to server..."
        send_server_command "stop" || log WARN "Failed to send 'stop' command"
        
        # Give the server a moment to process the stop command
        sleep 5
        
        # Do an immediate check to see if server stopped quickly
        if ! is_server_running; then
            log INFO "Server stopped quickly after stop command"
        else
            log INFO "Server still running, continuing to wait..."
        fi
        
        # Wait for server to stop (max 60 seconds)
        local count=0
        local max_wait=60
        
        log INFO "Waiting for server to stop (max $max_wait seconds)..."
        while [[ $count -lt $max_wait ]]; do
            # Check if server is still running
            if ! is_server_running; then
                log INFO "Server stopped after $count seconds"
                break
            fi
            
            sleep 1
            count=$((count + 1))
            
            # Show progress every 10 seconds
            if [[ $((count % 10)) -eq 0 ]]; then
                log INFO "Still waiting... ($count/$max_wait seconds)"
                
                # Additional debug info every 20 seconds
                if [[ $((count % 20)) -eq 0 ]]; then
                    local screen_status="not found"
                    local process_status="not found"
                    
                    if has_screen_session; then
                        screen_status="running"
                    fi
                    
                    if has_server_process; then
                        process_status="running"
                    fi
                    
                    log DEBUG "Status check - Screen: $screen_status, Process: $process_status"
                fi
            fi
        done
        
        if is_server_running; then
            log WARN "Server didn't stop gracefully within $max_wait seconds"
            log WARN "Killing screen session..."
            sudo -u "$SERVER_USER" screen -S "$SCREEN_SESSION_NAME" -X quit || {
                log WARN "Failed to kill screen session gracefully"
            }
            sleep 2
            
            # Also kill any remaining bedrock_server processes
            if has_server_process; then
                log WARN "Killing remaining bedrock_server processes..."
                signal_server TERM || {
                    log ERROR "Failed to kill bedrock_server processes"
                    return 1
                }
                sleep 2
            fi
            
            if is_server_running; then
                log ERROR "Failed to stop server"
                return 1
            else
                log WARN "Server stopped forcefully"
            fi
        else
            log INFO "Server stopped gracefully"
            # Clean up any lingering screen session
            if has_screen_session; then
                log INFO "Cleaning up screen session..."
                sudo -u "$SERVER_USER" screen -S "$SCREEN_SESSION_NAME" -X quit || true
            fi
        fi
    fi
    
    return 0
}

# Force stop the server (immediate)
force_stop() {
    log WARN "Force stopping configured server instance..."
    signal_server TERM || return 1
    local count
    for ((count=0; count<10; count++)); do
        has_server_process || break
        sleep 1
    done
    if has_server_process; then
        signal_server KILL || return 1
        sleep 2
    fi
    if has_screen_session; then
        sudo -u "$SERVER_USER" screen -S "$SCREEN_SESSION_NAME" -X quit || return 1
        sleep 1
    fi
    ! is_server_running
}

# Show server status
show_status() {
    echo ""
    log INFO "Server Status:"
    
    if is_server_running; then
        log INFO "✓ Server is running"
        
        # Show screen sessions
        echo ""
        log INFO "Active screen sessions:"
        sudo -u "$SERVER_USER" screen -list | grep "$SCREEN_SESSION_NAME" || true
    else
        log INFO "✗ Server is not running"
    fi
    
    echo ""
}

# Show usage information
show_usage() {
    echo "Usage: $0 [OPTIONS]"
    echo ""
    echo "Options:"
    echo "  -f, --force     Force stop the server immediately (no warnings)"
    echo "  -s, --status    Show server status only"
    echo "  -h, --help      Show this help message"
    echo ""
    echo "Examples:"
    echo "  $0              Stop server gracefully with player warnings"
    echo "  $0 --force      Stop server immediately"
    echo "  $0 --status     Check if server is running"
}

# Main function
main() {
    local action="stop"
    local force_mode="normal"
    
    # Parse command line arguments
    while [[ $# -gt 0 ]]; do
        case $1 in
            -f|--force)
                force_mode="force"
                shift
                ;;
            -s|--status)
                action="status"
                shift
                ;;
            -h|--help)
                show_usage
                exit 0
                ;;
            *)
                log ERROR "Unknown option: $1"
                show_usage
                exit 1
                ;;
        esac
    done
    
    log INFO "Minecraft Bedrock Server Stop Script"
    
    case "$action" in
        "status")
            show_status
            ;;
        "stop")
            if [[ $EUID -ne 0 ]]; then
                log ERROR "Run stop-server.sh as root"; return 1
            fi
            acquire_management_lock
            if [[ "$force_mode" == "force" ]]; then
                force_stop
            else
                graceful_stop "$force_mode"
            fi
            show_status
            ;;
    esac
}

# Run main function
main "$@"