/
create.go
121 lines (101 loc) · 2.98 KB
/
create.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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
package user
import (
"context"
"fmt"
"net/mail"
"connectrpc.com/connect"
"github.com/nyaruka/phonenumbers"
"github.com/rigdev/rig-go-api/api/v1/user"
"github.com/rigdev/rig/cmd/common"
"github.com/spf13/cobra"
)
func (c *Cmd) create(ctx context.Context, cmd *cobra.Command, _ []string) error {
updates, err := c.getUserAndPasswordUpdates("", email, password)
if err != nil {
return err
}
if role == "" {
_, role, err = c.Prompter.Select("What is the role of the user?",
[]string{"admin", "owner", "developer", "viewer"})
if err != nil {
return err
}
}
res, err := c.Rig.User().Create(ctx, &connect.Request[user.CreateRequest]{
Msg: &user.CreateRequest{
Initializers: updates,
InitialGroupId: role,
},
})
if err != nil {
return err
}
cmd.Println("Succesfully created user with ID:", res.Msg.GetUser().GetUserId())
return nil
}
func (c *Cmd) getUserAndPasswordUpdates(username, email, password string) ([]*user.Update, error) {
updates, err := c.getUserIdentifierUpdates(username, email)
if err != nil {
return nil, err
}
if password == "" {
password, err = c.Prompter.Password("Password:")
if err != nil {
return nil, err
}
}
updates = append(updates, makeUpdatePassword(password))
return updates, nil
}
func (c *Cmd) getUserIdentifierUpdates(username, email string) ([]*user.Update, error) {
if username == "" && email == "" {
update, err := c.userIndentifierUpdate()
if err != nil {
return nil, err
}
return []*user.Update{update}, nil
}
var updates []*user.Update
if username != "" {
updates = append(updates, makeUpdateUsername(username))
}
if email != "" {
updates = append(updates, makeUpdateUsername(email))
}
return updates, nil
}
func (c *Cmd) userIndentifierUpdate() (*user.Update, error) {
var err error
identifier, err := c.Prompter.Input("Username or email:", common.ValidateAllOpt)
if err != nil {
return nil, err
}
update, err := parseUserIdentifierUpdate(identifier)
if err != nil {
return nil, err
}
return update, nil
}
func parseUserIdentifierUpdate(identifier string) (*user.Update, error) {
var err error
if _, err = mail.ParseAddress(identifier); err == nil {
return makeUpdateEmail(identifier), nil
} else if _, err = phonenumbers.Parse(identifier, ""); err == nil {
return makeUpdatePhoneNumber(identifier), nil
} else if err = common.ValidateSystemName(identifier); err == nil {
return makeUpdateUsername(identifier), nil
}
return nil, fmt.Errorf("invalid user identifier")
}
func makeUpdateUsername(username string) *user.Update {
return &user.Update{Field: &user.Update_Username{Username: username}}
}
func makeUpdateEmail(email string) *user.Update {
return &user.Update{Field: &user.Update_Email{Email: email}}
}
func makeUpdatePhoneNumber(phoneNumber string) *user.Update {
return &user.Update{Field: &user.Update_PhoneNumber{PhoneNumber: phoneNumber}}
}
func makeUpdatePassword(password string) *user.Update {
return &user.Update{Field: &user.Update_Password{Password: password}}
}