mirror of
https://github.com/cheat/cheat.git
synced 2024-12-23 13:09:44 +01:00
46 lines
996 B
Go
46 lines
996 B
Go
|
package sheet
|
||
|
|
||
|
import (
|
||
|
"regexp"
|
||
|
"strings"
|
||
|
|
||
|
"github.com/mgutz/ansi"
|
||
|
)
|
||
|
|
||
|
// Search searches for regexp matches in a cheatsheet's text, and optionally
|
||
|
// colorizes matching strings.
|
||
|
func (s *Sheet) Search(reg *regexp.Regexp, colorize bool) []Match {
|
||
|
|
||
|
// record matches
|
||
|
matches := []Match{}
|
||
|
|
||
|
// search through the cheatsheet's text line by line
|
||
|
// TODO: searching line-by-line is surely the "naive" approach. Revisit this
|
||
|
// later with an eye for performance improvements.
|
||
|
for linenum, line := range strings.Split(s.Text, "\n") {
|
||
|
|
||
|
// exit early if the line doesn't match the regex
|
||
|
if !reg.MatchString(line) {
|
||
|
continue
|
||
|
}
|
||
|
|
||
|
// init the match
|
||
|
m := Match{
|
||
|
Line: linenum + 1,
|
||
|
Text: strings.TrimSpace(line),
|
||
|
}
|
||
|
|
||
|
// colorize the matching text if so configured
|
||
|
if colorize {
|
||
|
m.Text = reg.ReplaceAllStringFunc(m.Text, func(matched string) string {
|
||
|
return ansi.Color(matched, "red+b")
|
||
|
})
|
||
|
}
|
||
|
|
||
|
// record the match
|
||
|
matches = append(matches, m)
|
||
|
}
|
||
|
|
||
|
return matches
|
||
|
}
|