[refactor] add state listener to signal state transition to UI

2 files changed, 15 insertions, 3 deletions

diff --git a/src/leap/bitmask/vpn/management.py b/src/leap/bitmask/vpn/management.py
index 4aa0e00..e692dc0 100644
--- a/src/leap/bitmask/vpn/management.py
+++ b/src/leap/bitmask/vpn/management.py
@@ -62,13 +62,16 @@ class ManagementProtocol(LineReceiver):
         """
         self._state_listeners.add(listener)
 
+    # TODO -- this needs to be exposed by the API
+    # The UI needs this feature.
+
     def getStateHistory(self):
         return self._statelog
 
     def lineReceived(self, line):
         if self.verbose:
             if int(self.verbose) > 1:
-                print line 
+                print line
             elif line.startswith('>LOG'):
                 print line
 
@@ -152,7 +155,6 @@ class ManagementProtocol(LineReceiver):
             elif len(data) == 8:
                 ts, state = data[:2]
         except Exception as exc:
-            print "ERROR", exc
             log.error('Failure parsing data: %s' % exc)
 
         if state != self.state:
diff --git a/src/leap/bitmask/vpn/process.py b/src/leap/bitmask/vpn/process.py
index ecbc150..19115d6 100644
--- a/src/leap/bitmask/vpn/process.py
+++ b/src/leap/bitmask/vpn/process.py
@@ -33,14 +33,21 @@ from twisted.logger import Logger
 
 from leap.bitmask.vpn.utils import get_vpn_launcher
 from leap.bitmask.vpn.management import ManagementProtocol
-
 from leap.bitmask.vpn.launchers import darwin
 from leap.bitmask.vpn.constants import IS_MAC, IS_LINUX
 
+from leap.common.events import catalog, emit_async
+
 
 OPENVPN_VERBOSITY = 4
 
 
+class VPNStateListener(object):
+
+    def change_state(self, state):
+        emit_async(catalog.VPN_STATUS_CHANGED)
+
+
 class _VPNProcess(protocol.ProcessProtocol):
 
     """
@@ -99,6 +106,9 @@ class _VPNProcess(protocol.ProcessProtocol):
     @defer.inlineCallbacks
     def _got_management_protocol(self, proto):
         self.proto = proto
+        listener = VPNStateListener()
+        proto.addStateListener(listener)
+
         try:
             yield proto.logOn()
             yield proto.getVersion()