From e4e1c31e7d9742e7286cb43e168c9954de9bc5b3 Mon Sep 17 00:00:00 2001
From: Kali Kaneko <kali@leap.se>
Date: Wed, 18 Nov 2015 19:27:47 -0400
Subject: [feat] bonafide zmq service

---
 src/leap/bonafide/zmq_service.py | 97 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 97 insertions(+)
 create mode 100644 src/leap/bonafide/zmq_service.py

(limited to 'src/leap/bonafide/zmq_service.py')

diff --git a/src/leap/bonafide/zmq_service.py b/src/leap/bonafide/zmq_service.py
new file mode 100644
index 0000000..e33df3d
--- /dev/null
+++ b/src/leap/bonafide/zmq_service.py
@@ -0,0 +1,97 @@
+# -*- coding: utf-8 -*-
+# zmq_service.py
+# Copyright (C) 2015 LEAP
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+"""
+Bonafide ZMQ Service
+"""
+from leap.bonafide import config
+from leap.bonafide.service import BonafideService, COMMANDS
+
+from txzmq import ZmqEndpoint, ZmqFactory, ZmqREPConnection
+
+from twisted.python import log
+
+
+class BonafideZmqREPConnection(ZmqREPConnection):
+
+    def initialize(self):
+        self._service = BonafideService()
+
+    def do_greet(self):
+        print "[+] Bonafide service running..."
+
+    def do_bye(self):
+        print "[+] Bonafide service stopped. Have a nice day."
+        reactor.stop()
+
+    def gotMessage(self, msgId, *parts):
+        def defer_reply(response):
+            reactor.callLater(0, self.reply, msgId, str(response))
+
+        def log_err(failure):
+            log.err(failure)
+            print "FAILURE", failure
+            defer_reply("ERROR: %r" % failure)
+
+        cmd = parts[0]
+
+        if cmd == "shutdown":
+            defer_reply('ok, shutting down')
+            reactor.callLater(1, self.do_bye)
+
+        if cmd not in COMMANDS:
+            response = 'INVALID COMMAND'
+            defer_reply(response)
+
+        elif cmd == 'signup':
+            username, password = parts[1], parts[2]
+            d = self._service.do_signup(username, password)
+            d.addCallback(lambda response: defer_reply(
+                'REGISTERED -> %s' % response))
+            d.addErrback(log_err)
+
+        elif cmd == 'authenticate':
+            username, password = parts[1], parts[2]
+            d = self._service.do_authenticate(username, password)
+            d.addCallback(lambda response: defer_reply(
+                'TOKEN -> %s' % response))
+            d.addErrback(log_err)
+
+        elif cmd == 'logout':
+            username, password = parts[1], parts[2]
+            d = self._service.do_logout(username, password)
+            d.addCallback(lambda response: defer_reply(
+                'LOGOUT -> ok'))
+            d.addErrback(log_err)
+
+        elif cmd == 'stats':
+            response = self._service.do_stats()
+            defer_reply(response)
+
+
+def get_zmq_connection():
+    zf = ZmqFactory()
+    e = ZmqEndpoint("bind", config.ENDPOINT)
+    return BonafideZmqREPConnection(zf, e)
+
+
+if __name__ == "__main__":
+    from twisted.internet import reactor
+
+    s = get_zmq_connection()
+    reactor.callWhenRunning(s.initialize)
+    reactor.callWhenRunning(s.do_greet)
+    reactor.run()
-- 
cgit v1.2.3