mirror of
https://github.com/mark3labs/kit.git
synced 2026-06-14 03:30:26 +00:00
Ed Zynda
28d2de8f39
Moved leaf utility files to subpackages for better organization: - events.go -> core/ (core message types) - clipboard.go -> clipboard/ (clipboard operations) - commands.go -> commands/ (slash commands) - file_processor.go -> fileutil/ (file attachment processing) - preferences.go -> prefs/ (theme/model preferences) - enhanced_styles.go, styles.go, themes.go -> style/ (theming system) Added exports.go to re-export commonly used types for backward compatibility. External importers can still use ui.XXX without changes. All tests pass, basic smoke test successful.
84 lines
2.8 KiB
Go
84 lines
2.8 KiB
Go
package ui
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"charm.land/lipgloss/v2"
|
|
|
|
"github.com/mark3labs/kit/internal/ui/style"
|
|
)
|
|
|
|
// Renderer is the interface satisfied by MessageRenderer. It allows model.go
|
|
// and cli.go to call rendering methods uniformly.
|
|
type Renderer interface {
|
|
RenderUserMessage(content string, timestamp time.Time) UIMessage
|
|
RenderAssistantMessage(content string, timestamp time.Time, modelName string) UIMessage
|
|
RenderReasoningBlock(content string, timestamp time.Time) UIMessage
|
|
RenderToolMessage(toolName, toolArgs, toolResult string, isError bool) UIMessage
|
|
RenderSystemMessage(content string, timestamp time.Time) UIMessage
|
|
RenderErrorMessage(errorMsg string, timestamp time.Time) UIMessage
|
|
RenderDebugMessage(message string, timestamp time.Time) UIMessage
|
|
RenderDebugConfigMessage(config map[string]any, timestamp time.Time) UIMessage
|
|
SetWidth(width int)
|
|
UpdateTheme()
|
|
}
|
|
|
|
// Compile-time check that MessageRenderer satisfies the Renderer interface.
|
|
var _ Renderer = (*MessageRenderer)(nil)
|
|
|
|
// parseBashOutput parses <stdout>/<stderr> tagged output from bash tool
|
|
// results, styling stderr with the theme's error color. Returns the
|
|
// combined, styled output string with tags stripped.
|
|
//
|
|
// Shared by MessageRenderer.
|
|
func parseBashOutput(result string, theme style.Theme) string {
|
|
var formattedResult strings.Builder
|
|
remaining := result
|
|
|
|
for {
|
|
// Find stderr tags
|
|
stderrStart := strings.Index(remaining, "<stderr>")
|
|
stderrEnd := strings.Index(remaining, "</stderr>")
|
|
|
|
// Find stdout tags
|
|
stdoutStart := strings.Index(remaining, "<stdout>")
|
|
stdoutEnd := strings.Index(remaining, "</stdout>")
|
|
|
|
// Process whichever comes first
|
|
if stderrStart != -1 && stderrEnd != -1 && stderrEnd > stderrStart &&
|
|
(stdoutStart == -1 || stderrStart < stdoutStart) {
|
|
// Process stderr
|
|
if stderrStart > 0 {
|
|
formattedResult.WriteString(remaining[:stderrStart])
|
|
}
|
|
stderrContent := remaining[stderrStart+8 : stderrEnd]
|
|
stderrContent = strings.Trim(stderrContent, "\n")
|
|
if len(stderrContent) > 0 {
|
|
styledContent := lipgloss.NewStyle().Foreground(theme.Error).Render(stderrContent)
|
|
formattedResult.WriteString(styledContent)
|
|
}
|
|
remaining = remaining[stderrEnd+9:] // Skip past </stderr>
|
|
|
|
} else if stdoutStart != -1 && stdoutEnd != -1 && stdoutEnd > stdoutStart {
|
|
// Process stdout
|
|
if stdoutStart > 0 {
|
|
formattedResult.WriteString(remaining[:stdoutStart])
|
|
}
|
|
stdoutContent := remaining[stdoutStart+8 : stdoutEnd]
|
|
stdoutContent = strings.Trim(stdoutContent, "\n")
|
|
if len(stdoutContent) > 0 {
|
|
formattedResult.WriteString(stdoutContent)
|
|
}
|
|
remaining = remaining[stdoutEnd+9:] // Skip past </stdout>
|
|
|
|
} else {
|
|
// No more tags, add remaining content
|
|
formattedResult.WriteString(remaining)
|
|
break
|
|
}
|
|
}
|
|
|
|
return strings.TrimSpace(formattedResult.String())
|
|
}
|