package accounts

import (
	"fmt"
	"reflect"
	"strings"

	codectypes "github.com/cosmos/cosmos-sdk/codec/types"
	"github.com/cosmos/gogoproto/proto"

	"github.com/onsonr/sonr/internal/transaction"
)

// ProtoMsgG is a generic interface for protobuf messages.
type ProtoMsgG[T any] interface {
	*T
	transaction.Msg
}

type Any = codectypes.Any

func FindMessageByName(name string) (transaction.Msg, error) {
	typ := proto.MessageType(name)
	if typ == nil {
		return nil, fmt.Errorf("no message type found for %s", name)
	}
	return reflect.New(typ.Elem()).Interface().(transaction.Msg), nil
}

func MessageName(msg transaction.Msg) string {
	return proto.MessageName(msg)
}

// PackAny packs a proto message into an anypb.Any.
func PackAny(msg transaction.Msg) (*Any, error) {
	return codectypes.NewAnyWithValue(msg)
}

// UnpackAny unpacks an anypb.Any into a proto message.
func UnpackAny[T any, PT ProtoMsgG[T]](anyPB *Any) (PT, error) {
	to := new(T)
	return to, UnpackAnyTo(anyPB, PT(to))
}

func UnpackAnyTo(anyPB *Any, to transaction.Msg) error {
	return proto.Unmarshal(anyPB.Value, to)
}

func UnpackAnyRaw(anyPB *Any) (proto.Message, error) {
	split := strings.Split(anyPB.TypeUrl, "/")
	name := split[len(split)-1]
	typ := proto.MessageType(name)
	if typ == nil {
		return nil, fmt.Errorf("no message type found for %s", name)
	}
	to := reflect.New(typ.Elem()).Interface().(proto.Message)
	return to, UnpackAnyTo(anyPB, to)
}

func Merge(a, b transaction.Msg) {
	proto.Merge(a, b)
}

func Equal(a, b transaction.Msg) bool {
	return proto.Equal(a, b)
}