mirror of
https://github.com/cheat/cheat.git
synced 2024-11-22 14:01:36 +01:00
daa43d3867
Implements filtering by pattern with `-l`. Resolves #504.
104 lines
2.4 KiB
Go
104 lines
2.4 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"regexp"
|
|
"sort"
|
|
"strings"
|
|
"text/tabwriter"
|
|
|
|
"github.com/cheat/cheat/internal/config"
|
|
"github.com/cheat/cheat/internal/sheet"
|
|
"github.com/cheat/cheat/internal/sheets"
|
|
)
|
|
|
|
// cmdList lists all available cheatsheets.
|
|
func cmdList(opts map[string]interface{}, conf config.Config) {
|
|
|
|
// load the cheatsheets
|
|
cheatsheets, err := sheets.Load(conf.Cheatpaths)
|
|
if err != nil {
|
|
fmt.Fprintln(os.Stderr, fmt.Sprintf("failed to list cheatsheets: %v", err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
// filter cheatcheats by tag if --tag was provided
|
|
if opts["--tag"] != nil {
|
|
cheatsheets = sheets.Filter(
|
|
cheatsheets,
|
|
strings.Split(opts["--tag"].(string), ","),
|
|
)
|
|
}
|
|
|
|
// instead of "consolidating" all of the cheatsheets (ie, overwriting global
|
|
// sheets with local sheets), here we simply want to create a slice
|
|
// containing all sheets.
|
|
flattened := []sheet.Sheet{}
|
|
for _, pathSheets := range cheatsheets {
|
|
for _, s := range pathSheets {
|
|
flattened = append(flattened, s)
|
|
}
|
|
}
|
|
|
|
// sort the "flattened" sheets alphabetically
|
|
sort.Slice(flattened, func(i, j int) bool {
|
|
return flattened[i].Title < flattened[j].Title
|
|
})
|
|
|
|
// filter if <cheatsheet> was specified
|
|
// NB: our docopt specification is misleading here. When used in conjunction
|
|
// with `-l`, `<cheatsheet>` is really a pattern against which to filter
|
|
// sheet titles.
|
|
if opts["<cheatsheet>"] != nil {
|
|
|
|
// initialize a slice of filtered sheets
|
|
filtered := []sheet.Sheet{}
|
|
|
|
// initialize our filter pattern
|
|
pattern := "(?i)" + opts["<cheatsheet>"].(string)
|
|
|
|
// compile the regex
|
|
reg, err := regexp.Compile(pattern)
|
|
if err != nil {
|
|
fmt.Fprintln(
|
|
os.Stderr,
|
|
fmt.Sprintf("failed to compile regexp: %s, %v", pattern, err),
|
|
)
|
|
os.Exit(1)
|
|
}
|
|
|
|
// iterate over each cheatsheet, and pass-through those which match the
|
|
// filter pattern
|
|
for _, s := range flattened {
|
|
if reg.MatchString(s.Title) {
|
|
filtered = append(filtered, s)
|
|
}
|
|
}
|
|
|
|
flattened = filtered
|
|
}
|
|
|
|
// exit early if no cheatsheets are available
|
|
if len(flattened) == 0 {
|
|
os.Exit(0)
|
|
}
|
|
|
|
// initialize a tabwriter to produce cleanly columnized output
|
|
w := tabwriter.NewWriter(os.Stdout, 0, 0, 1, ' ', 0)
|
|
|
|
// generate sorted, columnized output
|
|
fmt.Fprintln(w, "title:\tfile:\ttags:")
|
|
for _, sheet := range flattened {
|
|
fmt.Fprintln(w, fmt.Sprintf(
|
|
"%s\t%s\t%s",
|
|
sheet.Title,
|
|
sheet.Path,
|
|
strings.Join(sheet.Tags, ","),
|
|
))
|
|
}
|
|
|
|
// write columnized output to stdout
|
|
w.Flush()
|
|
}
|