From 66e7fb0bfbf4c4a0392004321d11ce4aa1e151fc Mon Sep 17 00:00:00 2001
From: Parker Moore <parkrmoore@gmail.com>
Date: Sun, 10 Mar 2013 01:44:32 +0100
Subject: [PATCH] #highlight_failed in PygmentsCode

---
 plugins/pygments_code.rb | 24 +++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/plugins/pygments_code.rb b/plugins/pygments_code.rb
index 3c18aad..4b0aecb 100644
--- a/plugins/pygments_code.rb
+++ b/plugins/pygments_code.rb
@@ -16,17 +16,9 @@ module HighlightCode
   include TemplateWrapper
   include SiteConfig
   def pygments(code, lang)
-    begin
-      highlighted_code = Pygments.highlight(code, :lexer => lang, :formatter => 'html', :options => {:encoding => 'utf-8'})
-      highlighted_code = highlighted_code.gsub(/{{/, '&#x7b;&#x7b;').gsub(/{%/, '&#x7b;&#x25;')
-      highlighted_code.to_s
-    rescue
-      fail_message =  "Pygments couldn't highlight your code in lang '#{lang}':"
-      fail_message += "\n\n#{code}"
-      raise ArgumentError, fail_message.red
-    end
-  rescue
-    puts $!,$@
+    highlighted_code = Pygments.highlight(code, :lexer => lang, :formatter => 'html', :options => {:encoding => 'utf-8'})
+    highlighted_code = highlighted_code.gsub(/{{/, '&#x7b;&#x7b;').gsub(/{%/, '&#x7b;&#x25;')
+    highlighted_code.to_s
   end
 
   def highlight(code, options = {})
@@ -192,4 +184,14 @@ module HighlightCode
     code
   end
 
+  def highlight_failed(file, code, lang)
+    code_snippet = code.split("\n")[0..9].map{|l| ">> #{l}" }.join("\n")
+    fail_message =  "Pygments doesn't know how to highlight your code in lang '#{lang}'."
+    fail_message += "\nFailing code from #{file}:" if file
+    fail_message += "\n\n#{code_snippet}"
+    fail_message += "\n>> ..." if code.split("\n").size > 10
+    $stderr.puts fail_message.red
+    raise ArgumentError
+  end
+
 end