66 lines
2.6 KiB
Bash
Executable file
66 lines
2.6 KiB
Bash
Executable file
#!/usr/bin/env bash
|
|
# =============================================================================
|
|
# prediction-poll.sh — Cron wrapper for prediction-agent (per-project)
|
|
#
|
|
# Runs hourly. Guards against concurrent runs and low memory.
|
|
# Iterates over all registered projects and runs prediction-agent.sh for each.
|
|
#
|
|
# Cron: 0 * * * * /path/to/disinto/planner/prediction-poll.sh
|
|
# Log: tail -f /path/to/disinto/planner/prediction.log
|
|
# =============================================================================
|
|
set -euo pipefail
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
|
FACTORY_ROOT="$(dirname "$SCRIPT_DIR")"
|
|
|
|
# shellcheck source=../lib/env.sh
|
|
source "$FACTORY_ROOT/lib/env.sh"
|
|
|
|
LOG_FILE="$SCRIPT_DIR/prediction.log"
|
|
LOCK_FILE="/tmp/prediction-poll.lock"
|
|
PROJECTS_DIR="$FACTORY_ROOT/projects"
|
|
|
|
log() { echo "[$(date -u +%Y-%m-%dT%H:%M:%S)Z] $*" >> "$LOG_FILE"; }
|
|
|
|
# ── Lock ──────────────────────────────────────────────────────────────────
|
|
if [ -f "$LOCK_FILE" ]; then
|
|
LOCK_PID=$(cat "$LOCK_FILE" 2>/dev/null || true)
|
|
if [ -n "$LOCK_PID" ] && kill -0 "$LOCK_PID" 2>/dev/null; then
|
|
log "poll: prediction running (PID $LOCK_PID)"
|
|
exit 0
|
|
fi
|
|
rm -f "$LOCK_FILE"
|
|
fi
|
|
echo $$ > "$LOCK_FILE"
|
|
trap 'rm -f "$LOCK_FILE"' EXIT
|
|
|
|
# ── Memory guard ──────────────────────────────────────────────────────────
|
|
AVAIL_MB=$(free -m | awk '/Mem:/{print $7}')
|
|
if [ "${AVAIL_MB:-0}" -lt 2000 ]; then
|
|
log "poll: skipping — only ${AVAIL_MB}MB available (need 2000)"
|
|
exit 0
|
|
fi
|
|
|
|
log "--- Prediction poll start ---"
|
|
|
|
# ── Iterate over projects ─────────────────────────────────────────────────
|
|
PROJECT_COUNT=0
|
|
if [ -d "$PROJECTS_DIR" ]; then
|
|
for project_toml in "$PROJECTS_DIR"/*.toml; do
|
|
[ -f "$project_toml" ] || continue
|
|
PROJECT_COUNT=$((PROJECT_COUNT + 1))
|
|
log "starting prediction-agent for $(basename "$project_toml")"
|
|
if ! "$SCRIPT_DIR/prediction-agent.sh" "$project_toml"; then
|
|
log "prediction-agent exited non-zero for $(basename "$project_toml")"
|
|
fi
|
|
done
|
|
fi
|
|
|
|
if [ "$PROJECT_COUNT" -eq 0 ]; then
|
|
log "No projects/*.toml found — running prediction-agent with .env defaults"
|
|
if ! "$SCRIPT_DIR/prediction-agent.sh"; then
|
|
log "prediction-agent exited non-zero"
|
|
fi
|
|
fi
|
|
|
|
log "--- Prediction poll done ---"
|