From 4311c4ee1d5c7cb8ecf94ca95b5e5b344a97f7b6 Mon Sep 17 00:00:00 2001
From: nolash <dev@holbrook.no>
Date: Tue, 5 Jan 2021 14:22:34 +0100
Subject: [PATCH] Add declarator viewer cli

---
 python/eth_address_declarator/runnable/add.py |  4 +-
 .../eth_address_declarator/runnable/view.py   | 85 +++++++++++++++++++
 2 files changed, 87 insertions(+), 2 deletions(-)
 create mode 100644 python/eth_address_declarator/runnable/view.py

diff --git a/python/eth_address_declarator/runnable/add.py b/python/eth_address_declarator/runnable/add.py
index 6bbfc05..7483de5 100644
--- a/python/eth_address_declarator/runnable/add.py
+++ b/python/eth_address_declarator/runnable/add.py
@@ -26,8 +26,8 @@ data_dir = os.path.join(script_dir, '..', 'data')
 
 argparser = argparse.ArgumentParser()
 argparser.add_argument('-p', '--provider', dest='p', default='http://localhost:8545', type=str, help='Web3 provider url (http only)')
-argparser.add_argument('-r', '--contract-address', dest='r', type=str, help='Token endorsement contract address')
-argparser.add_argument('-o', '--endorser-address', dest='o', type=str, help='Address to use to sign endorsement transaction')
+argparser.add_argument('-r', '--contract-address', dest='r', type=str, help='Address declaration contract address')
+argparser.add_argument('-o', '--declarator-address', dest='o', type=str, help='Signing address for the declaration')
 argparser.add_argument('--abi-dir', dest='abi_dir', type=str, default=data_dir, help='Directory containing bytecode and abi (default: {})'.format(data_dir))
 argparser.add_argument('-v', action='store_true', help='Be verbose')
 argparser.add_argument('address', type=str, help='Ethereum address to add declaration to')
diff --git a/python/eth_address_declarator/runnable/view.py b/python/eth_address_declarator/runnable/view.py
new file mode 100644
index 0000000..e6cce9f
--- /dev/null
+++ b/python/eth_address_declarator/runnable/view.py
@@ -0,0 +1,85 @@
+"""View address declaration, and attempt to resolve contents
+
+.. moduleauthor:: Louis Holbrook <dev@holbrook.no>
+.. pgp:: 0826EDA1702D1E87C6E2875121D2E7BB88C2A746 
+
+"""
+
+# standard imports
+import urllib
+import os
+import json
+import argparse
+import logging
+
+# third-party imports
+import web3
+
+
+logging.basicConfig(level=logging.WARNING)
+logg = logging.getLogger()
+
+logging.getLogger('web3').setLevel(logging.WARNING)
+logging.getLogger('urllib3').setLevel(logging.WARNING)
+
+script_dir = os.path.dirname(__file__)
+data_dir = os.path.join(script_dir, '..', 'data')
+
+argparser = argparse.ArgumentParser()
+argparser.add_argument('-p', '--provider', dest='p', default='http://localhost:8545', type=str, help='Web3 provider url (http only)')
+argparser.add_argument('-r', '--contract-address', dest='r', type=str, help='Address declaration contract address')
+argparser.add_argument('-o', '--declarator-address', dest='o', type=str, help='Signing address for the declaration')
+argparser.add_argument('--resolve', action='store_true', help='Attempt to resolve the hashes to actual content')
+argparser.add_argument('--resolve-http', dest='resolve_http', type=str, help='Base url to look up content hashes')
+argparser.add_argument('--abi-dir', dest='abi_dir', type=str, default=data_dir, help='Directory containing bytecode and abi (default: {})'.format(data_dir))
+argparser.add_argument('-v', action='store_true', help='Be verbose')
+argparser.add_argument('address', type=str, help='Ethereum declaration address to look up')
+args = argparser.parse_args()
+
+if args.v:
+    logg.setLevel(logging.DEBUG)
+
+
+def try_sha256(s):
+    r = urllib.request.urlopen(os.path.join(args.resolve_http, s.hex()))
+    return r.read().decode('utf-8')
+
+def try_utf8(s):
+    return s.decode('utf-8')
+
+
+def main():
+    w3 = web3.Web3(web3.Web3.HTTPProvider(args.p))
+
+    f = open(os.path.join(args.abi_dir, 'AddressDeclarator.json'), 'r')
+    abi = json.load(f)
+    f.close()
+
+    w3.eth.defaultAccount = w3.eth.accounts[0]
+    if args.o != None:
+        w3.eth.defaultAccount = args.o
+    logg.debug('endorser address {}'.format(w3.eth.defaultAccount))
+
+    c = w3.eth.contract(abi=abi, address=args.r)
+
+    declarations = c.functions.declaration(w3.eth.defaultAccount, args.address).call()
+
+    for d in declarations:
+        if not args.resolve:
+            print(d.hex())
+            continue
+        if args.resolve_http:
+            try:
+                r = try_sha256(d)
+                print(r)
+                continue
+            except urllib.error.HTTPError:
+                pass
+        try:
+            print(try_utf8(d))
+        except UnicodeDecodeError:
+            pass
+
+
+if __name__ == '__main__':
+    main()