Your Name
112 lines
3.7 KiB
Bash
Executable File
112 lines
3.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT="$(cd "$(dirname "$0")/.." && pwd)"
|
|
NODE_BIN="${NODE_BIN:-/opt/homebrew/opt/node@24/bin/node}"
|
|
SCRIPT="${MEMIND_AGENT_RUN_WORKER_SCRIPT:-$ROOT/scripts/agent-run-worker.mjs}"
|
|
LABEL="${MEMIND_AGENT_RUN_WORKER_LABEL:-cn.tkmind.memind-agent-run-worker}"
|
|
PLIST="$HOME/Library/LaunchAgents/${LABEL}.plist"
|
|
LOG="${MEMIND_AGENT_RUN_WORKER_LOG:-$HOME/Library/Logs/memind-agent-run-worker.log}"
|
|
GUI="gui/$(id -u)"
|
|
POLL_MS="${MEMIND_AGENT_RUN_WORKER_POLL_MS:-3000}"
|
|
BATCH_SIZE="${MEMIND_AGENT_RUN_WORKER_BATCH_SIZE:-1}"
|
|
CONCURRENCY="${MEMIND_AGENT_RUN_QUEUE_CONCURRENCY:-1}"
|
|
RUN_TIMEOUT_MS="${MEMIND_AGENT_RUN_TIMEOUT_MS:-900000}"
|
|
TOOL_GATEWAY_ENABLED="${MEMIND_TOOL_GATEWAY_ENABLED:-0}"
|
|
TOOL_GATEWAY_DRY_RUN="${MEMIND_TOOL_GATEWAY_DRY_RUN:-0}"
|
|
WORKDIR_OVERRIDE="${MEMIND_AGENT_RUN_WORKDIR_OVERRIDE:-}"
|
|
WORKDIR_OVERRIDE_USER_ID="${MEMIND_AGENT_RUN_WORKDIR_USER_ID:-}"
|
|
START="${MEMIND_AGENT_RUN_WORKER_START:-0}"
|
|
|
|
mkdir -p "$HOME/Library/LaunchAgents" "$(dirname "$LOG")"
|
|
|
|
if [[ ! -x "$NODE_BIN" ]]; then
|
|
NODE_BIN="$(command -v node)"
|
|
fi
|
|
if [[ ! -f "$SCRIPT" ]]; then
|
|
echo "agent run worker script not found: $SCRIPT" >&2
|
|
exit 1
|
|
fi
|
|
|
|
cat > "$PLIST" <<EOF
|
|
<?xml version="1.0" encoding="UTF-8"?>
|
|
<!DOCTYPE plist PUBLIC "-//Apple//DTD PLIST 1.0//EN" "http://www.apple.com/DTDs/PropertyList-1.0.dtd">
|
|
<plist version="1.0">
|
|
<dict>
|
|
<key>Label</key>
|
|
<string>$LABEL</string>
|
|
<key>ProgramArguments</key>
|
|
<array>
|
|
<string>$NODE_BIN</string>
|
|
<string>$SCRIPT</string>
|
|
<string>--poll-ms</string>
|
|
<string>$POLL_MS</string>
|
|
<string>--limit</string>
|
|
<string>$BATCH_SIZE</string>
|
|
</array>
|
|
<key>WorkingDirectory</key>
|
|
<string>$ROOT</string>
|
|
<key>RunAtLoad</key>
|
|
<false/>
|
|
<key>KeepAlive</key>
|
|
<true/>
|
|
<key>ThrottleInterval</key>
|
|
<integer>10</integer>
|
|
<key>StandardOutPath</key>
|
|
<string>$LOG</string>
|
|
<key>StandardErrorPath</key>
|
|
<string>$LOG</string>
|
|
<key>EnvironmentVariables</key>
|
|
<dict>
|
|
<key>PATH</key>
|
|
<string>/opt/homebrew/bin:/opt/homebrew/opt/node@24/bin:/usr/local/bin:/usr/bin:/bin</string>
|
|
<key>MEMIND_AGENT_RUN_AUTODISPATCH</key>
|
|
<string>0</string>
|
|
<key>MEMIND_AGENT_RUN_WORKER_POLL_MS</key>
|
|
<string>$POLL_MS</string>
|
|
<key>MEMIND_AGENT_RUN_WORKER_BATCH_SIZE</key>
|
|
<string>$BATCH_SIZE</string>
|
|
<key>MEMIND_AGENT_RUN_QUEUE_CONCURRENCY</key>
|
|
<string>$CONCURRENCY</string>
|
|
<key>MEMIND_AGENT_RUN_TIMEOUT_MS</key>
|
|
<string>$RUN_TIMEOUT_MS</string>
|
|
<key>MEMIND_TOOL_GATEWAY_ENABLED</key>
|
|
<string>$TOOL_GATEWAY_ENABLED</string>
|
|
<key>MEMIND_TOOL_GATEWAY_DRY_RUN</key>
|
|
<string>$TOOL_GATEWAY_DRY_RUN</string>
|
|
<key>MEMIND_AGENT_RUN_WORKDIR_OVERRIDE</key>
|
|
<string>$WORKDIR_OVERRIDE</string>
|
|
<key>MEMIND_AGENT_RUN_WORKDIR_USER_ID</key>
|
|
<string>$WORKDIR_OVERRIDE_USER_ID</string>
|
|
</dict>
|
|
</dict>
|
|
</plist>
|
|
EOF
|
|
|
|
plutil -lint "$PLIST"
|
|
launchctl bootout "$GUI/$LABEL" 2>/dev/null || true
|
|
launchctl bootstrap "$GUI" "$PLIST"
|
|
|
|
if [[ "$START" == "1" || "$START" == "true" || "$START" == "yes" ]]; then
|
|
launchctl enable "$GUI/$LABEL"
|
|
launchctl kickstart -k "$GUI/$LABEL"
|
|
state="started"
|
|
else
|
|
launchctl disable "$GUI/$LABEL" 2>/dev/null || true
|
|
state="installed-disabled"
|
|
fi
|
|
|
|
echo "installed $PLIST"
|
|
echo "state: $state"
|
|
echo "script: $SCRIPT"
|
|
echo "poll_ms: $POLL_MS"
|
|
echo "batch_size: $BATCH_SIZE"
|
|
echo "concurrency: $CONCURRENCY"
|
|
echo "tool_gateway_enabled: $TOOL_GATEWAY_ENABLED"
|
|
echo "tool_gateway_dry_run: $TOOL_GATEWAY_DRY_RUN"
|
|
echo "workdir_override: $WORKDIR_OVERRIDE"
|
|
echo "workdir_override_user_id: $WORKDIR_OVERRIDE_USER_ID"
|
|
echo "log: $LOG"
|
|
echo "manual start: launchctl enable $GUI/$LABEL && launchctl kickstart -k $GUI/$LABEL"
|
|
echo "manual stop: launchctl bootout $GUI/$LABEL"
|