Source code embedded plugin for tDiary

codelist.css

/* codelist.css
 * Source code embedded plugin for tDiary
 * Original Copyright (C) 2005 bashi <http://www.cc.ariake-nct.ac.jp/~bashi/d/?date=20051010#p01>
 * Modified Copyright (C) 2007,2008 Koichiro Ohba <[email protected]>
 * License under GPL2.
 */

span.classdef, span.moduledef {
        color: #44aa44;
        font-weight: bold;
}

span.funcall, span.fundef {
        color: blue;
}

span.const {
        color: #44aa44;
}

span.comment {
        color: gray;
}

span.keyword {
        color: #8800cc;
}

span.size {
        font-size: smaller;
        color: gray;
}

span.string, span.character {
        color: #888800;
}

span.symbol {
        color: #ff6666;
}

span.type {
        color: #44aa44;
}

.line_numbers {
    background-color: #ececec;
    color: #aaa;
    padding: 1em .5em
    border-right: 1px solid #ddd;
    text-align: right;
}

div.body pre {
    margin-left: 0;
}

.code_data {
  padding-left: 0.5em;
  margin-left: 0;
}

.codelist table {
    margin: 0;
}

.codelist pre {
    font-family: 'Bitstream Vera Sans Mono', 'Courier', 'monospace';
}

codelist.rb

# -*- coding: utf-8 -*- codelist.rb
# codelist.rb
# Source code embedded plugin for tDiary.
# Original Copyright (C) 2005 bashi <http://www.cc.ariake-nct.ac.jp/~bashi/d/?date=20051010#p01>
# Modified Copyright (C) 2007,2008 Koichiro Ohba <[email protected]>
# License under GPL2.

require 'langscan'

module LangScan
  def self.choose_by_abbrev(hint)
    ::LangScan::LangScanRegistry.each_value {|scanner|
      return scanner if hint.downcase == scanner.abbrev
    }
    ::LangScan::Text
  end
end

Highlight_type = [
  :fundef, :funcall, :classdef, :moduledef, :comment, :string,
  :keyword, :const, :character, :type
]

add_header_proc do
  %Q[<link rel="stylesheet" href="#{theme_url}/codelist.css" type="text/css" media="all">\n]
end

def codelist(file)
  scanner = LangScan.choose(file)
  scanner = LangScan::Text unless scanner
  code(File.open(file).readlines.join, scanner)
end

def line_number(content)
  r = "<pre class='line_numbers'>\n"
  n = 1
  content.each_line do |line|
    r += %[<span id='LID#{n}'>#{n}</span>\n]
    n += 1
  end
  r += '</pre>'
end

def line_number_ref(code)
  r = ""
  n = 1
  code.each_line do |line|
    r += "<span class='line' id='L#{n}'>" + line + '</span>'
    n += 1
  end
  r
end

def highlighting( content, scanner )
  contents = []
  scanner.scan(content) {|f|
    text = if Highlight_type.include?(f.type)
             "<span class='#{f.type}'>#{CGI.escapeHTML(f.text)}</span>"
           else
             CGI.escapeHTML(f.text)
           end
    f.text = text
    contents.push(f)
  }
  contents
end

def code(content, scanner = ::LangScan::Text)
  scanner = LangScan.choose_by_abbrev(scanner) if scanner.kind_of? String
  contents = highlighting( content, scanner )
  contents.sort! {|a, b| a.byteno <=> b.byteno }
  r = <<-EOS
<div class="codelist">
<table cellpadding='0' cellspacing='0'>
  <tr>
    <td>
EOS
  r += line_number(content)
  r += <<EOS
    </td>
    <td width="100%"><pre class="code_data">
EOS

  code = ""
  contents.each {|f|
    code << f.text
  }
  code.chomp!
  r += line_number_ref(code)
  r.chomp!

  r += <<EOS
    </pre></td>
  </tr>
</table>
</div>
EOS
  return r
end