antcolony

4.3 KB

1. #!/bin/bash
2. # docker/entrypoint.sh — mission 028 (ticket antcolony#1): the container's main process, started by docker/pivot.sh
3. # INSIDE the host's filesystem as uid 1000 (mre). Runs `./colony run [--live] [$COLONY_RUN_ARGS]` with mre's login
4. # environment, tees every line to `docker logs` (stdout) AND a rotating file log, and maps docker's ONE stop signal
5. # onto the colony wrapper's two-signal logic:
6. # docker stop (SIGTERM) → 1st TERM to the wrapper = "finish" (nothing new starts; idle → stops at once)
7. # → after COLONY_STOP_FINISH_SECONDS (20) still running → 2nd TERM = "park" (sessions killed +
8. # parked, resumed by the next start) → exit. docker-compose.yml stop_grace_period (75 s) must
9. # cover finish window + park; a 2nd SIGTERM here parks at once.
10. set -u
11. if [ "$(id -u)" = 0 ]; then echo "entrypoint: REFUSED — running as root (the scheduler and its workers run as uid 1000)"; exit 1; fi
12. here="$(cd "$(dirname "$0")/.." && pwd)"
14. # the host user's login environment (the same toolchain as a login shell of mre on the host)
15. HOME="$(getent passwd "$(id -u)" | cut -d: -f6)"
16. USER="$(id -un)"
17. export HOME USER LOGNAME="$USER" SHELL=/bin/bash
18. export PATH="$HOME/.local/bin:/usr/local/sbin:/usr/local/bin:/usr/bin:/bin"
19. if [ -r /etc/locale.conf ]; then set -a; . /etc/locale.conf; set +a; fi
20. cd "$here" || exit 1
22. # empty settings (compose writes VAR= for "not set") are unset, so the scheduler's own defaults apply
23. for v in ${!COLONY_@}; do [ -z "${!v}" ] && unset "$v"; done
24. export COLONY_TOKEN_FILE="${COLONY_TOKEN_FILE:-$HOME/.config/antcolony/tickets-token}"
25. # CLAUDE_CONFIG_DIR: empty → unset (claude uses ~/.claude, the host login); set → the workers log in with that folder's account
26. if [ -z "${CLAUDE_CONFIG_DIR:-}" ]; then unset CLAUDE_CONFIG_DIR; else export CLAUDE_CONFIG_DIR; fi
28. log_dir="${COLONY_LOG_DIR:-$here/logs}"
29. log_max="${COLONY_LOG_MAX_BYTES:-10485760}"
30. log_keep="${COLONY_LOG_KEEP:-5}"
31. finish_s="${COLONY_STOP_FINISH_SECONDS:-20}"
32. case "$log_max$log_keep$finish_s" in *[!0-9]*) echo "entrypoint: REFUSED — COLONY_LOG_MAX_BYTES / COLONY_LOG_KEEP / COLONY_STOP_FINISH_SECONDS must be whole numbers"; exit 1;; esac
33. [ "$log_keep" -ge 1 ] || log_keep=1
34. mkdir -p "$log_dir"
35. log="$log_dir/colony.log"
37. # file log: <UTC time> <line>, rotated at COLONY_LOG_MAX_BYTES into colony.log.1 … .COLONY_LOG_KEEP (oldest dropped)
38. rotate() {
39. local i
40. rm -f "$log.$log_keep"
41. for ((i = log_keep - 1; i >= 1; i--)); do [ -e "$log.$i" ] && mv -f "$log.$i" "$log.$((i + 1))"; done
42. mv -f "$log" "$log.1"
43. }
44. logger() {
45. local LC_ALL=C line stamped size
46. size=$(stat -c %s "$log" 2>/dev/null || echo 0)
47. while IFS= read -r line; do
48. printf '%s\n' "$line"
49. stamped="$(date -u +%Y-%m-%dT%H:%M:%SZ) $line"
50. printf '%s\n' "$stamped" >> "$log"
51. size=$((size + ${#stamped} + 1))
52. if [ "$size" -ge "$log_max" ]; then rotate; size=0; fi
53. done
54. }
55. exec > >(logger) 2>&1
56. logger_pid=$!
58. args=(run)
59. case "${COLONY_LIVE:-0}" in 1|yes|true) args+=(--live);; esac
60. if [ -n "${COLONY_RUN_ARGS:-}" ]; then read -r -a extra <<< "$COLONY_RUN_ARGS"; args+=("${extra[@]}"); fi
61. settings=""
62. for v in $(compgen -v COLONY_ | sort); do settings="$settings $v=${!v}"; done
63. echo "entrypoint: $(date -u +%Y-%m-%dT%H:%M:%SZ) uid $(id -u) ($USER) on $(cat /etc/hostname), pid $$ → ./colony ${args[*]}"
64. echo "entrypoint: settings:$settings"
65. echo "entrypoint: file log $log (rotates at $log_max bytes, keeps $log_keep); docker stop → finish, park after ${finish_s}s"
67. pid=""
68. timer=""
69. signals=0
70. on_stop() {
71. signals=$((signals + 1))
72. [ -n "$pid" ] || exit 0
73. if [ "$signals" -eq 1 ]; then
74. echo "entrypoint: stop signal → finish (nothing new starts); park after ${finish_s}s if a session still runs"
75. kill -TERM "$pid" 2>/dev/null
76. ( sleep "$finish_s" < /dev/null > /dev/null 2>&1
77. if kill -0 "$pid" 2>/dev/null; then echo "entrypoint: ${finish_s}s over → park"; kill -TERM "$pid" 2>/dev/null; fi ) &
78. timer=$!
79. else
80. echo "entrypoint: stop signal $signals → park now"
81. kill -TERM "$pid" 2>/dev/null
82. fi
83. }
84. trap on_stop TERM INT
86. "$here/colony" "${args[@]}" < /dev/null &
87. pid=$!
88. while kill -0 "$pid" 2>/dev/null; do wait "$pid"; done
89. wait "$pid"; rc=$?
90. [ -n "$timer" ] && kill "$timer" 2>/dev/null
91. echo "entrypoint: $(date -u +%Y-%m-%dT%H:%M:%SZ) ./colony run ended (status $rc)"
92. exec > /dev/null 2>&1
93. wait "$logger_pid"
94. exit "$rc"