search.go

package main

import (
	"fmt"
	"os"
	"sort"
	"strings"
	"text/tabwriter"
	"text/template"

	"facette.io/natsort"
	_ "github.com/blevesearch/bleve/config"
	"github.com/urfave/cli/v2"

	"mkdeb.sh/catalog"
)

var searchCommand = &cli.Command{
	Name:      "search",
	Usage:     "Search for recipes",
	ArgsUsage: "[TERM]",
	Action:    execSearch,
	Flags: []cli.Flag{
		&cli.BoolFlag{
			Name:  "description, desc",
			Usage: "Include recipes description when searching",
		},
		&cli.StringFlag{
			Name:  "format",
			Usage: "Output template format",
		},
	},
}

func execSearch(ctx *cli.Context) error {
	if ctx.NArg() > 1 {
		cli.ShowCommandHelpAndExit(ctx, "search", 1)
	}

	if !catalog.Ready(catalogDir) {
		err := ctx.App.Run([]string{ctx.App.Name, "repo", "add", catalog.DefaultRepository})
		if err != nil {
			return err
		}
	}

	c, err := catalog.New(catalogDir)
	if err != nil {
		return fmt.Errorf("cannot initialize catalog: %w", err)
	}
	defer c.Close()

	term := ctx.Args().First()

	hits, err := c.Search(term, ctx.Bool("desc"))
	if err != nil {
		return fmt.Errorf("cannot search catalog: %w", err)
	}

	if len(hits) == 0 {
		if term != "" {
			fmt.Printf("No match found for %q\n", term)
		} else {
			fmt.Println("No match found")
		}
		return nil
	}

	sort.Slice(hits, func(i, j int) bool {
		return natsort.Compare(hits[i].Repository+"/"+hits[i].Name, hits[j].Repository+"/"+hits[j].Name)
	})

	format := ctx.String("format")
	if format == "" {
		format = "{{ if ne .Repository \"" + catalog.DefaultRepository +
			"\" }}{{ .Repository }}/{{ end }}{{ .Name }}\t{{ .Description }}\n"
	} else {
		format = strings.TrimSpace(format) + "\n"
	}

	tmpl, err := template.New("").Parse(format)
	if err != nil {
		return fmt.Errorf("invalid format: %w", err)
	}

	tr := tabwriter.NewWriter(os.Stdout, 0, 0, 3, ' ', 0)
	for _, hit := range hits {
		err = tmpl.Execute(tr, hit)
		if err != nil {
			return fmt.Errorf("cannot execute template: %w", err)
		}
	}
	tr.Flush()

	return nil
}