From a42303bbe8eb81395038f5f16af4eeff5c3504f2 Mon Sep 17 00:00:00 2001
From: Eric Holscher <eric@ericholscher.com>
Date: Mon, 14 Mar 2016 14:11:12 -0700
Subject: [PATCH] WIP: Add basic badge support.

---
 readthedocs/projects/views/public.py   | 55 +++++++++++++++-----------
 readthedocs/templates/badges/badge.svg | 10 +++++
 2 files changed, 42 insertions(+), 23 deletions(-)
 create mode 100644 readthedocs/templates/badges/badge.svg

diff --git a/readthedocs/projects/views/public.py b/readthedocs/projects/views/public.py
index 7f8c9f3b60f..9a4abd09937 100644
--- a/readthedocs/projects/views/public.py
+++ b/readthedocs/projects/views/public.py
@@ -7,6 +7,7 @@
 import mimetypes
 import md5
 
+from django.template import loader
 from django.core.urlresolvers import reverse
 from django.core.cache import cache
 from django.conf import settings
@@ -106,45 +107,53 @@ def get_context_data(self, **kwargs):
         return context
 
 
-def _badge_return(redirect, url):
-    if redirect:
-        return HttpResponseRedirect(url)
+def _badge_return(version, status):
+    template = loader.get_template('badges/badge.svg')
+    if status == 'passing':
+        fill = '#4c1'
+    elif status == 'failing':
+        fill = '#e05d44'
     else:
-        response = requests.get(url)
-        http_response = HttpResponse(response.content,
-                                     content_type="image/svg+xml")
-        http_response['Cache-Control'] = 'no-cache'
-        http_response['Etag'] = md5.new(url)
-        return http_response
+        fill = '#dfb317'
+    content = template.render({
+        'project': version.project.slug,
+        'project_length': len(version.project.slug) * 5,
+        'project_fill': len(version.project.slug) * 5 / 2 + 1,
+        'version': version.slug,
+        'version_length': len(version.slug) * 5,
+        'version_fill': len(version.slug) * 5 / 2 + 1,
+        'status': status,
+        'fill': fill,
+        'full_width': (len(version.project.slug) + len(version.slug)) * 5,
+    })
+    http_response = HttpResponse(content,
+                                 content_type="image/svg+xml")
+    http_response['Cache-Control'] = 'no-cache'
+    http_response['Etag'] = md5.new(content)
+    return http_response
 
 
 @cache_control(no_cache=True)
 def project_badge(request, project_slug, redirect=True):
     """Return a sweet badge for the project"""
     version_slug = request.GET.get('version', LATEST)
-    style = request.GET.get('style', 'flat')
+
+    status = 'passing'
+
     try:
         version = Version.objects.public(request.user).get(
             project__slug=project_slug, slug=version_slug)
     except Version.DoesNotExist:
-        url = (
-            'https://img.shields.io/badge/docs-unknown%20version-yellow.svg?style={style}'
-            .format(style=style))
-        return _badge_return(redirect, url)
+        status = 'unknown'
     version_builds = version.builds.filter(type='html', state='finished').order_by('-date')
     if not version_builds.exists():
-        url = (
-            'https://img.shields.io/badge/docs-no%20builds-yellow.svg?style={style}'
-            .format(style=style))
-        return _badge_return(redirect, url)
+        status = 'unknown'
     last_build = version_builds[0]
     if last_build.success:
-        color = 'brightgreen'
+        status = 'passing'
     else:
-        color = 'red'
-    url = 'https://img.shields.io/badge/docs-%s-%s.svg?style=%s' % (
-        version.slug.replace('-', '--'), color, style)
-    return _badge_return(redirect, url)
+        color = 'failing'
+    return _badge_return(version, status)
 
 
 def project_downloads(request, project_slug):
diff --git a/readthedocs/templates/badges/badge.svg b/readthedocs/templates/badges/badge.svg
new file mode 100644
index 00000000000..c38985bb5e7
--- /dev/null
+++ b/readthedocs/templates/badges/badge.svg
@@ -0,0 +1,10 @@
+<svg xmlns="http://www.w3.org/2000/svg" width="{{ full_width }}" height="20">
+  <g shape-rendering="crispEdges">
+    <rect width="{{ project_length }}" height="20" fill="{{ fill }}"/>
+    <rect x="{{ project_length }}" width="{{ version_length }}" height="20" fill="{{ fill }}"/>
+  </g>
+  <g fill="#fff" text-anchor="middle" font-family="DejaVu Sans,Verdana,Geneva,sans-serif" font-size="11">
+    <text x="{{ project_fill }}" y="14">{{ project }}</text>
+    <text x="{{ version_fill }}" y="14">{{ version }}</text>
+  </g>
+</svg>
\ No newline at end of file