selfpass/sp/commands/commands.go

package commands

import (
	"context"
	"errors"
	"fmt"
	"os"
	"time"

	"github.com/spf13/cobra"
	"gopkg.in/AlecAivazis/survey.v1"

	"github.com/mitchell/selfpass/services/credentials/types"
)

type credentialsClientInit func(ctx context.Context) (c types.CredentialsClient)

var errSourceNotFound = errors.New("source host not found")

type credentialFlagSet struct {
	includePasswordFlags bool
	includeCredFlags     bool

	sourceHost string
	primary    string
	noNumbers  bool
	noSpecials bool
	length     uint
}

func (set credentialFlagSet) withPasswordFlags() credentialFlagSet {
	set.includePasswordFlags = true
	return set
}

func (set credentialFlagSet) withCredFlags() credentialFlagSet {
	set.includeCredFlags = true
	return set
}

func (set *credentialFlagSet) register(cmd *cobra.Command) {
	if set.includeCredFlags {
		cmd.Flags().StringVarP(&set.sourceHost, "source-host", "s", "", "filter results to this source host")
		cmd.Flags().StringVarP(&set.primary, "primary", "p", "", "specify a primary user key (must include tag if applicable)")
	}

	if set.includePasswordFlags {
		cmd.Flags().BoolVarP(&set.noNumbers, "no-numbers", "n", false, "do not use numbers in the generated password")
		cmd.Flags().BoolVarP(&set.noSpecials, "no-specials", "e", false, "do not use special characters in the generated password")
		cmd.Flags().UintVarP(&set.length, "length", "l", 32, "length of the generated password")
	}
}

func (set *credentialFlagSet) resetValues() {
	set.sourceHost = ""
	set.primary = ""
	set.noNumbers = false
	set.noSpecials = false
	set.length = 32
}

var checkPromptMode = false

func check(err error) {
	if err != nil {
		if checkPromptMode {
			panic(err)
		}

		fmt.Fprintln(os.Stdout, err)
		os.Exit(1)
	}
}

func selectCredential(client types.CredentialsClient, sourceHost string, primary string) types.Credential {
	var prompt survey.Prompt

	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
	defer cancel()

	mdch, errch := client.GetAllMetadata(ctx, sourceHost)
	var mds []types.Metadata

	fmt.Println()

receive:
	for {
		select {
		case <-ctx.Done():
			check(ctx.Err())

		case err := <-errch:
			check(err)

		case md, ok := <-mdch:
			if !ok {
				break receive
			}

			mds = append(mds, md)
		}
	}

	var sources []string
	mdmap := map[string][]types.Metadata{}
	for _, md := range mds {
		tmds := mdmap[md.SourceHost]

		if tmds == nil {
			mdmap[md.SourceHost] = []types.Metadata{md}
			sources = append(sources, md.SourceHost)
			continue
		}

		mdmap[md.SourceHost] = append(mdmap[md.SourceHost], md)
	}

	if sourceHost == "" {
		prompt = &survey.Select{
			Message:  "Source host:",
			Options:  sources,
			PageSize: 20,
			VimMode:  true,
		}

		check(survey.AskOne(prompt, &sourceHost, nil))
	}

	if len(mdmap[sourceHost]) == 0 {
		check(errSourceNotFound)
	}

	keys := []string{}
	keyIDMap := map[string]string{}
	for _, md := range mdmap[sourceHost] {
		key := md.Primary
		if md.Tag != "" {
			key += "-" + md.Tag
		}
		keys = append(keys, key)
		keyIDMap[key] = md.ID
	}

	if len(keys) == 1 {
		primary = keys[0]
		fmt.Printf("Selecting sole credential for this source.\n\n")
	}

	if primary == "" {
		var idKey string
		prompt = &survey.Select{
			Message:  "Primary user key (and tag):",
			Options:  keys,
			PageSize: 20,
			VimMode:  true,
		}

		check(survey.AskOne(prompt, &idKey, nil))

		primary = idKey
	}

	ctx, cancel = context.WithTimeout(context.Background(), time.Second*5)
	defer cancel()

	cred, err := client.Get(ctx, keyIDMap[primary])
	check(err)

	return cred
}