Update pygments plugin to render unrecognized code langs without high…

sitegen/dispatch.lua

@@ -22,8 +22,7 @@ do
       local callbacks = self.callbacks
       for _index_0 = 1, #parts do
         local p = parts[_index_0]
-        local _update_0 = p
-        callbacks[_update_0] = callbacks[_update_0] or { }
+        callbacks[p] = callbacks[p] or { }
         callbacks = callbacks[p]
       end
       return table.insert(callbacks, callback)

sitegen/plugins/pygments.lua

@@ -12,15 +12,25 @@ do
   local _base_0 = {
     custom_highlighters = { },
     disable_indent_detect = false,
+    ignore_missing_lexer = true,
     highlight = function(self, lang, code)
-      local fname = os.tmpname()
+      local fname, errfname = os.tmpname(), os.tmpname()
       do
         local _with_0 = io.open(fname, "w")
         _with_0:write(code)
         _with_0:close()
       end
-      local p = io.popen(("pygmentize -f html -l %s %s"):format(lang, fname))
+      local p = io.popen(("pygmentize -f html -l %s %s 2>%s"):format(lang, fname, errfname))
       local out = p:read("*a")
+      local e = io.open(errfname, "r")
+      local errout = e:read("*a")
+      e:close()
+      if self.ignore_missing_lexer and errout:match("Error: no lexer for alias") then
+        if self.site then
+          self.site.logger:warn("Failed to find syntax highlighter for: " .. tostring(lang))
+        end
+        return html.escape(code)
+      end
       return assert(out:match('^<div class="highlight"><pre>(.-)\n?</pre></div>'), "Failed to parse pygmentize result, is pygments installed?")
     end,
     _highlight = function(self, lang, code, page)

sitegen/plugins/pygments.moon

@@ -16,17 +16,28 @@ import Plugin from require "sitegen.plugin"
 class PygmentsPlugin extends Plugin
   custom_highlighters: {}
   disable_indent_detect: false
+  ignore_missing_lexer: true
 
   -- highlight code with pygments
   highlight: (lang, code) =>
-    fname = os.tmpname!
+    fname, errfname = os.tmpname!, os.tmpname!
     with io.open fname, "w"
       \write code
       \close!
 
-    p = io.popen ("pygmentize -f html -l %s %s")\format lang, fname
+    p = io.popen ("pygmentize -f html -l %s %s 2>%s")\format lang, fname, errfname
     out = p\read"*a"
 
+    e = io.open errfname, "r"
+    errout = e\read"*a"
+    e\close!
+
+    if @ignore_missing_lexer and errout\match "Error: no lexer for alias"
+      if @site
+        @site.logger\warn "Failed to find syntax highlighter for: #{lang}"
+
+      return html.escape code
+
     -- get rid of the div and pre inserted by pygments
     assert out\match('^<div class="highlight"><pre>(.-)\n?</pre></div>'),
       "Failed to parse pygmentize result, is pygments installed?"

spec/plugins_spec.moon

@@ -122,3 +122,51 @@ $hello{[[
   print thing
   ```
 ]]}]=], out
+
+
+describe "sitegen.plugins.pygments", ->
+  local site, py
+
+  before_each ->
+    site = factory.Site {}
+    py = require("sitegen.plugins.pygments")
+
+  it "syntax highlights some code", ->
+    plugin = py site
+    out = flatten_html plugin\filter [[
+```lua
+print("hello world")
+```]]
+
+    assert.same [[<pre class="highlight lang_lua"><code><span></span><span class="nb">print</span><span class="p">(</span><span class="s2">&quot;hello world&quot;</span><span class="p">)</span></code></pre>]], out
+
+  it "doesnt highlight code inside of a cosmo template", ->
+    plugin = py site
+    out = flatten_html plugin\filter [=[
+```lua
+print 5
+```
+
+$hello{[[
+  ```lua
+  print thing
+  ```
+]]}]=]
+
+    assert.same [=[<pre class="highlight lang_lua"><code><span></span><span class="nb">print</span><span class="mi">5</span></code></pre>
+
+$hello{[[
+  ```lua
+  print thing
+  ```
+]]}]=], out
+
+  it "doesn't highlight unrecognized code languages", ->
+    plugin = py site
+    out = flatten_html plugin\filter [[
+```snickerdoodle
+this code block has an unknown label
+```]]
+
+    assert.same [[<pre class="highlight lang_snickerdoodle"><code>this code block has an unknown label
+</code></pre>]], out