/*
 * Copyright 2026 Oleg Borodin <onborodin@gmail.com>
 */
package grant

import (
	"context"
	"fmt"

	"github.com/spf13/cobra"
	"github.com/spf13/viper"
	"mbase/app/operator"
)

type Util struct {
	oper              *operator.Operator
	operID            string
	subCmd            *cobra.Command
	setGrantParams    SetGrantParams
	deleteGrantParams DeleteGrantParams
	commonParams      CommonParams
}

func NewUtil(oper *operator.Operator) *Util {
	return &Util{
		oper: oper,
	}
}

type CommonParams struct {
	Username string
	Password string
}

func (util *Util) GetRooCmd() *cobra.Command {
	return util.subCmd
}

func (util *Util) BuildCmds() *cobra.Command {
	subCmd := &cobra.Command{
		Use:               "grant",
		Short:             "\nGrant operations",
		SilenceUsage:      true,
		PersistentPreRunE: util.ValidateOper,
	}
	subCmd.CompletionOptions.DisableDefaultCmd = true

	subCmd.PersistentFlags().StringVarP(&util.commonParams.Username, "user", "U", util.commonParams.Username, "Username")
	subCmd.PersistentFlags().StringVarP(&util.commonParams.Password, "pass", "P", util.commonParams.Password, "Password")
	subCmd.MarkFlagsRequiredTogether("user", "pass")

	vi := viper.New()
	vi.SetEnvPrefix("mstore")
	vi.BindEnv("user")
	vi.BindEnv("pass")
	util.commonParams.Username = vi.GetString("user")
	util.commonParams.Password = vi.GetString("pass")

	var setGrantCmd = &cobra.Command{
		Use:   "set username grant",
		Short: "Set grant for account",
		Args:  cobra.ExactArgs(2),
		Run:   util.SetGrant,
	}
	subCmd.AddCommand(setGrantCmd)

	var deleteGrantCmd = &cobra.Command{
		Use:   "delete username|id grant",
		Short: "Delete grant for account",
		Args:  cobra.ExactArgs(2),
		Run:   util.DeleteGrant,
	}
	subCmd.AddCommand(deleteGrantCmd)

	util.subCmd = subCmd
	return util.subCmd
}

func (util *Util) ValidateOper(cmd *cobra.Command, args []string) error {
	var err error
	ctx := context.Background()
	authOk, operID, err := util.oper.ValidateAccount(ctx, util.commonParams.Username, util.commonParams.Password)
	if !authOk {
		err := fmt.Errorf("Authentification error")
		return err
	}
	util.operID = operID
	return err
}