jarvis-ai-assistant 0.7.16__py3-none-any.whl → 1.0.2__py3-none-any.whl

jarvis/jarvis_code_agent/diff_visualizer.py

@@ -0,0 +1,998 @@
+# -*- coding: utf-8 -*-
+"""改进的 Diff 可视化工具
+
+提供多种 diff 可视化方式，改善代码变更的可读性。
+"""
+
+import difflib
+from typing import List
+from typing import Optional
+from typing import Union
+
+from rich.console import Console
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.table import Table
+from rich.text import Text
+
+LANGUAGE_EXTENSION_MAPPING = {
+    "py": "python",
+    "js": "javascript",
+    "ts": "typescript",
+    "java": "java",
+    "cpp": "cpp",
+    "cc": "cpp",
+    "cxx": "cpp",
+    "c": "c",
+    "h": "c",
+    "hpp": "cpp",
+    "rs": "rust",
+    "go": "go",
+    "sh": "bash",
+    "bash": "bash",
+    "zsh": "zsh",
+    "html": "html",
+    "css": "css",
+    "json": "json",
+    "yaml": "yaml",
+    "yml": "yaml",
+    "xml": "xml",
+    "md": "markdown",
+    "markdown": "markdown",
+    "toml": "toml",
+    "dockerfile": "dockerfile",
+    "ps1": "powershell",
+    "ini": "ini",
+    "make": "makefile",
+    "mk": "makefile",
+}
+
+
+class DiffVisualizer:
+    """改进的 Diff 可视化工具"""
+
+    def __init__(self, console: Optional[Console] = None):
+        """初始化可视化器
+
+        参数:
+            console: Rich Console 实例，如果为 None 则创建新实例
+        """
+        self.console = console or Console()
+
+    def _get_language_by_extension(self, file_path: str) -> str:
+        """根据文件扩展名推断编程语言
+
+        参数:
+            file_path: 文件路径
+
+        返回:
+            编程语言名称（用于 Syntax 高亮）
+        """
+        if not file_path:
+            return "text"
+
+        ext = file_path.lower().split(".")[-1] if "." in file_path else ""
+        return LANGUAGE_EXTENSION_MAPPING.get(ext, "text")
+
+    def visualize_unified_diff(
+        self,
+        diff_text: str,
+        file_path: str = "",
+        show_line_numbers: bool = True,
+        context_lines: int = 3,
+    ) -> None:
+        """可视化统一格式的 diff（改进版）
+
+        参数:
+            diff_text: git diff 输出的文本
+            file_path: 文件路径（用于显示标题）
+            show_line_numbers: 是否显示行号
+            context_lines: 上下文行数
+        """
+        if not diff_text.strip():
+            return
+
+        lines = diff_text.split("\n")
+
+        # 创建表格显示
+        table = Table(
+            show_header=False,
+            header_style="bold magenta",
+            box=None,  # 无边框，更简洁
+            padding=(0, 1),
+        )
+
+        if show_line_numbers:
+            table.add_column("", style="red", width=8, justify="right")
+            table.add_column("", style="green", width=8, justify="right")
+        table.add_column("", width=4, justify="center")
+        table.add_column("", style="white", overflow="fold")
+
+        old_line_num = 0
+        new_line_num = 0
+        in_hunk = False
+        hunk_lines: list = []  # 存储当前 hunk 中的所有行（包括上下文和变更）
+
+        def flush_hunk_context():
+            """刷新当前 hunk，只显示 context_lines 数量的上下文"""
+            nonlocal hunk_lines
+            if not hunk_lines:
+                return
+
+            # 找到所有变更行的索引
+            change_indices = []
+            for idx, (line_type, _, _, _) in enumerate(hunk_lines):
+                if line_type in ("-", "+"):
+                    change_indices.append(idx)
+
+            if not change_indices:
+                # 没有变更，只显示最后 context_lines 行上下文
+                for line_type, old_ln, new_ln, content in hunk_lines[-context_lines:]:
+                    if show_line_numbers:
+                        table.add_row(
+                            str(old_ln),
+                            str(new_ln),
+                            " ",
+                            f"{content}",
+                        )
+                    else:
+                        table.add_row("", " ", f"{content}")
+            else:
+                # 有变更，只显示变更前后的上下文
+                first_change_idx = change_indices[0]
+                last_change_idx = change_indices[-1]
+
+                # 显示变更前的上下文（最多 context_lines 行）
+                pre_context_start = max(0, first_change_idx - context_lines)
+                for idx in range(pre_context_start, first_change_idx):
+                    _, old_ln, new_ln, content = hunk_lines[idx]
+                    if show_line_numbers:
+                        table.add_row(
+                            str(old_ln),
+                            str(new_ln),
+                            " ",
+                            f"{content}",
+                        )
+                    else:
+                        table.add_row("", " ", f"{content}")
+
+                # 显示所有变更行
+                for idx in range(first_change_idx, last_change_idx + 1):
+                    line_type, old_ln, new_ln, content = hunk_lines[idx]
+                    if line_type == "-":
+                        if show_line_numbers:
+                            table.add_row(
+                                str(old_ln) if old_ln else "",
+                                "",
+                                "[bold red]-[/bold red]",
+                                f"[bright_red]{content}[/bright_red]",
+                            )
+                        else:
+                            table.add_row(
+                                "",
+                                "[bold red]-[/bold red]",
+                                f"[bright_red]{content}[/bright_red]",
+                            )
+                    elif line_type == "+":
+                        if show_line_numbers:
+                            table.add_row(
+                                "",
+                                str(new_ln) if new_ln else "",
+                                "[bold green]+[/bold green]",
+                                f"[bright_green]{content}[/bright_green]",
+                            )
+                        else:
+                            table.add_row(
+                                "",
+                                "[bold green]+[/bold green]",
+                                f"[bright_green]{content}[/bright_green]",
+                            )
+
+                # 显示变更后的上下文（最多 context_lines 行）
+                post_context_end = min(
+                    len(hunk_lines), last_change_idx + 1 + context_lines
+                )
+                for idx in range(last_change_idx + 1, post_context_end):
+                    _, old_ln, new_ln, content = hunk_lines[idx]
+                    if show_line_numbers:
+                        table.add_row(
+                            str(old_ln),
+                            str(new_ln),
+                            " ",
+                            f"{content}",
+                        )
+                    else:
+                        table.add_row("", " ", f"{content}")
+
+            hunk_lines = []
+
+        for line in lines:
+            if line.startswith("diff --git") or line.startswith("index"):
+                # 跳过 diff 头部
+                continue
+            elif line.startswith("---"):
+                # 旧文件路径
+                old_path = line[4:].strip()
+                if not file_path and old_path != "/dev/null":
+                    file_path = old_path
+            elif line.startswith("+++"):
+                # 新文件路径
+                new_path = line[4:].strip()
+                if new_path != "/dev/null":
+                    file_path = new_path
+            elif line.startswith("@@"):
+                # Hunk 头部 - 刷新上一个 hunk
+                if in_hunk:
+                    flush_hunk_context()
+
+                in_hunk = True
+                # 解析行号信息
+                parts = line.split("@@")
+                if len(parts) >= 2:
+                    hunk_info = parts[1].strip()
+                    if hunk_info:
+                        # 解析格式: -old_start,old_count +new_start,new_count
+                        old_part = ""
+                        new_part = ""
+                        for token in hunk_info.split():
+                            if token.startswith("-"):
+                                old_part = token[1:].split(",")[0]
+                            elif token.startswith("+"):
+                                new_part = token[1:].split(",")[0]
+
+                        if old_part:
+                            try:
+                                old_line_num = int(old_part)
+                            except ValueError:
+                                pass
+                        if new_part:
+                            try:
+                                new_line_num = int(new_part)
+                            except ValueError:
+                                pass
+
+                # 显示 hunk 头部
+                hunk_text = Text(f"[bright_cyan]{line}[/bright_cyan]", style="cyan")
+                if show_line_numbers:
+                    table.add_row("", "", "", hunk_text)
+                else:
+                    table.add_row("", "", hunk_text)
+            elif in_hunk:
+                if line.startswith("-"):
+                    # 删除的行
+                    content = line[1:] if len(line) > 1 else ""
+                    hunk_lines.append(("-", old_line_num, None, content))
+                    old_line_num += 1
+                elif line.startswith("+"):
+                    # 新增的行
+                    content = line[1:] if len(line) > 1 else ""
+                    hunk_lines.append(("+", None, new_line_num, content))
+                    new_line_num += 1
+                elif line.startswith(" "):
+                    # 未更改的行（上下文）
+                    content = line[1:] if len(line) > 1 else ""
+                    hunk_lines.append((" ", old_line_num, new_line_num, content))
+                    old_line_num += 1
+                    new_line_num += 1
+                elif line.strip() == "\\":
+                    # 文件末尾换行符差异
+                    if show_line_numbers:
+                        table.add_row(
+                            "",
+                            "",
+                            "",
+                            "[bright_yellow]\[/bright_yellow]",
+                        )
+                    else:
+                        table.add_row(
+                            "",
+                            "",
+                            "[bright_yellow]\[/bright_yellow]",
+                        )
+
+        # 刷新最后一个 hunk
+        if in_hunk:
+            flush_hunk_context()
+
+        # 显示 diff 表格（包裹在 Panel 中）
+        if table.rows:
+            title = f"📝 {file_path}" if file_path else "Diff"
+            panel = Panel(table, title=title, border_style="cyan", padding=(0, 1))
+            self.console.print(panel)
+
+    def visualize_statistics(
+        self, file_path: str, additions: int, deletions: int, total_changes: int = 0
+    ) -> None:
+        """显示文件变更统计
+
+        参数:
+            file_path: 文件路径
+            additions: 新增行数
+            deletions: 删除行数
+            total_changes: 总变更行数（如果为0则自动计算）
+        """
+        if total_changes == 0:
+            total_changes = additions + deletions
+
+        # 创建统计文本
+        stats_text = Text()
+        stats_text.append(f"📊 {file_path}\n", style="bold cyan")
+        stats_text.append("  ", style="bright_white")
+        stats_text.append("➕ 新增: ", style="green")
+        stats_text.append(f"{additions} 行", style="bold green")
+        stats_text.append("  |  ", style="bright_white")
+        stats_text.append("➖ 删除: ", style="red")
+        stats_text.append(f"{deletions} 行", style="bold red")
+        if total_changes > 0:
+            stats_text.append("  |  ", style="bright_white")
+            stats_text.append("📈 总计: ", style="cyan")
+            stats_text.append(f"{total_changes} 行", style="bold cyan")
+
+        panel = Panel(stats_text, border_style="cyan", padding=(1, 2))
+        self.console.print(panel)
+
+    def visualize_syntax_highlighted(
+        self, diff_text: str, file_path: str = "", theme: str = "monokai"
+    ) -> None:
+        """使用语法高亮显示 diff（保持原有风格但改进）
+
+        参数:
+            diff_text: git diff 输出的文本
+            file_path: 文件路径
+            theme: 语法高亮主题
+        """
+        if not diff_text.strip():
+            return
+
+        # 使用 Rich 的 diff 语法高亮
+        syntax = Syntax(
+            diff_text,
+            "diff",
+            theme=theme,
+            line_numbers=True,
+            word_wrap=True,
+            background_color="default",
+        )
+
+        if file_path:
+            panel = Panel(
+                syntax,
+                title=f"📝 {file_path}",
+                border_style="cyan",
+                padding=(0, 1),
+            )
+            self.console.print(panel)
+        else:
+            self.console.print(syntax)
+
+    def visualize_compact(
+        self,
+        diff_text: str,
+        file_path: str = "",
+        max_lines: int = 50,
+    ) -> None:
+        """紧凑型 diff 显示（适合快速预览）
+
+        参数:
+            diff_text: git diff 输出的文本
+            file_path: 文件路径
+            max_lines: 最大显示行数
+        """
+        if not diff_text.strip():
+            return
+
+        lines = diff_text.split("\n")
+        display_lines = lines[:max_lines]
+
+        # 统计信息
+        additions = sum(
+            1
+            for line in display_lines
+            if line.startswith("+") and not line.startswith("+++")
+        )
+        deletions = sum(
+            1
+            for line in display_lines
+            if line.startswith("-") and not line.startswith("---")
+        )
+
+        # 显示 diff（使用语法高亮，包裹在 Panel 中）
+        if len(lines) > max_lines:
+            remaining = len(lines) - max_lines
+            display_text = "\n".join(display_lines)
+            display_text += f"\n... ({remaining} 行已省略)"
+        else:
+            display_text = "\n".join(display_lines)
+
+        syntax = Syntax(
+            display_text,
+            "diff",
+            theme="monokai",
+            line_numbers=False,
+            word_wrap=True,
+        )
+
+        # 构建标题（包含统计信息）
+        title = f"📝 {file_path}" if file_path else "Diff"
+        if additions > 0 or deletions > 0:
+            title += f"  [green]+{additions}[/green] / [red]-{deletions}[/red]"
+
+        panel = Panel(syntax, title=title, border_style="cyan", padding=(0, 1))
+        self.console.print(panel)
+
+    def visualize_side_by_side_summary(
+        self,
+        old_lines: List[str],
+        new_lines: List[str],
+        file_path: str = "",
+        context_lines: int = 3,
+        old_line_map: Optional[List[int]] = None,
+        new_line_map: Optional[List[int]] = None,
+    ) -> None:
+        """并排显示摘要（仅显示变更部分，智能配对）
+
+        参数:
+            old_lines: 旧文件行列表
+            new_lines: 新文件行列表
+            file_path: 文件路径
+            context_lines: 上下文行数
+            old_line_map: 旧文件行号映射（索引对应 old_lines 索引，值为文件中的绝对行号）
+            new_line_map: 新文件行号映射（索引对应 new_lines 索引，值为文件中的绝对行号）
+        """
+        # 使用 difflib.SequenceMatcher 进行更精确的匹配
+        matcher = difflib.SequenceMatcher(None, old_lines, new_lines)
+        opcodes = matcher.get_opcodes()
+
+        # 如果没有提供行号映射，使用索引+1作为行号（向后兼容）
+        if old_line_map is None:
+            old_line_map = [i + 1 for i in range(len(old_lines))]
+        if new_line_map is None:
+            new_line_map = [i + 1 for i in range(len(new_lines))]
+
+        # 获取语言类型用于语法高亮
+        language = self._get_language_by_extension(file_path)
+
+        # 创建并排表格 - 使用最大可用宽度
+        table = Table(
+            show_header=True,
+            header_style="bold magenta",
+            box=None,
+            padding=(0, 0),  # 移除内部padding让内容更紧凑
+            expand=True,  # 让表格占满可用宽度
+        )
+        table.add_column("", style="bright_cyan", width=6, justify="right")
+        table.add_column(
+            "", style="bright_white", overflow="fold", ratio=10
+        )  # 增加ratio值获得更多空间
+        table.add_column("", style="bright_cyan", width=6, justify="right")
+        table.add_column(
+            "", style="bright_white", overflow="fold", ratio=10
+        )  # 增加ratio值获得更多空间
+
+        additions = 0
+        deletions = 0
+        has_changes = False
+
+        # 用于跟踪前一个变更块的结束位置
+        prev_change_end_old = 0
+        prev_change_end_new = 0
+
+        for idx, (tag, i1, i2, j1, j2) in enumerate(opcodes):
+            # 检测是否需要在变更块之间插入分隔符
+            if idx > 0 and tag in ("replace", "delete", "insert"):
+                # 使用实际行号检查连续性
+                prev_old_line = (
+                    old_line_map[prev_change_end_old - 1]
+                    if prev_change_end_old > 0
+                    and prev_change_end_old - 1 < len(old_line_map)
+                    else None
+                )
+                curr_old_line = old_line_map[i1] if i1 < len(old_line_map) else None
+                prev_new_line = (
+                    new_line_map[prev_change_end_new - 1]
+                    if prev_change_end_new > 0
+                    and prev_change_end_new - 1 < len(new_line_map)
+                    else None
+                )
+                curr_new_line = new_line_map[j1] if j1 < len(new_line_map) else None
+
+                # 只有当所有行号都有效且都不连续时才添加分割线
+                should_add_separator = False
+                if (
+                    prev_old_line is not None
+                    and curr_old_line is not None
+                    and prev_new_line is not None
+                    and curr_new_line is not None
+                ):
+                    # 检查实际行号是否不连续（两个方向都不连续）
+                    old_not_continuous = curr_old_line != prev_old_line + 1
+                    new_not_continuous = curr_new_line != prev_new_line + 1
+
+                    # 只有当两个方向都不连续时才添加分割线
+                    should_add_separator = old_not_continuous and new_not_continuous
+
+                if should_add_separator:
+                    # 动态计算分隔符宽度（基于终端宽度，限制在合理范围）
+                    terminal_width = self.console.width or 120
+                    # 表格有两列内容区域（每列约占总宽度的45%，减去行号列）
+                    separator_width = max(20, min(50, int(terminal_width * 0.4)))
+                    separator_text = "─" * separator_width
+                    separator = Text(separator_text, style="bright_black dim")
+                    table.add_row(
+                        "",
+                        separator,
+                        "",
+                        separator,
+                    )
+
+            # 只有在处理变更块时才更新结束位置
+            if tag in ("replace", "delete", "insert"):
+                prev_change_end_old = i2
+                prev_change_end_new = j2
+
+            if tag == "equal":
+                # 显示未更改的行（灰色/dim样式），但只显示上下文行数
+                equal_chunk = old_lines[i1:i2]
+                equal_len = len(equal_chunk)
+
+                # 只显示最后 context_lines 行作为上下文
+                if equal_len > context_lines * 2:
+                    # 如果 equal 块太长，只显示开头和结尾的上下文
+                    if idx > 0:  # 不是第一个块
+                        # 显示开头的 context_lines 行
+                        for k in range(min(context_lines, equal_len)):
+                            old_line_num = (
+                                old_line_map[i1 + k]
+                                if i1 + k < len(old_line_map)
+                                else i1 + k + 1
+                            )
+                            new_line_num = (
+                                new_line_map[j1 + k]
+                                if j1 + k < len(new_line_map)
+                                else j1 + k + 1
+                            )
+                            old_syntax = Syntax(
+                                equal_chunk[k],
+                                language,
+                                theme="monokai",
+                                background_color="default",
+                            )
+                            new_syntax = Syntax(
+                                equal_chunk[k],
+                                language,
+                                theme="monokai",
+                                background_color="default",
+                            )
+                            table.add_row(
+                                f"[bright_cyan]{old_line_num}[/bright_cyan]",
+                                old_syntax,
+                                f"[bright_cyan]{new_line_num}[/bright_cyan]",
+                                new_syntax,
+                            )
+                        # 如果有省略，显示省略标记
+                        if equal_len > context_lines * 2:
+                            table.add_row(
+                                "",
+                                "[bright_yellow]... ({0} lines omitted) ...[/bright_yellow]".format(
+                                    equal_len - context_lines * 2
+                                ),
+                                "",
+                                "",
+                            )
+                        # 显示结尾的 context_lines 行
+                        for k in range(max(0, equal_len - context_lines), equal_len):
+                            old_line_num = (
+                                old_line_map[i1 + k]
+                                if i1 + k < len(old_line_map)
+                                else i1 + k + 1
+                            )
+                            new_line_num = (
+                                new_line_map[j1 + k]
+                                if j1 + k < len(new_line_map)
+                                else j1 + k + 1
+                            )
+                            old_syntax = Syntax(
+                                equal_chunk[k],
+                                language,
+                                theme="monokai",
+                                background_color="default",
+                            )
+                            new_syntax = Syntax(
+                                equal_chunk[k],
+                                language,
+                                theme="monokai",
+                                background_color="default",
+                            )
+                            table.add_row(
+                                f"[bright_cyan]{old_line_num}[/bright_cyan]",
+                                old_syntax,
+                                f"[bright_cyan]{new_line_num}[/bright_cyan]",
+                                new_syntax,
+                            )
+                    else:
+                        # 第一个块，只显示结尾的上下文
+                        start_idx = max(0, equal_len - context_lines)
+                        for k in range(start_idx, equal_len):
+                            old_line_num = (
+                                old_line_map[i1 + k]
+                                if i1 + k < len(old_line_map)
+                                else i1 + k + 1
+                            )
+                            new_line_num = (
+                                new_line_map[j1 + k]
+                                if j1 + k < len(new_line_map)
+                                else j1 + k + 1
+                            )
+                            old_syntax = Syntax(
+                                equal_chunk[k],
+                                language,
+                                theme="monokai",
+                                background_color="default",
+                            )
+                            new_syntax = Syntax(
+                                equal_chunk[k],
+                                language,
+                                theme="monokai",
+                                background_color="default",
+                            )
+                            table.add_row(
+                                f"[bright_cyan]{old_line_num}[/bright_cyan]",
+                                old_syntax,
+                                f"[bright_cyan]{new_line_num}[/bright_cyan]",
+                                new_syntax,
+                            )
+                else:
+                    # 如果 equal 块不长，显示所有行
+                    for k, line in enumerate(equal_chunk):
+                        old_line_num = (
+                            old_line_map[i1 + k]
+                            if i1 + k < len(old_line_map)
+                            else i1 + k + 1
+                        )
+                        new_line_num = (
+                            new_line_map[j1 + k]
+                            if j1 + k < len(new_line_map)
+                            else j1 + k + 1
+                        )
+                        old_syntax = Syntax(
+                            line,
+                            language,
+                            theme="monokai",
+                            background_color="default",
+                        )
+                        new_syntax = Syntax(
+                            line,
+                            language,
+                            theme="monokai",
+                            background_color="default",
+                        )
+                        table.add_row(
+                            f"[bright_cyan]{old_line_num}[/bright_cyan]",
+                            old_syntax,
+                            f"[bright_cyan]{new_line_num}[/bright_cyan]",
+                            new_syntax,
+                        )
+                continue
+            elif tag == "replace":
+                # 替换：删除的行和新增的行配对显示
+                old_chunk = old_lines[i1:i2]
+                new_chunk = new_lines[j1:j2]
+                deletions += len(old_chunk)
+                additions += len(new_chunk)
+                has_changes = True
+
+                # 配对显示
+                max_len = max(len(old_chunk), len(new_chunk))
+                for k in range(max_len):
+                    # 文件中的绝对行号
+                    if k < len(old_chunk):
+                        old_line_num_actual: Union[int, str] = (
+                            old_line_map[i1 + k]
+                            if i1 + k < len(old_line_map)
+                            else i1 + k + 1
+                        )
+                        old_replace_syntax: Union[Syntax, str] = Syntax(
+                            old_chunk[k],
+                            language,
+                            theme="monokai",
+                            background_color="#5c0000",
+                        )
+                    else:
+                        old_line_num_actual = ""
+                        old_replace_syntax = ""
+
+                    if k < len(new_chunk):
+                        new_line_num_actual: Union[int, str] = (
+                            new_line_map[j1 + k]
+                            if j1 + k < len(new_line_map)
+                            else j1 + k + 1
+                        )
+                        new_replace_syntax: Union[Syntax, str] = Syntax(
+                            new_chunk[k],
+                            language,
+                            theme="monokai",
+                            background_color="#004d00",
+                        )
+                    else:
+                        new_line_num_actual = ""
+                        new_replace_syntax = ""
+
+                    table.add_row(
+                        str(old_line_num_actual),
+                        old_replace_syntax,
+                        str(new_line_num_actual),
+                        new_replace_syntax,
+                    )
+            elif tag == "delete":
+                # 仅删除
+                old_chunk = old_lines[i1:i2]
+                deletions += len(old_chunk)
+                has_changes = True
+                for k, line in enumerate(old_chunk):
+                    # 文件中的绝对行号
+                    old_line_num = (
+                        old_line_map[i1 + k]
+                        if i1 + k < len(old_line_map)
+                        else i1 + k + 1
+                    )
+                    old_delete_syntax: Union[Syntax, str] = Syntax(
+                        line, language, theme="monokai", background_color="#5c0000"
+                    )
+                    table.add_row(
+                        str(old_line_num),
+                        old_delete_syntax,
+                        "",
+                        "",
+                    )
+            elif tag == "insert":
+                # 仅新增
+                new_chunk = new_lines[j1:j2]
+                additions += len(new_chunk)
+                has_changes = True
+                for k, line in enumerate(new_chunk):
+                    # 文件中的绝对行号
+                    new_line_num = (
+                        new_line_map[j1 + k]
+                        if j1 + k < len(new_line_map)
+                        else j1 + k + 1
+                    )
+                    new_insert_syntax: Union[Syntax, str] = Syntax(
+                        line, language, theme="monokai", background_color="#004d00"
+                    )
+                    table.add_row(
+                        "",
+                        "",
+                        str(new_line_num),
+                        new_insert_syntax,
+                    )
+
+        # 如果没有变更，显示提示
+        if not has_changes:
+            self.console.print("[dim]（无变更）[/dim]")
+            return
+
+        # 构建标题（包含统计信息）
+        title = f"📝 {file_path}" if file_path else "Side-by-Side Diff"
+        title += f"  [bright_green]+{additions}[/bright_green] / [bright_red]-{deletions}[/bright_red]"
+
+        # 包裹在 Panel 中显示 - 最小化padding以最大化内容区域
+        panel = Panel(table, title=title, border_style="bright_cyan", padding=(0, 0))
+        self.console.print(panel)
+
+
+def _split_diff_by_files(diff_text: str) -> List[tuple]:
+    """将包含多个文件的 diff 文本分割成单个文件的 diff
+
+    参数:
+        diff_text: git diff 输出的文本（可能包含多个文件）
+
+    返回:
+        List[tuple]: [(file_path, single_file_diff), ...] 列表
+    """
+    files = []
+    lines = diff_text.splitlines()
+    current_file_path = ""
+    current_file_lines: list = []
+
+    i = 0
+    while i < len(lines):
+        line = lines[i]
+
+        if line.startswith("diff --git"):
+            # 如果已经有文件在处理，先保存它
+            if current_file_lines:
+                files.append((current_file_path, "\n".join(current_file_lines)))
+
+            # 开始新文件
+            current_file_lines = [line]
+            # 尝试从 diff --git 行提取文件路径
+            # 格式: diff --git a/path b/path
+            parts = line.split()
+            if len(parts) >= 4:
+                # 取 b/path 部分，去掉 b/ 前缀
+                path_part = parts[3]
+                if path_part.startswith("b/"):
+                    current_file_path = path_part[2:]
+                else:
+                    current_file_path = path_part
+            else:
+                current_file_path = ""
+        elif line.startswith("---") or line.startswith("+++"):
+            # 更新文件路径（优先使用 +++ 行的路径）
+            current_file_lines.append(line)
+            if line.startswith("+++"):
+                path_part = line[4:].strip()
+                if path_part != "/dev/null":
+                    # 去掉 a/ 或 b/ 前缀
+                    if path_part.startswith("b/"):
+                        current_file_path = path_part[2:]
+                    elif path_part.startswith("a/"):
+                        current_file_path = path_part[2:]
+                    else:
+                        current_file_path = path_part
+        else:
+            # 其他行添加到当前文件
+            current_file_lines.append(line)
+
+        i += 1
+
+    # 保存最后一个文件
+    if current_file_lines:
+        files.append((current_file_path, "\n".join(current_file_lines)))
+
+    # 如果没有找到任何文件分隔符，整个 diff 作为一个文件处理
+    if not files:
+        # 尝试从 --- 或 +++ 行提取文件路径
+        file_path = ""
+        for line in lines:
+            if line.startswith("+++"):
+                path_part = line[4:].strip()
+                if path_part != "/dev/null":
+                    if path_part.startswith("b/"):
+                        file_path = path_part[2:]
+                    elif path_part.startswith("a/"):
+                        file_path = path_part[2:]
+                    else:
+                        file_path = path_part
+                    break
+
+        files.append((file_path, diff_text))
+
+    return files
+
+
+def _parse_diff_to_lines(diff_text: str) -> tuple:
+    """从 git diff 文本中解析出旧文件和新文件的行列表（带行号信息）
+
+    参数:
+        diff_text: git diff 输出的文本
+
+    返回:
+        (old_lines, new_lines, old_line_map, new_line_map):
+        旧文件行列表、新文件行列表、旧文件行号映射、新文件行号映射
+        行号映射是一个列表，索引对应行列表的索引，值是该行在文件中的绝对行号
+    """
+    old_lines = []
+    new_lines = []
+    old_line_map = []  # 映射 old_lines 索引到文件中的绝对行号
+    new_line_map = []  # 映射 new_lines 索引到文件中的绝对行号
+
+    old_line_num = 0
+    new_line_num = 0
+
+    for line in diff_text.splitlines():
+        if line.startswith("@@"):
+            # 解析 hunk 头，获取起始行号
+            parts = line.split("@@")
+            if len(parts) >= 2:
+                hunk_info = parts[1].strip()
+                if hunk_info:
+                    for token in hunk_info.split():
+                        if token.startswith("-"):
+                            try:
+                                old_line_num = int(token[1:].split(",")[0])
+                            except ValueError:
+                                pass
+                        elif token.startswith("+"):
+                            try:
+                                new_line_num = int(token[1:].split(",")[0])
+                            except ValueError:
+                                pass
+            continue
+        elif line.startswith("---") or line.startswith("+++"):
+            # 跳过文件头
+            continue
+        elif line.startswith("diff ") or line.startswith("index "):
+            # 跳过 diff 元信息
+            continue
+        elif line.startswith("-"):
+            # 删除的行
+            old_lines.append(line[1:])
+            old_line_map.append(old_line_num)
+            old_line_num += 1
+        elif line.startswith("+"):
+            # 新增的行
+            new_lines.append(line[1:])
+            new_line_map.append(new_line_num)
+            new_line_num += 1
+        elif line.startswith(" "):
+            # 未更改的行
+            old_lines.append(line[1:])
+            new_lines.append(line[1:])
+            old_line_map.append(old_line_num)
+            new_line_map.append(new_line_num)
+            old_line_num += 1
+            new_line_num += 1
+        else:
+            # 其他行（如空行）
+            old_lines.append(line)
+            new_lines.append(line)
+            old_line_map.append(old_line_num if old_line_num > 0 else 0)
+            new_line_map.append(new_line_num if new_line_num > 0 else 0)
+
+    return old_lines, new_lines, old_line_map, new_line_map
+
+
+def visualize_diff_enhanced(
+    diff_text: str,
+    file_path: str = "",
+    mode: str = "unified",
+    show_line_numbers: bool = True,
+    context_lines: int = 3,
+) -> None:
+    """增强的 diff 可视化函数（便捷接口）
+
+    参数:
+        diff_text: git diff 输出的文本
+        file_path: 文件路径
+        mode: 可视化模式 ("unified" | "syntax" | "compact" | "side_by_side" | "statistics")
+        show_line_numbers: 是否显示行号
+        context_lines: 上下文行数
+    """
+    visualizer = DiffVisualizer()
+
+    if mode == "unified":
+        visualizer.visualize_unified_diff(
+            diff_text,
+            file_path,
+            show_line_numbers=show_line_numbers,
+            context_lines=context_lines,
+        )
+    elif mode == "syntax":
+        visualizer.visualize_syntax_highlighted(diff_text, file_path)
+    elif mode == "compact":
+        visualizer.visualize_compact(diff_text, file_path)
+    elif mode == "side_by_side":
+        # 检查是否有多个文件
+        file_diffs = _split_diff_by_files(diff_text)
+
+        if len(file_diffs) > 1:
+            # 多个文件，为每个文件显示独立的 table
+            for single_file_path, single_file_diff in file_diffs:
+                old_lines, new_lines, old_line_map, new_line_map = _parse_diff_to_lines(
+                    single_file_diff
+                )
+                visualizer.visualize_side_by_side_summary(
+                    old_lines,
+                    new_lines,
+                    single_file_path if single_file_path else file_path,
+                    context_lines=context_lines,
+                    old_line_map=old_line_map,
+                    new_line_map=new_line_map,
+                )
+        else:
+            # 单个文件，使用原有逻辑
+            old_lines, new_lines, old_line_map, new_line_map = _parse_diff_to_lines(
+                diff_text
+            )
+            visualizer.visualize_side_by_side_summary(
+                old_lines,
+                new_lines,
+                file_path,
+                context_lines=context_lines,
+                old_line_map=old_line_map,
+                new_line_map=new_line_map,
+            )
+    else:
+        # 默认使用语法高亮
+        visualizer.visualize_syntax_highlighted(diff_text, file_path)