mirror of
https://github.com/affaan-m/everything-claude-code.git
synced 2026-04-10 03:13:29 +08:00
feat: add ecc2 diff viewer mode
This commit is contained in:
@@ -45,6 +45,7 @@ pub async fn run(db: StateStore, cfg: Config) -> Result<()> {
|
|||||||
(_, KeyCode::Char('i')) => dashboard.drain_inbox_selected().await,
|
(_, KeyCode::Char('i')) => dashboard.drain_inbox_selected().await,
|
||||||
(_, KeyCode::Char('g')) => dashboard.auto_dispatch_backlog().await,
|
(_, KeyCode::Char('g')) => dashboard.auto_dispatch_backlog().await,
|
||||||
(_, KeyCode::Char('G')) => dashboard.coordinate_backlog().await,
|
(_, KeyCode::Char('G')) => dashboard.coordinate_backlog().await,
|
||||||
|
(_, KeyCode::Char('v')) => dashboard.toggle_output_mode(),
|
||||||
(_, KeyCode::Char('p')) => dashboard.toggle_auto_dispatch_policy(),
|
(_, KeyCode::Char('p')) => dashboard.toggle_auto_dispatch_policy(),
|
||||||
(_, KeyCode::Char(',')) => dashboard.adjust_auto_dispatch_limit(-1),
|
(_, KeyCode::Char(',')) => dashboard.adjust_auto_dispatch_limit(-1),
|
||||||
(_, KeyCode::Char('.')) => dashboard.adjust_auto_dispatch_limit(1),
|
(_, KeyCode::Char('.')) => dashboard.adjust_auto_dispatch_limit(1),
|
||||||
|
|||||||
@@ -32,6 +32,7 @@ const MAX_PANE_SIZE_PERCENT: u16 = 80;
|
|||||||
const PANE_RESIZE_STEP_PERCENT: u16 = 5;
|
const PANE_RESIZE_STEP_PERCENT: u16 = 5;
|
||||||
const MAX_LOG_ENTRIES: u64 = 12;
|
const MAX_LOG_ENTRIES: u64 = 12;
|
||||||
const MAX_DIFF_PREVIEW_LINES: usize = 6;
|
const MAX_DIFF_PREVIEW_LINES: usize = 6;
|
||||||
|
const MAX_DIFF_PATCH_LINES: usize = 80;
|
||||||
|
|
||||||
pub struct Dashboard {
|
pub struct Dashboard {
|
||||||
db: StateStore,
|
db: StateStore,
|
||||||
@@ -53,6 +54,8 @@ pub struct Dashboard {
|
|||||||
logs: Vec<ToolLogEntry>,
|
logs: Vec<ToolLogEntry>,
|
||||||
selected_diff_summary: Option<String>,
|
selected_diff_summary: Option<String>,
|
||||||
selected_diff_preview: Vec<String>,
|
selected_diff_preview: Vec<String>,
|
||||||
|
selected_diff_patch: Option<String>,
|
||||||
|
output_mode: OutputMode,
|
||||||
selected_pane: Pane,
|
selected_pane: Pane,
|
||||||
selected_session: usize,
|
selected_session: usize,
|
||||||
show_help: bool,
|
show_help: bool,
|
||||||
@@ -85,6 +88,12 @@ enum Pane {
|
|||||||
Log,
|
Log,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
|
||||||
|
enum OutputMode {
|
||||||
|
SessionOutput,
|
||||||
|
WorktreeDiff,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Clone, Copy)]
|
#[derive(Debug, Clone, Copy)]
|
||||||
struct PaneAreas {
|
struct PaneAreas {
|
||||||
sessions: Rect,
|
sessions: Rect,
|
||||||
@@ -160,6 +169,8 @@ impl Dashboard {
|
|||||||
logs: Vec::new(),
|
logs: Vec::new(),
|
||||||
selected_diff_summary: None,
|
selected_diff_summary: None,
|
||||||
selected_diff_preview: Vec::new(),
|
selected_diff_preview: Vec::new(),
|
||||||
|
selected_diff_patch: None,
|
||||||
|
output_mode: OutputMode::SessionOutput,
|
||||||
selected_pane: Pane::Sessions,
|
selected_pane: Pane::Sessions,
|
||||||
selected_session: 0,
|
selected_session: 0,
|
||||||
show_help: false,
|
show_help: false,
|
||||||
@@ -319,27 +330,43 @@ impl Dashboard {
|
|||||||
fn render_output(&mut self, frame: &mut Frame, area: Rect) {
|
fn render_output(&mut self, frame: &mut Frame, area: Rect) {
|
||||||
self.sync_output_scroll(area.height.saturating_sub(2) as usize);
|
self.sync_output_scroll(area.height.saturating_sub(2) as usize);
|
||||||
|
|
||||||
let content = if self.sessions.get(self.selected_session).is_some() {
|
let (title, content) = if self.sessions.get(self.selected_session).is_some() {
|
||||||
let lines = self.selected_output_lines();
|
match self.output_mode {
|
||||||
|
OutputMode::SessionOutput => {
|
||||||
if lines.is_empty() {
|
let lines = self.selected_output_lines();
|
||||||
"Waiting for session output...".to_string()
|
let content = if lines.is_empty() {
|
||||||
} else {
|
"Waiting for session output...".to_string()
|
||||||
lines
|
} else {
|
||||||
.iter()
|
lines.iter().map(|line| line.text.as_str()).collect::<Vec<_>>().join("\n")
|
||||||
.map(|line| line.text.as_str())
|
};
|
||||||
.collect::<Vec<_>>()
|
(" Output ", content)
|
||||||
.join("\n")
|
}
|
||||||
|
OutputMode::WorktreeDiff => {
|
||||||
|
let content = self
|
||||||
|
.selected_diff_patch
|
||||||
|
.clone()
|
||||||
|
.or_else(|| {
|
||||||
|
self.selected_diff_summary.as_ref().map(|summary| {
|
||||||
|
format!(
|
||||||
|
"{summary}\n\nNo patch content to preview yet. The worktree may be clean or only have summary-level changes."
|
||||||
|
)
|
||||||
|
})
|
||||||
|
})
|
||||||
|
.unwrap_or_else(|| {
|
||||||
|
"No worktree diff available for the selected session.".to_string()
|
||||||
|
});
|
||||||
|
(" Diff ", content)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
"No sessions. Press 'n' to start one.".to_string()
|
(" Output ", "No sessions. Press 'n' to start one.".to_string())
|
||||||
};
|
};
|
||||||
|
|
||||||
let paragraph = Paragraph::new(content)
|
let paragraph = Paragraph::new(content)
|
||||||
.block(
|
.block(
|
||||||
Block::default()
|
Block::default()
|
||||||
.borders(Borders::ALL)
|
.borders(Borders::ALL)
|
||||||
.title(" Output ")
|
.title(title)
|
||||||
.border_style(self.pane_border_style(Pane::Output)),
|
.border_style(self.pane_border_style(Pane::Output)),
|
||||||
)
|
)
|
||||||
.scroll((self.output_scroll_offset as u16, 0));
|
.scroll((self.output_scroll_offset as u16, 0));
|
||||||
@@ -427,7 +454,7 @@ impl Dashboard {
|
|||||||
|
|
||||||
fn render_status_bar(&self, frame: &mut Frame, area: Rect) {
|
fn render_status_bar(&self, frame: &mut Frame, area: Rect) {
|
||||||
let text = format!(
|
let text = format!(
|
||||||
" [n]ew session [a]ssign re[b]alance global re[B]alance dra[i]n inbox [g]lobal dispatch coordinate [G]lobal toggle [p]olicy [,/.] dispatch limit [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 global re[B]alance dra[i]n inbox [g]lobal dispatch coordinate [G]lobal [v]iew diff toggle [p]olicy [,/.] dispatch limit [s]top [u]resume [x]cleanup [d]elete [r]efresh [Tab] switch pane [j/k] scroll [+/-] resize [{}] layout [?] help [q]uit ",
|
||||||
self.layout_label()
|
self.layout_label()
|
||||||
);
|
);
|
||||||
let text = if let Some(note) = self.operator_note.as_ref() {
|
let text = if let Some(note) = self.operator_note.as_ref() {
|
||||||
@@ -478,6 +505,7 @@ impl Dashboard {
|
|||||||
" i Drain unread task handoffs from selected lead",
|
" i Drain unread task handoffs from selected lead",
|
||||||
" g Auto-dispatch unread handoffs across lead sessions",
|
" g Auto-dispatch unread handoffs across lead sessions",
|
||||||
" G Dispatch then rebalance backlog across lead teams",
|
" G Dispatch then rebalance backlog across lead teams",
|
||||||
|
" v Toggle selected worktree diff in output pane",
|
||||||
" p Toggle daemon auto-dispatch policy and persist config",
|
" p Toggle daemon auto-dispatch policy and persist config",
|
||||||
" ,/. Decrease/increase auto-dispatch limit per lead",
|
" ,/. Decrease/increase auto-dispatch limit per lead",
|
||||||
" s Stop selected session",
|
" s Stop selected session",
|
||||||
@@ -669,6 +697,27 @@ impl Dashboard {
|
|||||||
self.refresh_logs();
|
self.refresh_logs();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn toggle_output_mode(&mut self) {
|
||||||
|
match self.output_mode {
|
||||||
|
OutputMode::SessionOutput => {
|
||||||
|
if self.selected_diff_patch.is_some() || self.selected_diff_summary.is_some() {
|
||||||
|
self.output_mode = OutputMode::WorktreeDiff;
|
||||||
|
self.selected_pane = Pane::Output;
|
||||||
|
self.output_follow = false;
|
||||||
|
self.output_scroll_offset = 0;
|
||||||
|
self.set_operator_note("showing selected worktree diff".to_string());
|
||||||
|
} else {
|
||||||
|
self.set_operator_note("no worktree diff for selected session".to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
OutputMode::WorktreeDiff => {
|
||||||
|
self.output_mode = OutputMode::SessionOutput;
|
||||||
|
self.reset_output_view();
|
||||||
|
self.set_operator_note("showing session output".to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub async fn assign_selected(&mut self) {
|
pub async fn assign_selected(&mut self) {
|
||||||
let Some(source_session) = self.sessions.get(self.selected_session) else {
|
let Some(source_session) = self.sessions.get(self.selected_session) else {
|
||||||
return;
|
return;
|
||||||
@@ -1270,6 +1319,11 @@ impl Dashboard {
|
|||||||
self.selected_diff_preview = worktree
|
self.selected_diff_preview = worktree
|
||||||
.and_then(|worktree| worktree::diff_file_preview(worktree, MAX_DIFF_PREVIEW_LINES).ok())
|
.and_then(|worktree| worktree::diff_file_preview(worktree, MAX_DIFF_PREVIEW_LINES).ok())
|
||||||
.unwrap_or_default();
|
.unwrap_or_default();
|
||||||
|
self.selected_diff_patch = worktree
|
||||||
|
.and_then(|worktree| worktree::diff_patch_preview(worktree, MAX_DIFF_PATCH_LINES).ok().flatten());
|
||||||
|
if self.output_mode == OutputMode::WorktreeDiff && self.selected_diff_patch.is_none() {
|
||||||
|
self.output_mode = OutputMode::SessionOutput;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn sync_selected_messages(&mut self) {
|
fn sync_selected_messages(&mut self) {
|
||||||
@@ -1950,6 +2004,20 @@ impl Dashboard {
|
|||||||
.collect::<Vec<_>>()
|
.collect::<Vec<_>>()
|
||||||
.join("\n")
|
.join("\n")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
fn rendered_output_text(&mut self, width: u16, height: u16) -> String {
|
||||||
|
let backend = ratatui::backend::TestBackend::new(width, height);
|
||||||
|
let mut terminal = ratatui::Terminal::new(backend).expect("terminal");
|
||||||
|
terminal.draw(|frame| self.render(frame)).expect("draw");
|
||||||
|
terminal
|
||||||
|
.backend()
|
||||||
|
.buffer()
|
||||||
|
.content()
|
||||||
|
.iter()
|
||||||
|
.map(|cell| cell.symbol())
|
||||||
|
.collect::<String>()
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Pane {
|
impl Pane {
|
||||||
@@ -2239,6 +2307,36 @@ mod tests {
|
|||||||
assert!(text.contains("Failed failed-8 | Render dashboard rows"));
|
assert!(text.contains("Failed failed-8 | Render dashboard rows"));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn toggle_output_mode_switches_to_worktree_diff_preview() {
|
||||||
|
let mut dashboard = test_dashboard(
|
||||||
|
vec![sample_session(
|
||||||
|
"focus-12345678",
|
||||||
|
"planner",
|
||||||
|
SessionState::Running,
|
||||||
|
Some("ecc/focus"),
|
||||||
|
512,
|
||||||
|
42,
|
||||||
|
)],
|
||||||
|
0,
|
||||||
|
);
|
||||||
|
dashboard.selected_diff_summary = Some("1 file changed".to_string());
|
||||||
|
dashboard.selected_diff_patch = Some(
|
||||||
|
"--- Branch diff vs main ---\ndiff --git a/src/lib.rs b/src/lib.rs\n+hello".to_string(),
|
||||||
|
);
|
||||||
|
|
||||||
|
dashboard.toggle_output_mode();
|
||||||
|
|
||||||
|
assert_eq!(dashboard.output_mode, OutputMode::WorktreeDiff);
|
||||||
|
assert_eq!(
|
||||||
|
dashboard.operator_note.as_deref(),
|
||||||
|
Some("showing selected worktree diff")
|
||||||
|
);
|
||||||
|
let rendered = dashboard.rendered_output_text(180, 30);
|
||||||
|
assert!(rendered.contains("Diff"));
|
||||||
|
assert!(rendered.contains("diff --git a/src/lib.rs b/src/lib.rs"));
|
||||||
|
}
|
||||||
|
|
||||||
#[test]
|
#[test]
|
||||||
fn selected_session_metrics_text_includes_team_capacity_summary() {
|
fn selected_session_metrics_text_includes_team_capacity_summary() {
|
||||||
let mut dashboard = test_dashboard(
|
let mut dashboard = test_dashboard(
|
||||||
@@ -3072,6 +3170,8 @@ mod tests {
|
|||||||
logs: Vec::new(),
|
logs: Vec::new(),
|
||||||
selected_diff_summary: None,
|
selected_diff_summary: None,
|
||||||
selected_diff_preview: Vec::new(),
|
selected_diff_preview: Vec::new(),
|
||||||
|
selected_diff_patch: None,
|
||||||
|
output_mode: OutputMode::SessionOutput,
|
||||||
selected_pane: Pane::Sessions,
|
selected_pane: Pane::Sessions,
|
||||||
selected_session,
|
selected_session,
|
||||||
show_help: false,
|
show_help: false,
|
||||||
|
|||||||
@@ -136,6 +136,34 @@ pub fn diff_file_preview(worktree: &WorktreeInfo, limit: usize) -> Result<Vec<St
|
|||||||
Ok(preview)
|
Ok(preview)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pub fn diff_patch_preview(worktree: &WorktreeInfo, max_lines: usize) -> Result<Option<String>> {
|
||||||
|
let mut remaining = max_lines.max(1);
|
||||||
|
let mut sections = Vec::new();
|
||||||
|
let base_ref = format!("{}...HEAD", worktree.base_branch);
|
||||||
|
|
||||||
|
let committed = git_diff_patch_lines(&worktree.path, &[&base_ref])?;
|
||||||
|
if !committed.is_empty() && remaining > 0 {
|
||||||
|
let taken = take_preview_lines(&committed, &mut remaining);
|
||||||
|
sections.push(format!(
|
||||||
|
"--- Branch diff vs {} ---\n{}",
|
||||||
|
worktree.base_branch,
|
||||||
|
taken.join("\n")
|
||||||
|
));
|
||||||
|
}
|
||||||
|
|
||||||
|
let working = git_diff_patch_lines(&worktree.path, &[])?;
|
||||||
|
if !working.is_empty() && remaining > 0 {
|
||||||
|
let taken = take_preview_lines(&working, &mut remaining);
|
||||||
|
sections.push(format!("--- Working tree diff ---\n{}", taken.join("\n")));
|
||||||
|
}
|
||||||
|
|
||||||
|
if sections.is_empty() {
|
||||||
|
Ok(None)
|
||||||
|
} else {
|
||||||
|
Ok(Some(sections.join("\n\n")))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
fn git_diff_shortstat(worktree_path: &Path, extra_args: &[&str]) -> Result<Option<String>> {
|
fn git_diff_shortstat(worktree_path: &Path, extra_args: &[&str]) -> Result<Option<String>> {
|
||||||
let mut command = Command::new("git");
|
let mut command = Command::new("git");
|
||||||
command
|
command
|
||||||
@@ -191,6 +219,31 @@ fn git_diff_name_status(worktree_path: &Path, extra_args: &[&str]) -> Result<Vec
|
|||||||
Ok(parse_nonempty_lines(&output.stdout))
|
Ok(parse_nonempty_lines(&output.stdout))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn git_diff_patch_lines(worktree_path: &Path, extra_args: &[&str]) -> Result<Vec<String>> {
|
||||||
|
let mut command = Command::new("git");
|
||||||
|
command
|
||||||
|
.arg("-C")
|
||||||
|
.arg(worktree_path)
|
||||||
|
.arg("diff")
|
||||||
|
.args(["--stat", "--patch", "--find-renames"]);
|
||||||
|
command.args(extra_args);
|
||||||
|
|
||||||
|
let output = command
|
||||||
|
.output()
|
||||||
|
.context("Failed to generate worktree patch preview")?;
|
||||||
|
|
||||||
|
if !output.status.success() {
|
||||||
|
let stderr = String::from_utf8_lossy(&output.stderr);
|
||||||
|
tracing::warn!(
|
||||||
|
"Worktree patch preview warning for {}: {stderr}",
|
||||||
|
worktree_path.display()
|
||||||
|
);
|
||||||
|
return Ok(Vec::new());
|
||||||
|
}
|
||||||
|
|
||||||
|
Ok(parse_nonempty_lines(&output.stdout))
|
||||||
|
}
|
||||||
|
|
||||||
fn git_status_short(worktree_path: &Path) -> Result<Vec<String>> {
|
fn git_status_short(worktree_path: &Path) -> Result<Vec<String>> {
|
||||||
let output = Command::new("git")
|
let output = Command::new("git")
|
||||||
.arg("-C")
|
.arg("-C")
|
||||||
@@ -220,6 +273,13 @@ fn parse_nonempty_lines(stdout: &[u8]) -> Vec<String> {
|
|||||||
.collect()
|
.collect()
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn take_preview_lines(lines: &[String], remaining: &mut usize) -> Vec<String> {
|
||||||
|
let count = (*remaining).min(lines.len());
|
||||||
|
let taken = lines.iter().take(count).cloned().collect::<Vec<_>>();
|
||||||
|
*remaining = remaining.saturating_sub(count);
|
||||||
|
taken
|
||||||
|
}
|
||||||
|
|
||||||
fn get_current_branch(repo_root: &Path) -> Result<String> {
|
fn get_current_branch(repo_root: &Path) -> Result<String> {
|
||||||
let output = Command::new("git")
|
let output = Command::new("git")
|
||||||
.arg("-C")
|
.arg("-C")
|
||||||
@@ -361,4 +421,57 @@ mod tests {
|
|||||||
let _ = fs::remove_dir_all(root);
|
let _ = fs::remove_dir_all(root);
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn diff_patch_preview_reports_branch_and_working_tree_sections() -> Result<()> {
|
||||||
|
let root = std::env::temp_dir().join(format!("ecc2-worktree-patch-{}", Uuid::new_v4()));
|
||||||
|
let repo = root.join("repo");
|
||||||
|
fs::create_dir_all(&repo)?;
|
||||||
|
|
||||||
|
run_git(&repo, &["init", "-b", "main"])?;
|
||||||
|
run_git(&repo, &["config", "user.email", "ecc@example.com"])?;
|
||||||
|
run_git(&repo, &["config", "user.name", "ECC"])?;
|
||||||
|
fs::write(repo.join("README.md"), "hello\n")?;
|
||||||
|
run_git(&repo, &["add", "README.md"])?;
|
||||||
|
run_git(&repo, &["commit", "-m", "init"])?;
|
||||||
|
|
||||||
|
let worktree_dir = root.join("wt-1");
|
||||||
|
run_git(
|
||||||
|
&repo,
|
||||||
|
&[
|
||||||
|
"worktree",
|
||||||
|
"add",
|
||||||
|
"-b",
|
||||||
|
"ecc/test",
|
||||||
|
worktree_dir.to_str().expect("utf8 path"),
|
||||||
|
"HEAD",
|
||||||
|
],
|
||||||
|
)?;
|
||||||
|
|
||||||
|
fs::write(worktree_dir.join("src.txt"), "branch\n")?;
|
||||||
|
run_git(&worktree_dir, &["add", "src.txt"])?;
|
||||||
|
run_git(&worktree_dir, &["commit", "-m", "branch file"])?;
|
||||||
|
fs::write(worktree_dir.join("README.md"), "hello\nworking\n")?;
|
||||||
|
|
||||||
|
let info = WorktreeInfo {
|
||||||
|
path: worktree_dir.clone(),
|
||||||
|
branch: "ecc/test".to_string(),
|
||||||
|
base_branch: "main".to_string(),
|
||||||
|
};
|
||||||
|
|
||||||
|
let preview = diff_patch_preview(&info, 40)?.expect("patch preview");
|
||||||
|
assert!(preview.contains("--- Branch diff vs main ---"));
|
||||||
|
assert!(preview.contains("--- Working tree diff ---"));
|
||||||
|
assert!(preview.contains("src.txt"));
|
||||||
|
assert!(preview.contains("README.md"));
|
||||||
|
|
||||||
|
let _ = Command::new("git")
|
||||||
|
.arg("-C")
|
||||||
|
.arg(&repo)
|
||||||
|
.args(["worktree", "remove", "--force"])
|
||||||
|
.arg(&worktree_dir)
|
||||||
|
.output();
|
||||||
|
let _ = fs::remove_dir_all(root);
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user