优化消息里读/写内容显示样式
This commit is contained in:
@@ -574,6 +574,131 @@ impl ClaudeParser {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
"tool_use" => {
|
||||||
|
// Top-level tool_use record (Claude Code JSONL format)
|
||||||
|
let timestamp = parse_timestamp(&value).unwrap_or_else(Utc::now);
|
||||||
|
let tool_name = value
|
||||||
|
.get("tool_name")
|
||||||
|
.and_then(|n| n.as_str())
|
||||||
|
.unwrap_or("unknown")
|
||||||
|
.to_string();
|
||||||
|
let input_preview = value.get("tool_input").map(|i| i.to_string());
|
||||||
|
let synthetic_id = format!("tl-tool-{}", messages.len());
|
||||||
|
|
||||||
|
// Attach to last assistant message, or create a synthetic one
|
||||||
|
if let Some(last) = messages
|
||||||
|
.iter_mut()
|
||||||
|
.rev()
|
||||||
|
.find(|m| matches!(m.role, MessageRole::Assistant))
|
||||||
|
{
|
||||||
|
last.content.push(ContentBlock::ToolUse {
|
||||||
|
tool_use_id: Some(synthetic_id),
|
||||||
|
tool_name,
|
||||||
|
input_preview,
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
messages.push(UnifiedMessage {
|
||||||
|
id: format!("synth-assistant-{}", messages.len()),
|
||||||
|
role: MessageRole::Assistant,
|
||||||
|
content: vec![ContentBlock::ToolUse {
|
||||||
|
tool_use_id: Some(synthetic_id),
|
||||||
|
tool_name,
|
||||||
|
input_preview,
|
||||||
|
}],
|
||||||
|
timestamp,
|
||||||
|
usage: None,
|
||||||
|
duration_ms: None,
|
||||||
|
model: None,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
"tool_result" => {
|
||||||
|
// Top-level tool_result record (Claude Code JSONL format)
|
||||||
|
let tool_output = value.get("tool_output");
|
||||||
|
let tool_name = value
|
||||||
|
.get("tool_name")
|
||||||
|
.and_then(|n| n.as_str())
|
||||||
|
.unwrap_or("");
|
||||||
|
let is_error = tool_output
|
||||||
|
.and_then(|o| o.get("exit"))
|
||||||
|
.and_then(|e| e.as_i64())
|
||||||
|
.is_some_and(|code| code != 0);
|
||||||
|
|
||||||
|
// Extract output text: prefer "preview" (read), then "output" (bash)
|
||||||
|
let output_text = tool_output
|
||||||
|
.and_then(|o| {
|
||||||
|
o.get("preview")
|
||||||
|
.or_else(|| o.get("output"))
|
||||||
|
.and_then(|v| v.as_str())
|
||||||
|
})
|
||||||
|
.map(|s| s.to_string());
|
||||||
|
|
||||||
|
// For read tools, structurize with start_line from tool_input.offset
|
||||||
|
let output_preview =
|
||||||
|
if tool_name == "read" || tool_name == "Read" {
|
||||||
|
let start_line = value
|
||||||
|
.get("tool_input")
|
||||||
|
.and_then(|i| i.get("offset"))
|
||||||
|
.and_then(|o| o.as_u64())
|
||||||
|
.map(|o| o + 1)
|
||||||
|
.unwrap_or(1);
|
||||||
|
output_text.map(|text| {
|
||||||
|
serde_json::json!({
|
||||||
|
"start_line": start_line,
|
||||||
|
"content": text
|
||||||
|
})
|
||||||
|
.to_string()
|
||||||
|
})
|
||||||
|
} else {
|
||||||
|
output_text
|
||||||
|
};
|
||||||
|
|
||||||
|
// Find matching ToolUse in the last assistant message and use its ID
|
||||||
|
let matching_id = messages
|
||||||
|
.iter()
|
||||||
|
.rev()
|
||||||
|
.find(|m| matches!(m.role, MessageRole::Assistant))
|
||||||
|
.and_then(|m| {
|
||||||
|
m.content.iter().rev().find_map(|b| {
|
||||||
|
if let ContentBlock::ToolUse {
|
||||||
|
tool_use_id: Some(ref id),
|
||||||
|
..
|
||||||
|
} = b
|
||||||
|
{
|
||||||
|
Some(id.clone())
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
})
|
||||||
|
});
|
||||||
|
|
||||||
|
// Append ToolResult to the same assistant message so they stay in the same turn
|
||||||
|
if let Some(last) = messages
|
||||||
|
.iter_mut()
|
||||||
|
.rev()
|
||||||
|
.find(|m| matches!(m.role, MessageRole::Assistant))
|
||||||
|
{
|
||||||
|
last.content.push(ContentBlock::ToolResult {
|
||||||
|
tool_use_id: matching_id,
|
||||||
|
output_preview,
|
||||||
|
is_error,
|
||||||
|
});
|
||||||
|
} else {
|
||||||
|
messages.push(UnifiedMessage {
|
||||||
|
id: format!("synth-result-{}", messages.len()),
|
||||||
|
role: MessageRole::Assistant,
|
||||||
|
content: vec![ContentBlock::ToolResult {
|
||||||
|
tool_use_id: matching_id,
|
||||||
|
output_preview,
|
||||||
|
is_error,
|
||||||
|
}],
|
||||||
|
timestamp: parse_timestamp(&value).unwrap_or_else(Utc::now),
|
||||||
|
usage: None,
|
||||||
|
duration_ms: None,
|
||||||
|
model: None,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -583,6 +708,7 @@ impl ClaudeParser {
|
|||||||
|
|
||||||
let mut turns = group_into_turns(messages);
|
let mut turns = group_into_turns(messages);
|
||||||
super::relocate_orphaned_tool_results(&mut turns);
|
super::relocate_orphaned_tool_results(&mut turns);
|
||||||
|
super::structurize_read_tool_output(&mut turns);
|
||||||
let context_window_used_tokens = latest_claude_context_window_used_tokens(&turns);
|
let context_window_used_tokens = latest_claude_context_window_used_tokens(&turns);
|
||||||
let context_window_max_tokens =
|
let context_window_max_tokens =
|
||||||
claude_context_window_max_tokens_for_model(model.as_deref());
|
claude_context_window_max_tokens_for_model(model.as_deref());
|
||||||
|
|||||||
@@ -773,6 +773,7 @@ impl CodexParser {
|
|||||||
|
|
||||||
let mut turns = group_into_turns(messages);
|
let mut turns = group_into_turns(messages);
|
||||||
super::relocate_orphaned_tool_results(&mut turns);
|
super::relocate_orphaned_tool_results(&mut turns);
|
||||||
|
super::structurize_read_tool_output(&mut turns);
|
||||||
let mut session_stats = super::compute_session_stats(&turns);
|
let mut session_stats = super::compute_session_stats(&turns);
|
||||||
session_stats =
|
session_stats =
|
||||||
merge_codex_total_usage_stats(session_stats, latest_total_usage, latest_total_tokens);
|
merge_codex_total_usage_stats(session_stats, latest_total_usage, latest_total_tokens);
|
||||||
|
|||||||
@@ -556,6 +556,7 @@ impl GeminiParser {
|
|||||||
|
|
||||||
let mut turns = group_into_turns(messages);
|
let mut turns = group_into_turns(messages);
|
||||||
super::relocate_orphaned_tool_results(&mut turns);
|
super::relocate_orphaned_tool_results(&mut turns);
|
||||||
|
super::structurize_read_tool_output(&mut turns);
|
||||||
summary.message_count = turns.len() as u32;
|
summary.message_count = turns.len() as u32;
|
||||||
summary.id = conversation_id.to_string();
|
summary.id = conversation_id.to_string();
|
||||||
let context_window_used_tokens = super::latest_turn_total_usage_tokens(&turns);
|
let context_window_used_tokens = super::latest_turn_total_usage_tokens(&turns);
|
||||||
|
|||||||
@@ -4,7 +4,7 @@ pub mod gemini;
|
|||||||
pub mod openclaw;
|
pub mod openclaw;
|
||||||
pub mod opencode;
|
pub mod opencode;
|
||||||
|
|
||||||
use std::collections::HashMap;
|
use std::collections::{HashMap, HashSet};
|
||||||
use std::sync::OnceLock;
|
use std::sync::OnceLock;
|
||||||
|
|
||||||
use regex::Regex;
|
use regex::Regex;
|
||||||
@@ -292,6 +292,117 @@ pub fn relocate_orphaned_tool_results(turns: &mut Vec<MessageTurn>) {
|
|||||||
turns.retain(|turn| !turn.blocks.is_empty());
|
turns.retain(|turn| !turn.blocks.is_empty());
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Convert Read tool output from numbered-line format to `{"start_line":N,"content":"..."}`.
|
||||||
|
///
|
||||||
|
/// Claude Code embeds line numbers in Read output like ` 115→content`.
|
||||||
|
/// This splits on the `→` delimiter (or tab for older `cat -n` format),
|
||||||
|
/// extracts the starting line number, and returns clean content.
|
||||||
|
pub fn structurize_read_tool_output(turns: &mut [MessageTurn]) {
|
||||||
|
let mut read_tool_ids: HashSet<String> = HashSet::new();
|
||||||
|
for turn in turns.iter() {
|
||||||
|
for block in &turn.blocks {
|
||||||
|
if let ContentBlock::ToolUse {
|
||||||
|
tool_use_id: Some(ref id),
|
||||||
|
ref tool_name,
|
||||||
|
..
|
||||||
|
} = block
|
||||||
|
{
|
||||||
|
let name = tool_name.to_lowercase();
|
||||||
|
if matches!(
|
||||||
|
name.as_str(),
|
||||||
|
"read" | "read_file" | "readfile" | "read file" | "cat" | "view"
|
||||||
|
) {
|
||||||
|
read_tool_ids.insert(id.clone());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for turn in turns.iter_mut() {
|
||||||
|
for block in turn.blocks.iter_mut() {
|
||||||
|
let is_read_result = matches!(
|
||||||
|
block,
|
||||||
|
ContentBlock::ToolResult { tool_use_id: Some(ref id), .. }
|
||||||
|
if read_tool_ids.contains(id)
|
||||||
|
);
|
||||||
|
if !is_read_result {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if let ContentBlock::ToolResult {
|
||||||
|
ref mut output_preview,
|
||||||
|
..
|
||||||
|
} = block
|
||||||
|
{
|
||||||
|
if let Some(ref text) = output_preview {
|
||||||
|
if let Some(json) = strip_numbered_lines(text) {
|
||||||
|
*output_preview = Some(json);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/// Known delimiters between line number and content.
|
||||||
|
const LINE_NUM_DELIMITERS: &[&str] = &["→", "\t"];
|
||||||
|
|
||||||
|
/// Try to split a line at a known delimiter, returning (line_number, content).
|
||||||
|
fn split_line_number(line: &str) -> Option<(u64, &str)> {
|
||||||
|
for delim in LINE_NUM_DELIMITERS {
|
||||||
|
if let Some(pos) = line.find(delim) {
|
||||||
|
let prefix = line[..pos].trim();
|
||||||
|
if let Ok(num) = prefix.parse::<u64>() {
|
||||||
|
let content_start = pos + delim.len();
|
||||||
|
return Some((num, &line[content_start..]));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
None
|
||||||
|
}
|
||||||
|
|
||||||
|
/// If most lines have a recognized line-number prefix, strip them all
|
||||||
|
/// and return `{"start_line":N,"content":"clean text"}`.
|
||||||
|
fn strip_numbered_lines(text: &str) -> Option<String> {
|
||||||
|
let raw_lines: Vec<&str> = text.lines().collect();
|
||||||
|
if raw_lines.len() < 2 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let matched = raw_lines
|
||||||
|
.iter()
|
||||||
|
.filter(|l| l.is_empty() || split_line_number(l).is_some())
|
||||||
|
.count();
|
||||||
|
if matched < raw_lines.len() * 4 / 5 {
|
||||||
|
return None;
|
||||||
|
}
|
||||||
|
|
||||||
|
let mut start_line: u64 = 1;
|
||||||
|
let mut first = true;
|
||||||
|
let stripped: Vec<&str> = raw_lines
|
||||||
|
.iter()
|
||||||
|
.map(|line| {
|
||||||
|
if let Some((num, content)) = split_line_number(line) {
|
||||||
|
if first {
|
||||||
|
start_line = num;
|
||||||
|
first = false;
|
||||||
|
}
|
||||||
|
content
|
||||||
|
} else {
|
||||||
|
first = false;
|
||||||
|
*line
|
||||||
|
}
|
||||||
|
})
|
||||||
|
.collect();
|
||||||
|
|
||||||
|
Some(
|
||||||
|
serde_json::json!({
|
||||||
|
"start_line": start_line,
|
||||||
|
"content": stripped.join("\n")
|
||||||
|
})
|
||||||
|
.to_string(),
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
/// Extract the last path component as the folder name.
|
/// Extract the last path component as the folder name.
|
||||||
pub fn folder_name_from_path(path: &str) -> String {
|
pub fn folder_name_from_path(path: &str) -> String {
|
||||||
path.rsplit(['/', '\\']).next().unwrap_or(path).to_string()
|
path.rsplit(['/', '\\']).next().unwrap_or(path).to_string()
|
||||||
|
|||||||
@@ -652,6 +652,7 @@ impl OpenClawParser {
|
|||||||
let folder_name = folder_path.as_ref().map(|p| folder_name_from_path(p));
|
let folder_name = folder_path.as_ref().map(|p| folder_name_from_path(p));
|
||||||
let mut turns = group_into_turns(messages);
|
let mut turns = group_into_turns(messages);
|
||||||
super::relocate_orphaned_tool_results(&mut turns);
|
super::relocate_orphaned_tool_results(&mut turns);
|
||||||
|
super::structurize_read_tool_output(&mut turns);
|
||||||
|
|
||||||
let context_window_used_tokens = latest_turn_total_usage_tokens(&turns);
|
let context_window_used_tokens = latest_turn_total_usage_tokens(&turns);
|
||||||
let context_window_max_tokens = session_meta
|
let context_window_max_tokens = session_meta
|
||||||
@@ -983,9 +984,15 @@ fn extract_assistant_content(value: &serde_json::Value) -> Vec<ContentBlock> {
|
|||||||
.and_then(|n| n.as_str())
|
.and_then(|n| n.as_str())
|
||||||
.unwrap_or("unknown")
|
.unwrap_or("unknown")
|
||||||
.to_string();
|
.to_string();
|
||||||
|
let is_edit_tool = matches!(
|
||||||
|
tool_name.to_lowercase().as_str(),
|
||||||
|
"edit" | "write" | "apply_patch" | "patch" | "applypatch"
|
||||||
|
| "edit_file" | "editfile"
|
||||||
|
);
|
||||||
|
let max_len = if is_edit_tool { 50000 } else { 500 };
|
||||||
let input_preview = item.get("arguments").map(|a| {
|
let input_preview = item.get("arguments").map(|a| {
|
||||||
let s = a.to_string();
|
let s = a.to_string();
|
||||||
truncate_str(&s, 500)
|
truncate_str(&s, max_len)
|
||||||
});
|
});
|
||||||
blocks.push(ContentBlock::ToolUse {
|
blocks.push(ContentBlock::ToolUse {
|
||||||
tool_use_id,
|
tool_use_id,
|
||||||
|
|||||||
@@ -189,6 +189,7 @@ impl OpenCodeParser {
|
|||||||
let messages = self.load_sqlite_messages(&conn, conversation_id).await?;
|
let messages = self.load_sqlite_messages(&conn, conversation_id).await?;
|
||||||
let mut turns = group_into_turns(messages);
|
let mut turns = group_into_turns(messages);
|
||||||
super::relocate_orphaned_tool_results(&mut turns);
|
super::relocate_orphaned_tool_results(&mut turns);
|
||||||
|
super::structurize_read_tool_output(&mut turns);
|
||||||
let context_window_used_tokens = super::latest_turn_total_usage_tokens(&turns);
|
let context_window_used_tokens = super::latest_turn_total_usage_tokens(&turns);
|
||||||
let context_window_max_tokens =
|
let context_window_max_tokens =
|
||||||
super::infer_context_window_max_tokens(summary.model.as_deref());
|
super::infer_context_window_max_tokens(summary.model.as_deref());
|
||||||
|
|||||||
@@ -22,6 +22,7 @@ import { useTranslations } from "next-intl"
|
|||||||
import { isValidElement } from "react"
|
import { isValidElement } from "react"
|
||||||
|
|
||||||
import { CodeBlock } from "./code-block"
|
import { CodeBlock } from "./code-block"
|
||||||
|
import { UnifiedDiffPreview } from "@/components/diff/unified-diff-preview"
|
||||||
import { MessageResponse } from "./message"
|
import { MessageResponse } from "./message"
|
||||||
|
|
||||||
export type ToolProps = ComponentProps<typeof Collapsible>
|
export type ToolProps = ComponentProps<typeof Collapsible>
|
||||||
@@ -380,9 +381,10 @@ export const ToolOutput = ({
|
|||||||
<MessageResponse>{output}</MessageResponse>
|
<MessageResponse>{output}</MessageResponse>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
|
} else if (lang === "diff") {
|
||||||
|
Output = <UnifiedDiffPreview diffText={output} />
|
||||||
} else {
|
} else {
|
||||||
const language = detectOutputLanguage(output)
|
Output = <CodeBlock code={output} language={lang} />
|
||||||
Output = <CodeBlock code={output} language={language} />
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,13 +1,9 @@
|
|||||||
"use client"
|
"use client"
|
||||||
|
|
||||||
import { useCallback, useEffect, useMemo, useRef } from "react"
|
import { useMemo } from "react"
|
||||||
import dynamic from "next/dynamic"
|
|
||||||
import type { editor as MonacoEditorNs } from "monaco-editor"
|
|
||||||
import { useTranslations } from "next-intl"
|
import { useTranslations } from "next-intl"
|
||||||
import { useFolderContext } from "@/contexts/folder-context"
|
import { useFolderContext } from "@/contexts/folder-context"
|
||||||
import { defineMonacoThemes, useMonacoThemeSync } from "@/lib/monaco-themes"
|
|
||||||
import { cn } from "@/lib/utils"
|
import { cn } from "@/lib/utils"
|
||||||
import "@/lib/monaco-local"
|
|
||||||
|
|
||||||
type RowMarker = "none" | "added" | "deleted" | "modified"
|
type RowMarker = "none" | "added" | "deleted" | "modified"
|
||||||
type DiffFileMode = "modified" | "added" | "deleted" | "renamed"
|
type DiffFileMode = "modified" | "added" | "deleted" | "renamed"
|
||||||
@@ -67,14 +63,18 @@ interface WorkingFile {
|
|||||||
hunks: WorkingHunk[]
|
hunks: WorkingHunk[]
|
||||||
}
|
}
|
||||||
|
|
||||||
interface HunkPreviewLine {
|
const ROW_CLASS: Record<RowMarker, string> = {
|
||||||
text: string
|
none: "",
|
||||||
marker: RowMarker
|
added: "bg-green-500/10 text-green-900 dark:text-green-300",
|
||||||
|
deleted: "bg-red-500/10 text-red-900 dark:text-red-300",
|
||||||
|
modified: "bg-blue-500/10 text-blue-900 dark:text-blue-300",
|
||||||
}
|
}
|
||||||
|
|
||||||
const MonacoEditor = dynamic(async () => import("@monaco-editor/react"), {
|
const SIGN_CLASS: Record<string, string> = {
|
||||||
ssr: false,
|
"+": "text-green-700 dark:text-green-400",
|
||||||
})
|
"-": "text-red-700 dark:text-red-400",
|
||||||
|
" ": "text-muted-foreground/50",
|
||||||
|
}
|
||||||
|
|
||||||
function normalizePath(raw: string): string | null {
|
function normalizePath(raw: string): string | null {
|
||||||
const trimmed = raw.trim().replace(/^"|"$/g, "")
|
const trimmed = raw.trim().replace(/^"|"$/g, "")
|
||||||
@@ -190,13 +190,11 @@ function classifyRows(rows: RawDiffRow[]): ParsedDiffRow[] {
|
|||||||
addEnd += 1
|
addEnd += 1
|
||||||
}
|
}
|
||||||
|
|
||||||
const delRows = rows.slice(index, delEnd)
|
for (let d = index; d < delEnd; d++) {
|
||||||
const addRows = rows.slice(delEnd, addEnd)
|
const row = rows[d]
|
||||||
const modifiedPairs = Math.min(delRows.length, addRows.length)
|
if (!row) continue
|
||||||
|
|
||||||
for (const [delta, row] of delRows.entries()) {
|
|
||||||
parsed.push({
|
parsed.push({
|
||||||
type: delta < modifiedPairs ? "modified" : "deleted",
|
type: "deleted",
|
||||||
text: row.text,
|
text: row.text,
|
||||||
sign: "-",
|
sign: "-",
|
||||||
oldLine: row.oldLine,
|
oldLine: row.oldLine,
|
||||||
@@ -204,9 +202,11 @@ function classifyRows(rows: RawDiffRow[]): ParsedDiffRow[] {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
for (const [delta, row] of addRows.entries()) {
|
for (let a = delEnd; a < addEnd; a++) {
|
||||||
|
const row = rows[a]
|
||||||
|
if (!row) continue
|
||||||
parsed.push({
|
parsed.push({
|
||||||
type: delta < modifiedPairs ? "modified" : "added",
|
type: "added",
|
||||||
text: row.text,
|
text: row.text,
|
||||||
sign: "+",
|
sign: "+",
|
||||||
oldLine: row.oldLine,
|
oldLine: row.oldLine,
|
||||||
@@ -440,171 +440,53 @@ function toDisplayPath(filePath: string, folderPath: string | null): string {
|
|||||||
return normalizedPath
|
return normalizedPath
|
||||||
}
|
}
|
||||||
|
|
||||||
function countHunkChanges(hunk: ParsedDiffHunk): {
|
function rowMarker(row: ParsedDiffRow): RowMarker {
|
||||||
additions: number
|
if (row.type === "added") return "added"
|
||||||
deletions: number
|
if (row.type === "deleted") return "deleted"
|
||||||
} {
|
return "none"
|
||||||
let additions = 0
|
|
||||||
let deletions = 0
|
|
||||||
|
|
||||||
for (const row of hunk.rows) {
|
|
||||||
if (row.sign === "+") additions += 1
|
|
||||||
if (row.sign === "-") deletions += 1
|
|
||||||
}
|
}
|
||||||
|
|
||||||
return { additions, deletions }
|
function HunkLines({ rows }: { rows: ParsedDiffRow[] }) {
|
||||||
}
|
|
||||||
|
|
||||||
function buildHunkPreviewLines(rows: ParsedDiffRow[]): {
|
|
||||||
lines: HunkPreviewLine[]
|
|
||||||
} {
|
|
||||||
const lines: HunkPreviewLine[] = rows.map((row) => {
|
|
||||||
let marker: RowMarker = "none"
|
|
||||||
if (row.type === "added") marker = "added"
|
|
||||||
else if (row.type === "deleted") marker = "deleted"
|
|
||||||
else if (row.type === "modified") marker = "modified"
|
|
||||||
|
|
||||||
return {
|
|
||||||
text: `${row.sign}${row.text}`,
|
|
||||||
marker,
|
|
||||||
}
|
|
||||||
})
|
|
||||||
|
|
||||||
return {
|
|
||||||
lines,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
function HunkMonacoPreview({
|
|
||||||
hunk,
|
|
||||||
modelId,
|
|
||||||
theme,
|
|
||||||
}: {
|
|
||||||
hunk: ParsedDiffHunk
|
|
||||||
modelId: string
|
|
||||||
theme: string
|
|
||||||
}) {
|
|
||||||
const t = useTranslations("Folder.diffPreview")
|
|
||||||
const editorRef = useRef<MonacoEditorNs.IStandaloneCodeEditor | null>(null)
|
|
||||||
const decorationsRef = useRef<string[]>([])
|
|
||||||
|
|
||||||
const { lines } = useMemo(() => buildHunkPreviewLines(hunk.rows), [hunk.rows])
|
|
||||||
|
|
||||||
const renderedContent = useMemo(
|
|
||||||
() => lines.map((line) => line.text).join("\n"),
|
|
||||||
[lines]
|
|
||||||
)
|
|
||||||
|
|
||||||
const applyDecorations = useCallback(() => {
|
|
||||||
const editor = editorRef.current
|
|
||||||
if (!editor) return
|
|
||||||
|
|
||||||
const model = editor.getModel()
|
|
||||||
if (!model) return
|
|
||||||
|
|
||||||
const maxLine = model.getLineCount()
|
|
||||||
const decorations: MonacoEditorNs.IModelDeltaDecoration[] = []
|
|
||||||
|
|
||||||
for (const [index, line] of lines.entries()) {
|
|
||||||
const lineNumber = index + 1
|
|
||||||
if (lineNumber > maxLine) continue
|
|
||||||
|
|
||||||
let cls: string | null = null
|
|
||||||
if (line.marker === "added") {
|
|
||||||
cls = "codeg-session-diff-line-added"
|
|
||||||
} else if (line.marker === "modified") {
|
|
||||||
cls = "codeg-session-diff-line-modified"
|
|
||||||
} else if (line.marker === "deleted") {
|
|
||||||
cls = "codeg-session-diff-line-deleted"
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!cls) continue
|
|
||||||
|
|
||||||
decorations.push({
|
|
||||||
range: {
|
|
||||||
startLineNumber: lineNumber,
|
|
||||||
startColumn: 1,
|
|
||||||
endLineNumber: lineNumber,
|
|
||||||
endColumn: 1,
|
|
||||||
},
|
|
||||||
options: {
|
|
||||||
isWholeLine: true,
|
|
||||||
className: cls,
|
|
||||||
},
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
decorationsRef.current = editor.deltaDecorations(
|
|
||||||
decorationsRef.current,
|
|
||||||
decorations
|
|
||||||
)
|
|
||||||
}, [lines])
|
|
||||||
|
|
||||||
useEffect(() => {
|
|
||||||
applyDecorations()
|
|
||||||
}, [applyDecorations])
|
|
||||||
|
|
||||||
useEffect(
|
|
||||||
() => () => {
|
|
||||||
const editor = editorRef.current
|
|
||||||
if (!editor) return
|
|
||||||
editor.deltaDecorations(decorationsRef.current, [])
|
|
||||||
decorationsRef.current = []
|
|
||||||
},
|
|
||||||
[]
|
|
||||||
)
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<MonacoEditor
|
<div className="inline-block min-w-full font-mono text-[12px] leading-[20px]">
|
||||||
beforeMount={defineMonacoThemes}
|
{rows.map((row, i) => {
|
||||||
onMount={(editor) => {
|
const marker = rowMarker(row)
|
||||||
editorRef.current = editor
|
return (
|
||||||
applyDecorations()
|
<div key={i} className={cn("flex", ROW_CLASS[marker])}>
|
||||||
}}
|
<span className="w-[3.5rem] shrink-0 select-none pr-1 text-right text-muted-foreground/40">
|
||||||
path={`inmemory://session-hunk/${encodeURIComponent(modelId)}`}
|
{row.oldLine ?? ""}
|
||||||
value={renderedContent}
|
</span>
|
||||||
language="plaintext"
|
<span className="w-[3.5rem] shrink-0 select-none pr-1 text-right text-muted-foreground/40">
|
||||||
theme={theme}
|
{row.newLine ?? ""}
|
||||||
loading={
|
</span>
|
||||||
<div className="h-28 flex items-center justify-center text-xs text-muted-foreground">
|
<span
|
||||||
{t("loadingHunk")}
|
className={cn(
|
||||||
|
"w-4 shrink-0 select-none text-center",
|
||||||
|
SIGN_CLASS[row.sign] ?? ""
|
||||||
|
)}
|
||||||
|
>
|
||||||
|
{row.sign === " " ? "" : row.sign}
|
||||||
|
</span>
|
||||||
|
<span className="flex-1 whitespace-pre pr-3">{row.text}</span>
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
})}
|
||||||
</div>
|
</div>
|
||||||
}
|
|
||||||
options={{
|
|
||||||
readOnly: true,
|
|
||||||
minimap: { enabled: false },
|
|
||||||
automaticLayout: true,
|
|
||||||
fontSize: 12,
|
|
||||||
lineNumbers: "off",
|
|
||||||
lineDecorationsWidth: 10,
|
|
||||||
glyphMargin: false,
|
|
||||||
wordWrap: "off",
|
|
||||||
scrollBeyondLastLine: false,
|
|
||||||
renderLineHighlight: "none",
|
|
||||||
contextmenu: false,
|
|
||||||
folding: false,
|
|
||||||
scrollbar: {
|
|
||||||
alwaysConsumeMouseWheel: false,
|
|
||||||
},
|
|
||||||
padding: { top: 6, bottom: 6 },
|
|
||||||
}}
|
|
||||||
/>
|
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
export function UnifiedDiffPreview({
|
export function UnifiedDiffPreview({
|
||||||
diffText,
|
diffText,
|
||||||
modelId,
|
|
||||||
className,
|
className,
|
||||||
}: {
|
}: {
|
||||||
diffText: string
|
diffText: string
|
||||||
|
/** @deprecated No longer used — kept for API compat */
|
||||||
modelId?: string
|
modelId?: string
|
||||||
className?: string
|
className?: string
|
||||||
}) {
|
}) {
|
||||||
const t = useTranslations("Folder.diffPreview")
|
const t = useTranslations("Folder.diffPreview")
|
||||||
const { folder } = useFolderContext()
|
const { folder } = useFolderContext()
|
||||||
const files = useMemo(() => parseUnifiedDiff(diffText), [diffText])
|
const files = useMemo(() => parseUnifiedDiff(diffText), [diffText])
|
||||||
const theme = useMonacoThemeSync()
|
|
||||||
|
|
||||||
if (!diffText.trim()) {
|
if (!diffText.trim()) {
|
||||||
return (
|
return (
|
||||||
@@ -621,8 +503,8 @@ export function UnifiedDiffPreview({
|
|||||||
|
|
||||||
if (files.length === 0) {
|
if (files.length === 0) {
|
||||||
return (
|
return (
|
||||||
<div className={cn("h-full overflow-auto p-3", className)}>
|
<div className={cn("h-full overflow-auto", className)}>
|
||||||
<pre className="font-mono text-[11px] leading-5 whitespace-pre-wrap text-muted-foreground">
|
<pre className="font-mono text-[11px] leading-5 whitespace-pre-wrap text-muted-foreground p-3">
|
||||||
{diffText}
|
{diffText}
|
||||||
</pre>
|
</pre>
|
||||||
</div>
|
</div>
|
||||||
@@ -630,14 +512,14 @@ export function UnifiedDiffPreview({
|
|||||||
}
|
}
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className={cn("h-full overflow-auto p-3", className)}>
|
<div className={cn("h-full overflow-auto", className)}>
|
||||||
<div className="space-y-3">
|
<div className="space-y-3">
|
||||||
{files.map((file) => (
|
{files.map((file) => (
|
||||||
<section
|
<section
|
||||||
key={file.key}
|
key={file.key}
|
||||||
className="overflow-hidden rounded-lg border border-border bg-background"
|
className="flex max-h-[480px] flex-col rounded-lg border border-border bg-background"
|
||||||
>
|
>
|
||||||
<header className="flex items-center gap-2 border-b border-border bg-muted/40 px-3 py-2 text-[11px]">
|
<header className="flex shrink-0 items-center gap-2 border-b border-border bg-muted/40 px-3 py-2 text-[11px]">
|
||||||
<span className="shrink-0 rounded border border-border bg-background px-1.5 py-0.5 text-[10px] text-muted-foreground">
|
<span className="shrink-0 rounded border border-border bg-background px-1.5 py-0.5 text-[10px] text-muted-foreground">
|
||||||
{t(modeKey(file.mode))}
|
{t(modeKey(file.mode))}
|
||||||
</span>
|
</span>
|
||||||
@@ -657,41 +539,10 @@ export function UnifiedDiffPreview({
|
|||||||
</span>
|
</span>
|
||||||
</header>
|
</header>
|
||||||
|
|
||||||
<div className="space-y-2 p-2">
|
<div className="overflow-auto">
|
||||||
{file.hunks.map((hunk, index) => {
|
{file.hunks.map((hunk) => (
|
||||||
const hunkStats = countHunkChanges(hunk)
|
<HunkLines key={hunk.key} rows={hunk.rows} />
|
||||||
|
))}
|
||||||
return (
|
|
||||||
<div
|
|
||||||
key={hunk.key}
|
|
||||||
className="rounded-md border border-border"
|
|
||||||
>
|
|
||||||
<div className="flex items-center gap-2 border-b border-border bg-muted/30 px-2 py-1 text-[10px] font-mono text-muted-foreground">
|
|
||||||
<span>{t("hunkLabel", { index: index + 1 })}</span>
|
|
||||||
<span className="ml-auto inline-flex items-center gap-2">
|
|
||||||
<span className="text-green-700 dark:text-green-400">
|
|
||||||
+{hunkStats.additions}
|
|
||||||
</span>
|
|
||||||
<span className="text-red-700 dark:text-red-400">
|
|
||||||
-{hunkStats.deletions}
|
|
||||||
</span>
|
|
||||||
</span>
|
|
||||||
</div>
|
|
||||||
<div
|
|
||||||
className="min-h-[7rem]"
|
|
||||||
style={{
|
|
||||||
height: `${Math.max(120, hunk.rows.length * 20 + 18)}px`,
|
|
||||||
}}
|
|
||||||
>
|
|
||||||
<HunkMonacoPreview
|
|
||||||
hunk={hunk}
|
|
||||||
modelId={`${modelId ?? "session"}:${file.key}:${hunk.key}`}
|
|
||||||
theme={theme}
|
|
||||||
/>
|
|
||||||
</div>
|
|
||||||
</div>
|
|
||||||
)
|
|
||||||
})}
|
|
||||||
</div>
|
</div>
|
||||||
</section>
|
</section>
|
||||||
))}
|
))}
|
||||||
|
|||||||
@@ -1270,8 +1270,7 @@ export function FileWorkspacePanel() {
|
|||||||
)}
|
)}
|
||||||
<UnifiedDiffPreview
|
<UnifiedDiffPreview
|
||||||
diffText={activeFileTab.content}
|
diffText={activeFileTab.content}
|
||||||
modelId={activeFileTab.id}
|
className="h-full p-3"
|
||||||
className="h-full"
|
|
||||||
/>
|
/>
|
||||||
</div>
|
</div>
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -1,5 +1,4 @@
|
|||||||
import { memo, useMemo, useState, type ReactNode } from "react"
|
import { memo, useMemo, useState, type ReactNode } from "react"
|
||||||
import type { BundledLanguage } from "shiki"
|
|
||||||
import type { AdaptedContentPart } from "@/lib/adapters/ai-elements-adapter"
|
import type { AdaptedContentPart } from "@/lib/adapters/ai-elements-adapter"
|
||||||
import type { MessageRole } from "@/lib/types"
|
import type { MessageRole } from "@/lib/types"
|
||||||
import { normalizeToolName } from "@/lib/tool-call-normalization"
|
import { normalizeToolName } from "@/lib/tool-call-normalization"
|
||||||
@@ -17,6 +16,8 @@ import {
|
|||||||
} from "@/components/ai-elements/tool"
|
} from "@/components/ai-elements/tool"
|
||||||
import { Terminal } from "@/components/ai-elements/terminal"
|
import { Terminal } from "@/components/ai-elements/terminal"
|
||||||
import { CodeBlock } from "@/components/ai-elements/code-block"
|
import { CodeBlock } from "@/components/ai-elements/code-block"
|
||||||
|
import { UnifiedDiffPreview } from "@/components/diff/unified-diff-preview"
|
||||||
|
import { generateUnifiedDiff } from "@/lib/unified-diff-generator"
|
||||||
import {
|
import {
|
||||||
Reasoning,
|
Reasoning,
|
||||||
ReasoningTrigger,
|
ReasoningTrigger,
|
||||||
@@ -374,62 +375,6 @@ function num(obj: Record<string, unknown>, key: string): number | undefined {
|
|||||||
return typeof v === "number" ? v : undefined
|
return typeof v === "number" ? v : undefined
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Guess shiki language from file path extension. */
|
|
||||||
const EXT_LANG_MAP: Record<string, BundledLanguage> = {
|
|
||||||
ts: "typescript",
|
|
||||||
tsx: "tsx",
|
|
||||||
js: "javascript",
|
|
||||||
jsx: "jsx",
|
|
||||||
mjs: "javascript",
|
|
||||||
cjs: "javascript",
|
|
||||||
py: "python",
|
|
||||||
rs: "rust",
|
|
||||||
go: "go",
|
|
||||||
java: "java",
|
|
||||||
css: "css",
|
|
||||||
scss: "scss",
|
|
||||||
less: "less",
|
|
||||||
html: "html",
|
|
||||||
json: "json",
|
|
||||||
jsonl: "json",
|
|
||||||
yaml: "yaml",
|
|
||||||
yml: "yaml",
|
|
||||||
md: "markdown",
|
|
||||||
mdx: "mdx",
|
|
||||||
sql: "sql",
|
|
||||||
sh: "bash",
|
|
||||||
bash: "bash",
|
|
||||||
zsh: "bash",
|
|
||||||
toml: "toml",
|
|
||||||
xml: "xml",
|
|
||||||
svg: "xml",
|
|
||||||
vue: "vue",
|
|
||||||
svelte: "svelte",
|
|
||||||
rb: "ruby",
|
|
||||||
php: "php",
|
|
||||||
swift: "swift",
|
|
||||||
kt: "kotlin",
|
|
||||||
c: "c",
|
|
||||||
cpp: "cpp",
|
|
||||||
h: "c",
|
|
||||||
hpp: "cpp",
|
|
||||||
cs: "csharp",
|
|
||||||
dart: "dart",
|
|
||||||
lua: "lua",
|
|
||||||
r: "r",
|
|
||||||
dockerfile: "dockerfile",
|
|
||||||
graphql: "graphql",
|
|
||||||
prisma: "prisma",
|
|
||||||
}
|
|
||||||
|
|
||||||
function guessLangFromPath(filePath: string): BundledLanguage {
|
|
||||||
const ext = filePath.split(".").pop()?.toLowerCase() ?? ""
|
|
||||||
// Handle dotfiles like "Dockerfile"
|
|
||||||
const basename = filePath.split("/").pop()?.toLowerCase() ?? ""
|
|
||||||
if (basename === "dockerfile") return "dockerfile"
|
|
||||||
return EXT_LANG_MAP[ext] ?? ("log" as BundledLanguage)
|
|
||||||
}
|
|
||||||
|
|
||||||
type ApplyPatchOp = "add" | "update" | "delete" | "move"
|
type ApplyPatchOp = "add" | "update" | "delete" | "move"
|
||||||
|
|
||||||
type ApplyPatchFile = {
|
type ApplyPatchFile = {
|
||||||
@@ -1215,115 +1160,46 @@ function localizeDerivedToolTitle(
|
|||||||
|
|
||||||
/** Edit tool: file path + unified diff view */
|
/** Edit tool: file path + unified diff view */
|
||||||
function EditToolInput({ input }: { input: Record<string, unknown> }) {
|
function EditToolInput({ input }: { input: Record<string, unknown> }) {
|
||||||
const t = useTranslations("Folder.chat.contentParts")
|
|
||||||
const filePath = str(input, "file_path")
|
const filePath = str(input, "file_path")
|
||||||
const oldString = str(input, "old_string") ?? ""
|
const oldString = str(input, "old_string") ?? ""
|
||||||
const newString = str(input, "new_string") ?? ""
|
const newString = str(input, "new_string") ?? ""
|
||||||
const replaceAll = input.replace_all === true
|
|
||||||
|
|
||||||
const diffCode = useMemo(() => {
|
const diffCode = useMemo(() => {
|
||||||
const parts: string[] = []
|
|
||||||
if (oldString) {
|
|
||||||
for (const line of oldString.split("\n")) {
|
|
||||||
parts.push(`- ${line}`)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (newString) {
|
|
||||||
for (const line of newString.split("\n")) {
|
|
||||||
parts.push(`+ ${line}`)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return parts.join("\n")
|
|
||||||
}, [oldString, newString])
|
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="space-y-2">
|
generateUnifiedDiff(oldString, newString, filePath ?? undefined) ?? ""
|
||||||
<div className="flex items-center gap-2 text-xs">
|
|
||||||
<FilePenLineIcon className="size-3.5 shrink-0 text-muted-foreground" />
|
|
||||||
<span className="break-all font-mono text-foreground">
|
|
||||||
{filePath ?? t("unknown")}
|
|
||||||
</span>
|
|
||||||
{replaceAll && (
|
|
||||||
<span className="shrink-0 rounded bg-muted px-1.5 py-0.5 text-[10px] font-medium text-muted-foreground">
|
|
||||||
{t("replaceAll")}
|
|
||||||
</span>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
{diffCode && <CodeBlock code={diffCode} language="diff" />}
|
|
||||||
</div>
|
|
||||||
)
|
)
|
||||||
|
}, [oldString, newString, filePath])
|
||||||
|
|
||||||
|
return diffCode ? <UnifiedDiffPreview diffText={diffCode} /> : null
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Edit tool (changes payload): file list + summary + combined diff view */
|
/** Edit tool (changes payload): combined diff view */
|
||||||
function EditChangesToolInput({ changes }: { changes: EditChangePreview[] }) {
|
function EditChangesToolInput({ changes }: { changes: EditChangePreview[] }) {
|
||||||
const t = useTranslations("Folder.chat.contentParts")
|
const diffCode = useMemo(() => {
|
||||||
const { additions, deletions, diffCode } = useMemo(() => {
|
|
||||||
let additions = 0
|
|
||||||
let deletions = 0
|
|
||||||
const diffParts: string[] = []
|
const diffParts: string[] = []
|
||||||
|
|
||||||
for (const change of changes) {
|
for (const change of changes) {
|
||||||
if (change.unifiedDiff && change.unifiedDiff.trim().length > 0) {
|
if (change.unifiedDiff && change.unifiedDiff.trim().length > 0) {
|
||||||
diffParts.push(change.unifiedDiff.trim())
|
diffParts.push(change.unifiedDiff.trim())
|
||||||
diffParts.push("")
|
diffParts.push("")
|
||||||
for (const line of change.unifiedDiff.split("\n")) {
|
|
||||||
if (line.startsWith("+") && !line.startsWith("+++")) additions += 1
|
|
||||||
if (line.startsWith("-") && !line.startsWith("---")) deletions += 1
|
|
||||||
}
|
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
const oldLines = change.oldText ? change.oldText.split("\n") : []
|
const generated = generateUnifiedDiff(
|
||||||
const newLines = change.newText ? change.newText.split("\n") : []
|
change.oldText,
|
||||||
|
change.newText,
|
||||||
deletions += oldLines.length
|
change.path
|
||||||
additions += newLines.length
|
)
|
||||||
|
if (generated) {
|
||||||
diffParts.push(`--- ${change.path}`)
|
diffParts.push(generated)
|
||||||
diffParts.push(`+++ ${change.path}`)
|
|
||||||
for (const line of oldLines) {
|
|
||||||
diffParts.push(`-${line}`)
|
|
||||||
}
|
|
||||||
for (const line of newLines) {
|
|
||||||
diffParts.push(`+${line}`)
|
|
||||||
}
|
|
||||||
diffParts.push("")
|
diffParts.push("")
|
||||||
}
|
}
|
||||||
|
|
||||||
return {
|
|
||||||
additions,
|
|
||||||
deletions,
|
|
||||||
diffCode: diffParts.join("\n").trim(),
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
return diffParts.join("\n").trim()
|
||||||
}, [changes])
|
}, [changes])
|
||||||
|
|
||||||
return (
|
return diffCode ? <UnifiedDiffPreview diffText={diffCode} /> : null
|
||||||
<div className="space-y-2">
|
|
||||||
<div className="flex flex-wrap gap-3 text-xs text-muted-foreground">
|
|
||||||
<span>{t("filesCount", { count: changes.length })}</span>
|
|
||||||
{additions > 0 && <span>+{additions}</span>}
|
|
||||||
{deletions > 0 && <span>-{deletions}</span>}
|
|
||||||
</div>
|
|
||||||
<div className="space-y-1 rounded-md bg-muted/40 p-2">
|
|
||||||
{changes.slice(0, 8).map((change, index) => (
|
|
||||||
<div key={`${change.path}-${index}`} className="flex gap-2 text-xs">
|
|
||||||
<span className="shrink-0 rounded bg-blue-500/15 px-1.5 py-0.5 font-medium uppercase text-blue-600">
|
|
||||||
{t("update")}
|
|
||||||
</span>
|
|
||||||
<span className="break-all font-mono text-foreground">
|
|
||||||
{change.path}
|
|
||||||
</span>
|
|
||||||
</div>
|
|
||||||
))}
|
|
||||||
{changes.length > 8 && (
|
|
||||||
<div className="text-xs text-muted-foreground">
|
|
||||||
{t("moreFiles", { count: changes.length - 8 })}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
{diffCode && <CodeBlock code={diffCode} language="diff" />}
|
|
||||||
</div>
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
/** Bash / exec_command: terminal-style command display */
|
/** Bash / exec_command: terminal-style command display */
|
||||||
@@ -1358,13 +1234,60 @@ function BashToolInput({ input }: { input: Record<string, unknown> }) {
|
|||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Parse structured read output from backend: `{"start_line":N,"content":"..."}`.
|
||||||
|
* Falls back to raw text with startLine=1 if not structured.
|
||||||
|
*/
|
||||||
|
function parseReadOutput(raw: string): { startLine: number; content: string } {
|
||||||
|
try {
|
||||||
|
const parsed = JSON.parse(raw)
|
||||||
|
if (
|
||||||
|
typeof parsed === "object" &&
|
||||||
|
parsed !== null &&
|
||||||
|
typeof parsed.start_line === "number" &&
|
||||||
|
typeof parsed.content === "string"
|
||||||
|
) {
|
||||||
|
return { startLine: parsed.start_line, content: parsed.content }
|
||||||
|
}
|
||||||
|
} catch {
|
||||||
|
// not JSON
|
||||||
|
}
|
||||||
|
return { startLine: 1, content: raw }
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Lightweight file content viewer with line numbers */
|
||||||
|
function FileContentLines({
|
||||||
|
content,
|
||||||
|
startLine = 1,
|
||||||
|
}: {
|
||||||
|
content: string
|
||||||
|
startLine?: number
|
||||||
|
}) {
|
||||||
|
const lines = useMemo(() => content.split("\n"), [content])
|
||||||
|
|
||||||
|
return (
|
||||||
|
<div className="inline-block min-w-full font-mono text-[12px] leading-[20px]">
|
||||||
|
{lines.map((line, i) => (
|
||||||
|
<div key={i} className="flex">
|
||||||
|
<span className="w-[3.5rem] shrink-0 select-none pr-1 text-right text-muted-foreground/40">
|
||||||
|
{startLine + i}
|
||||||
|
</span>
|
||||||
|
<span className="flex-1 whitespace-pre pr-3">{line}</span>
|
||||||
|
</div>
|
||||||
|
))}
|
||||||
|
</div>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
/** Read / Write / NotebookEdit: file-focused display */
|
/** Read / Write / NotebookEdit: file-focused display */
|
||||||
function FileToolInput({
|
function FileToolInput({
|
||||||
toolName,
|
toolName,
|
||||||
input,
|
input,
|
||||||
|
output,
|
||||||
}: {
|
}: {
|
||||||
toolName: string
|
toolName: string
|
||||||
input: Record<string, unknown>
|
input: Record<string, unknown>
|
||||||
|
output?: string | null
|
||||||
}) {
|
}) {
|
||||||
const t = useTranslations("Folder.chat.contentParts")
|
const t = useTranslations("Folder.chat.contentParts")
|
||||||
const name = toolName.toLowerCase()
|
const name = toolName.toLowerCase()
|
||||||
@@ -1377,48 +1300,52 @@ function FileToolInput({
|
|||||||
const pages = str(input, "pages")
|
const pages = str(input, "pages")
|
||||||
const cellType = str(input, "cell_type")
|
const cellType = str(input, "cell_type")
|
||||||
const editMode = str(input, "edit_mode")
|
const editMode = str(input, "edit_mode")
|
||||||
|
const isRead = name === "read" || name === "read file"
|
||||||
|
|
||||||
const lang = filePath
|
const badges: string[] = []
|
||||||
? guessLangFromPath(filePath)
|
if (offset != null) badges.push(t("offset", { offset }))
|
||||||
: ("log" as BundledLanguage)
|
if (limit != null) badges.push(t("limit", { limit }))
|
||||||
|
if (pages) badges.push(t("pages", { pages }))
|
||||||
|
if (editMode) badges.push(t("mode", { mode: editMode }))
|
||||||
|
if (cellType) badges.push(t("cell", { cell: cellType }))
|
||||||
|
|
||||||
|
const { displayContent, startLine } = useMemo(() => {
|
||||||
|
if (isRead && output) {
|
||||||
|
const parsed = parseReadOutput(output)
|
||||||
|
return { displayContent: parsed.content, startLine: parsed.startLine }
|
||||||
|
}
|
||||||
|
return {
|
||||||
|
displayContent: content ?? newSource ?? null,
|
||||||
|
startLine: 1,
|
||||||
|
}
|
||||||
|
}, [isRead, output, content, newSource])
|
||||||
|
|
||||||
return (
|
return (
|
||||||
<div className="space-y-2">
|
<section className="flex max-h-[480px] flex-col rounded-lg border border-border bg-background">
|
||||||
{filePath && (
|
<header className="flex shrink-0 items-center gap-2 border-b border-border bg-muted/40 px-3 py-2 text-[11px]">
|
||||||
<div className="flex items-center gap-2 text-xs">
|
<span className="shrink-0 rounded border border-border bg-background px-1.5 py-0.5 text-[10px] text-muted-foreground">
|
||||||
{name === "read" || name === "read file" ? (
|
{isRead ? "READ" : "WRITE"}
|
||||||
<FileTextIcon className="size-3.5 shrink-0 text-muted-foreground" />
|
|
||||||
) : (
|
|
||||||
<FilePlusIcon className="size-3.5 shrink-0 text-muted-foreground" />
|
|
||||||
)}
|
|
||||||
<span className="break-all font-mono text-foreground">
|
|
||||||
{filePath}
|
|
||||||
</span>
|
</span>
|
||||||
</div>
|
<span
|
||||||
)}
|
className="min-w-0 flex-1 truncate font-mono text-foreground"
|
||||||
{(offset != null || limit != null || pages) && (
|
title={filePath ?? undefined}
|
||||||
<div className="flex flex-wrap gap-3 text-xs text-muted-foreground">
|
>
|
||||||
{offset != null && <span>{t("offset", { offset })}</span>}
|
{filePath ?? t("unknown")}
|
||||||
{limit != null && <span>{t("limit", { limit })}</span>}
|
</span>
|
||||||
{pages && <span>{t("pages", { pages })}</span>}
|
{badges.length > 0 && (
|
||||||
</div>
|
<span className="ml-auto inline-flex shrink-0 items-center gap-2 text-[10px] text-muted-foreground">
|
||||||
)}
|
{badges.map((b) => (
|
||||||
{(cellType || editMode) && (
|
<span key={b}>{b}</span>
|
||||||
<div className="flex flex-wrap gap-3 text-xs text-muted-foreground">
|
|
||||||
{editMode && <span>{t("mode", { mode: editMode })}</span>}
|
|
||||||
{cellType && <span>{t("cell", { cell: cellType })}</span>}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
{(name === "write" || name === "notebookedit") &&
|
|
||||||
(content || newSource) &&
|
|
||||||
(lang === "markdown" || lang === "mdx" ? (
|
|
||||||
<div className="rounded-md border p-3 text-sm prose prose-sm dark:prose-invert max-w-none [&_ul]:list-inside [&_ol]:list-inside">
|
|
||||||
<MessageResponse>{content ?? newSource ?? ""}</MessageResponse>
|
|
||||||
</div>
|
|
||||||
) : (
|
|
||||||
<CodeBlock code={content ?? newSource ?? ""} language={lang} />
|
|
||||||
))}
|
))}
|
||||||
|
</span>
|
||||||
|
)}
|
||||||
|
</header>
|
||||||
|
{displayContent && (
|
||||||
|
<div className="overflow-auto">
|
||||||
|
<FileContentLines content={displayContent} startLine={startLine} />
|
||||||
</div>
|
</div>
|
||||||
|
)}
|
||||||
|
</section>
|
||||||
)
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1629,49 +1556,7 @@ function TodoWriteToolInput({ input }: { input: Record<string, unknown> }) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
function ApplyPatchToolInput({ input }: { input: string }) {
|
function ApplyPatchToolInput({ input }: { input: string }) {
|
||||||
const t = useTranslations("Folder.chat.contentParts")
|
return <UnifiedDiffPreview diffText={input} />
|
||||||
const { files, additions, deletions } = useMemo(
|
|
||||||
() => parseApplyPatchInput(input),
|
|
||||||
[input]
|
|
||||||
)
|
|
||||||
const opClass: Record<ApplyPatchOp, string> = {
|
|
||||||
add: "bg-green-500/15 text-green-600",
|
|
||||||
update: "bg-blue-500/15 text-blue-600",
|
|
||||||
delete: "bg-red-500/15 text-red-600",
|
|
||||||
move: "bg-purple-500/15 text-purple-600",
|
|
||||||
}
|
|
||||||
|
|
||||||
return (
|
|
||||||
<div className="space-y-2">
|
|
||||||
<div className="flex flex-wrap gap-3 text-xs text-muted-foreground">
|
|
||||||
<span>{t("filesCount", { count: files.length })}</span>
|
|
||||||
{additions > 0 && <span>+{additions}</span>}
|
|
||||||
{deletions > 0 && <span>-{deletions}</span>}
|
|
||||||
</div>
|
|
||||||
{files.length > 0 && (
|
|
||||||
<div className="space-y-1 rounded-md bg-muted/40 p-2">
|
|
||||||
{files.slice(0, 8).map((file, index) => (
|
|
||||||
<div key={`${file.path}-${index}`} className="flex gap-2 text-xs">
|
|
||||||
<span
|
|
||||||
className={`shrink-0 rounded px-1.5 py-0.5 font-medium uppercase ${opClass[file.op]}`}
|
|
||||||
>
|
|
||||||
{file.op}
|
|
||||||
</span>
|
|
||||||
<span className="break-all font-mono text-foreground">
|
|
||||||
{file.path}
|
|
||||||
</span>
|
|
||||||
</div>
|
|
||||||
))}
|
|
||||||
{files.length > 8 && (
|
|
||||||
<div className="text-xs text-muted-foreground">
|
|
||||||
{t("moreFiles", { count: files.length - 8 })}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
</div>
|
|
||||||
)}
|
|
||||||
<CodeBlock code={input} language="diff" />
|
|
||||||
</div>
|
|
||||||
)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// ── Switch mode (plan) input ──────────────────────────────────────────
|
// ── Switch mode (plan) input ──────────────────────────────────────────
|
||||||
@@ -1794,20 +1679,41 @@ function GenericToolInput({ input }: { input: string }) {
|
|||||||
|
|
||||||
// ── Dispatcher ───────────────────────────────────────────────────────
|
// ── Dispatcher ───────────────────────────────────────────────────────
|
||||||
|
|
||||||
|
function isTruncatedInput(input: string): boolean {
|
||||||
|
return input.endsWith('..."') || input.endsWith("...")
|
||||||
|
}
|
||||||
|
|
||||||
function StructuredToolInput({
|
function StructuredToolInput({
|
||||||
toolName,
|
toolName,
|
||||||
input,
|
input,
|
||||||
|
output,
|
||||||
}: {
|
}: {
|
||||||
toolName: string
|
toolName: string
|
||||||
input: string
|
input: string
|
||||||
|
output?: string | null
|
||||||
}) {
|
}) {
|
||||||
|
const t = useTranslations("Folder.chat.contentParts")
|
||||||
const name = toolName.toLowerCase()
|
const name = toolName.toLowerCase()
|
||||||
const parsed = tryParseJson(input)
|
const parsed = tryParseJson(input)
|
||||||
|
const truncated =
|
||||||
|
(name === "edit" || name === "write" || name === "apply_patch") &&
|
||||||
|
isTruncatedInput(input)
|
||||||
|
|
||||||
|
const truncationBanner = truncated ? (
|
||||||
|
<div className="rounded-md bg-yellow-500/10 px-2.5 py-1.5 text-[11px] text-yellow-700 dark:text-yellow-400">
|
||||||
|
{t("inputTruncated")}
|
||||||
|
</div>
|
||||||
|
) : null
|
||||||
|
|
||||||
if (name === "apply_patch") {
|
if (name === "apply_patch") {
|
||||||
const patchInput =
|
const patchInput =
|
||||||
extractApplyPatchTextFromUnknownInput(input, parsed) ?? input
|
extractApplyPatchTextFromUnknownInput(input, parsed) ?? input
|
||||||
return <ApplyPatchToolInput input={patchInput} />
|
return (
|
||||||
|
<>
|
||||||
|
{truncationBanner}
|
||||||
|
<ApplyPatchToolInput input={patchInput} />
|
||||||
|
</>
|
||||||
|
)
|
||||||
}
|
}
|
||||||
|
|
||||||
if (name === "bash" || name === "exec_command") {
|
if (name === "bash" || name === "exec_command") {
|
||||||
@@ -1831,16 +1737,31 @@ function StructuredToolInput({
|
|||||||
if (name === "edit") {
|
if (name === "edit") {
|
||||||
const patchInput = extractApplyPatchTextFromUnknownInput(input, parsed)
|
const patchInput = extractApplyPatchTextFromUnknownInput(input, parsed)
|
||||||
if (patchInput) {
|
if (patchInput) {
|
||||||
return <ApplyPatchToolInput input={patchInput} />
|
return (
|
||||||
|
<>
|
||||||
|
{truncationBanner}
|
||||||
|
<ApplyPatchToolInput input={patchInput} />
|
||||||
|
</>
|
||||||
|
)
|
||||||
}
|
}
|
||||||
if (parsed) {
|
if (parsed) {
|
||||||
const changesPayload = extractEditChangesPayload(parsed)
|
const changesPayload = extractEditChangesPayload(parsed)
|
||||||
if (changesPayload.length > 0) {
|
if (changesPayload.length > 0) {
|
||||||
return <EditChangesToolInput changes={changesPayload} />
|
return (
|
||||||
|
<>
|
||||||
|
{truncationBanner}
|
||||||
|
<EditChangesToolInput changes={changesPayload} />
|
||||||
|
</>
|
||||||
|
)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if (isCanonicalEditPayload(parsed)) {
|
if (isCanonicalEditPayload(parsed)) {
|
||||||
return <EditToolInput input={parsed} />
|
return (
|
||||||
|
<>
|
||||||
|
{truncationBanner}
|
||||||
|
<EditToolInput input={parsed} />
|
||||||
|
</>
|
||||||
|
)
|
||||||
}
|
}
|
||||||
return <GenericToolInput input={input} />
|
return <GenericToolInput input={input} />
|
||||||
}
|
}
|
||||||
@@ -1852,7 +1773,7 @@ function StructuredToolInput({
|
|||||||
name === "write" ||
|
name === "write" ||
|
||||||
name === "notebookedit"
|
name === "notebookedit"
|
||||||
)
|
)
|
||||||
return <FileToolInput toolName={toolName} input={parsed} />
|
return <FileToolInput toolName={toolName} input={parsed} output={output} />
|
||||||
if (name === "glob" || name === "grep")
|
if (name === "glob" || name === "grep")
|
||||||
return <SearchToolInput toolName={toolName} input={parsed} />
|
return <SearchToolInput toolName={toolName} input={parsed} />
|
||||||
if (name === "webfetch" || name === "websearch")
|
if (name === "webfetch" || name === "websearch")
|
||||||
@@ -2275,12 +2196,18 @@ const ToolCallPart = memo(function ToolCallPart({
|
|||||||
displayCommand,
|
displayCommand,
|
||||||
isRunning,
|
isRunning,
|
||||||
])
|
])
|
||||||
|
const isFileTool =
|
||||||
|
toolNameLower === "read" ||
|
||||||
|
toolNameLower === "read file" ||
|
||||||
|
toolNameLower === "write" ||
|
||||||
|
toolNameLower === "notebookedit"
|
||||||
const shouldHideDuplicateResult =
|
const shouldHideDuplicateResult =
|
||||||
(toolNameLower === "edit" ||
|
(toolNameLower === "edit" ||
|
||||||
toolNameLower === "apply_patch" ||
|
toolNameLower === "apply_patch" ||
|
||||||
toolNameLower === "switch_mode" ||
|
toolNameLower === "switch_mode" ||
|
||||||
toolNameLower === "enterplanmode" ||
|
toolNameLower === "enterplanmode" ||
|
||||||
toolNameLower === "exitplanmode") &&
|
toolNameLower === "exitplanmode" ||
|
||||||
|
isFileTool) &&
|
||||||
!part.errorText
|
!part.errorText
|
||||||
const open = (isRunning && (isCommandTool || hasLiveOutput)) || manualOpen
|
const open = (isRunning && (isCommandTool || hasLiveOutput)) || manualOpen
|
||||||
|
|
||||||
@@ -2299,6 +2226,7 @@ const ToolCallPart = memo(function ToolCallPart({
|
|||||||
<StructuredToolInput
|
<StructuredToolInput
|
||||||
toolName={normalizedToolName}
|
toolName={normalizedToolName}
|
||||||
input={part.input}
|
input={part.input}
|
||||||
|
output={part.output}
|
||||||
/>
|
/>
|
||||||
)}
|
)}
|
||||||
{(toolNameLower === "task" || toolNameLower === "agent") &&
|
{(toolNameLower === "task" || toolNameLower === "agent") &&
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "يتم عرض نهاية المخرجات أثناء البث لتحسين الأداء.",
|
"showingTailOutput": "يتم عرض نهاية المخرجات أثناء البث لتحسين الأداء.",
|
||||||
"result": "النتيجة",
|
"result": "النتيجة",
|
||||||
"unknown": "غير معروف",
|
"unknown": "غير معروف",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "استبدال الكل",
|
"replaceAll": "استبدال الكل",
|
||||||
"filesCount": "الملفات: {count}",
|
"filesCount": "الملفات: {count}",
|
||||||
"update": "تحديث",
|
"update": "تحديث",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "Zur besseren Performance wird während des Streamings nur die Endausgabe angezeigt.",
|
"showingTailOutput": "Zur besseren Performance wird während des Streamings nur die Endausgabe angezeigt.",
|
||||||
"result": "Ergebnis",
|
"result": "Ergebnis",
|
||||||
"unknown": "unbekannt",
|
"unknown": "unbekannt",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "ALLES ERSETZEN",
|
"replaceAll": "ALLES ERSETZEN",
|
||||||
"filesCount": "Dateien: {count}",
|
"filesCount": "Dateien: {count}",
|
||||||
"update": "aktualisieren",
|
"update": "aktualisieren",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "Showing tail output while streaming for performance.",
|
"showingTailOutput": "Showing tail output while streaming for performance.",
|
||||||
"result": "Result",
|
"result": "Result",
|
||||||
"unknown": "unknown",
|
"unknown": "unknown",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "REPLACE ALL",
|
"replaceAll": "REPLACE ALL",
|
||||||
"filesCount": "Files: {count}",
|
"filesCount": "Files: {count}",
|
||||||
"update": "update",
|
"update": "update",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "Mostrando la salida final durante el streaming para mejorar el rendimiento.",
|
"showingTailOutput": "Mostrando la salida final durante el streaming para mejorar el rendimiento.",
|
||||||
"result": "Resultado",
|
"result": "Resultado",
|
||||||
"unknown": "desconocido",
|
"unknown": "desconocido",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "REEMPLAZAR TODO",
|
"replaceAll": "REEMPLAZAR TODO",
|
||||||
"filesCount": "Archivos: {count}",
|
"filesCount": "Archivos: {count}",
|
||||||
"update": "actualizar",
|
"update": "actualizar",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "Affichage de la fin de la sortie pendant le streaming pour de meilleures performances.",
|
"showingTailOutput": "Affichage de la fin de la sortie pendant le streaming pour de meilleures performances.",
|
||||||
"result": "Résultat",
|
"result": "Résultat",
|
||||||
"unknown": "inconnu",
|
"unknown": "inconnu",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "TOUT REMPLACER",
|
"replaceAll": "TOUT REMPLACER",
|
||||||
"filesCount": "Fichiers : {count}",
|
"filesCount": "Fichiers : {count}",
|
||||||
"update": "mettre à jour",
|
"update": "mettre à jour",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "パフォーマンスのため、ストリーミング中は末尾出力を表示しています。",
|
"showingTailOutput": "パフォーマンスのため、ストリーミング中は末尾出力を表示しています。",
|
||||||
"result": "結果",
|
"result": "結果",
|
||||||
"unknown": "不明",
|
"unknown": "不明",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "すべて置換",
|
"replaceAll": "すべて置換",
|
||||||
"filesCount": "ファイル: {count}",
|
"filesCount": "ファイル: {count}",
|
||||||
"update": "更新",
|
"update": "更新",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "성능을 위해 스트리밍 중에는 출력의 끝부분만 표시합니다.",
|
"showingTailOutput": "성능을 위해 스트리밍 중에는 출력의 끝부분만 표시합니다.",
|
||||||
"result": "결과",
|
"result": "결과",
|
||||||
"unknown": "알 수 없음",
|
"unknown": "알 수 없음",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "모두 바꾸기",
|
"replaceAll": "모두 바꾸기",
|
||||||
"filesCount": "파일: {count}",
|
"filesCount": "파일: {count}",
|
||||||
"update": "업데이트",
|
"update": "업데이트",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "Mostrando a saída final durante o streaming para melhor desempenho.",
|
"showingTailOutput": "Mostrando a saída final durante o streaming para melhor desempenho.",
|
||||||
"result": "Resultado",
|
"result": "Resultado",
|
||||||
"unknown": "desconhecido",
|
"unknown": "desconhecido",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "SUBSTITUIR TUDO",
|
"replaceAll": "SUBSTITUIR TUDO",
|
||||||
"filesCount": "Arquivos: {count}",
|
"filesCount": "Arquivos: {count}",
|
||||||
"update": "atualizar",
|
"update": "atualizar",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "为保证性能,流式输出时仅显示尾部内容。",
|
"showingTailOutput": "为保证性能,流式输出时仅显示尾部内容。",
|
||||||
"result": "结果",
|
"result": "结果",
|
||||||
"unknown": "未知",
|
"unknown": "未知",
|
||||||
|
"inputTruncated": "输入已截断,diff 可能不完整。",
|
||||||
"replaceAll": "全部替换",
|
"replaceAll": "全部替换",
|
||||||
"filesCount": "文件:{count}",
|
"filesCount": "文件:{count}",
|
||||||
"update": "更新",
|
"update": "更新",
|
||||||
|
|||||||
@@ -1485,6 +1485,7 @@
|
|||||||
"showingTailOutput": "為確保效能,串流輸出時僅顯示尾端內容。",
|
"showingTailOutput": "為確保效能,串流輸出時僅顯示尾端內容。",
|
||||||
"result": "結果",
|
"result": "結果",
|
||||||
"unknown": "未知",
|
"unknown": "未知",
|
||||||
|
"inputTruncated": "Input was truncated — diff may be incomplete.",
|
||||||
"replaceAll": "全部替換",
|
"replaceAll": "全部替換",
|
||||||
"filesCount": "檔案:{count}",
|
"filesCount": "檔案:{count}",
|
||||||
"update": "更新",
|
"update": "更新",
|
||||||
|
|||||||
@@ -1,5 +1,6 @@
|
|||||||
import type { MessageTurn } from "./types"
|
import type { MessageTurn } from "./types"
|
||||||
import { normalizeToolName } from "./tool-call-normalization"
|
import { normalizeToolName } from "./tool-call-normalization"
|
||||||
|
import { generateUnifiedDiff } from "./unified-diff-generator"
|
||||||
|
|
||||||
export type FileOperation = "read" | "edit" | "write" | "apply_patch"
|
export type FileOperation = "read" | "edit" | "write" | "apply_patch"
|
||||||
|
|
||||||
@@ -266,32 +267,12 @@ function countDiffLines(text: string): DiffStat {
|
|||||||
return { additions, deletions }
|
return { additions, deletions }
|
||||||
}
|
}
|
||||||
|
|
||||||
function createHunkHeader(oldLineCount: number, newLineCount: number): string {
|
|
||||||
const oldStart = oldLineCount === 0 ? 0 : 1
|
|
||||||
const newStart = newLineCount === 0 ? 0 : 1
|
|
||||||
return `@@ -${oldStart},${oldLineCount} +${newStart},${newLineCount} @@`
|
|
||||||
}
|
|
||||||
|
|
||||||
function buildUnifiedDiff(
|
function buildUnifiedDiff(
|
||||||
filePath: string,
|
filePath: string,
|
||||||
oldText: string,
|
oldText: string,
|
||||||
newText: string
|
newText: string
|
||||||
): string | null {
|
): string | null {
|
||||||
if (!oldText && !newText) return null
|
return generateUnifiedDiff(oldText, newText, filePath)
|
||||||
|
|
||||||
const oldLines = oldText ? oldText.split("\n") : []
|
|
||||||
const newLines = newText ? newText.split("\n") : []
|
|
||||||
|
|
||||||
const lines: string[] = [
|
|
||||||
`--- a/${filePath}`,
|
|
||||||
`+++ b/${filePath}`,
|
|
||||||
createHunkHeader(oldLines.length, newLines.length),
|
|
||||||
]
|
|
||||||
|
|
||||||
for (const line of oldLines) lines.push(`-${line}`)
|
|
||||||
for (const line of newLines) lines.push(`+${line}`)
|
|
||||||
|
|
||||||
return lines.join("\n")
|
|
||||||
}
|
}
|
||||||
|
|
||||||
function parseEditChangeValue(value: unknown): EditChangePreview | null {
|
function parseEditChangeValue(value: unknown): EditChangePreview | null {
|
||||||
|
|||||||
180
src/lib/unified-diff-generator.ts
Normal file
180
src/lib/unified-diff-generator.ts
Normal file
@@ -0,0 +1,180 @@
|
|||||||
|
import { computeLineDiff, type DiffHunk } from "@/components/merge/merge-diff"
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Maximum product of line counts before falling back to naive diff.
|
||||||
|
* Avoids O(n*m) LCS blowup for very large inputs.
|
||||||
|
*/
|
||||||
|
const LCS_PAIR_BUDGET = 200_000
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Generate a unified diff string from old and new text.
|
||||||
|
*
|
||||||
|
* Uses LCS-based line diff when within budget, falls back to
|
||||||
|
* simple "all deletions then all additions" for very large inputs.
|
||||||
|
*/
|
||||||
|
export function generateUnifiedDiff(
|
||||||
|
oldText: string,
|
||||||
|
newText: string,
|
||||||
|
filePath?: string,
|
||||||
|
contextLines: number = 3
|
||||||
|
): string | null {
|
||||||
|
if (!oldText && !newText) return null
|
||||||
|
if (oldText === newText) return null
|
||||||
|
|
||||||
|
const oldLines = oldText ? splitLines(oldText) : []
|
||||||
|
const newLines = newText ? splitLines(newText) : []
|
||||||
|
|
||||||
|
const path = filePath ?? "file"
|
||||||
|
const header = `--- a/${path}\n+++ b/${path}`
|
||||||
|
|
||||||
|
// Performance gate: fall back to naive diff for large inputs
|
||||||
|
if (oldLines.length * newLines.length > LCS_PAIR_BUDGET) {
|
||||||
|
return buildNaiveDiff(header, oldLines, newLines)
|
||||||
|
}
|
||||||
|
|
||||||
|
const hunks = computeLineDiff(oldLines, newLines)
|
||||||
|
if (hunks.length === 0) return null
|
||||||
|
|
||||||
|
const unifiedHunks = buildUnifiedHunks(oldLines, hunks, contextLines)
|
||||||
|
|
||||||
|
return `${header}\n${unifiedHunks}`
|
||||||
|
}
|
||||||
|
|
||||||
|
function splitLines(text: string): string[] {
|
||||||
|
const lines = text.split("\n")
|
||||||
|
// Remove trailing empty line from trailing newline
|
||||||
|
if (lines.length > 0 && lines[lines.length - 1] === "") {
|
||||||
|
lines.pop()
|
||||||
|
}
|
||||||
|
return lines
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Naive diff: all deletions first, then all additions, with a single hunk header.
|
||||||
|
* Used as fallback when inputs are too large for LCS.
|
||||||
|
*/
|
||||||
|
function buildNaiveDiff(
|
||||||
|
header: string,
|
||||||
|
oldLines: string[],
|
||||||
|
newLines: string[]
|
||||||
|
): string {
|
||||||
|
const oldStart = oldLines.length === 0 ? 0 : 1
|
||||||
|
const newStart = newLines.length === 0 ? 0 : 1
|
||||||
|
const hunkHeader = `@@ -${oldStart},${oldLines.length} +${newStart},${newLines.length} @@`
|
||||||
|
|
||||||
|
const parts = [header, hunkHeader]
|
||||||
|
for (const line of oldLines) parts.push(`-${line}`)
|
||||||
|
for (const line of newLines) parts.push(`+${line}`)
|
||||||
|
return parts.join("\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Convert DiffHunk[] into unified diff text with context lines and hunk headers.
|
||||||
|
*
|
||||||
|
* Groups nearby hunks that overlap in their context windows into a single
|
||||||
|
* unified hunk, producing output similar to `diff -u`.
|
||||||
|
*/
|
||||||
|
function buildUnifiedHunks(
|
||||||
|
oldLines: string[],
|
||||||
|
hunks: DiffHunk[],
|
||||||
|
contextLines: number
|
||||||
|
): string {
|
||||||
|
// Build "change regions" with context, then merge overlapping ones
|
||||||
|
const regions = hunks.map((hunk) => ({
|
||||||
|
// Context-expanded range in old lines
|
||||||
|
ctxOldStart: Math.max(0, hunk.baseStart - contextLines),
|
||||||
|
ctxOldEnd: Math.min(
|
||||||
|
oldLines.length,
|
||||||
|
hunk.baseStart + hunk.baseCount + contextLines
|
||||||
|
),
|
||||||
|
hunk,
|
||||||
|
}))
|
||||||
|
|
||||||
|
// Merge overlapping regions
|
||||||
|
const merged: {
|
||||||
|
ctxOldStart: number
|
||||||
|
ctxOldEnd: number
|
||||||
|
hunks: DiffHunk[]
|
||||||
|
}[] = []
|
||||||
|
|
||||||
|
for (const region of regions) {
|
||||||
|
const last = merged[merged.length - 1]
|
||||||
|
if (last && region.ctxOldStart <= last.ctxOldEnd) {
|
||||||
|
// Overlapping — extend and add hunk
|
||||||
|
last.ctxOldEnd = Math.max(last.ctxOldEnd, region.ctxOldEnd)
|
||||||
|
last.hunks.push(region.hunk)
|
||||||
|
} else {
|
||||||
|
merged.push({
|
||||||
|
ctxOldStart: region.ctxOldStart,
|
||||||
|
ctxOldEnd: region.ctxOldEnd,
|
||||||
|
hunks: [region.hunk],
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Render each merged region as a unified hunk
|
||||||
|
const output: string[] = []
|
||||||
|
|
||||||
|
for (const group of merged) {
|
||||||
|
const lines: string[] = []
|
||||||
|
let oldCursor = group.ctxOldStart
|
||||||
|
let newLineCount = 0
|
||||||
|
const oldLineCount = group.ctxOldEnd - group.ctxOldStart
|
||||||
|
|
||||||
|
for (const hunk of group.hunks) {
|
||||||
|
// Context lines before this change
|
||||||
|
while (oldCursor < hunk.baseStart) {
|
||||||
|
lines.push(` ${oldLines[oldCursor]}`)
|
||||||
|
newLineCount++
|
||||||
|
oldCursor++
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deleted lines
|
||||||
|
for (let i = 0; i < hunk.baseCount; i++) {
|
||||||
|
lines.push(`-${oldLines[hunk.baseStart + i]}`)
|
||||||
|
oldCursor++
|
||||||
|
}
|
||||||
|
|
||||||
|
// Added lines
|
||||||
|
for (const newLine of hunk.newLines) {
|
||||||
|
lines.push(`+${newLine}`)
|
||||||
|
newLineCount++
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Trailing context
|
||||||
|
while (oldCursor < group.ctxOldEnd) {
|
||||||
|
lines.push(` ${oldLines[oldCursor]}`)
|
||||||
|
newLineCount++
|
||||||
|
oldCursor++
|
||||||
|
}
|
||||||
|
|
||||||
|
// Compute hunk header
|
||||||
|
const oldStart = oldLineCount === 0 ? 0 : group.ctxOldStart + 1
|
||||||
|
const newStart =
|
||||||
|
newLineCount === 0
|
||||||
|
? 0
|
||||||
|
: group.ctxOldStart +
|
||||||
|
1 +
|
||||||
|
computeNewOffset(group.hunks, group.ctxOldStart)
|
||||||
|
|
||||||
|
output.push(
|
||||||
|
`@@ -${oldStart},${oldLineCount} +${newStart},${newLineCount} @@`
|
||||||
|
)
|
||||||
|
output.push(...lines)
|
||||||
|
}
|
||||||
|
|
||||||
|
return output.join("\n")
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Compute the offset applied to new-line numbering by hunks before a given position.
|
||||||
|
*/
|
||||||
|
function computeNewOffset(hunks: DiffHunk[], beforeOldLine: number): number {
|
||||||
|
let offset = 0
|
||||||
|
for (const hunk of hunks) {
|
||||||
|
if (hunk.baseStart >= beforeOldLine) break
|
||||||
|
offset += hunk.newLines.length - hunk.baseCount
|
||||||
|
}
|
||||||
|
return offset
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user