From fcf3a01bd66748b95121956ab0928c0ddacd902f Mon Sep 17 00:00:00 2001
From: Wes Alvaro <hello@wesalvaro.com>
Date: Mon, 4 Apr 2016 11:53:18 +0900
Subject: [PATCH] Allow marking up of other languages.

This change allows users to load in other language definitions and use them with the Prism Highlighter element.
---
 prism-highlighter.html | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/prism-highlighter.html b/prism-highlighter.html
index c81d482..f18e306 100644
--- 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;
       }
     },