[refactor] reorganize API

so that whitelisting doesn't have to peek into the data.
added more documentation and some tests stubs too.

6 files changed, 82 insertions, 44 deletions

diff --git a/src/leap/bitmask/core/dispatcher.py b/src/leap/bitmask/core/dispatcher.py
index 59003906..b068683d 100644
--- a/src/leap/bitmask/core/dispatcher.py
+++ b/src/leap/bitmask/core/dispatcher.py
@@ -38,6 +38,10 @@ from .api import APICommand, register_method
 logger = Logger()
 
 
+class DispatchError(Exception):
+    pass
+
+
 class SubCommand(object):
 
     __metaclass__ = APICommand
@@ -112,6 +116,8 @@ class UserCmd(SubCommand):
 
     @register_method("{'signup': 'ok', 'user': str}")
     def do_CREATE(self, bonafide, *parts):
+        if len(parts) < 5:
+            raise DispatchError('Not enough parameters passed')
         # params are: [user, create, full_id, password, invite, autoconf]
         user, password, invite = parts[2], parts[3], parts[4]
 
@@ -219,7 +225,6 @@ class WebUICmd(SubCommand):
 
     @register_method('dict')
     def do_STATUS(self, webui, *parts, **kw):
-        print 'webui', webui
         d = webui.do_status()
         return d
 
@@ -310,7 +315,6 @@ class EventsCmd(SubCommand):
         self.waiting.append(d)
         return d
 
-    @register_method("")
     def _callback(self, event, *content):
         payload = (str(event), content)
         if not self.waiting:
@@ -322,15 +326,35 @@ class EventsCmd(SubCommand):
             d.callback(payload)
 
 
+class CoreCmd(SubCommand):
+
+    label = 'core'
+
+    @register_method("{'mem_usage': str}")
+    def do_STATS(self, core, *parts):
+        return core.do_stats()
+
+    @register_method("{version_core': '0.0.0'}")
+    def do_VERSION(self, core, *parts):
+        return core.do_version()
+
+    @register_method("{'mail': 'running'}")
+    def do_STATUS(self, core, *parts):
+        return core.do_status()
+
+    @register_method("{'stop': 'ok'}")
+    def do_STOP(self, core, *parts):
+        return core.do_stop()
+
+
 class CommandDispatcher(object):
 
     __metaclass__ = APICommand
 
-    label = 'core'
-
     def __init__(self, core):
 
         self.core = core
+        self.subcommand_core = CoreCmd()
         self.subcommand_bonafide = BonafideCmd()
         self.subcommand_eip = EIPCmd()
         self.subcommand_mail = MailCmd()
@@ -338,26 +362,10 @@ class CommandDispatcher(object):
         self.subcommand_events = EventsCmd()
         self.subcommand_webui = WebUICmd()
 
-    # XXX --------------------------------------------
-    # TODO move general services to another subclass
-
-    @register_method("{'mem_usage': str}")
-    def do_STATS(self, *parts):
-        return _format_result(self.core.do_stats())
-
-    @register_method("{version_core': '0.0.0'}")
-    def do_VERSION(self, *parts):
-        return _format_result(self.core.do_version())
-
-    @register_method("{'mail': 'running'}")
-    def do_STATUS(self, *parts):
-        return _format_result(self.core.do_status())
-
-    @register_method("{'stop': 'ok'}")
-    def do_STOP(self, *parts):
-        return _format_result(self.core.do_stop())
-
-    # -----------------------------------------------
+    def do_CORE(self, *parts):
+        d = self.subcommand_core.dispatch(self.core, *parts)
+        d.addCallbacks(_format_result, _format_error)
+        return d
 
     def do_BONAFIDE(self, *parts):
         bonafide = self._get_service('bonafide')
@@ -446,7 +454,6 @@ class CommandDispatcher(object):
 
     def dispatch(self, msg):
         cmd = msg[0]
-
         _method = getattr(self, 'do_' + cmd.upper(), None)
 
         if not _method:
diff --git a/src/leap/bitmask/core/dummy.py b/src/leap/bitmask/core/dummy.py
index 455756c4..2037b81f 100644
--- a/src/leap/bitmask/core/dummy.py
+++ b/src/leap/bitmask/core/dummy.py
@@ -22,6 +22,34 @@ import json
 from leap.bitmask.hooks import HookableService
 
 
