From 3a6763a25c399eb07c641d8c3136f57649cd19e6 Mon Sep 17 00:00:00 2001
From: Glavin Wiechert <glavin.wiechert@gmail.com>
Date: Thu, 25 Feb 2016 23:30:34 -0400
Subject: [PATCH] See #828, #757, #731, #750. Fix Riot.js support,

add support to Pretty diff beautifier
---
 .../riotjs/expected/test.tag                  | 24 +++++++++++++------
 src/beautifiers/prettydiff.coffee             |  3 ++-
 2 files changed, 19 insertions(+), 8 deletions(-)

diff --git a/examples/nested-jsbeautifyrc/riotjs/expected/test.tag b/examples/nested-jsbeautifyrc/riotjs/expected/test.tag
index 4baf564..3f135b0 100644
--- a/examples/nested-jsbeautifyrc/riotjs/expected/test.tag
+++ b/examples/nested-jsbeautifyrc/riotjs/expected/test.tag
@@ -7,15 +7,25 @@
   <script>
     this.message = 'Hello, Riot!'
     this.techs = [
-      { name: 'HTML' },
-      { name: 'JavaScript' },
-      { name: 'CSS' }
+      {
+        name: 'HTML'
+      }, {
+        name: 'JavaScript'
+      }, {
+        name: 'CSS'
+      }
     ]
   </script>
 
   <style scoped>
-    :scope { font-size: 2rem }
-    h3 { color: #444 }
-    ul { color: #999 }
+    :scope {
+      font-size: 2rem;
+    }
+    h3 {
+      color: #444;
+    }
+    ul {
+      color: #999;
+    }
   </style>
-</sample>
+</sample>
\ No newline at end of file
diff --git a/src/beautifiers/prettydiff.coffee b/src/beautifiers/prettydiff.coffee
index 67f4053..81236fd 100644
--- a/src/beautifiers/prettydiff.coffee
+++ b/src/beautifiers/prettydiff.coffee
@@ -54,6 +54,7 @@ module.exports = class PrettyDiff extends Beautifier
     LESS: true
     Swig: true
     Visualforce: true
+    "Riot.js": true
   }
 
   beautify: (text, language, options) ->
@@ -73,7 +74,7 @@ module.exports = class PrettyDiff extends Beautifier
           lang = "ejs"
         when "ERB"
           lang = "html_ruby"
-        when "Handlebars", "Mustache", "Spacebars", "Swig"
+        when "Handlebars", "Mustache", "Spacebars", "Swig", "Riot.js"
           lang = "handlebars"
         when "SGML"
           lang = "markup"