diff options
Diffstat (limited to 'lib/irb')
| -rw-r--r-- | lib/irb/init.rb | 17 | ||||
| -rw-r--r-- | lib/irb/input-method.rb | 37 |
2 files changed, 47 insertions, 7 deletions
diff --git a/lib/irb/init.rb b/lib/irb/init.rb index 9fd86e94a2..1df2926fa5 100644 --- a/lib/irb/init.rb +++ b/lib/irb/init.rb @@ -180,17 +180,20 @@ module IRB end end + # enumerate possible rc files + def IRB.rc_files(rc) + yield File.expand_path("~/.irb#{rc}") if ENV.key?("HOME") + yield ".irb#{rc}" + yield "irb#{rc.sub(/\A_?/, '.')}" + yield "_irb#{rc}" + yield "$irb#{rc}" + end + # running config def IRB.run_config if @CONF[:RC] - rcs = [] - rcs.push File.expand_path("~/.irbrc") if ENV.key?("HOME") - rcs.push ".irbrc" - rcs.push "irb.rc" - rcs.push "_irbrc" - rcs.push "$irbrc" catch(:EXIT) do - for rc in rcs + rc_files("rc") do |rc| begin load rc throw :EXIT diff --git a/lib/irb/input-method.rb b/lib/irb/input-method.rb index ec834b17bc..d1fc83a975 100644 --- a/lib/irb/input-method.rb +++ b/lib/irb/input-method.rb @@ -85,12 +85,49 @@ module IRB require "readline" class ReadlineInputMethod < InputMethod include Readline + + def ReadlineInputMethod.create_finalizer(hist, file) + proc do + if num = IRB.conf[:SAVE_HISTORY] and (num = num.to_i) > 0 + if hf = IRB.conf[:HISTORY_FILE] + file = File.expand_path(hf) + end + if file + open(file, 'w' ) do |f| + hist = hist.to_a + f.puts(hist[-num..-1] || hist) + end + end + end + end + end + def initialize super @line_no = 0 @line = [] @eof = false + + loader = proc {|f| f.each {|l| HISTORY << l.chomp}} + if hist = IRB.conf[:HISTORY_FILE] + hist = File.expand_path(hist) + begin + open(hist, &loader) + rescue + end + else + IRB.rc_files("_history") do |hist| + begin + open(hist, &loader) + rescue + hist = nil + else + break + end + end + end + ObjectSpace.define_finalizer(self, ReadlineInputMethod.create_finalizer(HISTORY, hist)) end def gets |