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 = "#{$1}#{$4}"
@@ -64,7 +69,7 @@ module Jekyll
@file = $1
@caption = "#{$1}\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 = "