selfpass/sp/commands/commands.go

166 lines
3.4 KiB
Go
Raw Normal View History

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 = 0
}
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 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
}