feat(side-dag): implement gen_keys CLI command

Author: glevco

hathor/cli/generate_keys.py

@@ -0,0 +1,42 @@
+# Copyright 2021 Hathor Labs
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import os
+import sys
+
+from cryptography.hazmat.primitives.asymmetric import ec
+from cryptography.hazmat.primitives.asymmetric.ec import EllipticCurvePrivateKeyWithSerialization
+
+
+def main():
+    from hathor.cli.util import create_parser
+    from hathor.crypto.util import (
+        get_address_b58_from_public_key,
+        get_private_key_bytes,
+        get_public_key_bytes_compressed,
+    )
+    parser = create_parser()
+    parser.add_argument('--config-yaml', type=str, help='Configuration yaml filepath')
+    args = parser.parse_args(sys.argv[1:])
+    if not args.config_yaml:
+        raise Exception('`--config-yaml` is required')
+    os.environ['HATHOR_CONFIG_YAML'] = args.config_yaml
+
+    private_key = ec.generate_private_key(ec.SECP256K1())
+    public_key = private_key.public_key()
+    assert isinstance(private_key, EllipticCurvePrivateKeyWithSerialization)
+
+    print('Private key (hex):', get_private_key_bytes(private_key=private_key).hex())
+    print('Public key (hex):', get_public_key_bytes_compressed(public_key=public_key).hex())
+    print('Address (base58):', get_address_b58_from_public_key(public_key=public_key))

hathor/cli/main.py

@@ -34,6 +34,7 @@ def __init__(self) -> None:
         from . import (
             db_export,
             db_import,
+            generate_keys,
             generate_valid_words,
             merged_mining,
             mining,
@@ -78,6 +79,7 @@ def __init__(self) -> None:
         self.add_cmd('tests', 'gen_twin_tx', twin_tx, 'Generate a twin transaction from a transaction hash')
         self.add_cmd('wallet', 'gen_kp_wallet', wallet, 'Generate a new KeyPair wallet')
         self.add_cmd('wallet', 'gen_hd_words', generate_valid_words, 'Generate random words for HD wallet')
+        self.add_cmd('wallet', 'gen_keys', generate_keys, 'Generate a private/public key pair and its address')
         self.add_cmd('oracle', 'oracle-create-key', oracle_create_key, 'Create an oracle private/public key')
         self.add_cmd('oracle', 'oracle-get-pubkey', oracle_get_pubkey,
                      'Read an oracle private key and output public key hash')