/
init.go
72 lines (60 loc) · 1.58 KB
/
init.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package main
import (
"bufio"
"context"
"flag"
"fmt"
"os"
"strings"
"syscall"
noteofcli "github.com/NoteOf/noteof-cli"
sdk "github.com/NoteOf/sdk-go"
"github.com/google/subcommands"
"golang.org/x/crypto/ssh/terminal"
)
type InitCmd struct {
config *noteofcli.Config
api *sdk.UnauthenticatedAPI
}
func (*InitCmd) Name() string { return "init" }
func (*InitCmd) Synopsis() string { return "initialize the noteof cli" }
func (*InitCmd) Usage() string {
return `init:
Initialize the noteof cli.
`
}
func (p *InitCmd) SetFlags(f *flag.FlagSet) {}
func (p *InitCmd) Execute(_ context.Context, _ *flag.FlagSet, _ ...interface{}) subcommands.ExitStatus {
for {
username, password, err := credentials()
if err != nil {
os.Stdout.WriteString(err.Error() + "\n")
return subcommands.ExitFailure
} else if username == "" && password == "" {
os.Stdout.WriteString("exiting unsuccessfully\n")
return subcommands.ExitFailure
}
tr, err := p.api.DoAuth(username, password, "noteof-cli")
if err != nil {
fmt.Println(err)
continue
}
token := tr.APIToken
p.config.SetToken(token)
fmt.Println("token saved successfully")
return subcommands.ExitSuccess
}
}
func credentials() (string, string, error) {
reader := bufio.NewReader(os.Stdin)
fmt.Print("Enter Username: ")
username, _ := reader.ReadString('\n')
fmt.Print("Enter Password: ")
bytePassword, err := terminal.ReadPassword(int(syscall.Stdin))
if err != nil {
return "", "", err
}
fmt.Println()
password := string(bytePassword)
return strings.TrimSpace(username), strings.TrimSpace(password), nil
}