Add support for aliased styles

unexpectedjs · Jan 29, 2018 · f4bb2e7 · f4bb2e7
1 parent d192c5e
commit f4bb2e7
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/magicPenPrism.js b/lib/magicPenPrism.js
@@ -85,6 +85,7 @@ module.exports = {
 
             var that = this;
             var capitalizedLanguage = upperCamelCase(language);
+            var languageDefinition = prism.languages[language];
 
             function printTokens(token, parentStyle) {
                 if (Array.isArray(token)) {
@@ -98,6 +99,8 @@ module.exports = {
                         that['prism' + capitalizedLanguage + upperCamelCasedParentStyle](token);
                     } else if (that['prism' + upperCamelCasedParentStyle]) {
                         that['prism' + upperCamelCasedParentStyle](token);
+                    } else if (languageDefinition[parentStyle] && languageDefinition[parentStyle].alias) {
+                        printTokens(token, languageDefinition[parentStyle].alias);
                     } else {
                         that.text(token);
                     }