Merge pull request #24 from b-open-io/add-identity-key-to-key-deriver

feat: add identity key and identity key hex to KeyDeriver

Author: rohenaz

wallet/key_deriver.go

@@ -25,6 +25,14 @@ type KeyDeriver struct {
 	rootKey *ec.PrivateKey
 }
 
+func (kd *KeyDeriver) IdentityKey() *ec.PublicKey {
+	return kd.rootKey.PubKey()
+}
+
+func (kd *KeyDeriver) IdentityKeyHex() string {
+	return kd.IdentityKey().ToDERHex()
+}
+
 // NewKeyDeriver creates a new KeyDeriver instance with a root private key.
 // The root key can be either a specific private key or the special 'anyone' key.
 func NewKeyDeriver(privateKey *ec.PrivateKey) *KeyDeriver {

wallet/key_deriver_test.go

@@ -27,6 +27,16 @@ func TestKeyDeriver(t *testing.T) {
 
 	keyDeriver := NewKeyDeriver(rootPrivateKey)
 
+	t.Run("should return public key for root key as identity key", func(t *testing.T) {
+		identityKey := keyDeriver.IdentityKey()
+		assert.Equalf(t, rootPublicKey, identityKey, "identity key should match root public key")
+	})
+
+	t.Run("should return DER HEX from public key for root key as identity key hex", func(t *testing.T) {
+		identityKey := keyDeriver.IdentityKeyHex()
+		assert.Equalf(t, rootPublicKey.ToDERHex(), identityKey, "identity key hex should match root public key hex")
+	})
+
 	t.Run("should compute the correct invoice number", func(t *testing.T) {
 		invoiceNumber, err := keyDeriver.computeInvoiceNumber(protocolID, keyID)
 		assert.NoError(t, err, "computing invoice number should not error")