cmd_search.go

package main

import (
	"bytes"
	"fmt"
	"io"
	"os"
	"os/exec"
	"runtime"
	"strings"

	"github.com/olekukonko/tablewriter"
)

var cmdSearch = &Command{
	Run:       runSearch,
	UsageLine: "search",
	Short:     "Search keywords",
	Long:      `Search keywords interactively (default filtering tool: peco)`,
}

func runSearch(ctx context, args []string) error {
	cfg, err := GetConfig()
	if err != nil {
		return err
	}

	err = Initialize(cfg)
	if err != nil {
		return err
	}

	is, err := LoadItemsWithConfig(cfg)
	if err != nil {
		return err
	}

	if is.HasMaster() {
		if err = confirmMasterPassword(is.Master()); err != nil {
			return err
		}
	}

	if len(is) == 0 {
		_, err = fmt.Fprintln(ctx.out, "no password.")
		return err
	}

	buf := &bytes.Buffer{}
	err = runCommand(cfg.FilteringCommand, strings.NewReader(filteringText(is)), buf)
	if err != nil {
		return err
	}

	name := strings.TrimSpace(strings.Split(buf.String(), "|")[0])
	err = findAndCopy(ctx, is, name)
	if err != nil {
		return err
	}

	PrintSuccess(ctx.out, "password of '%s' is copied to clipboard successfully", name)
	return nil
}

func filteringText(is Items) string {
	buf := &bytes.Buffer{}
	tw := tablewriter.NewWriter(buf)
	tw.SetColumnSeparator("|")
	tw.SetBorder(false)
	tw.AppendBulk(is.ToDataTable())
	tw.Render()
	return buf.String()
}

func runCommand(cmdText string, r io.Reader, w io.Writer) error {
	var cmd *exec.Cmd
	if runtime.GOOS == "windows" {
		cmd = exec.Command("cmd", "/c", cmdText)
	} else {
		cmd = exec.Command("sh", "-c", cmdText)
	}
	cmd.Stderr = os.Stderr
	cmd.Stdout = w
	cmd.Stdin = r
	return cmd.Run()
}