Created
March 21, 2009 12:20
-
-
Save febuiles/82824 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
## | |
# Wrapper for the Pygments command line tool, pygmentize. | |
# | |
# Pygments: http://pygments.org/ | |
# | |
# Assumes pygmentize is in the path. If not, set its location | |
# with Albino.bin = '/path/to/pygmentize' | |
# | |
# Use like so: | |
# | |
# @syntaxer = Albino.new('/some/file.rb', :ruby) | |
# puts @syntaxer.colorize | |
# | |
# This'll print out an HTMLized, Ruby-highlighted version | |
# of '/some/file.rb'. | |
# | |
# To use another formatter, pass it as the third argument: | |
# | |
# @syntaxer = Albino.new('/some/file.rb', :ruby, :bbcode) | |
# puts @syntaxer.colorize | |
# | |
# You can also use the #colorize class method: | |
# | |
# puts Albino.colorize('/some/file.rb', :ruby) | |
# | |
# Another also: you get a #to_s, for somewhat nicer use in Rails views. | |
# | |
# ... helper file ... | |
# def highlight(text) | |
# Albino.new(text, :ruby) | |
# end | |
# | |
# ... view file ... | |
# <%= highlight text %> | |
# | |
# The default lexer is 'text'. You need to specify a lexer yourself; | |
# because we are using STDIN there is no auto-detect. | |
# | |
# To see all lexers and formatters available, run `pygmentize -L`. | |
# | |
# Chris Wanstrath // chris@ozmm.org | |
# GitHub // http://github.com | |
# | |
require 'rubygems' | |
require 'open4' | |
class Albino | |
@@bin = Rails.development? ? 'pygmentize' : '/usr/bin/pygmentize' rescue 'pygmentize' | |
def self.bin=(path) | |
@@bin = path | |
end | |
def self.colorize(*args) | |
new(*args).colorize | |
end | |
def initialize(target, lexer = :text, format = :html) | |
@target = File.exists?(target) ? File.read(target) : target rescue target | |
@options = { :l => lexer, :f => format } | |
end | |
def execute(command) | |
pid, stdin, stdout, stderr = Open4.popen4(command) | |
stdin.puts @target | |
stdin.close | |
err = stderr.read.strip | |
err.empty? ? stdout.read.strip : err | |
end | |
def colorize(options = {}) | |
execute @@bin + convert_options(options) | |
end | |
alias_method :to_s, :colorize | |
def convert_options(options = {}) | |
@options.merge(options).inject('') do |string, (flag, value)| | |
string + " -#{flag} #{value}" | |
end | |
end | |
end | |
if $0 == __FILE__ | |
require 'rubygems' | |
require 'test/spec' | |
require 'mocha' | |
begin require 'redgreen'; rescue LoadError; end | |
context "Albino" do | |
setup do | |
@syntaxer = Albino.new(__FILE__, :ruby) | |
end | |
specify "defaults to text" do | |
syntaxer = Albino.new(__FILE__) | |
syntaxer.expects(:execute).with('pygmentize -l text -f html').returns(true) | |
syntaxer.colorize | |
end | |
specify "accepts options" do | |
@syntaxer.expects(:execute).with('pygmentize -l ruby -f html').returns(true) | |
@syntaxer.colorize | |
end | |
specify "works with strings" do | |
syntaxer = Albino.new('class New; end', :ruby) | |
assert_match %r(highlight), syntaxer.colorize | |
end | |
specify "aliases to_s" do | |
assert_equal @syntaxer.colorize, @syntaxer.to_s | |
end | |
specify "class method colorize" do | |
assert_equal @syntaxer.colorize, Albino.colorize(__FILE__, :ruby) | |
end | |
specify "returns an error string for unknown lexers" do | |
syntaxer = Albino.new('class New; end', "something weird") | |
assert_equal syntaxer.colorize, "Error: no lexer for alias 'something' found" | |
end | |
end | |
end |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment