feat(hive): add HiveGetPublicKeys, HiveSignAccountCreate, HiveSignAccountUpdate

- messages_hive_pb2.py: regenerated from updated proto; now includes all 10
  message types (HiveGetPublicKey/Keys, HivePublicKey/Keys, HiveSignTx/ed,
  HiveSignAccountCreate/ed, HiveSignAccountUpdate/ed). Added role field to
  HiveGetPublicKey.

- mapping.py: register wire IDs 1604-1609 for the six new message types.

- hive.py: add get_public_keys(), sign_account_create(), sign_account_update()
  helpers. get_public_key() gains optional role parameter.

- client.py: add hive_get_public_keys(), hive_sign_account_create(),
  hive_sign_account_update() mixin methods with @expect decorators.

Author: BitHighlander

keepkeylib/client.py

@@ -1855,9 +1855,16 @@ def zcash_sign_pczt(self, address_n, actions, account=None,
 
     # ── Hive ────────────────────────────────────────────────────
     @expect(hive_proto.HivePublicKey)
-    def hive_get_public_key(self, address_n, show_display=False):
+    def hive_get_public_key(self, address_n, show_display=False, role=None):
+        kwargs = dict(address_n=address_n, show_display=show_display)
+        if role is not None:
+            kwargs['role'] = role
+        return self.call(hive_proto.HiveGetPublicKey(**kwargs))
+
+    @expect(hive_proto.HivePublicKeys)
+    def hive_get_public_keys(self, account_index=0, show_display=False):
         return self.call(
-            hive_proto.HiveGetPublicKey(address_n=address_n, show_display=show_display)
+            hive_proto.HiveGetPublicKeys(account_index=account_index, show_display=show_display)
         )
 
     @expect(hive_proto.HiveSignedTx)
@@ -1877,6 +1884,43 @@ def hive_sign_tx(self, address_n, chain_id, ref_block_num, ref_block_prefix,
             'memo': memo,
         }))
 
+    @expect(hive_proto.HiveSignedAccountCreate)
+    def hive_sign_account_create(self, address_n, chain_id, ref_block_num, ref_block_prefix,
+                                 expiration, creator, new_account_name, fee_amount=3000,
+                                 owner_key='', active_key='', posting_key='', memo_key=''):
+        return self.call(hive_proto.HiveSignAccountCreate(
+            address_n=address_n,
+            chain_id=chain_id,
+            ref_block_num=ref_block_num,
+            ref_block_prefix=ref_block_prefix,
+            expiration=expiration,
+            creator=creator,
+            new_account_name=new_account_name,
+            fee_amount=fee_amount,
+            owner_key=owner_key,
+            active_key=active_key,
+            posting_key=posting_key,
+            memo_key=memo_key,
+        ))
+
+    @expect(hive_proto.HiveSignedAccountUpdate)
+    def hive_sign_account_update(self, address_n, chain_id, ref_block_num, ref_block_prefix,
+                                 expiration, account,
+                                 new_owner_key='', new_active_key='',
+                                 new_posting_key='', new_memo_key=''):
+        return self.call(hive_proto.HiveSignAccountUpdate(
+            address_n=address_n,
+            chain_id=chain_id,
+            ref_block_num=ref_block_num,
+            ref_block_prefix=ref_block_prefix,
+            expiration=expiration,
+            account=account,
+            new_owner_key=new_owner_key,
+            new_active_key=new_active_key,
+            new_posting_key=new_posting_key,
+            new_memo_key=new_memo_key,
+        ))
+
 class KeepKeyClient(ProtocolMixin, TextUIMixin, BaseClient):
     pass
 

keepkeylib/hive.py

@@ -1,9 +1,16 @@
 from . import messages_hive_pb2 as proto
 
 
-def get_public_key(client, address_n, show_display=False):
+def get_public_key(client, address_n, show_display=False, role=None):
+    kwargs = dict(address_n=address_n, show_display=show_display)
+    if role is not None:
+        kwargs['role'] = role
+    return client.call(proto.HiveGetPublicKey(**kwargs))
+
+
+def get_public_keys(client, account_index=0, show_display=False):
     return client.call(
-        proto.HiveGetPublicKey(address_n=address_n, show_display=show_display)
+        proto.HiveGetPublicKeys(account_index=account_index, show_display=show_display)
     )
 
 
@@ -23,3 +30,40 @@ def sign_tx(client, address_n, chain_id, ref_block_num, ref_block_prefix,
         'asset_symbol': asset_symbol,
         'memo': memo,
     }))
+
+
+def sign_account_create(client, address_n, chain_id, ref_block_num, ref_block_prefix,
+                        expiration, creator, new_account_name, fee_amount=3000,
+                        owner_key='', active_key='', posting_key='', memo_key=''):
+    return client.call(proto.HiveSignAccountCreate(
+        address_n=address_n,
+        chain_id=chain_id,
+        ref_block_num=ref_block_num,
+        ref_block_prefix=ref_block_prefix,
+        expiration=expiration,
+        creator=creator,
+        new_account_name=new_account_name,
+        fee_amount=fee_amount,
+        owner_key=owner_key,
+        active_key=active_key,
+        posting_key=posting_key,
+        memo_key=memo_key,
+    ))
+
+
+def sign_account_update(client, address_n, chain_id, ref_block_num, ref_block_prefix,
+                        expiration, account,
+                        new_owner_key='', new_active_key='',
+                        new_posting_key='', new_memo_key=''):
+    return client.call(proto.HiveSignAccountUpdate(
+        address_n=address_n,
+        chain_id=chain_id,
+        ref_block_num=ref_block_num,
+        ref_block_prefix=ref_block_prefix,
+        expiration=expiration,
+        account=account,
+        new_owner_key=new_owner_key,
+        new_active_key=new_active_key,
+        new_posting_key=new_posting_key,
+        new_memo_key=new_memo_key,
+    ))

keepkeylib/mapping.py

@@ -100,10 +100,16 @@ def check_missing():
 
 # Manually register Hive messages (not in the old messages_pb2.py enum)
 _hive_wire_ids = {
-    1600: ('HiveGetPublicKey', hive_proto),
-    1601: ('HivePublicKey', hive_proto),
-    1602: ('HiveSignTx', hive_proto),
-    1603: ('HiveSignedTx', hive_proto),
+    1600: ('HiveGetPublicKey',        hive_proto),
+    1601: ('HivePublicKey',           hive_proto),
+    1602: ('HiveSignTx',              hive_proto),
+    1603: ('HiveSignedTx',            hive_proto),
+    1604: ('HiveGetPublicKeys',       hive_proto),
+    1605: ('HivePublicKeys',          hive_proto),
+    1606: ('HiveSignAccountCreate',   hive_proto),
+    1607: ('HiveSignedAccountCreate', hive_proto),
+    1608: ('HiveSignAccountUpdate',   hive_proto),
+    1609: ('HiveSignedAccountUpdate', hive_proto),
 }
 for wire_id, (msg_name, mod) in _hive_wire_ids.items():
     msg_class = getattr(mod, msg_name, None)