Wall

.gitignore

@@ -36,5 +36,6 @@ nosetests.xml
 .mr.developer.cfg
 .project
 .pydevproject
+.idea
 
 tests/mibs/.index

requirements.txt

@@ -4,7 +4,7 @@ MySQL-python==1.2.5
 PyYAML==3.10
 SQLAlchemy==0.9.1
 backports.ssl-match-hostname==3.4.0.2
-distribute==0.6.49
+distribute==0.6.35
 expvar==0.0.2
 oid-translate==0.2.2
 pyasn1==0.1.7
@@ -13,3 +13,4 @@ pysnmp==4.2.5
 pytz==2014.1
 tornado==3.2
 wsgiref==0.1.2
+tornado-redis==2.4.18

trapdoor/handlers.py

@@ -1,10 +1,22 @@
 from datetime import datetime
 import json
-from sqlalchemy import desc, func, or_
+import logging
+import string
+import random
+import sys
+from threading import Timer
+
+
+import tornado
+from sqlalchemy import desc, or_
+
+from tornado import gen
 
 from trapdoor.utils import TrapdoorHandler
 from trapperkeeper.models import Notification, VarBind
 
+import tornadoredis
+
 def filter_query(query, host, oid, severity):
     if host is not None:
         query = query.filter(Notification.host == host)
@@ -74,6 +86,7 @@ def get(self):
             "index.html", traps=traps, now=now, num_active=num_active,
             host=host, oid=oid, severity=severity, offset=offset, limit=limit)
 
+
 class Resolve(TrapdoorHandler):
     def post(self):
         host = self.get_argument("host")
@@ -176,3 +189,37 @@ def get(self):
         traps, num_active = _get_traps(self.db, offset, limit, host, oid, severity)
 
         self.write(json.dumps([trap.to_dict() for trap in traps]))
+
+class MessageHandler(tornado.websocket.WebSocketHandler):
+
+    def __init__(self, *args, **kwargs):
+        super(MessageHandler, self).__init__(*args, **kwargs)
+        self.listen()
+
+    @tornado.gen.engine
+    def listen(self):
+        self.client = tornadoredis.Client()
+        self.client.connect()
+        yield tornado.gen.Task(self.client.subscribe, 'root')
+        self.client.listen(self.on_message)
+
+    def on_message(self, msg):
+        logging.info("received message %s", msg)
+        if msg.kind == 'message':
+            self.write_message(str(msg.body))
+        if msg.kind == 'disconnect':
+            # Do not try to reconnect, just send a message back
+            # to the client and close the client connection
+            self.write_message('The connection terminated '
+                               'due to a Redis server error.')
+            self.close()
+
+    def on_close(self):
+        """
+        Callback when the socket is closed. Frees up resource related to this socket.
+        """
+        logging.info("socket closed, cleaning up resources now")
+        if self.client.subscribed:
+            self.client.unsubscribe('root')
+            self.client.disconnect()
+

trapdoor/routes.py

@@ -1,11 +1,14 @@
 
-from trapdoor import handlers
+from trapdoor import handlers,utils
 
 HANDLERS = [
     (r"/", handlers.Index),
     (r"/resolve/?", handlers.Resolve),
     (r"/resolve_all/?", handlers.ResolveAll),
 
+    #websocket
+    (r"/message_wall", handlers.MessageHandler),
+
     # API
     (r"/api/varbinds/(?P<notification_id>\d+)", handlers.ApiVarBinds),
     (r"/api/activetraps/?", handlers.ApiActiveTraps),

trapdoor/static/js/app.js

@@ -0,0 +1,37 @@
+/**
+ * Initiate global websocket object.
+ * @todo: Add user cookie for authentication.
+ */
+var host = location.origin.replace(/^http/, 'ws')
+var ws = new WebSocket(host +"/message_wall" );
+
+(function($){
+  $.extend({
+    playSound: function(){
+      return $("<embed src='"+arguments[0]+".mp3' hidden='true' autostart='true' loop='false' class='playSound'>" + "<audio autoplay='autoplay' style='display:none;' controls='controls'><source src='"+arguments[0]+".mp3' /><source src='"+arguments[0]+".ogg' /></audio>").appendTo('body');
+    }
+  });
+})(jQuery);
+
+$(document).ready(function() {
+    if (!window.console) window.console = {};
+    if (!window.console.log) window.console.log = function() {};
+
+    // Websocket callbacks:
+    ws.onopen = function() {
+        console.log("Connected...");
+    };
+    ws.onmessage = function (event) {
+        data = JSON.parse(event.data);
+        if (data.severity == "warning" || data.severity == "critical") {
+            setTimeout($.playSound('../static/mp3/msg'), 650);
+        }
+        console.log("New Message", data);
+        $('#table').load(document.URL + '# .table');
+    };
+    ws.onclose = function() {
+        // @todo: Implement reconnect.
+        console.log("Closed!");
+        location.reload();
+    };
+});