From 52b10975e9ae4462aae52641de26d2ea198657e4 Mon Sep 17 00:00:00 2001
From: Christophe Simonis <chs@odoo.com>
Date: Thu, 12 Jun 2014 11:08:31 +0200
Subject: [PATCH] [IMP] runbot: badge is now directly rendered via a qweb
 template instead of depending on shields.io service

---
 runbot/runbot.py  | 76 +++++++++++++++++++++++++++++++----------------
 runbot/runbot.xml | 41 +++++++++++++++++++++++++
 2 files changed, 91 insertions(+), 26 deletions(-)

diff --git a/runbot/runbot.py b/runbot/runbot.py
index 533c2f76..dea8bbca 100644
--- a/runbot/runbot.py
+++ b/runbot/runbot.py
@@ -3,6 +3,7 @@
 import datetime
 import fcntl
 import glob
+import hashlib
 import logging
 import os
 import re
@@ -12,6 +13,8 @@ import signal
 import simplejson
 import subprocess
 import time
+import Tkinter
+import tkFont
 
 import dateutil.parser
 import requests
@@ -897,9 +900,11 @@ class RunbotController(http.Controller):
                     _logger.exception("github error while adding label %s" % label_name)
         return werkzeug.utils.redirect('/runbot/repo/%s' % build.repo_id.id)
 
-    @http.route('/runbot/badge/<model("runbot.repo"):repo>/<branch>.<any(svg,png):ext>',
-                type="http", auth="public", methods=['GET', 'HEAD'])
-    def badge(self, repo, branch, ext):
+    @http.route([
+        '/runbot/badge/<model("runbot.repo"):repo>/<branch>.svg',
+        '/runbot/badge/<any(default,flat):theme>/<model("runbot.repo"):repo>/<branch>.svg',
+    ], type="http", auth="public", methods=['GET', 'HEAD'])
+    def badge(self, repo, branch, theme='default'):
 
         domain = [('repo_id', '=', repo.id),
                   ('branch_id.branch_name', '=', branch),
@@ -907,38 +912,57 @@ class RunbotController(http.Controller):
                   ('state', 'in', ['testing', 'running', 'done']),
                   ]
 
-        builds = request.registry['runbot.build'].search_read(request.cr, request.uid,
-                                                              domain, ['state', 'result'],
-                                                              order='id desc', limit=1)
+        last_update = '__last_update'
+        builds = request.registry['runbot.build'].search_read(
+            request.cr, request.uid,
+            domain, ['state', 'result', 'job_age', last_update],
+            order='id desc', limit=1)
 
         if not builds:
             return request.not_found()
+
         build = builds[0]
+        etag = request.httprequest.headers.get('If-None-Match')
+        retag = hashlib.md5(build[last_update]).hexdigest()
+
+        if etag == retag:
+            return werkzeug.wrappers.Response(status=304)
 
-        badge = {
-            'name': branch.replace('-', '--'),
-            'format': ext,
-        }
         if build['state'] == 'testing':
-            badge['status'] = 'testing'
+            state = 'testing'
+            cache_factor = 1
         else:
-            badge['status'] = 'success' if build['result'] == 'ok' else 'failed'
+            state = 'success' if build['result'] == 'ok' else 'failed'
+            cache_factor = 2
 
-        badge['color'] = {
-            'testing': 'yellow',
-            'success': 'brightgreen',
-            'failed': 'red',
-            # 'warning': 'orange',
-        }[badge['status']]
+        # from https://github.com/badges/shields/blob/master/colorscheme.json
+        color = {
+            'testing': "#dfb317",
+            'success': "#4c1",
+            'failed': "#e05d44",
+            'warning': "#fe7d37",
+        }[state]
 
