mirror of
https://github.com/tradecatlabs/vibe-coding-cn.git
synced 2026-08-08 16:47:47 +00:00
162 lines
4.1 KiB
Bash
162 lines
4.1 KiB
Bash
#!/usr/bin/env bash
|
|
# swarm-results: render terminal worker results as a delivery summary.
|
|
set -euo pipefail
|
|
|
|
usage() {
|
|
cat <<'EOF'
|
|
swarm-results: summarize DONE/FAIL/BLOCKED worker results
|
|
|
|
Usage:
|
|
swarm-results.sh [--dir DIR] [--out FILE] [--jsonl FILE] [--limit N]
|
|
|
|
Defaults:
|
|
--dir AUTO_TMUX_SWARM_DIR or /tmp/ai_swarm
|
|
--out stdout
|
|
--jsonl optional JSONL output for terminal results
|
|
--limit 80 result rows
|
|
|
|
This script is read-only. It does not mutate swarm state.
|
|
EOF
|
|
}
|
|
|
|
die() {
|
|
printf 'error: %s\n' "$*" >&2
|
|
exit 1
|
|
}
|
|
|
|
json_escape() {
|
|
local value="$1"
|
|
value="${value//\\/\\\\}"
|
|
value="${value//\"/\\\"}"
|
|
value="${value//$'\n'/\\n}"
|
|
value="${value//$'\r'/\\r}"
|
|
value="${value//$'\t'/\\t}"
|
|
printf '%s' "$value"
|
|
}
|
|
|
|
swarm_dir="${AUTO_TMUX_SWARM_DIR:-/tmp/ai_swarm}"
|
|
out_file=""
|
|
jsonl_file=""
|
|
limit=80
|
|
|
|
while [[ $# -gt 0 ]]; do
|
|
case "$1" in
|
|
--dir|--swarm-dir)
|
|
swarm_dir="${2:-}"
|
|
shift 2
|
|
;;
|
|
--out)
|
|
out_file="${2:-}"
|
|
shift 2
|
|
;;
|
|
--jsonl)
|
|
jsonl_file="${2:-}"
|
|
shift 2
|
|
;;
|
|
--limit)
|
|
limit="${2:-}"
|
|
shift 2
|
|
;;
|
|
-h|--help)
|
|
usage
|
|
exit 0
|
|
;;
|
|
*)
|
|
die "unknown option: $1"
|
|
;;
|
|
esac
|
|
done
|
|
|
|
[[ "$limit" =~ ^[0-9]+$ ]] || die "--limit must be a number"
|
|
|
|
tasks_tsv="$swarm_dir/tasks.tsv"
|
|
results_dir="$swarm_dir/results"
|
|
status_log="$swarm_dir/status.log"
|
|
[[ -f "$tasks_tsv" ]] || die "tasks file not found: $tasks_tsv"
|
|
|
|
render_results() {
|
|
printf '# auto-tmux Swarm Results\n\n'
|
|
printf -- '- dir: `%s`\n' "$swarm_dir"
|
|
printf -- '- created_at: `%s`\n\n' "$(date -Is)"
|
|
|
|
printf '## Summary\n\n'
|
|
awk -F '\t' '
|
|
NR > 1 {count[$2]++}
|
|
END {
|
|
print "| status | count |"
|
|
print "|:---|---:|"
|
|
n = split("DONE FAIL BLOCKED DOING TODO", order, " ")
|
|
for (i = 1; i <= n; i++) {
|
|
status = order[i]
|
|
print "| `" status "` | " (count[status] + 0) " |"
|
|
}
|
|
}
|
|
' "$tasks_tsv"
|
|
|
|
printf '\n## Terminal Results\n\n'
|
|
printf '| status | id | owner | text | result |\n'
|
|
printf '|:---|:---|:---|:---|:---|\n'
|
|
awk -F '\t' -v limit="$limit" '
|
|
function esc(value) {
|
|
gsub(/\|/, "\\|", value)
|
|
return value
|
|
}
|
|
NR > 1 && ($2 == "DONE" || $2 == "FAIL" || $2 == "BLOCKED") {
|
|
print "| `" esc($2) "` | `" esc($1) "` | `" esc($3) "` | " esc($4) " | " esc($5) " |"
|
|
shown++
|
|
if (shown >= limit) exit
|
|
}
|
|
END {if (shown == 0) print "| - | - | - | - | - |"}
|
|
' "$tasks_tsv"
|
|
|
|
printf '\n## Result Files\n\n'
|
|
if [[ -d "$results_dir" ]] && find "$results_dir" -mindepth 1 -maxdepth 1 -type f -name '*.txt' -print -quit | grep -q .; then
|
|
find "$results_dir" -mindepth 1 -maxdepth 1 -type f -name '*.txt' -print | sort | while read -r path; do
|
|
printf -- '- `%s`: %s\n' "$(basename "$path")" "$(head -n 1 "$path")"
|
|
done
|
|
else
|
|
printf -- '- no result files\n'
|
|
fi
|
|
|
|
printf '\n## Recent Terminal Status\n\n'
|
|
if [[ -s "$status_log" ]]; then
|
|
tail -n 30 "$status_log" | grep -E '\[(DONE|FAIL|BLOCKED)\]' | sed 's/^/- /' || printf -- '- no recent terminal status\n'
|
|
else
|
|
printf -- '- no status log\n'
|
|
fi
|
|
}
|
|
|
|
write_jsonl() {
|
|
local created_at id status owner text result
|
|
created_at="$(date -Is)"
|
|
mkdir -p "$(dirname "$jsonl_file")"
|
|
while IFS=$'\t' read -r id status owner text result; do
|
|
[[ "$id" == "id" ]] && continue
|
|
case "$status" in
|
|
DONE|FAIL|BLOCKED)
|
|
printf '{"type":"swarm-result","created_at":"%s","swarm_dir":"%s","id":"%s","status":"%s","owner":"%s","text":"%s","result":"%s"}\n' \
|
|
"$(json_escape "$created_at")" \
|
|
"$(json_escape "$swarm_dir")" \
|
|
"$(json_escape "$id")" \
|
|
"$(json_escape "$status")" \
|
|
"$(json_escape "$owner")" \
|
|
"$(json_escape "$text")" \
|
|
"$(json_escape "$result")"
|
|
;;
|
|
esac
|
|
done < "$tasks_tsv" > "$jsonl_file"
|
|
}
|
|
|
|
if [[ -n "$out_file" ]]; then
|
|
mkdir -p "$(dirname "$out_file")"
|
|
render_results > "$out_file"
|
|
printf 'swarm results written: %s\n' "$out_file"
|
|
else
|
|
render_results
|
|
fi
|
|
|
|
if [[ -n "$jsonl_file" ]]; then
|
|
write_jsonl
|
|
printf 'swarm results jsonl written: %s\n' "$jsonl_file"
|
|
fi
|