mirror of https://github.com/mitchell/selfpass.git
101 lines
1.7 KiB
Go
101 lines
1.7 KiB
Go
package commands
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"sort"
|
|
"time"
|
|
|
|
"gopkg.in/AlecAivazis/survey.v1"
|
|
|
|
"github.com/mitchell/selfpass/services/credentials/types"
|
|
)
|
|
|
|
type CredentialClientInit func(ctx context.Context) (c types.CredentialClient)
|
|
|
|
func check(err error) {
|
|
if err != nil {
|
|
fmt.Println(err)
|
|
os.Exit(1)
|
|
}
|
|
}
|
|
|
|
func selectCredential(client types.CredentialClient) 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
|
|
}
|