103 lines
2.4 KiB
Bash
Executable File
103 lines
2.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
# shellcheck source=./common.sh
|
|
source "$SCRIPT_DIR/common.sh"
|
|
|
|
ensure_repo_root
|
|
|
|
MOSAIC_HOME="${MOSAIC_HOME:-$HOME/.config/mosaic}"
|
|
ORCH_DIR=".mosaic/orchestrator"
|
|
PID_FILE="$ORCH_DIR/orchestrator.pid"
|
|
LOG_FILE="$ORCH_DIR/logs/daemon.log"
|
|
|
|
usage() {
|
|
cat <<USAGE
|
|
Usage: $(basename "$0") <start|drain|stop|status> [--poll-sec N] [--no-sync]
|
|
|
|
Commands:
|
|
start Run orchestrator drain loop in background (detached)
|
|
drain Run orchestrator drain loop in foreground (until queue drained)
|
|
stop Stop background orchestrator if running
|
|
status Show background orchestrator status
|
|
|
|
Options:
|
|
--poll-sec N Poll interval (default: 15)
|
|
--no-sync Skip docs/tasks.md -> orchestrator queue sync before run
|
|
USAGE
|
|
}
|
|
|
|
cmd="${1:-status}"
|
|
if [[ $# -gt 0 ]]; then
|
|
shift
|
|
fi
|
|
|
|
poll_sec=15
|
|
sync_arg=""
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--poll-sec)
|
|
poll_sec="${2:-15}"
|
|
shift 2
|
|
;;
|
|
--no-sync)
|
|
sync_arg="--no-sync"
|
|
shift
|
|
;;
|
|
*)
|
|
echo "[agent-framework] unknown argument: $1" >&2
|
|
usage
|
|
exit 1
|
|
;;
|
|
esac
|
|
done
|
|
|
|
mkdir -p "$ORCH_DIR/logs" "$ORCH_DIR/results"
|
|
|
|
is_running() {
|
|
[[ -f "$PID_FILE" ]] || return 1
|
|
local pid
|
|
pid="$(cat "$PID_FILE" 2>/dev/null || true)"
|
|
[[ -n "$pid" ]] || return 1
|
|
kill -0 "$pid" 2>/dev/null
|
|
}
|
|
|
|
case "$cmd" in
|
|
start)
|
|
if is_running; then
|
|
echo "[agent-framework] orchestrator already running (pid=$(cat "$PID_FILE"))"
|
|
exit 0
|
|
fi
|
|
nohup "$MOSAIC_HOME/bin/mosaic-orchestrator-drain" --poll-sec "$poll_sec" $sync_arg >"$LOG_FILE" 2>&1 &
|
|
echo "$!" > "$PID_FILE"
|
|
echo "[agent-framework] orchestrator started (pid=$!, log=$LOG_FILE)"
|
|
;;
|
|
drain)
|
|
exec "$MOSAIC_HOME/bin/mosaic-orchestrator-drain" --poll-sec "$poll_sec" $sync_arg
|
|
;;
|
|
stop)
|
|
if ! is_running; then
|
|
echo "[agent-framework] orchestrator not running"
|
|
rm -f "$PID_FILE"
|
|
exit 0
|
|
fi
|
|
pid="$(cat "$PID_FILE")"
|
|
kill "$pid" || true
|
|
rm -f "$PID_FILE"
|
|
echo "[agent-framework] orchestrator stopped (pid=$pid)"
|
|
;;
|
|
status)
|
|
if is_running; then
|
|
echo "[agent-framework] orchestrator running (pid=$(cat "$PID_FILE"), log=$LOG_FILE)"
|
|
else
|
|
echo "[agent-framework] orchestrator not running"
|
|
rm -f "$PID_FILE"
|
|
fi
|
|
;;
|
|
*)
|
|
usage
|
|
exit 1
|
|
;;
|
|
esac
|