From 38cf7a60b58dd7660b862be8069da65006f2b180 Mon Sep 17 00:00:00 2001
From: Thaddee Tyl <thaddee.tyl@gmail.com>
Date: Sun, 19 Feb 2017 22:59:41 +0100
Subject: [PATCH] Use path.extname() to fetch template file name extension

---
 badge.js | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/badge.js b/badge.js
index e47fadc..956bde4 100644
--- a/badge.js
+++ b/badge.js
@@ -12,7 +12,7 @@ templateFiles.forEach(function(filename) {
   if (filename[0] === '.') { return; }
   var templateData = fs.readFileSync(
     path.join(__dirname, 'templates', filename)).toString();
-  var extension = filename.split('.').pop();
+  var extension = path.extname(filename).slice(1);
   var style = filename.slice(0, -(('-template.' + extension).length));
   templates[style + '-' + extension] = dot.template(templateData);
 });