package mpc

import (
	"crypto/ecdsa"

	"github.com/onsonr/sonr/crypto/core/protocol"
	"github.com/onsonr/sonr/crypto/tecdsa/dklsv1/dkg"
)

// BaseKeyshare contains common fields and methods for both validator and user keyshares
type BaseKeyshare struct {
	Message            *protocol.Message `json:"message"`
	Role               int               `json:"role"`
	UncompressedPubKey []byte            `json:"public_key"`
	CompressedPubKey   []byte            `json:"compressed_public_key"`
}

func initFromAlice(aliceOut *dkg.AliceOutput, originalMsg *protocol.Message) BaseKeyshare {
	return BaseKeyshare{
		Message:            originalMsg,
		Role:               1,
		UncompressedPubKey: aliceOut.PublicKey.ToAffineUncompressed(),
		CompressedPubKey:   aliceOut.PublicKey.ToAffineCompressed(),
	}
}

func initFromBob(bobOut *dkg.BobOutput, originalMsg *protocol.Message) BaseKeyshare {
	return BaseKeyshare{
		Message:            originalMsg,
		Role:               2,
		UncompressedPubKey: bobOut.PublicKey.ToAffineUncompressed(),
		CompressedPubKey:   bobOut.PublicKey.ToAffineCompressed(),
	}
}

func (b *BaseKeyshare) GetPayloads() map[string][]byte {
	return b.Message.Payloads
}

func (b *BaseKeyshare) GetMetadata() map[string]string {
	return b.Message.Metadata
}

func (b *BaseKeyshare) GetPublicKey() []byte {
	return b.UncompressedPubKey
}

func (b *BaseKeyshare) GetProtocol() string {
	return b.Message.Protocol
}

func (b *BaseKeyshare) GetRole() int32 {
	return int32(b.Role)
}

func (b *BaseKeyshare) GetVersion() uint32 {
	return uint32(b.Message.Version)
}

func (b *BaseKeyshare) ECDSAPublicKey() (*ecdsa.PublicKey, error) {
	return ComputeEcdsaPublicKey(b.UncompressedPubKey)
}

func (b *BaseKeyshare) ExtractMessage() *protocol.Message {
	return &protocol.Message{
		Payloads: b.GetPayloads(),
		Metadata: b.GetMetadata(),
		Protocol: b.GetProtocol(),
		Version:  uint(b.GetVersion()),
	}
}