selfpass/sp/commands/commands.go

package commands

import (
	"context"
	"fmt"
	"os"
	"sort"
	"time"

	"gopkg.in/AlecAivazis/survey.v1"

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

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

func check(err error) {
	if err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

func selectCredential(client types.CredentialsClient) types.Credential {
	var (
		idKey  string
		source string
		prompt survey.Prompt
	)

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

	mdch, errch := client.GetAllMetadata(ctx, "")
	mds := map[string][]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[md.SourceHost] = append(mds[md.SourceHost], md)
		}
	}

	sources := []string{}
	for source := range mds {
		sources = append(sources, source)
	}

	sort.Strings(sources)

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

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

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

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

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

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

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

	return cred
}