SoundSoftware Code Site

#!/usr/bin/env ruby

# CodeRay Executable

#

# Version: 0.2

# Author: murphy

require 'coderay'

if ARGV.empty?

  $stderr.puts <<-USAGE

CodeRay #{CodeRay::VERSION} (http://coderay.rubychan.de)

Usage:

  coderay file [-<format>]

  coderay -<lang> [-<format>] [< file] [> output]

Defaults:

  lang:   based on file extension

  format: ANSI colorized output for terminal, HTML page for files

Examples:

  coderay foo.rb                         # colorized output to terminal, based on file extension

  coderay foo.rb -loc                    # print LOC count, based on file extension and format

  coderay foo.rb > foo.html              # HTML page output to file, based on extension

  coderay -ruby < foo.rb                 # colorized output to terminal, based on lang

  coderay -ruby -loc < foo.rb            # print LOC count, based on lang

  coderay -ruby -page foo.rb             # HTML page output to terminal, based on lang and format

  coderay -ruby -page foo.rb > foo.html  # HTML page output to file, based on lang and format

  USAGE

end

first, second = ARGV

def read

  file = ARGV.grep(/^(?!-)/).last

  if file

    if File.exist?(file)

      File.read file

    else

      $stderr.puts "No such file: #{file}"

    end

  else

    $stdin.read

  end

end

if first

  if first[/-(\w+)/] == first

    lang = $1

    input = read

    tokens = :scan

  else

    file = first

    unless File.exist? file

      $stderr.puts "No such file: #{file}"

      exit 2

    end

    tokens = CodeRay.scan_file file

  end

else

  $stderr.puts 'No lang/file given.'

  exit 1

end

if second

  if second[/-(\w+)/] == second

    format = $1.to_sym

  else

    raise 'invalid format (must be -xxx)'

  end

else

  if $stdout.tty?

    format = :term

  else

    $stderr.puts 'No format given; setting to default (HTML Page).'

    format = :page

  end

end

if tokens == :scan

  output = CodeRay::Duo[lang => format].highlight input

else

  output = tokens.encode format

end

out = $stdout

out.puts output