Import RDoc 3

git-svn-id: svn+ssh://ci.ruby-lang.org/ruby/trunk@30249 b2dd03c8-39d4-4d8f-98ff-823fe69b080e

1 files changed, 37 insertions, 0 deletions

diff --git a/lib/rdoc/erbio.rb b/lib/rdoc/erbio.rb
new file mode 100644
index 0000000000..04a89fbd34
--- /dev/null
+++ b/lib/rdoc/erbio.rb
@@ -0,0 +1,37 @@
+require 'erb'
+
+##
+# A subclass of ERB that writes directly to an IO.  Credit to Aaron Patterson
+# and Masatoshi SEKI.
+#
+# To use:
+#
+#   erbio = RDoc::ERBIO.new '<%= "hello world" %>', nil, nil
+#
+#   open 'hello.txt', 'w' do |io|
+#     erbio.result binding
+#   end
+#
+# Note that binding must enclose the io you wish to output on.
+
+class RDoc::ERBIO < ERB
+
+  ##
+  # Defaults +eoutvar+ to 'io', otherwise is identical to ERB's initialize
+
+  def initialize str, safe_level = nil, trim_mode = nil, eoutvar = 'io'
+    super
+  end
+
+  ##
+  # Instructs +compiler+ how to write to +io_variable+
+
+  def set_eoutvar compiler, io_variable
+    compiler.put_cmd    = "#{io_variable}.write"
+    compiler.insert_cmd = "#{io_variable}.write"
+    compiler.pre_cmd    = []
+    compiler.post_cmd   = []
+  end
+
+end
+