Workspace Services

#!/bin/bash
# Manage workspace services (start/stop)

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
PLUGIN_DIR="$(cd "$SCRIPT_DIR/../../.." && pwd)"

# Source utilities
source "$PLUGIN_DIR/lib/workspace-utils.sh"

# Usage
usage() {
    echo "Usage: $0 <start|stop|status> [project|all]"
    echo ""
    echo "Commands:"
    echo "  start   Start services"
    echo "  stop    Stop services"
    echo "  status  Show service status"
    echo ""
    echo "Arguments:"
    echo "  project  Name of a specific project"
    echo "  all      All projects (default)"
    echo ""
    exit 1
}

# Parse arguments
ACTION=""
TARGET="all"

while [[ $# -gt 0 ]]; do
    case $1 in
        start|stop|status)
            ACTION="$1"
            shift
            ;;
        -h|--help)
            usage
            ;;
        *)
            TARGET="$1"
            shift
            ;;
    esac
done

if [ -z "$ACTION" ]; then
    print_error "Action required (start, stop, or status)"
    usage
fi

# Detect workspace
detected=$(detect_workspace)
if [ -z "$detected" ]; then
    print_error "No workspace detected"
    exit 1
fi

WORKSPACE_NAME=$(echo "$detected" | cut -d: -f1)
CONFIG_FILE=$(get_workspace_config "$WORKSPACE_NAME")

# Get workspace display name
DISPLAY_NAME=$(grep -E "^name:" "$CONFIG_FILE" | head -1 | sed 's/^name:[ ]*//')
[ -z "$DISPLAY_NAME" ] && DISPLAY_NAME="$WORKSPACE_NAME"

# Get workspace-level commands
get_ws_command() {
    local cmd="$1"
    grep -E "^${cmd}:" "$CONFIG_FILE" 2>/dev/null | sed "s/^${cmd}:[ ]*//"
}

# Check port status
check_ports() {
    echo ""
    echo "Port status:"
    while IFS='|' read -r name repo path stack port; do
        [ -z "$name" ] && continue
        [ -z "$port" ] || [ "$port" = "-" ] && continue

        if port_in_use "$port"; then
            echo -e "  • $name: :$port ${GREEN}✓ running${NC}"
        else
            echo -e "  • $name: :$port ${YELLOW}○ stopped${NC}"
        fi
    done < <(parse_projects "$CONFIG_FILE")
}

# Start services
do_start() {
    local target="$1"

    if [ "$target" = "all" ]; then
        echo "🚀 Starting workspace services..."
        echo ""

        # Check for workspace-level start command
        START_ALL=$(get_ws_command "start-all")
        if [ -n "$START_ALL" ]; then
            print_info "Using workspace command: $START_ALL"
            eval "$START_ALL"
            print_success "Services started"
            check_ports
            return
        fi

        # Start individual projects
        while IFS='|' read -r name repo path stack port; do
            [ -z "$name" ] && continue

            expanded_path=$(expand_path "$path")
            if [ ! -d "$expanded_path" ]; then
                print_warning "$name not cloned, skipping"
                continue
            fi

            if [ -z "$port" ] || [ "$port" = "-" ]; then
                continue
            fi

            if port_in_use "$port"; then
                print_info "$name already running on port $port"
                continue
            fi

            echo "Starting $name..."
            # Try to find a start script
            if [ -f "$expanded_path/package.json" ]; then
                if grep -q '"dev"' "$expanded_path/package.json"; then
                    echo "  cd $expanded_path && npm run dev"
                    (cd "$expanded_path" && npm run dev > /dev/null 2>&1 &)
                elif grep -q '"start"' "$expanded_path/package.json"; then
                    echo "  cd $expanded_path && npm start"
                    (cd "$expanded_path" && npm start > /dev/null 2>&1 &)
                fi
                sleep 2  # Give it time to start
                if port_in_use "$port"; then
                    print_success "$name started on port $port"
                else
                    print_warning "$name may not have started correctly"
                fi
            else
                print_warning "No start script found for $name"
            fi
        done < <(parse_projects "$CONFIG_FILE")
    else
        # Start specific project
        project_info=$(get_project_info "$CONFIG_FILE" "$target")
        if [ -z "$project_info" ]; then
            print_error "Project '$target' not found"
            exit 1
        fi

        port=$(echo "$project_info" | cut -d'|' -f5)
        path=$(echo "$project_info" | cut -d'|' -f3)
        expanded_path=$(expand_path "$path")

        echo "🚀 Starting $target..."

        if [ ! -d "$expanded_path" ]; then
            print_error "$target not cloned"
            exit 1
        fi

        if port_in_use "$port"; then
            print_info "$target already running on port $port"
            exit 0
        fi

        if [ -f "$expanded_path/package.json" ]; then
            if grep -q '"dev"' "$expanded_path/package.json"; then
                (cd "$expanded_path" && npm run dev > /dev/null 2>&1 &)
            elif grep -q '"start"' "$expanded_path/package.json"; then
                (cd "$expanded_path" && npm start > /dev/null 2>&1 &)
            fi
            sleep 2
            if port_in_use "$port"; then
                print_success "$target started on port $port"
            else
                print_warning "$target may not have started correctly"
            fi
        fi
    fi

    check_ports
}

# Stop services
do_stop() {
    local target="$1"

    if [ "$target" = "all" ]; then
        echo "🛑 Stopping workspace services..."
        echo ""

        # Check for workspace-level stop command
        STOP_ALL=$(get_ws_command "stop-all")
        if [ -n "$STOP_ALL" ]; then
            print_info "Using workspace command: $STOP_ALL"
            eval "$STOP_ALL"
            print_success "Services stopped"
            check_ports
            return
        fi

        # Stop individual projects by port
        while IFS='|' read -r name repo path stack port; do
            [ -z "$name" ] && continue
            [ -z "$port" ] || [ "$port" = "-" ] && continue

            echo "Stopping $name (port $port)..."
            if port_in_use "$port"; then
                pid=$(lsof -ti :"$port" 2>/dev/null | head -1)
                if [ -n "$pid" ]; then
                    kill -15 "$pid" 2>/dev/null || true
                    sleep 1
                    if port_in_use "$port"; then
                        kill -9 "$pid" 2>/dev/null || true
                    fi
                fi
                print_success "Stopped"
            else
                print_info "Was not running"
            fi
        done < <(parse_projects "$CONFIG_FILE")
    else
        # Stop specific project
        project_info=$(get_project_info "$CONFIG_FILE" "$target")
        if [ -z "$project_info" ]; then
            print_error "Project '$target' not found"
            exit 1
        fi

        port=$(echo "$project_info" | cut -d'|' -f5)

        echo "🛑 Stopping $target (port $port)..."

        if port_in_use "$port"; then
            pid=$(lsof -ti :"$port" 2>/dev/null | head -1)
            if [ -n "$pid" ]; then
                kill -15 "$pid" 2>/dev/null || true
                sleep 1
            fi
            print_success "Stopped"
        else
            print_info "Was not running"
        fi
    fi

    check_ports
}

# Status
do_status() {
    echo "📊 Service status for $DISPLAY_NAME"
    check_ports
}

# Execute action
case $ACTION in
    start)
        do_start "$TARGET"
        ;;
    stop)
        do_stop "$TARGET"
        ;;
    status)
        do_status
        ;;
esac