From 8ae35b5e2f63cf6ec009a707458d4c779a635178 Mon Sep 17 00:00:00 2001
From: Matt Ketmo <matthieu@moquet.net>
Date: Fri, 7 Jun 2024 10:08:30 +0200
Subject: [PATCH] feat: register pubkey type cometbft/PubKeyBls12_381 (#73)

---
 pkg/app/run.go          |  1 +
 pkg/crypto/bls12_318.go | 55 +++++++++++++++++++++++++++++++++++++++++
 2 files changed, 56 insertions(+)
 create mode 100644 pkg/crypto/bls12_318.go

diff --git a/pkg/app/run.go b/pkg/app/run.go
index 9acc60b..d4fc13e 100644
--- a/pkg/app/run.go
+++ b/pkg/app/run.go
@@ -14,6 +14,7 @@ import (
 	"github.com/cosmos/cosmos-sdk/types/query"
 	staking "github.com/cosmos/cosmos-sdk/x/staking/types"
 	"github.com/fatih/color"
+	_ "github.com/kilnfi/cosmos-validator-watcher/pkg/crypto"
 	"github.com/kilnfi/cosmos-validator-watcher/pkg/metrics"
 	"github.com/kilnfi/cosmos-validator-watcher/pkg/rpc"
 	"github.com/kilnfi/cosmos-validator-watcher/pkg/watcher"
diff --git a/pkg/crypto/bls12_318.go b/pkg/crypto/bls12_318.go
new file mode 100644
index 0000000..c1263df
--- /dev/null
+++ b/pkg/crypto/bls12_318.go
@@ -0,0 +1,55 @@
+package crypto
+
+import (
+	"bytes"
+	"fmt"
+
+	"github.com/cometbft/cometbft/crypto"
+	"github.com/cometbft/cometbft/crypto/tmhash"
+	cmtjson "github.com/cometbft/cometbft/libs/json"
+)
+
+const (
+	PubKeyName = "cometbft/PubKeyBls12_381"
+)
+
+func init() {
+	cmtjson.RegisterType(Bls12PubKey{}, PubKeyName)
+}
+
+var _ crypto.PubKey = Bls12PubKey{}
+
+type Bls12PubKey []byte
+
+// Address is the SHA256-20 of the raw pubkey bytes.
+func (pubKey Bls12PubKey) Address() crypto.Address {
+	// if len(pubKey) != PubKeySize {
+	// 	panic("pubkey is incorrect size")
+	// }
+	return crypto.Address(tmhash.SumTruncated(pubKey))
+}
+
+// Bytes returns the PubKey byte format.
+func (pubKey Bls12PubKey) Bytes() []byte {
+	return []byte(pubKey)
+}
+
+func (pubKey Bls12PubKey) VerifySignature(msg []byte, sig []byte) bool {
+	return false
+}
+
+func (pubKey Bls12PubKey) String() string {
+	return fmt.Sprintf("PubKeyBls12_381{%X}", []byte(pubKey))
+}
+
+func (Bls12PubKey) Type() string {
+	return "bls12_381"
+}
+
+func (pubKey Bls12PubKey) Equals(other crypto.PubKey) bool {
+	if otherEd, ok := other.(Bls12PubKey); ok {
+		return bytes.Equal(pubKey[:], otherEd[:])
+	}
+
+	return false
+}