Add token auth infrastructure.

1 files changed, 12 insertions, 0 deletions

diff --git a/src/leap/soledad/backends/leap_backend.py b/src/leap/soledad/backends/leap_backend.py
index 05c27bc1..51c471eb 100644
--- a/src/leap/soledad/backends/leap_backend.py
+++ b/src/leap/soledad/backends/leap_backend.py
@@ -394,3 +394,15 @@ class LeapSyncTarget(HTTPSyncTarget):
         res = self._parse_sync_stream(data, return_doc_cb, ensure_callback)
         data = None
         return res['new_generation'], res['new_transaction_id']
+
+    def set_token_credentials(self, address, token):
+        self._creds = {'token': (address, token)}
+
+    def _sign_request(self, method, url_query, params):
+        if 'token' in self._creds:
+            address, token = self._creds['token']
+            auth = '%s:%s' % (address, token)
+            return [('Authorization', 'Token %s' % auth.encode('base64'))]
+        else:
+            return HTTPSyncTarget._sign_request(
+                self, method, url_query, params)