package local

import (
	"fmt"
	"math/big"

	"github.com/ChainSafe/chainbridge-core/chains/evm/calls/evmclient"
	"github.com/ChainSafe/chainbridge-core/chains/evm/calls/evmtransaction"

	"github.com/spf13/cobra"
)

var LocalSetupCmd = &cobra.Command{
	Use:   "local-setup",
	Short: "Deploy and prefund a local bridge for testing",
	Long:  "The local-setup command deploys a bridge, ERC20, ERC721 and generic handler contracts with preconfigured accounts and appropriate handlers",
	RunE:  localSetup,
}

// configuration
var (
	ethEndpoint1 = "http://localhost:8545"
	ethEndpoint2 = "http://localhost:8547"
	fabric1      = evmtransaction.NewTransaction
	fabric2      = evmtransaction.NewTransaction
)

func BindLocalSetupFlags(cmd *cobra.Command) {
	cmd.Flags().StringVar(&ethEndpoint1, "endpoint1", "", "RPC endpoint of the first network")
	cmd.Flags().StringVar(&ethEndpoint2, "endpoint2", "", "RPC endpoint of the second network")
}

func init() {
	BindLocalSetupFlags(LocalSetupCmd)
}

func localSetup(cmd *cobra.Command, args []string) error {
	// init client1
	ethClient, err := evmclient.NewEVMClient(ethEndpoint1, EveKp)
	if err != nil {
		return err
	}

	// init client2
	ethClient2, err := evmclient.NewEVMClient(ethEndpoint2, EveKp)
	if err != nil {
		return err
	}

	// chain 1
	// domainsId: 0
	config, err := SetupEVMBridge(ethClient, fabric1, 1, big.NewInt(1), EveKp.CommonAddress(), DefaultRelayerAddresses)
	if err != nil {
		return err
	}

	// chain 2
	// domainId: 1
	config2, err := SetupEVMBridge(ethClient2, fabric2, 2, big.NewInt(1), EveKp.CommonAddress(), DefaultRelayerAddresses)
	if err != nil {
		return err
	}

	prettyPrint(config, config2)

	return nil
}

func prettyPrint(config, config2 BridgeConfig) {
	fmt.Printf(`
===============================================
🎉🎉🎉 ChainBridge Successfully Deployed 🎉🎉🎉

- Chain 1 -
Bridge: %s
ERC20: %s
ERC20 Handler: %s
ERC721: %s
ERC721 Handler: %s
Generic Handler: %s
Asset Store: %s
ERC20 resourceId: %s
ERC721 resourceId %s
Generic resourceId %s

- Chain 2 -
Bridge: %s
ERC20: %s
ERC20 Handler: %s
ERC721: %s
ERC721 Handler: %s
Generic Handler: %s
Asset Store: %s
ERC20 resourceId: %s
ERC721 resourceId %s
Generic resourceId %s

===============================================
`,
		// config
		config.BridgeAddr,
		config.Erc20Addr,
		config.Erc20HandlerAddr,
		config.Erc721Addr,
		config.Erc721HandlerAddr,
		config.GenericHandlerAddr,
		config.AssetStoreAddr,
		config.Erc20ResourceID,
		config.Erc721ResourceID,
		config.GenericResourceID,
		// config2
		config2.BridgeAddr,
		config2.Erc20Addr,
		config2.Erc20HandlerAddr,
		config.Erc721Addr,
		config.Erc721HandlerAddr,
		config2.GenericHandlerAddr,
		config2.AssetStoreAddr,
		config2.Erc20ResourceID,
		config2.Erc721ResourceID,
		config2.GenericResourceID,
	)
}
