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