loveuer
86f5ca4aef
Add visible activity state for running turns. Move model metadata below the input area. Render assistant Markdown with theme-aware styling.
69 lines
1.5 KiB
Go
69 lines
1.5 KiB
Go
package tui
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/charmbracelet/glamour"
|
|
"github.com/charmbracelet/glamour/ansi"
|
|
glamstyles "github.com/charmbracelet/glamour/styles"
|
|
)
|
|
|
|
const minMarkdownWidth = 20
|
|
|
|
func renderMarkdown(content string, width int, th theme) string {
|
|
width = max(minMarkdownWidth, width)
|
|
renderer, err := glamour.NewTermRenderer(
|
|
glamour.WithStyles(markdownStyle(th)),
|
|
glamour.WithWordWrap(width),
|
|
)
|
|
if err != nil {
|
|
return content
|
|
}
|
|
|
|
rendered, err := renderer.Render(content)
|
|
if err != nil {
|
|
return content
|
|
}
|
|
return strings.TrimRight(rendered, "\n")
|
|
}
|
|
|
|
func markdownStyle(th theme) ansi.StyleConfig {
|
|
cfg := glamstyles.LightStyleConfig
|
|
if th.Mode == ThemeDark {
|
|
cfg = glamstyles.DarkStyleConfig
|
|
}
|
|
|
|
zero := uint(0)
|
|
cfg.Document.Margin = &zero
|
|
cfg.Document.BlockPrefix = ""
|
|
cfg.Document.BlockSuffix = ""
|
|
cfg.Document.Color = stringPtr(th.Text)
|
|
|
|
cfg.Heading.Color = stringPtr(th.Assistant)
|
|
cfg.H1.Prefix = ""
|
|
cfg.H1.Suffix = ""
|
|
cfg.H1.Color = stringPtr(th.Assistant)
|
|
cfg.H1.BackgroundColor = nil
|
|
cfg.H1.Bold = boolPtr(true)
|
|
|
|
cfg.BlockQuote.Color = stringPtr(th.Muted)
|
|
cfg.Link.Color = stringPtr(th.User)
|
|
cfg.LinkText.Color = stringPtr(th.User)
|
|
|
|
cfg.Code.Color = stringPtr(th.Tool)
|
|
cfg.Code.BackgroundColor = stringPtr(th.SurfaceAlt)
|
|
cfg.CodeBlock.Color = stringPtr(th.Text)
|
|
cfg.CodeBlock.BackgroundColor = nil
|
|
cfg.CodeBlock.Margin = &zero
|
|
|
|
return cfg
|
|
}
|
|
|
|
func stringPtr(value string) *string {
|
|
return &value
|
|
}
|
|
|
|
func boolPtr(value bool) *bool {
|
|
return &value
|
|
}
|