dkg

type BlameOutput struct {
	Valid        bool
	BlameMessage []byte
}

type Config struct {
	// Protocol the DKG protocol implementation
	KeygenProtocol      func(network Network, operatorID types.OperatorID, identifier RequestID, init *Init) Protocol
	ReshareProtocol     func(network Network, operatorID types.OperatorID, identifier RequestID, reshare *Reshare, output *KeyGenOutput) Protocol
	Network             Network
	Storage             Storage
	SignatureDomainType types.DomainType
	Signer              types.DKGSigner
}

type Init struct {
	// OperatorIDs are the operators selected for the DKG
	OperatorIDs []types.OperatorID
	// Threshold DKG threshold for signature reconstruction
	Threshold uint16
	// WithdrawalCredentials used when signing the deposit data
	WithdrawalCredentials []byte
	// Fork is eth2 fork version
	Fork phase0.Version
}

type KeyGenOutput struct {
	Share           *bls.SecretKey
	OperatorPubKeys map[types.OperatorID]*bls.PublicKey
	ValidatorPK     types.ValidatorPK
	Threshold       uint64
}

type Message struct {
	MsgType    MsgType
	Identifier RequestID
	Data       []byte
}

type MsgType int

type Network interface {
	// StreamDKGBlame will stream to any subscriber the blame result of the DKG
	StreamDKGBlame(blame *BlameOutput) error
	// StreamDKGOutput will stream to any subscriber the result of the DKG
	StreamDKGOutput(output map[types.OperatorID]*SignedOutput) error
	// BroadcastDKGMessage will broadcast a msg to the dkg network
	BroadcastDKGMessage(msg *SignedMessage) error
}

type Node struct {
	// contains filtered or unexported fields
}

type Operator struct {
	// OperatorID the node's Operator ID
	OperatorID types.OperatorID
	// ETHAddress the operator's eth address used to sign and verify messages against
	ETHAddress common.Address
	// EncryptionPubKey encryption pubkey for shares
	EncryptionPubKey *rsa.PublicKey
}

type Output struct {
	// RequestID for the DKG instance (not used for signing)
	RequestID RequestID
	// EncryptedShare standard SSV encrypted shares
	EncryptedShare []byte
	// SharePubKey is the share's BLS pubkey
	SharePubKey []byte
	// ValidatorPubKey the resulting public key corresponding to the shared private key
	ValidatorPubKey types.ValidatorPK
	// DepositDataSignature reconstructed signature of DepositMessage according to eth2 spec
	DepositDataSignature types.Signature
}

type PartialDepositData struct {
	Signer    types.OperatorID
	Root      []byte
	Signature types.Signature
}

type Protocol interface {
	Start() error
	// ProcessMsg returns true and a bls share if finished
	ProcessMsg(msg *SignedMessage) (bool, *ProtocolOutcome, error)
}

type ProtocolOutcome struct {
	ProtocolOutput *KeyGenOutput
	BlameOutput    *BlameOutput
}

type RequestID [24]byte

type Reshare struct {
	// ValidatorPK is the the public key to be reshared
	ValidatorPK types.ValidatorPK
	// OperatorIDs are the operators in the new set
	OperatorIDs []types.OperatorID
	// Threshold is the threshold of the new set
	Threshold uint16
}

type Runner interface {
	ProcessMsg(msg *SignedMessage) (bool, error)
}

type Runners map[string]Runner

type SignedMessage struct {
	Message   *Message
	Signer    types.OperatorID
	Signature types.Signature
}

type SignedOutput struct {
	// Data signed
	Data *Output
	// Signer Operator ID which signed
	Signer types.OperatorID
	// Signature over Data.GetRoot()
	Signature types.Signature
}

type Storage interface {
	// GetDKGOperator returns true and operator object if found by operator ID
	GetDKGOperator(operatorID types.OperatorID) (bool, *Operator, error)
	SaveKeyGenOutput(output *KeyGenOutput) error
	GetKeyGenOutput(pk types.ValidatorPK) (*KeyGenOutput, error)
}