oestats-client.bbclass: initial version of an oe-specific tinderbox tool

diff --git a/classes/oestats-client.bbclass b/classes/oestats-client.bbclass
new file mode 100644 (file)
index 0000000..da70a77
--- /dev/null
+++ b/classes/oestats-client.bbclass
@@ -0,0 +1,108 @@
+# Integration with the oestats build statistics server, see:
+#
+# http://opensource.bolloretelecom.eu/projects/oestats
+#
+# To make use of this class, add to your local.conf:
+#
+# INHERIT += "oestats-client"
+# OESTATS_SERVER = "some.server.org:8000"
+# OESTATS_BUILDER = "some title"
+
+def oestats_revision(dir):
+       import re
+       try:
+               f = file("%s/_MTN/revision" % dir)
+               m = re.search(r"old_revision \[(.*)\]", f.read())
+               return m.group(1)
+       except:
+               return
+
+def oestats_send(server, action, vars = {}):
+       import httplib, urllib
+
+       params = urllib.urlencode(vars)
+       headers = {"Content-type": "application/x-www-form-urlencoded",
+               "Accept": "text/plain"}
+       conn = httplib.HTTPConnection(server)
+       conn.request("POST", action, params, headers)
+       response = conn.getresponse()
+       conn.close()
+       return response
+
+def oestats_start(server, builder, d):
+       import bb
+       import os.path
+
+       # collect information about revisions
+       path_bb = bb.data.getVar('BBPATH', d, 1)
+       for p in (path_bb or "").split(':'):
+               revision = oestats_revision(p)
+               if revision:
+                       break
+
+       # send report
+       response = oestats_send(server, "/builds/start/", {
+               'builder': builder,
+               'revision': revision,
+               'machine': bb.data.getVar( 'MACHINE', d, True ),
+               'distro': bb.data.getVar( 'DISTRO', d, True ),
+       })
+       id = response.read()
+
+       # save the build id
+       bb.note("Stats id: %s" % id)
+       f = file(bb.data.getVar('TMPDIR', d, True)+"/oestats.id", 'w')
+       f.write(id)
+
+def oestats_stop(server, d, status):
+       import bb
+
+       # retrieve build id
+       f = file(bb.data.getVar('TMPDIR',d,True)+"/oestats.id", 'r')
+       id = f.read()
+
+       # send report
+       response = oestats_send(server, "/builds/stop/%s/" % id, {
+               'status': status,
+       })
+
+def oestats_task(server, d, task, status):
+       import bb
+
+       # retrieve build id
+       f = file(bb.data.getVar('TMPDIR',d,True)+"/oestats.id", 'r')
+       id = f.read()
+
+       # send report
+       response = oestats_send(server, "/builds/task/%s/" % id, {
+               'package': bb.data.getVar('PN', d, True),
+               'version': bb.data.getVar('PV', d, True),
+               'revision': bb.data.getVar('PR', d, True),
+               'task': task,
+               'status': status,
+       })
+
+addhandler oestats_eventhandler
+python oestats_eventhandler () {
+       from bb.event import getName
+       import bb
+
+       if e.data is None or getName(e) == "MsgNote":
+               return NotHandled
+
+       server = bb.data.getVar('OESTATS_SERVER', e.data, True)
+       builder = bb.data.getVar('OESTATS_BUILDER', e.data, True)
+       if not server or not builder:
+               return NotHandled
+
+       if getName(e) == 'BuildStarted':
+               oestats_start(server, builder, e.data)
+       elif getName(e) == 'BuildCompleted':
+               oestats_stop(server, e.data, 'Completed')
+       elif getName(e) == 'TaskSucceeded':
+               oestats_task(server, e.data, e.task, 'Succeeded')
+       elif getName(e) == 'TaskFailed':
+               oestats_task(server, e.data, e.task, 'Failed')
+
+       return NotHandled
+}