Initial Commit

Author: zikphil

.gitignore

@@ -0,0 +1,2 @@
+*.pyc
+.idea

README.md

@@ -0,0 +1,14 @@
+# chatkit-server-python
+
+The Unofficial Python server SDK for Pusher Chatkit.
+
+**WORK IN PROGRESS**
+
+## Installation
+
+```sh
+$ python3 setup.py install
+```
+
+## Credits
+This work is sponsored by [LedgerX](https://ledgerx.com)

pusher_chatkit/__init__.py

@@ -0,0 +1,27 @@
+import requests
+import json
+
+
+from pusher_chatkit.client import process_response
+
+
+class RequestsBackend(object):
+
+    def __init__(self):
+        self.http = requests
+        self.session = requests.Session()
+
+    def process_request(self, method, endpoint, body=None, token=None):
+        headers = {'Content-Type': 'application/json'}
+
+        if token:
+            headers['Authorization'] = 'Bearer {}'.format(token['token'])
+
+        resp = self.session.request(
+            method,
+            endpoint,
+            headers=headers,
+            data=json.dumps(body),
+            timeout=30)
+
+        return process_response(resp.status_code, resp.text)

pusher_chatkit/backends/Requests.py

@@ -0,0 +1,42 @@
+import json
+import tornado
+import tornado.httpclient
+
+from tornado.concurrent import Future
+from pusher_chatkit.client import process_response
+
+
+class TornadoBackend(object):
+
+    def __init__(self):
+        self.http = tornado.httpclient.AsyncHTTPClient()
+
+    def process_request(self, method, endpoint, body=None, token=None):
+        data = body
+        headers = {'Content-Type': 'application/json'}
+        future = Future()
+
+        headers['Authorization'] = 'Bearer {}'.format(token['token'])
+
+        def process_response_future(response):
+            if response.exception() is not None:
+                future.set_exception(response.exception())
+
+            else:
+                result = response.result()
+                code = result.code
+                body = (result.body or b'').decode('utf8')
+                future.set_result(process_response(code, body))
+
+        request = tornado.httpclient.HTTPRequest(
+            endpoint,
+            method=method,
+            body=json.dumps(data),
+            headers=headers,
+            request_timeout=30)
+
+        response_future = self.http.fetch(request, raise_error=False)
+        response_future.add_done_callback(process_response_future)
+
+        return future
+

pusher_chatkit/backends/Tornado.py

@@ -0,0 +1,2 @@
+from .Requests import RequestsBackend
+from .Tornado import TornadoBackend

pusher_chatkit/backends/__init__.py

@@ -0,0 +1,90 @@
+import json
+
+from pusher_chatkit.exceptions import PusherBadAuth, PusherBadRequest, PusherBadStatus, PusherForbidden
+from urllib.parse import urlencode, quote_plus
+
+
+class PusherChatKitClient(object):
+
+    def __init__(self, backend, instance_locator):
+        self.http = backend()
+        self.instance_locator = instance_locator.split(':')
+        self.scheme = 'https'
+        self.host = self.instance_locator[1] + '.pusherplatform.io'
+        self.instance_id = self.instance_locator[2]
+        self.services = {
+            'api': {
+                'service_name': 'chatkit',
+                'service_version': 'v2'
+            },
+            'authorizer': {
+                'service_name': 'chatkit_authorizer',
+                'service_version': 'v2'
+            },
+            'cursors': {
+                'service_name': 'chatkit_cursors',
+                'service_version': 'v2'
+            }
+        }
+
+    def build_endpoint(self, service, api_endpoint, query):
+        service_path_fragment = self.services[service]['service_name'] + '/' + self.services[service]['service_version']
+        full_path = '{}://{}/services/{}/{}{}'.format(
+            self.scheme,
+            self.host,
+            service_path_fragment,
+            self.instance_id,
+            api_endpoint
+        )
+        query = '?' + urlencode(query, quote_via=quote_plus) if query else ""
+
+        return full_path + query
+
+    def get(self, service, endpoint, query=None, **kwargs):
+        return self.http.process_request(
+            'GET',
+            self.build_endpoint(service, endpoint, query),
+            kwargs.get('body', None),
+            kwargs.get('token', None),
+        )
+
+    def put(self, service, endpoint, query=None, **kwargs):
+        return self.http.process_request(
+            'PUT',
+            self.build_endpoint(service, endpoint, query),
+            kwargs.get('body', None),
+            kwargs.get('token', None),
+        )
+
+    def post(self, service, endpoint, query=None, **kwargs):
+        return self.http.process_request(
+            'POST',
+            self.build_endpoint(service, endpoint, query),
+            kwargs.get('body', None),
+            kwargs.get('token', None),
+        )
+
+    def delete(self, service, endpoint, query=None, **kwargs):
+        return self.http.process_request(
+            'DELETE',
+            self.build_endpoint(service, endpoint, query),
+            kwargs.get('body', None),
+            kwargs.get('token', None),
+        )
+
+
+def process_response(status, body):
+    if status >= 200 and status <= 299:
+        return json.loads(body)
+
+    elif status == 400:
+        raise PusherBadRequest(body)
+
+    elif status == 401:
+        raise PusherBadAuth(body)
+
+    elif status == 403:
+        raise PusherForbidden(body)
+
+    else:
+        raise PusherBadStatus("%s: %s" % (status, body))

pusher_chatkit/client.py

@@ -0,0 +1,2 @@
+GLOBAL_SCOPE = 'global'
+ROOM_SCOPE = 'room'

pusher_chatkit/constants.py

@@ -0,0 +1,14 @@
+class PusherBadRequest(Exception):
+    pass
+
+
+class PusherBadAuth(Exception):
+    pass
+
+
+class PusherForbidden(Exception):
+    pass
+
+
+class PusherBadStatus(Exception):
+    pass