package cmd

import (
	"bufio"
	"fmt"
	"os"

	"github.com/spf13/cobra"
	"gitrepo.ru/neonxp/gorum/db"
	"gitrepo.ru/neonxp/gorum/models"
	"gitrepo.ru/neonxp/gorum/repository"
)

var userCmd = &cobra.Command{
	Use:   "user",
	Short: "User managment",
}

var createUserCmd = &cobra.Command{
	Use:        "add",
	Args:       cobra.ExactArgs(3),
	ArgAliases: []string{"username", "email", "role"},
	RunE: func(cmd *cobra.Command, args []string) error {
		orm, err := db.GetDB(dbFile)
		if err != nil {
			return fmt.Errorf("failed init db: %w", err)
		}
		username := args[0]
		email := args[1]
		role := args[2]
		iRole := models.RoleUser
		switch role {
		case "admin":
			iRole = models.RoleAdmin
		case "moderator":
			iRole = models.RoleModerator
		}
		reader := bufio.NewReader(os.Stdin)
		fmt.Printf("Enter password for user %s: ", username)
		password, _ := reader.ReadString('\n')

		ur := repository.NewUser(orm)
		id, err := ur.Create(cmd.Context(), email, password, username, iRole)
		if err != nil {
			return fmt.Errorf("failed create user: %w", err)
		}

		fmt.Printf("Created user %s (id=%d, email=%s, role_id=%d)\n", username, id, email, iRole)

		return nil
	},
}

var listUserCmd = &cobra.Command{
	Use: "list",
	RunE: func(cmd *cobra.Command, args []string) error {
		orm, err := db.GetDB(dbFile)
		if err != nil {
			return fmt.Errorf("failed init db: %w", err)
		}

		ur := repository.NewUser(orm)
		users, err := ur.List(cmd.Context())
		if err != nil {
			return err
		}

		fmt.Printf("ID\tUsername\tEmail\tRole\n")
		for _, u := range users {
			fmt.Printf("%d\t%s\t%s\t%d\n", u.ID, u.Username, u.Email, u.Role)
		}

		return nil
	},
}

func init() {
	userCmd.AddCommand(createUserCmd)
	userCmd.AddCommand(listUserCmd)
}