Fix #1: allow to define shield style

jirutka · Jun 7, 2014 · ecc253d · ecc253d
1 parent 8673db0
commit ecc253d
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -11,14 +11,17 @@ Well, I put together [shields.io](http://shields.io/), [Heroku](https://heroku.c
 Usage
 -----
 
-    https://maven-badges.herokuapp.com/maven-central/{group_id}/{artifact_id}/badge.(svg|png)
+    https://maven-badges.herokuapp.com/maven-central/{group_id}/{artifact_id}/badge.(svg|png)?style={style}
 
-For an example:
+For example:
 
 	[![Maven Central](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser/badge.svg)](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser)
 
 [![Maven Central](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser/badge.svg)](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser)
 
+	[![Maven Central](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser/badge.svg?style=flat)](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser)
+
+[![Maven Central](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser/badge.svg?style=flat)](https://maven-badges.herokuapp.com/maven-central/cz.jirutka.rsql/rsql-parser)
 
 License
 -------

diff --git a/lib/shields.rb b/lib/shields.rb
@@ -7,8 +7,8 @@ class Shields
 
   base_uri 'img.shields.io'
 
-  def self.badge_image(subject, status, color, format='svg')
-    resp = get("/badge/#{encode(subject)}-#{encode(status)}-#{color}.#{format}")
+  def self.badge_image(subject, status, color, format='svg', style='default')
+    resp = get("/badge/#{encode(subject)}-#{encode(status)}-#{color}.#{format}?style=#{style}")
 
     raise HTTParty::ResponseError.new(resp) unless resp.code == 200
     resp.body

diff --git a/lib/web.rb b/lib/web.rb
@@ -3,6 +3,7 @@
 require_relative 'shields'
 
 DEFAULT_SUBJECT = 'maven central'
+DEFAULT_STYLE = 'default'
 PROJECT_SITE = 'https://github.com/jirutka/maven-badges'
 
 configure :production do
@@ -30,7 +31,9 @@
     color = :lightgray
   end
 
-  Shields.badge_image(subject, version, color, format)
+  style = params[:style] || DEFAULT_STYLE
+
+  Shields.badge_image(subject, version, color, format, style)
 end
 
 # Returns the artifact's last version number in plain text