#!/usr/bin/env bash
# Atomizer Cluster Management Script
set -euo pipefail

AGENTS=(manager tech-lead secretary auditor optimizer study-builder nx-expert webster)
SERVICE_PREFIX="openclaw-atomizer@"

case "${1:-help}" in
  start)
    for a in "${AGENTS[@]}"; do
      echo "Starting ${a}..."
      systemctl --user enable --now "${SERVICE_PREFIX}${a}.service"
    done
    echo "All agents started."
    ;;
  stop)
    for a in "${AGENTS[@]}"; do
      echo "Stopping ${a}..."
      systemctl --user stop "${SERVICE_PREFIX}${a}.service" || true
    done
    echo "All agents stopped."
    ;;
  restart)
    for a in "${AGENTS[@]}"; do
      echo "Restarting ${a}..."
      systemctl --user restart "${SERVICE_PREFIX}${a}.service"
    done
    echo "All agents restarted."
    ;;
  status)
    for a in "${AGENTS[@]}"; do
      systemctl --user status "${SERVICE_PREFIX}${a}.service" --no-pager -l 2>/dev/null | head -3
      echo "---"
    done
    ;;
  logs)
    agent="${2:-manager}"
    journalctl --user -u "${SERVICE_PREFIX}${agent}.service" -f --no-pager
    ;;
  *)
    echo "Usage: $0 {start|stop|restart|status|logs [agent]}"
    exit 1
    ;;
esac