[feat] remove active user from get_token

We need to update the UI to use the new API.

- Resolves: #8974

2 files changed, 9 insertions, 10 deletions

diff --git a/src/leap/bitmask/core/dispatcher.py b/src/leap/bitmask/core/dispatcher.py
index 519ec9df..1e364ec1 100644
--- a/src/leap/bitmask/core/dispatcher.py
+++ b/src/leap/bitmask/core/dispatcher.py
@@ -264,8 +264,12 @@ class MailCmd(SubCommand):
 
     @register_method('dict')
     def do_GET_TOKEN(self, mail, *parts, **kw):
-        d = mail.get_token()
-        return d
+        try:
+            userid = parts[2]
+        except IndexError:
+            raise DispatchError(
+                'wrong number of arguments: expected 1, got none')
+        return mail.get_token(userid)
 
     @register_method('dict')
     def do_MIXNET_STATUS(self, mail, *parts, **kw):
diff --git a/src/leap/bitmask/core/mail_services.py b/src/leap/bitmask/core/mail_services.py
index be84b3cf..043f8487 100644
--- a/src/leap/bitmask/core/mail_services.py
+++ b/src/leap/bitmask/core/mail_services.py
@@ -475,7 +475,6 @@ class StandardMailService(service.MultiService, HookableService):
         self._keymanager_sessions = {}
         self._sendmail_opts = {}
         self._service_tokens = {}
-        self._active_user = None
         self._mixnet_enabled = mixnet_enabled
         super(StandardMailService, self).__init__()
         self.initializeChildrenServices()
@@ -508,7 +507,6 @@ class StandardMailService(service.MultiService, HookableService):
 
         def registerToken(token):
             self._service_tokens[userid] = token
-            self._active_user = userid
             return token
 
         incoming = self.getServiceNamed('incoming_mail')
@@ -612,12 +610,9 @@ class StandardMailService(service.MultiService, HookableService):
             status = 'ok'
         return {'status': status}
 
-    def get_token(self):
-        active_user = self._active_user
-        if not active_user:
-            return defer.succeed({'user': None})
-        token = self._service_tokens.get(active_user)
-        return defer.succeed({'user': active_user, 'token': token})
+    def get_token(self, userid):
+        token = self._service_tokens.get(userid)
+        return {'user': userid, 'token': token}
 
     # access to containers