1 files changed, 81 insertions, 0 deletions
diff --git a/pyload/remote/wsbackend/ApiHandler.py b/pyload/remote/wsbackend/ApiHandler.py
new file mode 100644
index 000000000..4685121d4
--- /dev/null
+++ b/pyload/remote/wsbackend/ApiHandler.py
@@ -0,0 +1,81 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+###############################################################################
+#   Copyright(c) 2008-2012 pyLoad Team
+#   http://www.pyload.org
+#
+#   This file is part of pyLoad.
+#   pyLoad is free software: you can redistribute it and/or modify
+#   it under the terms of the GNU Affero General Public License as
+#   published by the Free Software Foundation, either version 3 of the
+#   License, or (at your option) any later version.
+#
+#   Subjected to the terms and conditions in LICENSE
+#
+#   @author: RaNaN
+###############################################################################
+
+from mod_pywebsocket.msgutil import receive_message
+
+from pyload.Api import ExceptionObject
+
+from AbstractHandler import AbstractHandler
+
+class ApiHandler(AbstractHandler):
+    """Provides access to the API.
+
+    Send your request as json encoded string in the following manner:
+    ["function", [*args]] or ["function", {**kwargs}]
+
+    the result will be:
+
+    [code, result]
+
+    Don't forget to login first.
+    Non json request will be ignored.
+    """
+
+    PATH = "/api"
+
+    def transfer_data(self, req):
+        while True:
+            try:
+                line = receive_message(req)
+            except TypeError, e: # connection closed
+                self.log.debug("WS Error: %s" % e)
+                return self.passive_closing_handshake(req)
+
+            self.handle_message(line, req)
+
+    def handle_message(self, msg, req):
+
+        func, args, kwargs = self.handle_call(msg, req)
+        if not func:
+            return # handle_call already sent the result
+
+        if func == 'login':
+            return self.do_login(req, args, kwargs)
+        elif func == 'logout':
+            return self.do_logout(req)
+        else:
+            if not req.api:
+                return self.send_result(req, self.FORBIDDEN, "Forbidden")
+
+            if not self.api.isAuthorized(func, req.api.user):
+                return self.send_result(req, self.UNAUTHORIZED, "Unauthorized")
+
+            try:
+                result = getattr(req.api, func)(*args, **kwargs)
+            except ExceptionObject, e:
+                return self.send_result(req, self.BAD_REQUEST, e)
+            except AttributeError:
+                return self.send_result(req, self.NOT_FOUND, "Not Found")
+            except Exception, e:
+                self.core.print_exc()
+                return self.send_result(req, self.ERROR, str(e))
+
+            # None is invalid json type
+            if result is None: result = True
+
+            return self.send_result(req, self.OK, result)
+\ No newline at end of file