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

1 files changed, 108 insertions, 0 deletions

diff --git a/classes/oestats-client.bbclass b/classes/oestats-client.bbclass
new file mode 100644
index 0000000000..da70a77825
--- /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
+}