From 86e1258ebb21ade53404f107553905f222ff490f Mon Sep 17 00:00:00 2001
From: Alex Yatskov <alex@foosoft.net>
Date: Sun, 18 Jul 2021 20:19:37 -0700
Subject: [PATCH] Refactor

---
 plugin/api/gui.py |  3 ++-
 plugin/host.py    | 37 ++++++++++++++++++++++++++++---------
 plugin/web.py     | 22 ++++------------------
 3 files changed, 34 insertions(+), 28 deletions(-)

diff --git a/plugin/api/gui.py b/plugin/api/gui.py
index 615c05e..b7a8153 100644
--- a/plugin/api/gui.py
+++ b/plugin/api/gui.py
@@ -17,8 +17,9 @@ from . import util
 
 
 @util.api
-def guiBrowse(self, query=None):
+def guiBrowse(query=None):
     print(query)
+    return 'hi'
     # browser = aqt.dialogs.open('Browser', self.window())
     # browser.activateWindow()
     #
diff --git a/plugin/host.py b/plugin/host.py
index c8ebd56..8632a9f 100644
--- a/plugin/host.py
+++ b/plugin/host.py
@@ -13,6 +13,7 @@
 # You should have received a copy of the GNU General Public License
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 
+import inspect
 import PyQt5
 
 from . import web
@@ -20,9 +21,10 @@ from . import web
 
 class ApiHost:
     def __init__(self, origins, key, address, port):
+        self.origins = origins
         self.key = key
         self.modules = []
-        self.server = web.WebServer(self.handler, origins)
+        self.server = web.WebServer(self.handler)
         self.server.bindAndListen(address, port)
 
 
@@ -39,19 +41,36 @@ class ApiHost:
     def handler(self, call):
         action = call.get('action')
         params = call.get('params', {})
-        allowed = call.get('allowed', False)
-        key = call.get('key')
 
         try:
-            if key != self.key and action != 'requestPermission':
-                raise Exception('valid api key must be provided')
-
             for module in self.modules:
-                for methodName, methodInstance in inspect.getmembers(module, predicate=inspect.ismethod):
-                    if methodName == action and getattr(methodInstance, 'api', False):
-                        return {'error': None, 'result': methodInstance(**params)}
+                for funcName, funcInstance in inspect.getmembers(module, predicate=inspect.isfunction):
+                    if funcName == action and getattr(funcInstance, 'api', False):
+                        return {'error': None, 'result': funcInstance(**params)}
             else:
                 raise Exception('unsupported action')
 
         except Exception as e:
             return {'error': str(e), 'result': None}
+
+# if '*' in self.origins:
+#     origin = '*'
+#     allowed = True
+# else:
+#     origin = request.headers.get('origin', 'http://127.0.0.1:')
+#     for prefix in self.origins:
+#         if origin.startswith(prefix):
+#             allowed = True
+#             break
+#
+# try:
+#     if request.body:
+#         call = json.loads(request.body)
+#         call['allowed'] = allowed
+#         call['origin'] = origin
+#         body = json.dumps(self.handler(call))
+#     else:
+#         body = 'AnkiConnect'
+# except Exception as e:
+#     body = str(e)
+
diff --git a/plugin/web.py b/plugin/web.py
index 76397a3..a2999d9 100644
--- a/plugin/web.py
+++ b/plugin/web.py
@@ -116,9 +116,8 @@ class WebClient:
 #
 
 class WebServer:
-    def __init__(self, handler, origins):
+    def __init__(self, handler):
         self.handler = handler
-        self.origins = origins
         self.clients = []
         self.socket = None
 
@@ -153,22 +152,9 @@ class WebServer:
 
 
     def handlerWrapper(self, request):
-        if '*' in self.origins:
-            origin = '*'
-            allowed = True
-        else:
-            origin = request.headers.get('origin', 'http://127.0.0.1:')
-            for prefix in self.origins:
-                if origin.startswith(prefix):
-                    allowed = True
-                    break
-
         try:
             if request.body:
-                call = json.loads(request.body)
-                call['allowed'] = allowed
-                call['origin'] = origin
-                body = json.dumps(self.handler(call))
+                body = json.dumps(self.handler(json.loads(request.body)))
             else:
                 body = 'AnkiConnect'
         except Exception as e:
@@ -177,8 +163,8 @@ class WebServer:
         headers = [
             ['HTTP/1.1 200 OK', None],
             ['Content-Type', 'text/json'],
-            ['Access-Control-Allow-Origin', origin],
-            ['Access-Control-Allow-Headers', '*'],
+            # ['Access-Control-Allow-Origin', origin],
+            # ['Access-Control-Allow-Headers', '*'],
             ['Content-Length', len(body.encode('utf-8'))]
         ]