-        url = 'http://img.shields.io/badge/{name}-{status}-{color}.{format}'.format(**badge)
-        image = requests.get(url)
-        content = image.content
-        headers = werkzeug.datastructures.Headers()
-        headers.extend(image.headers.items())
-        headers['Content-Length'] = len(content)
-        headers.remove('content-encoding')
-        return werkzeug.wrappers.Response(content, status=image.status_code, headers=headers.items())
+        text_width = tkFont.Font(Tkinter.Tk(), family='Verdana', size=11).measure
+
+        class Text(object):
+            __slot__ = ['text', 'color', 'width']
+            def __init__(self, text, color):
+                self.text = text
+                self.color = color
+                self.width = text_width(text) + 10
+
+        data = {
+            'left': Text(branch, '#555'),
+            'right': Text(state, color),
+        }
+        five_minutes = 5 * 60
+        headers = [
+            ('Content-Type', 'image/svg+xml'),
+            ('Cache-Control', 'max-age=%d' % (five_minutes * cache_factor,)),
+            ('ETag', retag),
+        ]
+        return request.render("runbot.badge_" + theme, data, headers=headers)
 
 
 LABELS = {
diff --git a/runbot/runbot.xml b/runbot/runbot.xml
index cfd84385..cb3c14c3 100644
--- a/runbot/runbot.xml
+++ b/runbot/runbot.xml
@@ -534,6 +534,47 @@ http {
 }
     </template>
 
+    <template id="runbot.badge_default">
+      <!-- from https://github.com/badges/shields/tree/master/templates -->
+      <svg xmlns="http://www.w3.org/2000/svg" t-attf-width="{{left.width + right.width}}" height="18">
+        <linearGradient id="smooth" x2="0" y2="100%">
+          <stop offset="0"  stop-color="#fff" stop-opacity=".7"/>
+          <stop offset=".1" stop-color="#aaa" stop-opacity=".1"/>
+          <stop offset=".9" stop-color="#000" stop-opacity=".3"/>
+          <stop offset="1"  stop-color="#000" stop-opacity=".5"/>
+        </linearGradient>
+        <rect rx="4" t-attf-width="{{ left.width + right.width }}" height="18" t-att-fill="left.color"/>
+        <rect rx="4" t-att-x="left.width" t-att-width="right.width" height="18" t-att-fill="right.color"/>
+        <rect t-att-x="left.width" width="4" height="18" t-att-fill="right.color"/>
+        <rect rx="4" t-attf-width="{{ left.width + right.width }}" height="18" fill="url(#smooth)"/>
+        <g fill="#fff" text-anchor="middle" font-family="DejaVu Sans,Verdana,Geneva,sans-serif" font-size="11">
+          <text t-attf-x="{{left.width/2+1}}" y="13" fill="#010101" fill-opacity=".3"><t t-esc="left.text"/></text>
+          <text t-attf-x="{{left.width/2+1}}" y="12"><t t-esc="left.text"/></text>
+          <text t-attf-x="{{left.width+right.width/2-1}}" y="13" fill="#010101" fill-opacity=".3"><t t-esc="right.text"/></text>
+          <text t-attf-x="{{left.width+right.width/2-1}}" y="12"><t t-esc="right.text"/></text>
+        </g>
+      </svg>
+    </template>
+    <template id="runbot.badge_flat">
+      <!-- from https://github.com/badges/shields/tree/master/templates -->
+      <svg xmlns="http://www.w3.org/2000/svg" t-attf-width="{{left.width + right.width}}" height="20">
+        <linearGradient id="smooth" x2="0" y2="100%">
+          <stop offset="0"  stop-color="#fff" stop-opacity=".1"/>
+          <stop offset=".1" stop-color="#fff" stop-opacity=".1"/>
+          <stop offset=".9" stop-color="#fff" stop-opacity=".1"/>
+          <stop offset="1"  stop-color="#fff" stop-opacity=".1"/>
+        </linearGradient>
+        <rect t-attf-width="{{ left.width + right.width }}" height="20" t-att-fill="left.color"/>
+        <rect t-att-x="left.width" t-att-width="right.width" height="20" t-att-fill="right.color"/>
+        <rect t-att-x="left.width" width="4" height="20" t-att-fill="right.color"/>
+        <rect t-attf-width="{{ left.width + right.width }}" height="20" fill="url(#smooth)"/>
+        <g fill="#fff" text-anchor="middle" font-family="DejaVu Sans,Verdana,Geneva,sans-serif" font-size="11">
+          <text t-attf-x="{{left.width/2+1}}" y="13"><t t-esc="left.text"/></text>
+          <text t-attf-x="{{left.width+right.width/2-1}}" y="13"><t t-esc="right.text"/></text>
+        </g>
+      </svg>
+    </template>
+
     <!-- Cron -->
     <record model="ir.cron" id="repo_cron">
         <field name='name'>Runbot Cron</field>