mirror of
https://github.com/mark3labs/kit.git
synced 2026-06-13 19:20:06 +00:00
Ed Zynda
0703dd1602
Each spinner created a new tea.NewProgram which sent DECRQM queries for synchronized output mode 2026. When the program exited and restored cooked terminal mode, the terminal's DECRPM response leaked as visible ^[[?2026;2$y characters. Replace Bubble Tea spinner with a simple goroutine animation loop writing directly to stderr via lipgloss.
87 lines
3.3 KiB
Go
87 lines
3.3 KiB
Go
package ui
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
// CLIDebugLogger implements the tools.DebugLogger interface using CLI rendering.
|
|
// It provides debug logging functionality that integrates with the CLI's display
|
|
// system, ensuring debug messages are properly formatted and displayed alongside
|
|
// other conversation content.
|
|
type CLIDebugLogger struct {
|
|
cli *CLI
|
|
}
|
|
|
|
// NewCLIDebugLogger creates and returns a new CLIDebugLogger instance that routes
|
|
// debug output through the provided CLI instance. The logger will respect the CLI's
|
|
// debug mode setting and display format preferences.
|
|
func NewCLIDebugLogger(cli *CLI) *CLIDebugLogger {
|
|
return &CLIDebugLogger{cli: cli}
|
|
}
|
|
|
|
// LogDebug processes and displays a debug message through the CLI's rendering system.
|
|
// Messages are formatted with appropriate emojis and tags based on their content type
|
|
// (DEBUG, POOL, etc.) and only displayed when debug mode is enabled. The method handles
|
|
// multi-line debug output and connection pool status messages with context-aware formatting.
|
|
func (l *CLIDebugLogger) LogDebug(message string) {
|
|
if l.cli == nil || !l.cli.debug {
|
|
return
|
|
}
|
|
|
|
// Format the message to include all the debug info in a structured way
|
|
var formattedMessage string
|
|
|
|
// Check if this is a multi-line debug output (like connection info)
|
|
if strings.Contains(message, "[DEBUG]") || strings.Contains(message, "[POOL]") {
|
|
// Extract the tag and content
|
|
if after, ok := strings.CutPrefix(message, "[DEBUG]"); ok {
|
|
content := after
|
|
content = strings.TrimSpace(content)
|
|
formattedMessage = fmt.Sprintf("🔍 DEBUG: %s", content)
|
|
} else if after, ok := strings.CutPrefix(message, "[POOL]"); ok {
|
|
content := after
|
|
content = strings.TrimSpace(content)
|
|
|
|
// Add appropriate emoji based on the message content
|
|
if strings.Contains(content, "Creating new connection") {
|
|
formattedMessage = fmt.Sprintf("🆕 POOL: %s", content)
|
|
} else if strings.Contains(content, "Created connection") || strings.Contains(content, "Initialized") {
|
|
formattedMessage = fmt.Sprintf("✅ POOL: %s", content)
|
|
} else if strings.Contains(content, "Reusing") {
|
|
formattedMessage = fmt.Sprintf("🔄 POOL: %s", content)
|
|
} else if strings.Contains(content, "unhealthy") || strings.Contains(content, "failed") {
|
|
formattedMessage = fmt.Sprintf("❌ POOL: %s", content)
|
|
} else if strings.Contains(content, "closed") {
|
|
formattedMessage = fmt.Sprintf("🛑 POOL: %s", content)
|
|
} else if strings.Contains(content, "Failed to close") {
|
|
formattedMessage = fmt.Sprintf("⚠️ POOL: %s", content)
|
|
} else {
|
|
formattedMessage = fmt.Sprintf("🔍 POOL: %s", content)
|
|
}
|
|
} else {
|
|
formattedMessage = message
|
|
}
|
|
} else {
|
|
formattedMessage = message
|
|
}
|
|
|
|
// Use the CLI's debug message rendering
|
|
var msg UIMessage
|
|
if l.cli.compactMode {
|
|
msg = l.cli.compactRenderer.RenderDebugMessage(formattedMessage, time.Now())
|
|
} else {
|
|
msg = l.cli.messageRenderer.RenderDebugMessage(formattedMessage, time.Now())
|
|
}
|
|
l.cli.messageContainer.AddMessage(msg)
|
|
l.cli.displayContainer()
|
|
}
|
|
|
|
// IsDebugEnabled checks whether debug logging is currently active. Returns true
|
|
// if the CLI instance exists and has debug mode enabled, allowing callers to
|
|
// conditionally perform expensive debug operations only when necessary.
|
|
func (l *CLIDebugLogger) IsDebugEnabled() bool {
|
|
return l.cli != nil && l.cli.debug
|
|
}
|