feat: update wallet creation and signing processes to support multiple protocols

Author: vietddude

examples/generate/kms/main.go

@@ -117,7 +117,11 @@ func main() {
 	for _, walletID := range walletIDs {
 		wg.Add(1) // Add to WaitGroup BEFORE attempting to create wallet
 
-		if err := mpcClient.CreateWallet(walletID); err != nil {
+		if err := mpcClient.CreateWallet(&types.GenerateKeyMessage{
+			WalletID:      walletID,
+			ECDSAProtocol: types.ProtocolCGGMP21,
+			EdDSAProtocol: types.ProtocolGG18,
+		}); err != nil {
 			logger.Error("CreateWallet failed", err)
 			walletStartTimes.Delete(walletID)
 			wg.Done() // Now this is safe since we added 1 above

examples/presign/main.go

@@ -12,6 +12,7 @@ import (
 	"github.com/fystack/mpcium/pkg/event"
 	"github.com/fystack/mpcium/pkg/logger"
 	"github.com/fystack/mpcium/pkg/types"
+	"github.com/google/uuid"
 	"github.com/nats-io/nats.go"
 	"github.com/spf13/viper"
 )
@@ -65,8 +66,10 @@ func main() {
 	})
 
 	txMsg := &types.PresignTxMessage{
-		KeyType:  types.KeyTypeCGGMP21,
+		KeyType:  types.KeyTypeSecp256k1,
+		Protocol: types.ProtocolCGGMP21,
 		WalletID: "196c6858-30de-4a49-9134-8bc825d40764", // Use the generated wallet ID
+		TxID:     uuid.New().String(),
 	}
 	err = mpcClient.PresignTransaction(txMsg)
 	if err != nil {

examples/reshare/main.go

@@ -88,6 +88,7 @@ func main() {
 
 		NewThreshold: 1, // t+1 <= len(NodeIDs)
 		KeyType:      types.KeyTypeEd25519,
+		Protocol:     types.ProtocolFROST,
 	}
 	err = mpcClient.Resharing(resharingMsg)
 	if err != nil {

examples/sign/main.go

@@ -70,8 +70,9 @@ func main() {
 	dummyTx := []byte("deadbeef") // replace with real transaction bytes
 
 	txMsg := &types.SignTxMessage{
-		KeyType:             types.KeyTypeEd25519,
-		WalletID:            "ad24f678-b04b-4149-bcf6-bf9c90df8e63", // Use the generated wallet ID
+		KeyType:             types.KeyTypeSecp256k1,
+		Protocol:            types.ProtocolFROST,
+		WalletID:            "6d553e80-a1dc-4894-9eaf-b81e3fe0c94a", // Use the generated wallet ID
 		NetworkInternalCode: "solana-devnet",
 		TxID:                txID,
 		Tx:                  dummyTx,
@@ -87,6 +88,8 @@ func main() {
 		logger.Info("Signing result received",
 			"txID", evt.TxID,
 			"signature", fmt.Sprintf("%x", evt.Signature),
+			"error", evt.ErrorReason,
+			"errorCode", evt.ErrorCode,
 		)
 	})
 	if err != nil {

go.mod

@@ -11,9 +11,10 @@ require (
 	github.com/aws/aws-sdk-go-v2/credentials v1.18.8
 	github.com/aws/aws-sdk-go-v2/service/kms v1.45.0
 	github.com/bnb-chain/tss-lib/v2 v2.0.2
+	github.com/btcsuite/btcd/btcec/v2 v2.3.2
 	github.com/decred/dcrd/dcrec/edwards/v2 v2.0.3
-	github.com/decred/dcrd/dcrec/secp256k1/v4 v4.2.0
 	github.com/dgraph-io/badger/v4 v4.7.0
+	github.com/fxamacker/cbor/v2 v2.4.0
 	github.com/google/uuid v1.6.0
 	github.com/hashicorp/consul/api v1.32.1
 	github.com/mitchellh/mapstructure v1.5.0
@@ -25,6 +26,7 @@ require (
 	github.com/taurusgroup/multi-party-sig v0.7.0-alpha-2025-01-28
 	github.com/urfave/cli/v3 v3.3.2
 	golang.org/x/crypto v0.37.0
+	golang.org/x/sync v0.13.0
 	golang.org/x/term v0.31.0
 )
 
@@ -43,17 +45,16 @@ require (
 	github.com/aws/aws-sdk-go-v2/service/sts v1.38.1 // indirect
 	github.com/aws/smithy-go v1.23.0 // indirect
 	github.com/btcsuite/btcd v0.24.2 // indirect
-	github.com/btcsuite/btcd/btcec/v2 v2.3.2 // indirect
 	github.com/btcsuite/btcd/chaincfg/chainhash v1.1.0 // indirect
 	github.com/btcsuite/btcutil v1.0.2 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/cronokirby/saferith v0.33.0 // indirect
 	github.com/davecgh/go-spew v1.1.2-0.20180830191138-d8f796af33cc // indirect
+	github.com/decred/dcrd/dcrec/secp256k1/v4 v4.2.0 // indirect
 	github.com/dgraph-io/ristretto/v2 v2.2.0 // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
 	github.com/fatih/color v1.18.0 // indirect
 	github.com/fsnotify/fsnotify v1.7.0 // indirect
-	github.com/fxamacker/cbor/v2 v2.4.0 // indirect
 	github.com/go-logr/logr v1.4.2 // indirect
 	github.com/go-logr/stdr v1.2.2 // indirect
 	github.com/gogo/protobuf v1.3.2 // indirect

pkg/client/client.go

@@ -19,7 +19,7 @@ const (
 )
 
 type MPCClient interface {
-	CreateWallet(walletID string) error
+	CreateWallet(msg *types.GenerateKeyMessage) error
 	OnWalletCreationResult(callback func(event event.KeygenResultEvent)) error
 
 	SignTransaction(msg *types.SignTxMessage) error
@@ -110,11 +110,7 @@ func NewMPCClient(opts Options) MPCClient {
 }
 
 // CreateWallet generates a GenerateKeyMessage, signs it, and publishes it.
-func (c *mpcClient) CreateWallet(walletID string) error {
-	// build the message
-	msg := &types.GenerateKeyMessage{
-		WalletID: walletID,
-	}
+func (c *mpcClient) CreateWallet(msg *types.GenerateKeyMessage) error {
 	// compute the canonical raw bytes
 	raw, err := msg.Raw()
 	if err != nil {

pkg/event/keygen.go

@@ -7,11 +7,9 @@ const (
 )
 
 type KeygenResultEvent struct {
-	WalletID      string `json:"wallet_id"`
-	ECDSAPubKey   []byte `json:"ecdsa_pub_key"`
-	EDDSAPubKey   []byte `json:"eddsa_pub_key"`
-	CGGMP21PubKey []byte `json:"cggmp21_pub_key"`
-	TaprootPubKey []byte `json:"taproot_pub_key"`
+	WalletID    string `json:"wallet_id"`
+	ECDSAPubKey []byte `json:"ecdsa_pub_key"`
+	EDDSAPubKey []byte `json:"eddsa_pub_key"`
 
 	ResultType  ResultType `json:"result_type"`
 	ErrorReason string     `json:"error_reason"`

pkg/event/presign.go

@@ -13,5 +13,6 @@ type PresignResultEvent struct {
 	ErrorReason string     `json:"error_reason"`
 	IsTimeout   bool       `json:"is_timeout"`
 	WalletID    string     `json:"wallet_id"`
+	TxID        string     `json:"tx_id"`
 	Status      string     `json:"status"`
 }