From b7be635aa32b3ababef341b274ac8da4c8c10fc0 Mon Sep 17 00:00:00 2001
From: Bob Mottram <bob@freedombone.net>
Date: Tue, 11 Aug 2020 17:18:22 +0000
Subject: [PATCH] Return devices for a handle

---
 daemon.py | 75 +++++++++++++++++++++++++++++++++++++++++++++++++++++--
 tests.py  |  2 +-
 2 files changed, 74 insertions(+), 3 deletions(-)

diff --git a/daemon.py b/daemon.py
index fdc7c2ed..736d57f5 100644
--- a/daemon.py
+++ b/daemon.py
@@ -5781,6 +5781,49 @@ class PubServer(BaseHTTPRequestHandler):
                                             postBytes, boundary)
         return pageNumber
 
+    def _cryptoAPIreadHandle(self):
+        """Reads handle
+        """
+        messageBytes = None
+        maxDeviceIdLength = 2048
+        length = int(self.headers['Content-length'])
+        if length >= maxDeviceIdLength:
+            print('WARN: handle post to crypto API is too long ' +
+                  str(length) + ' bytes')
+            return {}
+        try:
+            messageBytes = self.rfile.read(length)
+        except SocketError as e:
+            if e.errno == errno.ECONNRESET:
+                print('WARN: handle POST messageBytes ' +
+                      'connection reset by peer')
+            else:
+                print('WARN: handle POST messageBytes socket error')
+            return {}
+        except ValueError as e:
+            print('ERROR: handle POST messageBytes rfile.read failed')
+            print(e)
+            return {}
+
+        lenMessage = len(messageBytes)
+        if lenMessage > 2048:
+            print('WARN: handle post to crypto API is too long ' +
+                  str(lenMessage) + ' bytes')
+            return {}
+
+        handle = messageBytes.decode("utf-8")
+        if not handle:
+            return None
+        if '@' not in handle:
+            return None
+        if '[' in handle:
+            return json.loads(messageBytes)
+        if handle.startswith('@'):
+            handle = handle[1:]
+        if '@' not in handle:
+            return None
+        return handle.strip()
+
     def _cryptoAPIreadJson(self) -> {}:
         """Obtains json from POST to the crypto API
         """
@@ -5813,6 +5856,34 @@ class PubServer(BaseHTTPRequestHandler):
 
         return json.loads(messageBytes)
 
+    def _cryptoAPIQuery(self, callingDomain: str) -> bool:
+        handle = self._cryptoAPIreadHandle()
+        if not handle:
+            return False
+        if isinstance(handle, str):
+            personDir = self.server.baseDir + '/accounts/' + handle
+            if not os.path.isdir(personDir + '/devices'):
+                return False
+            devicesList = []
+            for subdir, dirs, files in os.walk(personDir + '/devices'):
+                for f in files:
+                    deviceFilename = os.path.join(personDir + '/devices', f)
+                    if not os.path.isfile(deviceFilename):
+                        continue
+                    contentJson = loadJson(deviceFilename)
+                    if contentJson:
+                        devicesList.append(contentJson)
+            # return the list of devices for this handle
+            msg = \
+                json.dumps(devicesList,
+                           ensure_ascii=False).encode('utf-8')
+            self._set_headers('application/json',
+                              len(msg),
+                              None, callingDomain)
+            self._write(msg)
+            return True
+        return False
+
     def _cryptoAPI(self, path: str, authorized: bool) -> None:
         """POST or GET with the crypto API
         """
@@ -5862,8 +5933,8 @@ class PubServer(BaseHTTPRequestHandler):
                 return
             self._400()
         elif path.startswith('/api/v1/crypto/keys/query'):
-            # TODO
-            self._200()
+            if not self._cryptoAPIQuery():
+                self._400()
         elif path.startswith('/api/v1/crypto/keys/claim'):
             # TODO
             self._200()
diff --git a/tests.py b/tests.py
index 70da4a6a..947d9d98 100644
--- a/tests.py
+++ b/tests.py
@@ -1685,7 +1685,7 @@ def testWebLinks():
         'ＴｅｍｐｌａｔｅＡｔｔａｃｈｍｅｎｔＲｉｃｈＰｏｐｏｕｔ'
 
     exampleText = \
-        '<p>Test1 test2 #YetAnotherExcessivelyLongwindedAndBoringHashtag<\p>'
+        '<p>Test1 test2 #YetAnotherExcessivelyLongwindedAndBoringHashtag</p>'
     resultText = removeLongWords(addWebLinks(exampleText), 40, [])
     assert(resultText ==
            '<p>Test1 test2 '