[Sommaire] [Imprimer]

   1  require 'erb'
   2  
   3  class String
   4    def lines
   5      split $/
   6    end
   7    
   8    def strip_whitespace_at_line_ends
   9      lines.map {|line| line.gsub(/\s+$/, '')} * $/
  10    end
  11  end
  12  
  13  module Protodoc
  14    module Environment
  15      def include(*filenames)
  16        filenames.map {|filename| Preprocessor.new(filename).to_s}.join("\n")
  17      end
  18    end
  19    
  20    class Preprocessor
  21      include Environment
  22      
  23      def initialize(filename)
  24        @filename = File.expand_path(filename)
  25        @template = ERB.new(IO.read(@filename), nil, '%')
  26      end
  27      
  28      def to_s
  29        @template.result(binding).strip_whitespace_at_line_ends
  30      end
  31    end  
  32  end
  33  
  34  if __FILE__ == $0
  35    print Protodoc::Preprocessor.new(ARGV.first)
  36  end