Replace list_fonts with choose-fonts kitten

kittens/choose_fonts/family_list.go

@@ -0,0 +1,133 @@
+package choose_fonts
+
+import (
+	"fmt"
+	"kitty/tools/tui/subseq"
+	"kitty/tools/utils"
+	"kitty/tools/wcswidth"
+)
+
+var _ = fmt.Print
+
+type FamilyList struct {
+	families, all_families []string
+	current_search         string
+	display_strings        []string
+	widths                 []int
+	max_width, current_idx int
+}
+
+func (self *FamilyList) Len() int {
+	return len(self.families)
+}
+
+func (self *FamilyList) Next(delta int, allow_wrapping bool) bool {
+	if len(self.display_strings) == 0 {
+		return false
+	}
+	idx := self.current_idx + delta
+	if !allow_wrapping && (idx < 0 || idx > self.Len()) {
+		return false
+	}
+	for idx < 0 {
+		idx += self.Len()
+	}
+	self.current_idx = idx % self.Len()
+	return true
+}
+
+func limit_lengths(text string) string {
+	t, _ := wcswidth.TruncateToVisualLengthWithWidth(text, 31)
+	if len(t) >= len(text) {
+		return text
+	}
+	return t + "…"
+}
+
+func match(expression string, items []string) []*subseq.Match {
+	matches := subseq.ScoreItems(expression, items, subseq.Options{Level1: " "})
+	matches = utils.StableSort(matches, func(a, b *subseq.Match) int {
+		if b.Score < a.Score {
+			return -1
+		}
+		if b.Score > a.Score {
+			return 1
+		}
+		return 0
+	})
+	return matches
+}
+
+const (
+	MARK_BEFORE = "\033[33m"
+	MARK_AFTER  = "\033[39m"
+)
+
+func apply_search(families []string, expression string, marks ...string) (matched_families []string, display_strings []string) {
+	mark_before, mark_after := MARK_BEFORE, MARK_AFTER
+	if len(marks) == 2 {
+		mark_before, mark_after = marks[0], marks[1]
+	}
+	results := utils.Filter(match(expression, families), func(x *subseq.Match) bool { return x.Score > 0 })
+	matched_families = make([]string, 0, len(results))
+	display_strings = make([]string, 0, len(results))
+	for _, m := range results {
+		text := m.Text
+		positions := m.Positions
+		for i := len(positions) - 1; i >= 0; i-- {
+			p := positions[i]
+			text = text[:p] + mark_before + text[p:p+1] + mark_after + text[p+1:]
+		}
+		display_strings = append(display_strings, text)
+		matched_families = append(matched_families, m.Text)
+	}
+	return
+}
+
+func (self *FamilyList) UpdateFamilies(families []string) {
+	self.families, self.all_families = families, families
+	if self.current_search != "" {
+		self.families, self.display_strings = apply_search(self.all_families, self.current_search)
+		self.display_strings = utils.Map(limit_lengths, self.display_strings)
+	} else {
+		self.display_strings = utils.Map(limit_lengths, families)
+	}
+	self.widths = utils.Map(wcswidth.Stringwidth, self.display_strings)
+	self.max_width = utils.Max(0, self.widths...)
+	self.current_idx = 0
+}
+
+func (self *FamilyList) UpdateSearch(query string) bool {
+	if query == self.current_search || len(self.all_families) == 0 {
+		return false
+	}
+	self.current_search = query
+	self.UpdateFamilies(self.all_families)
+	return true
+}
+
+type Line struct {
+	text       string
+	width      int
+	is_current bool
+}
+
+func (self *FamilyList) Lines(num_rows int) []Line {
+	if num_rows < 1 {
+		return nil
+	}
+	ans := make([]Line, 0, len(self.display_strings))
+	before_num := utils.Min(self.current_idx, num_rows-1)
+	start := self.current_idx - before_num
+	for i := start; i < utils.Min(start+num_rows, len(self.display_strings)); i++ {
+		ans = append(ans, Line{self.display_strings[i], self.widths[i], i == self.current_idx})
+	}
+	return ans
+}
+
+func (self *FamilyList) CurrentFamily() string {
+	if self.current_idx >= 0 && self.current_idx < len(self.families) {
+		return self.families[self.current_idx]
+	}
+	return ""
+}