+class CannedData:
+
+    class backend:
+        status = {
+            'soledad': 'running',
+            'keymanager': 'running',
+            'mail': 'running',
+            'eip': 'stopped',
+            'backend': 'dummy'}
+        version = {'version_core': '0.0.1'}
+        stop = {'stop': 'ok'}
+        stats = {'mem_usage': '01 KB'}
+
+    class bonafide:
+        auth = {
+            u'srp_token': u'deadbeef123456789012345678901234567890123',
+            u'uuid': u'01234567890abcde01234567890abcde'}
+        signup = {
+            'signup': 'ok',
+            'user': 'dummyuser@provider.example.org'}
+        list_users = {
+            'userid': 'testuser',
+            'authenticated': False}
+        logout = {
+            'logout': 'ok'}
+        get_active_user = 'dummyuser@provider.example.org'
+
+
 class BackendCommands(object):
 
     """
@@ -30,23 +58,19 @@ class BackendCommands(object):
 
     def __init__(self, core):
         self.core = core
+        self.canned = CannedData
 
     def do_status(self):
-        return json.dumps(
-            {'soledad': 'running',
-             'keymanager': 'running',
-             'mail': 'running',
-             'eip': 'stopped',
-             'backend': 'dummy'})
+        return json.dumps(self.canned.backend.stats)
 
     def do_version(self):
-        return {'version_core': '0.0.1'}
+        return self.canned.backend.version
 
     def do_stats(self):
-        return {'mem_usage': '01 KB'}
+        return self.canned.backend.stats
 
     def do_stop(self):
-        return {'stop': 'ok'}
+        return self.canned.backend.stop
 
 
 class mail_services(object):
@@ -64,17 +88,19 @@ class mail_services(object):
 class BonafideService(HookableService):
 
     def __init__(self, basedir):
-        pass
+        self.canned = CannedData
 
     def do_authenticate(self, user, password):
-        return {u'srp_token': u'deadbeef123456789012345678901234567890123',
-                u'uuid': u'01234567890abcde01234567890abcde'}
+        return self.canned.bonafide.auth
 
     def do_signup(self, user, password):
-        return {'signup': 'ok', 'user': 'dummyuser@provider.example.org'}
+        return self.canned.bonafide.signup
+
+    def do_list_users(self):
+        return self.canned.bonafide.list_users
 
     def do_logout(self, user):
-        return {'logout': 'ok'}
+        return self.canned.bonafide.logout
 
     def do_get_active_user(self):
-        return 'dummyuser@provider.example.org'
+        return self.canned.bonafide.get_active_user
diff --git a/src/leap/bitmask/core/launcher.py b/src/leap/bitmask/core/launcher.py
index a1c8690f..14d8e607 100644
--- a/src/leap/bitmask/core/launcher.py
+++ b/src/leap/bitmask/core/launcher.py
@@ -45,6 +45,7 @@ def here(module=None):
 
 
 def run_bitmaskd():
+
     # TODO --- configure where to put the logs... (get --logfile, --logdir
     # from bitmaskctl
     for (index, arg) in enumerate(sys.argv):
diff --git a/src/leap/bitmask/core/service.py b/src/leap/bitmask/core/service.py
index 9682c18c..c3e97f72 100644
--- a/src/leap/bitmask/core/service.py
+++ b/src/leap/bitmask/core/service.py
@@ -266,6 +266,7 @@ class BackendCommands(object):
         return {'version_core': __version__}
 
     def do_stats(self):
+        print "DO STATS"
         logger.debug('BitmaskCore Service STATS')
         mem = resource.getrusage(resource.RUSAGE_SELF).ru_maxrss
         return {'mem_usage': '%s MB' % (mem / 1024)}
diff --git a/src/leap/bitmask/core/web/_auth.py b/src/leap/bitmask/core/web/_auth.py
index 6a5e3621..3eb4fa13 100644
--- a/src/leap/bitmask/core/web/_auth.py
+++ b/src/leap/bitmask/core/web/_auth.py
@@ -18,9 +18,6 @@ class WhitelistHTTPAuthSessionWrapper(HTTPAuthSessionWrapper):
     It doesn't apply the enforcement to routes included in a whitelist.
     """
 
-    # TODO extend this to inspect the data -- so that we pass a tuple
-    # with the action
-
     whitelist = (None,)
 
     def __init__(self, *args, **kw):
diff --git a/src/leap/bitmask/core/web/service.py b/src/leap/bitmask/core/web/service.py
index 2437d2d6..77e1c729 100644
--- a/src/leap/bitmask/core/web/service.py
+++ b/src/leap/bitmask/core/web/service.py
@@ -59,7 +59,13 @@ class HTTPDispatcherService(service.Service):
     """
 
     API_WHITELIST = (
-        '/API/bonafide/user',
+        '/API/core/version',
+        '/API/core/stats',
+        '/API/bonafide/user/create',
+        '/API/bonafide/user/authenticate',
+        '/API/bonafide/provider/list',
+        '/API/bonafide/provider/create',
+        '/API/bonafide/provider/read',
     )
 
     def __init__(self, core, port=7070, debug=False, onion=False):