From bde186d987d806d274e47f8ed0827b9cc3652aca Mon Sep 17 00:00:00 2001
From: Affaan Mustafa <affaan@dcube.ai>
Date: Tue, 7 Apr 2026 13:16:14 -0700
Subject: [PATCH] feat: add ecc2 dashboard rebalance action

---
 ecc2/src/tui/app.rs       |  1 +
 ecc2/src/tui/dashboard.rs | 88 ++++++++++++++++++++++++++++++++++++++-
 2 files changed, 88 insertions(+), 1 deletion(-)

diff --git a/ecc2/src/tui/app.rs b/ecc2/src/tui/app.rs
index 13aeeb81..5b86edfd 100644
--- a/ecc2/src/tui/app.rs
+++ b/ecc2/src/tui/app.rs
@@ -40,6 +40,7 @@ pub async fn run(db: StateStore, cfg: Config) -> Result<()> {
                     (_, KeyCode::Char('k')) | (_, KeyCode::Up) => dashboard.scroll_up(),
                     (_, KeyCode::Char('n')) => dashboard.new_session().await,
                     (_, KeyCode::Char('a')) => dashboard.assign_selected().await,
+                    (_, KeyCode::Char('b')) => dashboard.rebalance_selected_team().await,
                     (_, KeyCode::Char('i')) => dashboard.drain_inbox_selected().await,
                     (_, KeyCode::Char('g')) => dashboard.auto_dispatch_backlog().await,
                     (_, KeyCode::Char('s')) => dashboard.stop_selected().await,
diff --git a/ecc2/src/tui/dashboard.rs b/ecc2/src/tui/dashboard.rs
index cbabc31a..dcbb412d 100644
--- a/ecc2/src/tui/dashboard.rs
+++ b/ecc2/src/tui/dashboard.rs
@@ -400,7 +400,7 @@ impl Dashboard {
 
     fn render_status_bar(&self, frame: &mut Frame, area: Rect) {
         let text = format!(
-            " [n]ew session  [a]ssign  dra[i]n inbox  [g]lobal dispatch  [s]top  [u]resume  [x]cleanup  [d]elete  [r]efresh  [Tab] switch pane  [j/k] scroll  [+/-] resize  [{}] layout  [?] help  [q]uit ",
+            " [n]ew session  [a]ssign  re[b]alance  dra[i]n inbox  [g]lobal dispatch  [s]top  [u]resume  [x]cleanup  [d]elete  [r]efresh  [Tab] switch pane  [j/k] scroll  [+/-] resize  [{}] layout  [?] help  [q]uit ",
             self.layout_label()
         );
         let text = if let Some(note) = self.operator_note.as_ref() {
@@ -446,6 +446,7 @@ impl Dashboard {
             "",
             "  n       New session",
             "  a       Assign follow-up work from selected session",
+            "  b       Rebalance backed-up delegate inboxes for selected lead",
             "  i       Drain unread task handoffs from selected session inbox",
             "  g       Auto-dispatch unread handoffs across lead sessions",
             "  s       Stop selected session",
@@ -677,6 +678,59 @@ impl Dashboard {
         self.refresh_logs();
     }
 
+    pub async fn rebalance_selected_team(&mut self) {
+        let Some(source_session) = self.sessions.get(self.selected_session) else {
+            return;
+        };
+
+        let agent = self.cfg.default_agent.clone();
+        let source_session_id = source_session.id.clone();
+        let outcomes = match manager::rebalance_team_backlog(
+            &self.db,
+            &self.cfg,
+            &source_session_id,
+            &agent,
+            true,
+            self.cfg.auto_dispatch_limit_per_session,
+        )
+        .await
+        {
+            Ok(outcomes) => outcomes,
+            Err(error) => {
+                tracing::warn!(
+                    "Failed to rebalance team backlog for session {}: {error}",
+                    source_session_id
+                );
+                self.set_operator_note(format!(
+                    "rebalance failed for {}: {error}",
+                    format_session_id(&source_session_id)
+                ));
+                return;
+            }
+        };
+
+        self.refresh();
+        self.sync_selection_by_id(Some(&source_session_id));
+        self.sync_selected_output();
+        self.sync_selected_diff();
+        self.sync_selected_messages();
+        self.sync_selected_lineage();
+        self.refresh_logs();
+
+        if outcomes.is_empty() {
+            self.set_operator_note(format!(
+                "no delegate backlog needed rebalancing for {}",
+                format_session_id(&source_session_id)
+            ));
+        } else {
+            self.set_operator_note(format!(
+                "rebalanced {} delegate handoff(s) for {}",
+                outcomes.len(),
+                format_session_id(&source_session_id)
+            ));
+        }
+    }
+
     pub async fn drain_inbox_selected(&mut self) {
         let Some(source_session) = self.sessions.get(self.selected_session) else {
             return;
@@ -2167,6 +2221,38 @@ mod tests {
         Ok(())
     }
 
+    #[tokio::test]
+    async fn rebalance_selected_team_sets_operator_note_when_clear() -> Result<()> {
+        let db_path = std::env::temp_dir().join(format!("ecc2-dashboard-{}.db", Uuid::new_v4()));
+        let db = StateStore::open(&db_path)?;
+        let now = Utc::now();
+
+        db.insert_session(&Session {
+            id: "lead-1".to_string(),
+            task: "coordinate".to_string(),
+            agent_type: "claude".to_string(),
+            working_dir: PathBuf::from("/tmp"),
+            state: SessionState::Running,
+            pid: None,
+            worktree: None,
+            created_at: now,
+            updated_at: now,
+            metrics: SessionMetrics::default(),
+        })?;
+
+        let dashboard_store = StateStore::open(&db_path)?;
+        let mut dashboard = Dashboard::new(dashboard_store, Config::default());
+        dashboard.rebalance_selected_team().await;
+
+        assert_eq!(
+            dashboard.operator_note.as_deref(),
+            Some("no delegate backlog needed rebalancing for lead-1")
+        );
+
+        let _ = std::fs::remove_file(db_path);
+        Ok(())
+    }
+
     #[test]
     fn grid_layout_renders_four_panes() {
         let mut dashboard = test_dashboard(vec![sample_session("grid-1", "claude", SessionState::Running, None, 1, 1)], 0);