Added -j option to send jabber update messages.

2 files changed, 41 insertions, 1 deletions

diff --git a/run/experiment.py b/run/experiment.py
index ecb0241..03d6ab6 100644
--- a/run/experiment.py
+++ b/run/experiment.py
@@ -52,7 +52,7 @@ class Experiment(object):
             self.log("Enabling sched_trace")
             self.tracers.append( SchedTracer(working_dir) )
         if LinuxTracer.enabled():
-            self.log("Enabling trace-cmd / ftrace / kernelshark")
+            self.log("Enabling trace-cmd")
             self.tracers.append( LinuxTracer(working_dir) )
         if LogTracer.enabled():
             self.log("Enabling logging")
diff --git a/run/jabber.py b/run/jabber.py
new file mode 100644
index 0000000..c9dd097
--- /dev/null
+++ b/run/jabber.py
@@ -0,0 +1,40 @@
+#!/usr/bin/env python
+import os
+import sys
+import xmpp
+
+class Jabber(object):
+    def __init__(self, target):
+        self.target = target
+        self.client = self.__create_client()
+
+    def __create_client(self):
+        params = {}
+        conf_file = os.environ['HOME'] + '/.xsend'
+
+        if os.access(conf_file, os.R_OK):
+            for line in open(conf_file, 'r').readlines():
+                key, val = line.strip().split('=', 1)
+                params[key.lower()] = val
+
+        for field in ['login', 'password']:
+            if field not in params.keys():
+                with open(conf_file, 'wc') as f:
+                    f.write('#LOGIN=romeo@montague.net\n#PASSWORD=juliet\n')
+                raise IOError("Please ensure the ~/.xsend file has valid "+\
+                              "credentials for sending messages.")
+
+        jid = xmpp.protocol.JID(params['login'])
+        client = xmpp.Client(jid.getDomain(), debug=[])
+
+        client.connect(server=(jid.getDomain(), 5223))
+        client.auth(jid.getNode(), params['password'])
+
+        sys.stderr.write(("Jabber client loaded. Add '%s' to your friends " + \
+                          "or Jabber messages will not send") % params['login'])
+
+        return client
+
+    def send(self, text):
+        message = xmpp.protocol.Message(to=self.target, body=text, typ='chat')
+        self.client.send(message)