package cmd

import (
	"context"
	"encoding/hex"
	"fmt"
	"math/big"
	"os"
	"strings"

	"github.com/shopspring/decimal"
	"github.com/spf13/cobra"
	"source.quilibrium.com/quilibrium/monorepo/node/protobufs"
)

var splitCmd = &cobra.Command{
	Use:   "split",
	Short: "Splits a coin into multiple coins",
	Long: `Splits a coin into multiple coins:
	
	split <OfCoin> <Amounts>...
	
	OfCoin - the address of the coin to split
	Amounts - the sets of amounts to split
	`,
	Run: func(cmd *cobra.Command, args []string) {
		if len(args) < 3 {
			fmt.Println("invalid command")
			os.Exit(1)
		}

		payload := []byte("split")
		coinaddrHex, _ := strings.CutPrefix(args[0], "0x")
		coinaddr, err := hex.DecodeString(coinaddrHex)
		if err != nil {
			panic(err)
		}
		coin := &protobufs.CoinRef{
			Address: coinaddr,
		}
		payload = append(payload, coinaddr...)

		conversionFactor, _ := new(big.Int).SetString("1DCD65000", 16)
		amounts := [][]byte{}
		for _, amt := range args[1:] {
			amount, err := decimal.NewFromString(amt)
			if err != nil {
				fmt.Println("invalid amount")
				os.Exit(1)
			}
			amount.Mul(decimal.NewFromBigInt(conversionFactor, 0))
			amountBytes := amount.BigInt().FillBytes(make([]byte, 32))
			amounts = append(amounts, amountBytes)
			payload = append(payload, amountBytes...)
		}

		conn, err := GetGRPCClient()
		if err != nil {
			panic(err)
		}
		defer conn.Close()

		client := protobufs.NewNodeServiceClient(conn)
		key, err := GetPrivKeyFromConfig(NodeConfig)
		if err != nil {
			panic(err)
		}

		sig, err := key.Sign(payload)
		if err != nil {
			panic(err)
		}

		pub, err := key.GetPublic().Raw()
		if err != nil {
			panic(err)
		}

		_, err = client.SendMessage(
			context.Background(),
			&protobufs.TokenRequest{
				Request: &protobufs.TokenRequest_Split{
					Split: &protobufs.SplitCoinRequest{
						OfCoin:  coin,
						Amounts: amounts,
						Signature: &protobufs.Ed448Signature{
							Signature: sig,
							PublicKey: &protobufs.Ed448PublicKey{
								KeyValue: pub,
							},
						},
					},
				},
			},
		)
		if err != nil {
			panic(err)
		}
	},
}

func init() {
	tokenCmd.AddCommand(splitCmd)
}