agent-control-plane 0.2.0 → 0.3.0

package/tools/bin/heartbeat-loop-counting-lib.sh

@@ -0,0 +1,306 @@
+#!/usr/bin/env bash
+# heartbeat-loop-counting-lib.sh — worker counting, pending launch counts, capacity queries
+
+running_heavy_issue_workers() {
+  local session issue_id is_heavy count=0
+  ensure_running_issue_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    issue_id="$(issue_id_from_session "$session" || true)"
+    [[ -n "$issue_id" ]] || continue
+    is_heavy="$(cached_issue_attr heavy "$issue_id")"
+    if [[ "$is_heavy" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_issue_workers_cache"
+  printf '%s\n' "$count"
+}
+
+pending_issue_launch_count() {
+  local pending_file issue_id count=0
+  for pending_file in "${pending_launch_dir}"/issue-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    issue_id="${pending_file##*/issue-}"
+    issue_id="${issue_id%.pid}"
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_counts_toward_capacity "$issue_id"; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_pr_launch_count() {
+  local pending_file pr_id count=0
+  for pending_file in "${pending_launch_dir}"/pr-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    pr_id="${pending_file##*/pr-}"
+    pr_id="${pr_id%.pid}"
+    [[ -n "$pr_id" ]] || continue
+    if tmux has-session -t "${pr_prefix}${pr_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_pr_launch_active "$pr_id"; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_heavy_issue_launch_count() {
+  local pending_file issue_id count=0
+  for pending_file in "${pending_launch_dir}"/issue-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    issue_id="${pending_file##*/issue-}"
+    issue_id="${issue_id%.pid}"
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_counts_toward_capacity "$issue_id" && [[ "$(cached_issue_attr heavy "$issue_id")" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_scheduled_issue_launch_count() {
+  local pending_file issue_id count=0
+  for pending_file in "${pending_launch_dir}"/issue-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    issue_id="${pending_file##*/issue-}"
+    issue_id="${issue_id%.pid}"
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_counts_toward_capacity "$issue_id" && [[ "$(cached_issue_attr scheduled "$issue_id")" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_scheduled_heavy_issue_launch_count() {
+  local pending_file issue_id count=0
+  for pending_file in "${pending_launch_dir}"/issue-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    issue_id="${pending_file##*/issue-}"
+    issue_id="${issue_id%.pid}"
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_counts_toward_capacity "$issue_id" \
+      && [[ "$(cached_issue_attr scheduled "$issue_id")" == "yes" ]] \
+      && [[ "$(cached_issue_attr heavy "$issue_id")" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_recurring_issue_launch_count() {
+  local pending_file issue_id count=0
+  for pending_file in "${pending_launch_dir}"/issue-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    issue_id="${pending_file##*/issue-}"
+    issue_id="${issue_id%.pid}"
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_counts_toward_capacity "$issue_id" \
+      && [[ "$(cached_issue_attr scheduled "$issue_id")" != "yes" ]] \
+      && [[ "$(cached_issue_attr recurring "$issue_id")" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_blocked_recovery_issue_launch_count() {
+  local pending_file issue_id count=0
+  for pending_file in "${pending_launch_dir}"/issue-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    issue_id="${pending_file##*/issue-}"
+    issue_id="${issue_id%.pid}"
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_counts_toward_capacity "$issue_id" && blocked_recovery_issue_has_state "$issue_id"; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_exclusive_issue_launch_count() {
+  local pending_file issue_id count=0
+  for pending_file in "${pending_launch_dir}"/issue-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    issue_id="${pending_file##*/issue-}"
+    issue_id="${issue_id%.pid}"
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_counts_toward_capacity "$issue_id" && [[ "$(cached_issue_attr exclusive "$issue_id")" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+pending_exclusive_pr_launch_count() {
+  local pending_file pr_id count=0
+  for pending_file in "${pending_launch_dir}"/pr-*.pid; do
+    [[ -f "$pending_file" ]] || continue
+    pr_id="${pending_file##*/pr-}"
+    pr_id="${pr_id%.pid}"
+    [[ -n "$pr_id" ]] || continue
+    if tmux has-session -t "${pr_prefix}${pr_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_pr_launch_active "$pr_id" && [[ "$(cached_pr_is_exclusive "$pr_id")" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done
+  printf '%s\n' "$count"
+}
+
+running_non_recurring_issue_workers() {
+  local session issue_id is_recurring is_scheduled count=0
+  ensure_running_issue_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    issue_id="$(issue_id_from_session "$session" || true)"
+    [[ -n "$issue_id" ]] || continue
+    is_scheduled="$(cached_issue_attr scheduled "$issue_id")"
+    if [[ "$is_scheduled" == "yes" ]]; then
+      continue
+    fi
+    is_recurring="$(cached_issue_attr recurring "$issue_id")"
+    if [[ "$is_recurring" != "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_issue_workers_cache"
+  printf '%s\n' "$count"
+}
+
+running_recurring_issue_workers() {
+  local session issue_id is_recurring is_scheduled count=0
+  ensure_running_issue_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    issue_id="$(issue_id_from_session "$session" || true)"
+    [[ -n "$issue_id" ]] || continue
+    is_scheduled="$(cached_issue_attr scheduled "$issue_id")"
+    if [[ "$is_scheduled" == "yes" ]]; then
+      continue
+    fi
+    is_recurring="$(cached_issue_attr recurring "$issue_id")"
+    if [[ "$is_recurring" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_issue_workers_cache"
+  # Also count pending recurring launches that are still in progress
+  # (prevents infinite respawning when workers die before creating tmux sessions)
+  count=$((count + $(pending_recurring_issue_launch_count)))
+  printf '%s\n' "$count"
+}
+
+running_blocked_recovery_issue_workers() {
+  local session issue_id count=0
+  ensure_running_issue_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    issue_id="$(issue_id_from_session "$session" || true)"
+    [[ -n "$issue_id" ]] || continue
+    if blocked_recovery_issue_has_state "$issue_id"; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_issue_workers_cache"
+  printf '%s\n' "$count"
+}
+
+running_exclusive_issue_workers() {
+  local session issue_id is_exclusive count=0
+  ensure_running_issue_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    issue_id="$(issue_id_from_session "$session" || true)"
+    [[ -n "$issue_id" ]] || continue
+    is_exclusive="$(cached_issue_attr exclusive "$issue_id")"
+    if [[ "$is_exclusive" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_issue_workers_cache"
+  printf '%s\n' "$count"
+}
+
+running_exclusive_pr_workers() {
+  local session pr_id is_exclusive count=0
+  ensure_running_pr_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    pr_id="$(pr_id_from_session "$session" || true)"
+    [[ -n "$pr_id" ]] || continue
+    is_exclusive="$(cached_pr_is_exclusive "$pr_id")"
+    if [[ "$is_exclusive" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_pr_workers_cache"
+  printf '%s\n' "$count"
+}
+
+running_scheduled_issue_workers() {
+  local session issue_id is_scheduled count=0
+  ensure_running_issue_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    issue_id="$(issue_id_from_session "$session" || true)"
+    [[ -n "$issue_id" ]] || continue
+    is_scheduled="$(cached_issue_attr scheduled "$issue_id")"
+    if [[ "$is_scheduled" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_issue_workers_cache"
+  printf '%s\n' "$count"
+}
+
+running_scheduled_heavy_issue_workers() {
+  local session issue_id is_scheduled is_heavy count=0
+  ensure_running_issue_workers_cache
+  while IFS= read -r session; do
+    [[ -n "$session" ]] || continue
+    issue_id="$(issue_id_from_session "$session" || true)"
+    [[ -n "$issue_id" ]] || continue
+    is_scheduled="$(cached_issue_attr scheduled "$issue_id")"
+    is_heavy="$(cached_issue_attr heavy "$issue_id")"
+    if [[ "$is_scheduled" == "yes" && "$is_heavy" == "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$running_issue_workers_cache"
+  printf '%s\n' "$count"
+}
+
+ready_non_recurring_issue_count() {
+  local issue_id is_recurring count=0
+  ensure_ready_issue_ids_cache
+  while IFS= read -r issue_id; do
+    [[ -n "$issue_id" ]] || continue
+    if [[ "$(cached_issue_attr scheduled "$issue_id")" == "yes" ]]; then
+      continue
+    fi
+    is_recurring="$(cached_issue_attr recurring "$issue_id")"
+    if [[ "$is_recurring" != "yes" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$ready_issue_ids_cache"
+  printf '%s\n' "$count"
+}

package/tools/bin/heartbeat-loop-pr-strategy-lib.sh

@@ -0,0 +1,199 @@
+#!/usr/bin/env bash
+# heartbeat-loop-pr-strategy-lib.sh — PR candidate selection and strategy.
+#
+# Implements the priority-ordered lane dispatch for PR workers:
+# double-check-2 > double-check-1 > automerge > merge-repair > fix > ci-refresh.
+#
+# Depends on: heartbeat-loop-worker-lib.sh, heartbeat-loop-cache-lib.sh,
+#             heartbeat-loop-scheduling-lib.sh
+
+next_pr_candidate_json() {
+  local target_lane pr_number risk_json lane
+  ensure_open_agent_pr_ids_cache
+  for target_lane in double-check-2 double-check-1 automerge merge-repair fix ci-refresh; do
+    while IFS= read -r pr_number; do
+      [[ -n "$pr_number" ]] || continue
+      if tmux has-session -t "${pr_prefix}${pr_number}" 2>/dev/null; then
+        continue
+      fi
+      if pr_launch_reserved "$pr_number"; then
+        continue
+      fi
+      if pending_pr_launch_active "$pr_number"; then
+        continue
+      fi
+      if ! retry_ready pr "$pr_number"; then
+        continue
+      fi
+      risk_json="$(cached_pr_risk_json "$pr_number")"
+      lane="$(jq -r '.agentLane' <<<"$risk_json")"
+      if [[ "$lane" == "$target_lane" ]]; then
+        printf '%s\n' "$risk_json"
+        return 0
+      fi
+    done <<<"$open_agent_pr_ids_cache"
+  done
+}
+
+next_priority_review_pr_candidate_json() {
+  local target_lane pr_number risk_json lane
+  ensure_open_agent_pr_ids_cache
+  for target_lane in double-check-2 double-check-1; do
+    while IFS= read -r pr_number; do
+      [[ -n "$pr_number" ]] || continue
+      if tmux has-session -t "${pr_prefix}${pr_number}" 2>/dev/null; then
+        continue
+      fi
+      if pr_launch_reserved "$pr_number"; then
+        continue
+      fi
+      if pending_pr_launch_active "$pr_number"; then
+        continue
+      fi
+      if ! retry_ready pr "$pr_number"; then
+        continue
+      fi
+      risk_json="$(cached_pr_risk_json "$pr_number")"
+      lane="$(jq -r '.agentLane' <<<"$risk_json")"
+      if [[ "$lane" == "$target_lane" ]]; then
+        printf '%s\n' "$risk_json"
+        return 0
+      fi
+    done <<<"$open_agent_pr_ids_cache"
+  done
+}
+
+eligible_pr_backlog_count() {
+  local pr_number risk_json lane count=0
+  ensure_open_agent_pr_ids_cache
+  while IFS= read -r pr_number; do
+    [[ -n "$pr_number" ]] || continue
+    if tmux has-session -t "${pr_prefix}${pr_number}" 2>/dev/null; then
+      continue
+    fi
+    if pr_launch_reserved "$pr_number"; then
+      continue
+    fi
+    if pending_pr_launch_active "$pr_number"; then
+      continue
+    fi
+    if ! retry_ready pr "$pr_number"; then
+      continue
+    fi
+    risk_json="$(cached_pr_risk_json "$pr_number")"
+    lane="$(jq -r '.agentLane' <<<"$risk_json")"
+    case "$lane" in
+      double-check-1|double-check-2|automerge|merge-repair|fix)
+        count=$((count + 1))
+        ;;
+    esac
+  done <<<"$open_agent_pr_ids_cache"
+  printf '%s\n' "$count"
+}
+
+priority_review_backlog_count() {
+  local pr_number risk_json lane count=0
+  ensure_open_agent_pr_ids_cache
+  while IFS= read -r pr_number; do
+    [[ -n "$pr_number" ]] || continue
+    if tmux has-session -t "${pr_prefix}${pr_number}" 2>/dev/null; then
+      continue
+    fi
+    if pr_launch_reserved "$pr_number"; then
+      continue
+    fi
+    if pending_pr_launch_active "$pr_number"; then
+      continue
+    fi
+    if ! retry_ready pr "$pr_number"; then
+      continue
+    fi
+    risk_json="$(cached_pr_risk_json "$pr_number")"
+    lane="$(jq -r '.agentLane' <<<"$risk_json")"
+    case "$lane" in
+      double-check-1|double-check-2)
+        count=$((count + 1))
+        ;;
+    esac
+  done <<<"$open_agent_pr_ids_cache"
+  printf '%s\n' "$count"
+}
+
+next_exclusive_pr_candidate_json() {
+  local target_lane pr_number risk_json lane
+  ensure_exclusive_pr_ids_cache
+  for target_lane in double-check-2 double-check-1 automerge merge-repair fix ci-refresh; do
+    while IFS= read -r pr_number; do
+      [[ -n "$pr_number" ]] || continue
+      if tmux has-session -t "${pr_prefix}${pr_number}" 2>/dev/null; then
+        continue
+      fi
+      if pr_launch_reserved "$pr_number"; then
+        continue
+      fi
+      if pending_pr_launch_active "$pr_number"; then
+        continue
+      fi
+      if ! retry_ready pr "$pr_number"; then
+        continue
+      fi
+      risk_json="$(cached_pr_risk_json "$pr_number")"
+      lane="$(jq -r '.agentLane' <<<"$risk_json")"
+      # Skip PRs requiring human review; they should not hold exclusive lock
+      if [[ "$lane" == "human-review" ]]; then
+        continue
+      fi
+      if [[ "$lane" == "$target_lane" ]]; then
+        printf '%s\n' "$risk_json"
+        return 0
+      fi
+    done <<<"$exclusive_pr_ids_cache"
+  done
+}
+
+next_exclusive_issue_id() {
+  local issue_id
+  ensure_exclusive_issue_ids_cache
+  while IFS= read -r issue_id; do
+    [[ -n "$issue_id" ]] || continue
+    if tmux has-session -t "${issue_prefix}${issue_id}" 2>/dev/null; then
+      continue
+    fi
+    if pending_issue_launch_active "$issue_id"; then
+      continue
+    fi
+    if ! retry_ready issue "$issue_id"; then
+      continue
+    fi
+    printf '%s\n' "$issue_id"
+    return 0
+  done <<<"$exclusive_issue_ids_cache"
+}
+
+count_pr_lane() {
+  local target_lane="${1:?target lane required}"
+  local pr_number risk_json lane count=0
+  ensure_open_agent_pr_ids_cache
+  while IFS= read -r pr_number; do
+    [[ -n "$pr_number" ]] || continue
+    risk_json="$(cached_pr_risk_json "$pr_number")"
+    lane="$(jq -r '.agentLane' <<<"$risk_json")"
+    if [[ "$lane" == "$target_lane" ]]; then
+      count=$((count + 1))
+    fi
+  done <<<"$open_agent_pr_ids_cache"
+  printf '%s\n' "$count"
+}
+
+human_review_pr_ids() {
+  local pr_number risk_json lane
+  ensure_open_agent_pr_ids_cache
+  while IFS= read -r pr_number; do
+    [[ -n "$pr_number" ]] || continue
+    risk_json="$(cached_pr_risk_json "$pr_number")"
+    lane="$(jq -r '.agentLane' <<<"$risk_json")"
+    if [[ "$lane" == "human-review" ]]; then
+      printf '%s\n' "$pr_number"
+    fi
+  done <<<"$open_agent_pr_ids_cache"
+}