1package usercmd
 2
 3import (
 4	"fmt"
 5
 6	"github.com/spf13/cobra"
 7
 8	"github.com/MichaelMure/git-bug/cache"
 9	"github.com/MichaelMure/git-bug/commands/cmdjson"
10	"github.com/MichaelMure/git-bug/commands/completion"
11	"github.com/MichaelMure/git-bug/commands/execenv"
12	"github.com/MichaelMure/git-bug/util/colors"
13)
14
15type userOptions struct {
16	format string
17}
18
19func NewUserCommand() *cobra.Command {
20	env := execenv.NewEnv()
21	options := userOptions{}
22
23	cmd := &cobra.Command{
24		Use:     "user",
25		Short:   "List identities",
26		PreRunE: execenv.LoadBackend(env),
27		RunE: execenv.CloseBackend(env, func(cmd *cobra.Command, args []string) error {
28			return runUser(env, options)
29		}),
30	}
31
32	cmd.AddCommand(newUserNewCommand())
33	cmd.AddCommand(newUserShowCommand())
34	cmd.AddCommand(newUserAdoptCommand())
35
36	flags := cmd.Flags()
37	flags.SortFlags = false
38
39	flags.StringVarP(&options.format, "format", "f", "default",
40		"Select the output formatting style. Valid values are [default,json]")
41	cmd.RegisterFlagCompletionFunc("format", completion.From([]string{"default", "json"}))
42
43	return cmd
44}
45
46func runUser(env *execenv.Env, opts userOptions) error {
47	ids := env.Backend.Identities().AllIds()
48	var users []*cache.IdentityExcerpt
49	for _, id := range ids {
50		user, err := env.Backend.Identities().ResolveExcerpt(id)
51		if err != nil {
52			return err
53		}
54		users = append(users, user)
55	}
56
57	switch opts.format {
58	case "json":
59		return userJsonFormatter(env, users)
60	case "default":
61		return userDefaultFormatter(env, users)
62	default:
63		return fmt.Errorf("unknown format %s", opts.format)
64	}
65}
66
67func userDefaultFormatter(env *execenv.Env, users []*cache.IdentityExcerpt) error {
68	for _, user := range users {
69		env.Out.Printf("%s %s\n",
70			colors.Cyan(user.Id().Human()),
71			user.DisplayName(),
72		)
73	}
74
75	return nil
76}
77
78func userJsonFormatter(env *execenv.Env, users []*cache.IdentityExcerpt) error {
79	jsonUsers := make([]cmdjson.Identity, len(users))
80	for i, user := range users {
81		jsonUsers[i] = cmdjson.NewIdentityFromExcerpt(user)
82	}
83
84	return env.Out.PrintJSON(jsonUsers)
85}