package ignitecmd

import (
	"fmt"

	"github.com/spf13/cobra"

	"github.com/cosmos/cosmos-sdk/codec/address"
	sdk "github.com/cosmos/cosmos-sdk/types"

	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/cliui"
	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/cosmosaccount"
	"git.cw.tr/mukan-network/mukan-ignite/ignite/pkg/errors"
	"git.cw.tr/mukan-network/mukan-ignite/ignite/services/chain"
)

func NewTestnetInPlace() *cobra.Command {
	c := &cobra.Command{
		Use:   "in-place",
		Short: "Create and start a testnet from current local net state",
		Long: `Testnet in-place command is used to create and start a testnet from current local net state(including mainnet).
After using this command in the repo containing the config.yml file, the network will start.
We can create a testnet from the local network state and mint additional coins for the desired accounts from the config.yml file.`,
		Args: cobra.NoArgs,
		RunE: testnetInPlaceHandler,
	}
	flagSetPath(c)
	flagSetClearCache(c)
	c.Flags().AddFlagSet(flagSetHome())
	c.Flags().AddFlagSet(flagSetCheckDependencies())
	c.Flags().AddFlagSet(flagSetSkipProto())
	c.Flags().AddFlagSet(flagSetVerbose())
	c.Flags().AddFlagSet(flagSetAccountPrefixes())
	c.Flags().AddFlagSet(flagSetCoinType())

	return c
}

func testnetInPlaceHandler(cmd *cobra.Command, _ []string) error {
	session := cliui.New(
		cliui.WithVerbosity(getVerbosity(cmd)),
		cliui.WithoutUserInteraction(getYes(cmd)),
	)
	defer session.End()

	// Otherwise run the serve command directly
	return testnetInplace(cmd, session)
}

func testnetInplace(cmd *cobra.Command, session *cliui.Session) error {
	chainOption := []chain.Option{
		chain.WithOutputer(session),
		chain.CollectEvents(session.EventBus()),
		chain.CheckCosmosSDKVersion(),
	}

	if flagGetCheckDependencies(cmd) {
		chainOption = append(chainOption, chain.CheckDependencies())
	}

	// check if custom config is defined
	config, _ := cmd.Flags().GetString(flagConfig)
	if config != "" {
		chainOption = append(chainOption, chain.ConfigFile(config))
	}

	c, err := chain.NewWithHomeFlags(cmd, chainOption...)
	if err != nil {
		return err
	}

	cfg, err := c.Config()
	if err != nil {
		return err
	}

	home, err := c.Home()
	if err != nil {
		return err
	}

	keyringBackend, err := c.KeyringBackend()
	if err != nil {
		return err
	}

	prefix := getAddressPrefix(cmd)
	addressCodec := address.NewBech32Codec(prefix)
	valAddressCodec := address.NewBech32Codec(prefix + "valoper")
	coinType := getCoinType(cmd)

	ca, err := cosmosaccount.New(
		cosmosaccount.WithKeyringBackend(cosmosaccount.KeyringBackend(keyringBackend)),
		cosmosaccount.WithHome(home),
		cosmosaccount.WithBech32Prefix(prefix),
		cosmosaccount.WithCoinType(coinType),
	)
	if err != nil {
		return err
	}

	var (
		operatorAddress sdk.ValAddress
		accounts        string
		accErr          *cosmosaccount.AccountDoesNotExistError
	)

	for _, acc := range cfg.Accounts {
		sdkAcc, err := ca.GetByName(acc.Name)
		if errors.As(err, &accErr) {
			sdkAcc, _, err = ca.Create(acc.Name)
		}
		if err != nil {
			return err
		}

		sdkAddr, err := sdkAcc.Address(prefix)
		if err != nil {
			return err
		}
		if len(cfg.Validators) == 0 {
			return errors.Errorf("no validators found for account %s", sdkAcc.Name)
		}

		if cfg.Validators[0].Name == acc.Name {
			accAddr, err := addressCodec.StringToBytes(sdkAddr)
			if err != nil {
				return err
			}
			operatorAddress = accAddr
		}
		accounts = accounts + "," + sdkAddr
	}

	chainID, err := c.ID()
	if err != nil {
		return err
	}

	operatorAddressStr, err := valAddressCodec.BytesToString(operatorAddress)
	if err != nil {
		return err
	}

	args := chain.InPlaceArgs{
		NewChainID:         fmt.Sprintf("local%s", chainID),
		NewOperatorAddress: operatorAddressStr,
		AccountsToFund:     accounts,
	}

	return c.TestnetInPlace(cmd.Context(), args)
}