171 lines
3.5 KiB
Go
171 lines
3.5 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strconv"
|
|
"syscall"
|
|
|
|
"github.com/spf13/cobra"
|
|
"github.com/titpetric/factory"
|
|
"golang.org/x/crypto/ssh/terminal"
|
|
|
|
"github.com/crusttech/crust/internal/auth"
|
|
"github.com/crusttech/crust/system/internal/repository"
|
|
"github.com/crusttech/crust/system/internal/service"
|
|
"github.com/crusttech/crust/system/types"
|
|
)
|
|
|
|
func usersCmd(ctx context.Context, db *factory.DB) *cobra.Command {
|
|
// User management commands.
|
|
cmd := &cobra.Command{
|
|
Use: "users",
|
|
Short: "User management",
|
|
}
|
|
|
|
// List users.
|
|
listCmd := &cobra.Command{
|
|
Use: "list",
|
|
Short: "List users",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
userRepo := repository.User(ctx, db)
|
|
uf := &types.UserFilter{
|
|
OrderBy: "updated_at",
|
|
}
|
|
|
|
users, err := userRepo.Find(uf)
|
|
if err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
|
|
fmt.Println(" Created Updated EmailAddress")
|
|
for _, u := range users {
|
|
upd := "---- -- --"
|
|
|
|
if u.UpdatedAt != nil {
|
|
upd = u.UpdatedAt.Format("2006-01-02")
|
|
}
|
|
|
|
fmt.Printf(
|
|
"%20d %s %s %-100s %s\n",
|
|
u.ID,
|
|
u.CreatedAt.Format("2006-01-02"),
|
|
upd,
|
|
u.Email,
|
|
u.Name)
|
|
}
|
|
},
|
|
}
|
|
|
|
addCmd := &cobra.Command{
|
|
Use: "add [email]",
|
|
Short: "Add new user",
|
|
Args: cobra.MinimumNArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
var (
|
|
userRepo = repository.User(ctx, db)
|
|
authSvc = service.Auth(ctx)
|
|
|
|
// @todo email validation
|
|
user = &types.User{Email: args[0]}
|
|
|
|
err error
|
|
password []byte
|
|
)
|
|
|
|
if user, err = userRepo.Create(user); err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
|
|
cmd.Printf("User created [%d].\n", user.ID)
|
|
|
|
cmd.Print("Set password: ")
|
|
if password, err = terminal.ReadPassword(syscall.Stdin); err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
|
|
if len(password) == 0 {
|
|
// Password not set, that's ok too.
|
|
exit(cmd, nil)
|
|
}
|
|
|
|
if err = authSvc.SetPassword(user.ID, string(password)); err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
},
|
|
}
|
|
|
|
pwdCmd := &cobra.Command{
|
|
Use: "password [email]",
|
|
Short: "Add new user",
|
|
Args: cobra.MinimumNArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
var (
|
|
userRepo = repository.User(ctx, db)
|
|
authSvc = service.Auth(ctx)
|
|
|
|
user *types.User
|
|
err error
|
|
password []byte
|
|
)
|
|
|
|
if user, err = userRepo.FindByEmail(args[0]); err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
|
|
cmd.Print("Set password: ")
|
|
if password, err = terminal.ReadPassword(syscall.Stdin); err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
|
|
if len(password) == 0 {
|
|
// Password not set, that's ok too.
|
|
exit(cmd, nil)
|
|
}
|
|
|
|
if err = authSvc.SetPassword(user.ID, string(password)); err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
},
|
|
}
|
|
|
|
jwtCmd := &cobra.Command{
|
|
Use: "jwt [email-or-id]",
|
|
Short: "Generates new JWT for a user",
|
|
Args: cobra.MinimumNArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
var (
|
|
userRepo = repository.User(ctx, db)
|
|
// authSvc = service.Auth(ctx)
|
|
|
|
user *types.User
|
|
err error
|
|
ID uint64
|
|
|
|
userStr = args[0]
|
|
)
|
|
|
|
if user, err = userRepo.FindByEmail(userStr); err != nil && err != repository.ErrUserNotFound {
|
|
exit(cmd, err)
|
|
} else if user == nil || user.ID == 0 {
|
|
if ID, err = strconv.ParseUint(userStr, 10, 64); err != nil {
|
|
exit(cmd, err)
|
|
} else if user, err = userRepo.FindByID(ID); err != nil {
|
|
exit(cmd, err)
|
|
}
|
|
}
|
|
|
|
cmd.Println(auth.DefaultJwtHandler.Encode(user))
|
|
},
|
|
}
|
|
|
|
cmd.AddCommand(
|
|
listCmd,
|
|
addCmd,
|
|
pwdCmd,
|
|
jwtCmd,
|
|
)
|
|
|
|
return cmd
|
|
}
|