github.com/line/ostracon@v1.0.10-0.20230328032236-7f20145f065d/privval/msgs.go (about)

     1  package privval
     2  
     3  import (
     4  	"fmt"
     5  
     6  	"github.com/gogo/protobuf/proto"
     7  
     8  	privvalproto "github.com/tendermint/tendermint/proto/tendermint/privval"
     9  
    10  	ocprivvalproto "github.com/line/ostracon/proto/ostracon/privval"
    11  )
    12  
    13  // TODO: Add ChainIDRequest
    14  
    15  func mustWrapMsg(pb proto.Message) ocprivvalproto.Message {
    16  	msg := ocprivvalproto.Message{}
    17  
    18  	switch pb := pb.(type) {
    19  	case *ocprivvalproto.Message:
    20  		msg = *pb
    21  	case *privvalproto.PubKeyRequest:
    22  		msg.Sum = &ocprivvalproto.Message_PubKeyRequest{PubKeyRequest: pb}
    23  	case *privvalproto.PubKeyResponse:
    24  		msg.Sum = &ocprivvalproto.Message_PubKeyResponse{PubKeyResponse: pb}
    25  	case *privvalproto.SignVoteRequest:
    26  		msg.Sum = &ocprivvalproto.Message_SignVoteRequest{SignVoteRequest: pb}
    27  	case *privvalproto.SignedVoteResponse:
    28  		msg.Sum = &ocprivvalproto.Message_SignedVoteResponse{SignedVoteResponse: pb}
    29  	case *privvalproto.SignedProposalResponse:
    30  		msg.Sum = &ocprivvalproto.Message_SignedProposalResponse{SignedProposalResponse: pb}
    31  	case *privvalproto.SignProposalRequest:
    32  		msg.Sum = &ocprivvalproto.Message_SignProposalRequest{SignProposalRequest: pb}
    33  	case *ocprivvalproto.VRFProofRequest:
    34  		msg.Sum = &ocprivvalproto.Message_VrfProofRequest{VrfProofRequest: pb}
    35  	case *ocprivvalproto.VRFProofResponse:
    36  		msg.Sum = &ocprivvalproto.Message_VrfProofResponse{VrfProofResponse: pb}
    37  	case *privvalproto.PingRequest:
    38  		msg.Sum = &ocprivvalproto.Message_PingRequest{PingRequest: pb}
    39  	case *privvalproto.PingResponse:
    40  		msg.Sum = &ocprivvalproto.Message_PingResponse{PingResponse: pb}
    41  	default:
    42  		panic(fmt.Errorf("unknown message type %T", pb))
    43  	}
    44  
    45  	return msg
    46  }