package codec_test

import (
	"testing"

	"github.com/stretchr/testify/require"
	protov2 "google.golang.org/protobuf/proto"
	"google.golang.org/protobuf/types/dynamicpb"

	bankv1beta1 "cosmossdk.io/api/cosmos/bank/v1beta1"

	codectestutil "github.com/cosmos/cosmos-sdk/codec/testutil"
	codectypes "github.com/cosmos/cosmos-sdk/codec/types"
	"github.com/cosmos/cosmos-sdk/testutil/testdata"
	sdk "github.com/cosmos/cosmos-sdk/types"
	banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
)

type bankSendWrapper struct {
	*banktypes.MsgSend
}

func (msg bankSendWrapper) GetSigners() []sdk.AccAddress {
	fromAddress, _ := sdk.AccAddressFromBech32(msg.FromAddress)
	return []sdk.AccAddress{fromAddress}
}

func BenchmarkLegacyGetSigners(b *testing.B) {
	_, _, addr := testdata.KeyTestPubAddr()
	msg := bankSendWrapper{&banktypes.MsgSend{
		FromAddress: addr.String(),
		ToAddress:   "",
		Amount:      nil,
	}}

	for b.Loop() {
		_ = msg.GetSigners()
	}
}

func BenchmarkProtoreflectGetSigners(b *testing.B) {
	cdc := codectestutil.CodecOptions{}.NewCodec()
	signingCtx := cdc.InterfaceRegistry().SigningContext()
	_, _, addr := testdata.KeyTestPubAddr()
	// use a pulsar message
	msg := &bankv1beta1.MsgSend{
		FromAddress: addr.String(),
		ToAddress:   "",
		Amount:      nil,
	}

	for b.Loop() {
		_, err := signingCtx.GetSigners(msg)
		if err != nil {
			panic(err)
		}
	}
}

func BenchmarkProtoreflectGetSignersWithUnmarshal(b *testing.B) {
	cdc := codectestutil.CodecOptions{}.NewCodec()
	_, _, addr := testdata.KeyTestPubAddr()
	// start with a protoreflect message
	msg := &banktypes.MsgSend{
		FromAddress: addr.String(),
		ToAddress:   "",
		Amount:      nil,
	}
	// marshal to an any first because this is what we get from the wire
	a, err := codectypes.NewAnyWithValue(msg)
	require.NoError(b, err)

	for b.Loop() {
		_, _, err := cdc.GetMsgAnySigners(a)
		if err != nil {
			panic(err)
		}
	}
}

func BenchmarkProtoreflectGetSignersDynamicpb(b *testing.B) {
	cdc := codectestutil.CodecOptions{}.NewCodec()
	signingCtx := cdc.InterfaceRegistry().SigningContext()
	_, _, addr := testdata.KeyTestPubAddr()
	msg := &bankv1beta1.MsgSend{
		FromAddress: addr.String(),
		ToAddress:   "",
		Amount:      nil,
	}
	bz, err := protov2.Marshal(msg)
	require.NoError(b, err)

	dynamicmsg := dynamicpb.NewMessage(msg.ProtoReflect().Descriptor())
	err = protov2.Unmarshal(bz, dynamicmsg)
	require.NoError(b, err)

	for b.Loop() {
		_, err := signingCtx.GetSigners(dynamicmsg)
		if err != nil {
			panic(err)
		}
	}
}