lib/rdoc/erbio.rb

   1 # frozen_string_literal: true
   2 require 'erb'
   3
   4 ##
   5 # A subclass of ERB that writes directly to an IO.  Credit to Aaron Patterson
   6 # and Masatoshi SEKI.
   7 #
   8 # To use:
   9 #
  10 #   erbio = RDoc::ERBIO.new '<%= "hello world" %>', nil, nil
  11 #
  12 #   File.open 'hello.txt', 'w' do |io|
  13 #     erbio.result binding
  14 #   end
  15 #
  16 # Note that binding must enclose the io you wish to output on.
  17
  18 class RDoc::ERBIO < ERB
  19
  20   ##
  21   # Defaults +eoutvar+ to 'io', otherwise is identical to ERB's initialize
  22
  23   def initialize str, trim_mode: nil, eoutvar: 'io'
  24     super(str, trim_mode: trim_mode, eoutvar: eoutvar)
  25   end
  26
  27   ##
  28   # Instructs +compiler+ how to write to +io_variable+
  29
  30   def set_eoutvar compiler, io_variable
  31     compiler.put_cmd    = "#{io_variable}.write"
  32     compiler.insert_cmd = "#{io_variable}.write"
  33     compiler.pre_cmd    = []
  34     compiler.post_cmd   = []
  35   end
  36
  37 end