Merge pull request #600 from creatang/main

feat(tui): 优化 Markdown 渲染与折叠性能，并新增 /web 启动入口

Author: phantom5099

internal/tui/core/app/app.go

@@ -163,6 +163,7 @@ type appRuntimeState struct {
 	logPersistDirty                  bool
 	logPersistVersion                int
 	transcriptContent                string
+	transcriptBlockRenderCache       map[string]string
 	transcriptProcessFoldAvailable   bool
 	transcriptProcessExpanded        bool
 	transcriptProcessExpandedOrdinal int
@@ -395,12 +396,13 @@ func newApp(container tuibootstrap.Container) (App, error) {
 			markdownRenderer: markdownRenderer,
 		},
 		appRuntimeState: appRuntimeState{
-			nowFn:        time.Now,
-			focus:        panelInput,
-			todoFilter:   todoFilterAll,
-			layoutCached: true,
-			cachedWidth:  128,
-			cachedHeight: 40,
+			nowFn:                      time.Now,
+			focus:                      panelInput,
+			todoFilter:                 todoFilterAll,
+			layoutCached:               true,
+			cachedWidth:                128,
+			cachedHeight:               40,
+			transcriptBlockRenderCache: make(map[string]string),
 			// 初始进入草稿态时锁定启动页，直到发送或切换 session 才退出。
 			startupScreenLocked: true,
 		},

internal/tui/core/app/commands.go

@@ -36,6 +36,7 @@ const (
 	slashCommandSkills      = "/skills"
 	slashCommandSkill       = "/skill"
 	slashCommandCheckpoint  = "/checkpoint"
+	slashCommandWeb         = "/web"
 
 	slashUsageHelp              = "/help"
 	slashUsageExit              = "/exit"
@@ -56,6 +57,7 @@ const (
 	slashUsageCheckpointRestore = "/checkpoint restore <id>"
 	slashUsageCheckpointUndo    = "/checkpoint undo"
 	slashUsageCheckpointDiff    = "/checkpoint diff <id>"
+	slashUsageWeb               = "/web"
 
 	commandMenuTitle        = "Suggestions"
 	providerPickerTitle     = "Select Provider"
@@ -158,6 +160,7 @@ var builtinSlashCommands = []slashCommand{
 	{Usage: slashUsageProviderAdd, Description: "Add a new custom provider"},
 	{Usage: slashUsageModel, Description: "Open the interactive model picker"},
 	{Usage: slashUsageSession, Description: "Switch to another session"},
+	{Usage: slashUsageWeb, Description: "Start Web UI in browser"},
 	{Usage: slashUsageExit, Description: "Exit NeoCode"},
 }
 

internal/tui/core/app/commands_test.go

@@ -23,6 +23,7 @@ func TestBuiltinSlashCommands(t *testing.T) {
 	foundSkills := false
 	foundSkillUse := false
 	foundCheckpoint := false
+	foundWeb := false
 	foundStatus := false
 	for _, cmd := range builtinSlashCommands {
 		if cmd.Usage == slashUsageHelp {
@@ -40,6 +41,9 @@ func TestBuiltinSlashCommands(t *testing.T) {
 		if cmd.Usage == slashUsageCheckpoint {
 			foundCheckpoint = true
 		}
+		if cmd.Usage == slashUsageWeb {
+			foundWeb = true
+		}
 		if strings.EqualFold(cmd.Usage, "/status") {
 			foundStatus = true
 		}
@@ -59,6 +63,9 @@ func TestBuiltinSlashCommands(t *testing.T) {
 	if !foundCheckpoint {
 		t.Error("expected to find /checkpoint command")
 	}
+	if !foundWeb {
+		t.Error("expected to find /web command")
+	}
 	if foundStatus {
 		t.Error("did not expect /status command in builtin slash commands")
 	}

internal/tui/core/app/markdown_renderer.go

@@ -4,7 +4,7 @@ import tuiinfra "neo-code/internal/tui/infra"
 
 const (
 	defaultMarkdownStyle    = "dark"
-	defaultMarkdownCacheMax = 128
+	defaultMarkdownCacheMax = 512
 )
 
 type markdownContentRenderer interface {

internal/tui/core/app/styles.go

@@ -4,6 +4,7 @@ import (
 	"strings"
 
 	"github.com/charmbracelet/lipgloss"
+	"github.com/charmbracelet/x/ansi"
 )
 
 const (
@@ -171,10 +172,12 @@ func newStyles() styles {
 		messageAgentTag: tagStyle(neoBadge),
 		messageBody: lipgloss.NewStyle().
 			Foreground(lipgloss.Color(neoText)).
-			MarginLeft(3),
+			MarginLeft(2).
+			PaddingLeft(1),
 		messageUserBody: lipgloss.NewStyle().
 			Foreground(lipgloss.Color(youText)).
-			MarginLeft(3),
+			MarginLeft(2).
+			PaddingLeft(1),
 		inlineNotice: lipgloss.NewStyle().
 			Foreground(lipgloss.Color(oliveGray)).
 			Italic(true),
@@ -294,45 +297,14 @@ func wrapPlain(text string, width int) string {
 	if width <= 0 {
 		return text
 	}
-
-	lines := strings.Split(text, "\n")
-	out := make([]string, 0, len(lines))
-	for _, line := range lines {
-		runes := []rune(line)
-		if len(runes) == 0 {
-			out = append(out, "")
-			continue
-		}
-		for len(runes) > width {
-			out = append(out, string(runes[:width]))
-			runes = runes[width:]
-		}
-		out = append(out, string(runes))
-	}
-	return strings.Join(out, "\n")
+	return wrapByDisplayWidth(text, width, false)
 }
 
 func wrapCodeBlock(text string, width int) string {
 	if width <= 0 {
 		return text
 	}
-
-	lines := strings.Split(strings.ReplaceAll(text, "\r\n", "\n"), "\n")
-	out := make([]string, 0, len(lines))
-	for _, line := range lines {
-		expanded := strings.ReplaceAll(line, "\t", "    ")
-		runes := []rune(expanded)
-		if len(runes) == 0 {
-			out = append(out, "")
-			continue
-		}
-		for len(runes) > width {
-			out = append(out, string(runes[:width]))
-			runes = runes[width:]
-		}
-		out = append(out, string(runes))
-	}
-	return strings.Join(out, "\n")
+	return wrapByDisplayWidth(text, width, true)
 }
 
 func preview(text string, width int, lines int) string {
@@ -351,9 +323,45 @@ func preview(text string, width int, lines int) string {
 		return "(empty)"
 	}
 	joined := strings.Join(out, "\n")
-	runes := []rune(joined)
-	if len(runes) > width*lines {
-		return string(runes[:width*lines-3]) + "..."
+	maxWidth := width * lines
+	if maxWidth <= 0 {
+		return joined
+	}
+	if ansi.StringWidth(joined) > maxWidth {
+		if maxWidth <= 3 {
+			return ansi.Cut(joined, 0, maxWidth)
+		}
+		return ansi.Cut(joined, 0, maxWidth-3) + "..."
 	}
 	return joined
 }
+
+func wrapByDisplayWidth(text string, width int, expandTabs bool) string {
+	if width <= 0 {
+		return text
+	}
+
+	normalized := strings.ReplaceAll(text, "\r\n", "\n")
+	lines := strings.Split(normalized, "\n")
+	out := make([]string, 0, len(lines))
+	for _, line := range lines {
+		if expandTabs {
+			line = strings.ReplaceAll(line, "\t", "    ")
+		}
+		if line == "" {
+			out = append(out, "")
+			continue
+		}
+		remaining := line
+		for ansi.StringWidth(remaining) > width {
+			chunk := ansi.Cut(remaining, 0, width)
+			if chunk == "" {
+				break
+			}
+			out = append(out, chunk)
+			remaining = ansi.Cut(remaining, width, ansi.StringWidth(remaining))
+		}
+		out = append(out, remaining)
+	}
+	return strings.Join(out, "\n")
+}

internal/tui/core/app/update.go

@@ -69,6 +69,7 @@ const pasteSessionPerLineGuard = 8 * time.Millisecond
 const inlineLogMarker = "[[neo-log]] "
 const sessionWorkdirMissingWarning = "Session workspace not found, keeping current workspace."
 const localLogViewerPersistDir = "log-viewer"
+const transcriptBlockRenderCacheMax = 2048
 
 type sessionLogPersistenceRuntime interface {
 	LoadSessionLogEntries(ctx context.Context, sessionID string) ([]tuiservices.SessionLogEntry, error)
@@ -5591,20 +5592,63 @@ func (a *App) normalizeComposerHeight() {
 }
 
 func (a *App) rebuildTranscript() {
+	a.rebuildTranscriptInternal(false)
+}
+
+func (a *App) rebuildTranscriptForFoldToggle() {
+	a.rebuildTranscriptInternal(true)
+}
+
+func (a *App) rebuildTranscriptInternal(foldToggleOnly bool) {
 	width := max(24, a.transcript.Width)
 	if len(a.activeMessages) == 0 {
 		queued := a.renderQueuedInterventionBlock(width)
 		if strings.TrimSpace(queued) == "" {
 			a.setTranscriptContent(a.styles.empty.Width(width).Render(emptyConversationText))
 			a.transcript.GotoTop()
+			if foldToggleOnly {
+				a.transcriptScrollbarDrag = false
+				a.clearTextSelection()
+			}
 			return
 		}
 		a.setTranscriptContent(queued)
 		a.transcript.GotoTop()
+		if foldToggleOnly {
+			a.transcriptScrollbarDrag = false
+			a.clearTextSelection()
+		}
 		return
 	}
 
 	atBottom := a.transcript.AtBottom()
+	content, hasBlock := a.composeTranscriptContent(width)
+	if !hasBlock {
+		a.setTranscriptContent(a.styles.empty.Width(width).Render(emptyConversationText))
+		a.transcript.GotoTop()
+		if foldToggleOnly {
+			a.transcriptScrollbarDrag = false
+			a.clearTextSelection()
+		}
+		return
+	}
+
+	a.setTranscriptContent(content)
+	if atBottom {
+		a.transcript.GotoBottom()
+	}
+
+	if foldToggleOnly {
+		a.transcriptScrollbarDrag = false
+		a.clearTextSelection()
+		maxOffset := a.transcriptMaxOffset()
+		if a.transcript.YOffset > maxOffset {
+			a.transcript.SetYOffset(maxOffset)
+		}
+	}
+}
+
+func (a *App) composeTranscriptContent(width int) (string, bool) {
 	foldSegments := findTranscriptProcessFoldSegments(a.activeMessages)
 	foldExists := len(foldSegments) > 0
 	a.transcriptProcessFoldAvailable = foldExists
@@ -5667,7 +5711,7 @@ func (a *App) rebuildTranscript() {
 		if inlineLog && lastRenderedRole == roleAssistant {
 			continuation = true
 		}
-		rendered, _ := a.renderMessageBlockWithCopy(message, width, 0, !continuation)
+		rendered := a.renderMessageBlockForTranscript(message, width, !continuation)
 		if rendered == "" {
 			continue
 		}
@@ -5695,15 +5739,36 @@ func (a *App) rebuildTranscript() {
 	}
 
 	if !hasBlock {
-		a.setTranscriptContent(a.styles.empty.Width(width).Render(emptyConversationText))
-		a.transcript.GotoTop()
-		return
+		return "", false
 	}
 
-	a.setTranscriptContent(builder.String())
-	if atBottom {
-		a.transcript.GotoBottom()
+	return builder.String(), true
+}
+
+func (a *App) renderMessageBlockForTranscript(message providertypes.Message, width int, includeTag bool) string {
+	if a.transcriptBlockRenderCache == nil {
+		a.transcriptBlockRenderCache = make(map[string]string)
+	}
+	key := transcriptBlockRenderCacheKey(message, width, includeTag)
+	if cached, ok := a.transcriptBlockRenderCache[key]; ok {
+		return cached
+	}
+
+	rendered, _ := a.renderMessageBlockWithCopy(message, width, 0, includeTag)
+	if rendered == "" {
+		return ""
 	}
+	if len(a.transcriptBlockRenderCache) >= transcriptBlockRenderCacheMax {
+		clear(a.transcriptBlockRenderCache)
+	}
+	a.transcriptBlockRenderCache[key] = rendered
+	return rendered
+}
+
+func transcriptBlockRenderCacheKey(message providertypes.Message, width int, includeTag bool) string {
+	content := renderMessagePartsForDisplay(message.Parts)
+	sum := sha256.Sum256([]byte(content))
+	return fmt.Sprintf("%s|%t|%d|%t|%x", message.Role, message.IsError, width, includeTag, sum[:8])
 }
 
 func (a *App) toggleTranscriptProcessExpansion() bool {
@@ -5736,7 +5801,7 @@ func (a *App) toggleTranscriptProcessExpansionWithAnchor(anchorViewportRow int,
 	} else {
 		a.state.StatusText = "Process output collapsed"
 	}
-	a.rebuildTranscript()
+	a.rebuildTranscriptForFoldToggle()
 	if anchorViewportRow >= 0 {
 		a.pinTranscriptProcessControlRow(anchorViewportRow, controlOrdinal)
 	}
@@ -6024,6 +6089,8 @@ func (a *App) handleImmediateSlashCommand(input string) (bool, tea.Cmd) {
 		return true, a.handleSkillCommand(rest)
 	case slashCommandCheckpoint:
 		return true, a.handleCheckpointCommand(rest)
+	case slashCommandWeb:
+		return true, a.handleWebCommand(rest)
 	case slashCommandSession:
 		if err := a.ensureSessionSwitchAllowed(""); err != nil {
 			a.state.ExecutionError = err.Error()
@@ -6101,6 +6168,7 @@ func (a *App) startDraftSession() {
 	a.startupScreenLocked = false
 	a.state.ActiveSessionTitle = draftSessionTitle
 	a.activeMessages = nil
+	clear(a.transcriptBlockRenderCache)
 	a.transcriptProcessFoldAvailable = false
 	a.transcriptProcessExpanded = false
 	a.clearActivities()