Implement auto indent for multiline

1 files changed, 5 insertions, 0 deletions

diff --git a/lib/irb/input-method.rb b/lib/irb/input-method.rb
index 68d2ad280c..dbed7da14b 100644
--- a/lib/irb/input-method.rb
+++ b/lib/irb/input-method.rb
@@ -244,6 +244,10 @@ module IRB
       @prompt_proc = block
     end
 
+    def auto_indent(&block)
+      @auto_indent_proc = block
+    end
+
     # Reads the next line from this input method.
     #
     # See IO#gets for more information.
@@ -251,6 +255,7 @@ module IRB
       Reline.input = @stdin
       Reline.output = @stdout
       Reline.prompt_proc = @prompt_proc
+      Reline.auto_indent_proc = @auto_indent_proc
       if l = readmultiline(@prompt, false, &@check_termination_proc)
         HISTORY.push(l) if !l.empty?
         @line[@line_no += 1] = l + "\n"