Added get_keys_accounts API method

Author: aaroncox

README.md

@@ -44,6 +44,7 @@ Each API service will bind to a different IP address on the localhost, which can
 - `/v1/chain/get_currency_balances` runs on port 8901
 - `/v1/chain/get_accounts` runs on port 8902
 - `/v1/chain/get_blocks` runs on port 8903
+- `/v1/history/get_keys_accounts` runs on port 8904
 - `/v1/node/get_supported_apis` runs on port 8905
 
 ### nginx configuration
@@ -69,6 +70,10 @@ location /v1/chain/get_blocks$ {
     proxy_pass http://127.0.0.1:8903;
 }
 
+location /v1/history/get_keys_accounts$ {
+    proxy_pass http://127.0.0.1:8904;
+}
+
 location /v1/node/get_supported_apis$ {
     proxy_pass http://127.0.0.1:8905;
 }

config/get_keys_accounts/.env.example

@@ -0,0 +1,3 @@
+GET_KEYS_ACCOUNTS_WORKERS=4
+GET_KEYS_ACCOUNTS_UPSTREAM_WORKERS=4
+GET_KEYS_ACCOUNTS_LIMIT=50

docker-compose.yaml

@@ -12,6 +12,18 @@ services:
     restart: on-failure
     volumes:
         - .:/src
+  get_keys_accounts:
+    build: ./services/get_keys_accounts
+    env_file:
+        - ./config/.env
+        - ./config/get_keys_accounts/.env
+    networks:
+        - eosio_api_network
+    ports:
+        - "8904:8000"
+    restart: on-failure
+    volumes:
+        - .:/src
   get_available_endpoints:
     build: ./services/get_available_endpoints
     env_file:
@@ -61,6 +73,7 @@ services:
     volumes:
         - .:/src
 
+
 networks:
     eosio_api_network:
         external: true

services/get_keys_accounts/Dockerfile

@@ -0,0 +1,8 @@
+FROM python:3.7.0-slim
+
+COPY requirements.txt /tmp/requirements.txt
+RUN pip3 install -r /tmp/requirements.txt
+
+WORKDIR /src/services/get_keys_accounts
+
+CMD ["gunicorn", "--reload", "-c", "config.py", "main:app"]

services/get_keys_accounts/config.py

@@ -0,0 +1,4 @@
+import os
+
+bind = '0.0.0.0:8000'
+workers = os.environ['GET_KEYS_ACCOUNTS_WORKERS']

services/get_keys_accounts/main.py

@@ -0,0 +1,91 @@
+import asyncio
+import concurrent.futures
+import json
+import requests
+import os
+import sys
+import falcon
+
+from functools import partial
+
+# Load Environmental Variables
+max_workers = int(os.environ['GET_KEYS_ACCOUNTS_UPSTREAM_WORKERS'])
+max_results = int(os.environ['GET_KEYS_ACCOUNTS_LIMIT'])
+upstream = os.environ['UPSTREAM_API']
+
+async def get_keys_accounts(public_keys):
+    account_cache = {}
+    with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as pool:
+        loop = asyncio.get_event_loop()
+        futures = []
+        # Create a future execution of the post request in the futures pool
+        for public_key in public_keys:
+            futures.append(
+                loop.run_in_executor(
+                    pool,
+                    partial(
+                        requests.post,
+                        upstream + '/v1/history/get_key_accounts',
+                        json={
+                            "public_key": public_key
+                        }
+                    )
+                )
+            )
+        results = {
+            'account_names': [],
+            'map': {}
+        }
+        # Await for all processes to complete
+        for r in await asyncio.gather(*futures):
+            if r.status_code == 200:
+                body = json.loads(r.request.body)
+                for account_name in r.json()['account_names']:
+                    if account_name not in account_cache:
+                        sys.stdout.flush()
+                        r = requests.post(upstream + '/v1/chain/get_account', json={
+                            "account_name": account_name
+                        })
+                        if r.status_code == 200:
+                            account_cache[account_name] = r.json()
+                    permissions = account_cache[account_name].get('permissions')
+                    for permission in permissions:
+                        for key in permission['required_auth']['keys']:
+                            if key['key'] == body['public_key']:
+                                account_authority = account_name + '@' + permission['perm_name']
+                                results['account_names'].append(account_authority)
+                                if body['public_key'] in results['map']:
+                                    results['map'][body['public_key']].append(account_authority)
+                                else:
+                                    results['map'][body['public_key']] = [account_authority]
+            pass
+        # Return results
+        return results
+
+class GetKeysAccounts:
+    def on_get(self, req, resp):
+        self.on_post(req, resp);
+    def on_post(self, req, resp):
+        # Process the request to retrieve the account name
+        request = json.loads(req.stream.read())
+        # Accounts to load
+        public_keys = request.get('public_keys')
+        # Throw exception if exceeding limit
+        requested_public_keys = len(public_keys)
+        if requested_public_keys > max_results:
+            resp.body = json.dumps({
+                "code": 500,
+                "message": "Exceeded maximum number of requested accounts per request (requested {}, limit {})".format(requested_accounts, max_results)
+            })
+        else:
+            # Establish session for retrieval
+            with requests.Session() as session:
+                # Launch async event loop for results
+                loop = asyncio.get_event_loop()
+                results = loop.run_until_complete(get_keys_accounts(public_keys))
+                # Server the response
+                resp.body = json.dumps(results)
+
+# Launch falcon API
+app = falcon.API()
+app.add_route('/v1/history/get_keys_accounts', GetKeysAccounts())

services/get_keys_accounts/requirements.txt

@@ -0,0 +1,3 @@
+falcon==1.4.1
+gunicorn==19.9.0
+requests==2.20.0

services/get_supported_apis/main.py

@@ -25,6 +25,9 @@
     ('/v1/chain/get_currency_balances', { 'account': 'developjesta' }, [
         ('.', 'len')
     ]),
+    ('/v1/history/get_keys_accounts', { 'public_keys': ['EOS81fk5ACMV4VTW9j2HGsVQaqnB7nhjCBNSpFax2Sp9PuEzf1eJQ'] }, [
+        ('account_names', 'exists')
+    ]),
 ]
 
 upstream = os.environ['UPSTREAM_API']