stop_service.sh

#!/bin/bash
# stop_service.sh - Stops dual-node llama servers and nginx
# Simplified to dual-mode only. No mode params. Error handling on ALL functions.
# Uses centralized .env for model configuration. Zero-sudo.

set -e

# Load configuration from .env (centralized model settings)
load_env() {
    local env_file="$(dirname "${BASH_SOURCE[0]}")/.env"
    if [[ -f "$env_file" ]]; then
        set -a
        # shellcheck source=.env
        source "$env_file"
        set +a
    fi
}

load_env

# ─────────────────────────────────────────────
# ERROR HANDLING FOR ALL FUNCTIONS
# ─────────────────────────────────────────────
error() {
    echo "❌ ERROR: $1" >&2
    exit 1
}

success() {
    echo "✅ $1"
}

log() {
    echo "[$(date '+%Y-%m-%d %H:%M:%S')] $1"
}

stop_nginx() {
    log "🛑 Stopping user-level nginx..."
    systemctl --user stop user-nginx 2>/dev/null || log "Nginx was not running"
    success "Nginx stopped"
}

stop_nodes() {
    log "Stopping dual-node servers (llama-node1 and llama-node2)..."
    systemctl --user stop llama-node1 2>/dev/null || log "Node1 already stopped"
    systemctl --user stop llama-node2 2>/dev/null || log "Node2 already stopped"
    success "Dual nodes stopped"
}

cleanup() {
    log "Cleaning up..."
    # Optional: remove runtime if desired but not required
    success "Cleanup done"
}

show_summary() {
    echo ""
    echo "✓ All services stopped."
    echo "To restart: ./start_service.sh"
    echo "To re-register: ./setup-llama.sh"
    echo "Model config (${MODEL_NAME:-Unknown}, alias: coder, ctx: ${CONTEXT_LENGTH:-200000}) is ready for next start."
}

# Main
main() {
    log "Stopping dual-mode LLM serving system (${MODEL_NAME:-Unknown})..."
    stop_nginx
    stop_nodes
    cleanup
    show_summary
    success "Stop operation completed."
}

main "$@"