Merge pull request #12 from wesalvaro/patch-1

Allow marking up of other languages.
PolymerElements · May 25, 2016 · db31f02 · db31f02
2 parents c7ce4bc + fcf3a01
commit db31f02
Showing 1 changed file with 12 additions and 9 deletions.
diff --git a/prism-highlighter.html b/prism-highlighter.html
@@ -77,15 +77,18 @@
         return code.match(/^\s*</) ? Prism.languages.markup : Prism.languages.javascript;
       }
 
-      if (lang === 'js' || lang.substr(0, 2) === 'es') {
-        return Prism.languages.javascript;
-      } else if (lang === 'css') {
-        return Prism.languages.css;
-      } else if (lang === 'c') {
-        return Prism.languages.clike;
-      } else {
-        // The assumption is that you're mostly documenting HTML when in HTML.
-        return Prism.languages.markup;
+      if (Prism.languages[lang]) {
+        return Prism.languages[lang];
+      }
+      switch (lang.substr(0, 2)) {
+        case 'js':
+        case 'es':
+          return Prism.languages.javascript;
+        case 'c':
+          return Prism.languages.clike;
+        default:
+          // The assumption is that you're mostly documenting HTML when in HTML.
+          return Prism.languages.markup;
       }
     },