From cbd6f8d8dee52930b923dc2277f0677e95dd6bc8 Mon Sep 17 00:00:00 2001
From: Brandon Mathis <brandon@imathis.com>
Date: Sat, 20 Aug 2011 20:24:51 -0400
Subject: [PATCH] added option to force syntax highlighting language, example:
 {% include_code file lang:ruby %}. Fixes #108

---
 plugins/code_block.rb   | 7 ++++++-
 plugins/include_code.rb | 6 +++++-
 2 files changed, 11 insertions(+), 2 deletions(-)

diff --git a/plugins/code_block.rb b/plugins/code_block.rb
index d82893e..4d7ca6b 100644
--- a/plugins/code_block.rb
+++ b/plugins/code_block.rb
@@ -53,7 +53,12 @@ module Jekyll
     def initialize(tag_name, markup, tokens)
       @title = nil
       @caption = nil
+      @filetype = nil
       @highlight = true
+      if markup =~ /\s+lang:(\w+)/i
+        @filetype = $1
+        markup = markup.sub(/\s+lang:\w+\s*/i,'')
+      end
       if markup =~ CaptionUrlTitle
         @file = $1
         @caption = "<figcaption><span>#{$1}</span><a href='#{$2 + $3}'>#{$4}</a></figcaption>"
@@ -64,7 +69,7 @@ module Jekyll
         @file = $1
         @caption = "<figcaption><span>#{$1}</span></figcaption>\n"
       end
-      if @file =~ /\S[\S\s]*\w+\.(\w+)/
+      if @file =~ /\S[\S\s]*\w+\.(\w+)/ && @filetype.nil?
         @filetype = $1
       end
       super
diff --git a/plugins/include_code.rb b/plugins/include_code.rb
index d910c3e..046f624 100644
--- a/plugins/include_code.rb
+++ b/plugins/include_code.rb
@@ -30,6 +30,10 @@ module Jekyll
     def initialize(tag_name, markup, tokens)
       @title = nil
       @file = nil
+      if markup.strip =~ /\s+lang:(\w+)/i
+        @filetype = $1
+        markup = markup.strip.sub(/\s+lang:\w+\s*/i,'')
+      end
       if markup.strip =~ /(.*)?(\s+|^)(\/*\S+)/i
         @title = $1 || nil
         @file = $3
@@ -52,7 +56,7 @@ module Jekyll
 
       Dir.chdir(code_path) do
         code = file.read
-        @filetype = file.extname.sub('.','')
+        @filetype = file.extname.sub('.','') if @filetype.nil?
         title = @title ? "#{@title} (#{file.basename})" : file.basename
         url = "#{context.registers[:site].config['url']}/#{code_dir}/#{@file}"
         source = "<div><figure role=code><figcaption><span>#{title}</span> <a href='#{url}'>download</a></figcaption>\n"