From 145c1e0691d4a6ca8eda40a42e584a1f6899fae2 Mon Sep 17 00:00:00 2001 From: aycabta Date: Mon, 20 Dec 2021 14:36:37 +0900 Subject: [ruby/reline] Add support for overwriting dialog proc with the same name https://github.com/ruby/reline/commit/16aa20c380 --- lib/reline.rb | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) (limited to 'lib/reline.rb') diff --git a/lib/reline.rb b/lib/reline.rb index b92004c1ab..e5c722fe53 100644 --- a/lib/reline.rb +++ b/lib/reline.rb @@ -60,7 +60,7 @@ module Reline def initialize self.output = STDOUT - @dialog_proc_list = [] + @dialog_proc_list = {} yield self @completion_quote_character = nil @bracketed_paste_finished = false @@ -155,16 +155,15 @@ module Reline @dig_perfect_match_proc = p end + DialogProc = Struct.new(:dialog_proc, :context) def add_dialog_proc(name_sym, p, context = nil) raise ArgumentError unless p.respond_to?(:call) or p.nil? raise ArgumentError unless name_sym.instance_of?(Symbol) - @dialog_proc_list << [name_sym, p, context] + @dialog_proc_list[name_sym] = DialogProc.new(p, context) end def dialog_proc(name_sym) - dialog = @dialog_proc_list.find { |d| d[0] == name_sym } - dialog.nil? ? nil : dialog[1] - #@dialog_proc_list[name_sym] + @dialog_proc_list[name_sym] end def input=(val) @@ -307,9 +306,8 @@ module Reline line_editor.auto_indent_proc = auto_indent_proc line_editor.dig_perfect_match_proc = dig_perfect_match_proc line_editor.pre_input_hook = pre_input_hook - @dialog_proc_list.each do |d| - name_sym, dialog_proc, context = d - line_editor.add_dialog_proc(name_sym, dialog_proc, context) + @dialog_proc_list.each_pair do |name_sym, d| + line_editor.add_dialog_proc(name_sym, d.dialog_proc, d.context) end unless config.test_mode -- cgit v1.2.3