mirror of
https://github.com/charmbracelet/gum.git
synced 2024-10-26 15:20:05 +03:00
b23ebce896
There are a couple of gotchas in LipGlossTruncate: * len() returns the number of bytes in a string * slicing a string slices it on a byte level The fix would normally be to convert the string to a slice of runes and operate on that new slice: r := []rune(str) However, reflow already contains an ansi-aware truncate feature that we can use instead as a drop-in replacement for the LipGlossTruncate function.
165 lines
4.5 KiB
Go
165 lines
4.5 KiB
Go
package pager
|
|
|
|
import (
|
|
"fmt"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/charmbracelet/bubbles/textinput"
|
|
"github.com/charmbracelet/gum/internal/utils"
|
|
"github.com/charmbracelet/lipgloss"
|
|
"github.com/muesli/reflow/truncate"
|
|
)
|
|
|
|
type search struct {
|
|
active bool
|
|
input textinput.Model
|
|
query *regexp.Regexp
|
|
matchIndex int
|
|
matchLipglossStr string
|
|
matchString string
|
|
}
|
|
|
|
func (s *search) new() {
|
|
input := textinput.New()
|
|
input.Placeholder = "search"
|
|
input.Prompt = "/"
|
|
input.PromptStyle = lipgloss.NewStyle().Foreground(lipgloss.Color("240"))
|
|
s.input = input
|
|
}
|
|
|
|
func (s *search) Begin() {
|
|
s.new()
|
|
s.active = true
|
|
s.input.Focus()
|
|
}
|
|
|
|
// Execute find all lines in the model with a match.
|
|
func (s *search) Execute(m *model) {
|
|
defer s.Done()
|
|
if s.input.Value() == "" {
|
|
s.query = nil
|
|
return
|
|
}
|
|
|
|
var err error
|
|
s.query, err = regexp.Compile(s.input.Value())
|
|
if err != nil {
|
|
s.query = nil
|
|
return
|
|
}
|
|
query := regexp.MustCompile(fmt.Sprintf("(%s)", s.query.String()))
|
|
m.content = query.ReplaceAllString(m.content, m.matchStyle.Render("$1"))
|
|
|
|
// Recompile the regex to match the an replace the highlights.
|
|
leftPad, _ := utils.LipglossPadding(m.matchStyle)
|
|
matchingString := regexp.QuoteMeta(m.matchStyle.Render()[:leftPad]) + s.query.String() + regexp.QuoteMeta(m.matchStyle.Render()[leftPad:])
|
|
s.query, err = regexp.Compile(matchingString)
|
|
if err != nil {
|
|
s.query = nil
|
|
}
|
|
}
|
|
|
|
func (s *search) Done() {
|
|
s.active = false
|
|
|
|
// To account for the first match is always executed.
|
|
s.matchIndex = -1
|
|
}
|
|
|
|
func (s *search) NextMatch(m *model) {
|
|
// Check that we are within bounds.
|
|
if s.query == nil {
|
|
return
|
|
}
|
|
|
|
// Remove previous highlight.
|
|
m.content = strings.Replace(m.content, s.matchLipglossStr, s.matchString, 1)
|
|
|
|
// Highlight the next match.
|
|
allMatches := s.query.FindAllStringIndex(m.content, -1)
|
|
if len(allMatches) == 0 {
|
|
return
|
|
}
|
|
|
|
leftPad, rightPad := utils.LipglossPadding(m.matchStyle)
|
|
s.matchIndex = (s.matchIndex + 1) % len(allMatches)
|
|
match := allMatches[s.matchIndex]
|
|
lhs := m.content[:match[0]]
|
|
rhs := m.content[match[0]:]
|
|
s.matchString = m.content[match[0]:match[1]]
|
|
s.matchLipglossStr = m.matchHighlightStyle.Render(s.matchString[leftPad : len(s.matchString)-rightPad])
|
|
m.content = lhs + strings.Replace(rhs, m.content[match[0]:match[1]], s.matchLipglossStr, 1)
|
|
|
|
// Update the viewport position.
|
|
var line int
|
|
formatStr := softWrapEm(m.content, m.maxWidth, m.softWrap)
|
|
index := strings.Index(formatStr, s.matchLipglossStr)
|
|
if index != -1 {
|
|
line = strings.Count(formatStr[:index], "\n")
|
|
}
|
|
|
|
// Only update if the match is not within the viewport.
|
|
if index != -1 && (line > m.viewport.YOffset-1+m.viewport.VisibleLineCount()-1 || line < m.viewport.YOffset) {
|
|
m.viewport.SetYOffset(line)
|
|
}
|
|
}
|
|
|
|
func (s *search) PrevMatch(m *model) {
|
|
// Check that we are within bounds.
|
|
if s.query == nil {
|
|
return
|
|
}
|
|
|
|
// Remove previous highlight.
|
|
m.content = strings.Replace(m.content, s.matchLipglossStr, s.matchString, 1)
|
|
|
|
// Highlight the previous match.
|
|
allMatches := s.query.FindAllStringIndex(m.content, -1)
|
|
if len(allMatches) == 0 {
|
|
return
|
|
}
|
|
|
|
s.matchIndex = (s.matchIndex - 1) % len(allMatches)
|
|
if s.matchIndex < 0 {
|
|
s.matchIndex = len(allMatches) - 1
|
|
}
|
|
|
|
leftPad, rightPad := utils.LipglossPadding(m.matchStyle)
|
|
match := allMatches[s.matchIndex]
|
|
lhs := m.content[:match[0]]
|
|
rhs := m.content[match[0]:]
|
|
s.matchString = m.content[match[0]:match[1]]
|
|
s.matchLipglossStr = m.matchHighlightStyle.Render(s.matchString[leftPad : len(s.matchString)-rightPad])
|
|
m.content = lhs + strings.Replace(rhs, m.content[match[0]:match[1]], s.matchLipglossStr, 1)
|
|
|
|
// Update the viewport position.
|
|
var line int
|
|
formatStr := softWrapEm(m.content, m.maxWidth, m.softWrap)
|
|
index := strings.Index(formatStr, s.matchLipglossStr)
|
|
if index != -1 {
|
|
line = strings.Count(formatStr[:index], "\n")
|
|
}
|
|
|
|
// Only update if the match is not within the viewport.
|
|
if index != -1 && (line > m.viewport.YOffset-1+m.viewport.VisibleLineCount()-1 || line < m.viewport.YOffset) {
|
|
m.viewport.SetYOffset(line)
|
|
}
|
|
}
|
|
|
|
func softWrapEm(str string, maxWidth int, softWrap bool) string {
|
|
var text strings.Builder
|
|
for _, line := range strings.Split(str, "\n") {
|
|
for softWrap && lipgloss.Width(line) > maxWidth {
|
|
truncatedLine := truncate.String(line, uint(maxWidth))
|
|
text.WriteString(truncatedLine)
|
|
text.WriteString("\n")
|
|
line = strings.Replace(line, truncatedLine, "", 1)
|
|
}
|
|
text.WriteString(truncate.String(line, uint(maxWidth)))
|
|
text.WriteString("\n")
|
|
}
|
|
|
|
return text.String()
|
|
}
|