netmaker/cli/cmd/user/create.go

package user

import (
	"strings"

	"github.com/gravitl/netmaker/cli/functions"
	"github.com/gravitl/netmaker/models"
	"github.com/spf13/cobra"
)

var userCreateCmd = &cobra.Command{
	Use:   "create",
	Args:  cobra.NoArgs,
	Short: "Create a new user",
	Long:  `Create a new user`,
	Run: func(cmd *cobra.Command, args []string) {
		user := &models.User{UserName: username, Password: password, PlatformRoleID: models.UserRoleID(platformID)}
		if len(networkRoles) > 0 {
			netRolesMap := make(map[models.NetworkID]map[models.UserRoleID]struct{})
			for netID, netRoles := range networkRoles {
				roleMap := make(map[models.UserRoleID]struct{})
				for _, roleID := range strings.Split(netRoles, " ") {
					roleMap[models.UserRoleID(roleID)] = struct{}{}
				}
				netRolesMap[models.NetworkID(netID)] = roleMap
			}
			user.NetworkRoles = netRolesMap
		}
		if len(groups) > 0 {
			grMap := make(map[models.UserGroupID]struct{})
			for _, groupID := range groups {
				grMap[models.UserGroupID(groupID)] = struct{}{}
			}
			user.UserGroups = grMap
		}

		functions.PrettyPrint(functions.CreateUser(user))
	},
}

func init() {

	userCreateCmd.Flags().StringVar(&username, "name", "", "Name of the user")
	userCreateCmd.Flags().StringVar(&password, "password", "", "Password of the user")
	userCreateCmd.Flags().StringVarP(&platformID, "platform-role", "r", models.ServiceUser.String(),
		"Platform Role of the user; run `nmctl roles list` to see available user roles")
	userCreateCmd.MarkFlagRequired("name")
	userCreateCmd.MarkFlagRequired("password")
	userCreateCmd.PersistentFlags().StringToStringVarP(&networkRoles, "network-roles", "n", nil,
		"Mapping of networkID and list of roles user will be part of (comma separated)")
	userCreateCmd.Flags().BoolVar(&admin, "admin", false, "Make the user an admin ? (deprecated v0.25.0 onwards)")
	userCreateCmd.Flags().StringArrayVarP(&groups, "groups", "g", nil, "List of user groups the user will be part of (comma separated)")
	rootCmd.AddCommand(userCreateCmd)
}