/
get_user.go
62 lines (56 loc) · 1.34 KB
/
get_user.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
package cmd
import (
"io"
"os"
"github.com/nwidger/lighthouse/users"
"github.com/spf13/cobra"
)
type userCmdOpts struct {
memberships bool
avatar bool
}
var userCmdFlags userCmdOpts
// userCmd represents the user command
var userCmd = &cobra.Command{
Use: "user [id-or-name]",
Short: "Get information about a Lighthouse user",
Run: func(cmd *cobra.Command, args []string) {
flags := userCmdFlags
u := users.NewService(service)
if len(args) == 0 {
FatalUsage(cmd, "must supply user ID or name")
}
if flags.memberships {
memberships, err := u.Memberships(args[0])
if err != nil {
FatalUsage(cmd, err)
}
JSON(memberships)
} else if flags.avatar {
user, err := u.Get(args[0])
if err != nil {
FatalUsage(cmd, err)
}
if len(user.AvatarURL) == 0 {
FatalUsage(cmd, "user has no avatar")
}
r, _, err := u.GetAvatar(user)
if err != nil {
FatalUsage(cmd, err)
}
defer r.Close()
io.Copy(os.Stdout, r)
} else {
user, err := u.Get(args[0])
if err != nil {
FatalUsage(cmd, err)
}
JSON(user)
}
},
}
func init() {
getCmd.AddCommand(userCmd)
userCmd.Flags().BoolVar(&userCmdFlags.memberships, "memberships", false, "Show user's memberships")
userCmd.Flags().BoolVar(&userCmdFlags.avatar, "avatar", false, "Download user avatar image (prints image to standard out)")
}