From 33b9bec954f13002ae2445e1b1abadb57820e110 Mon Sep 17 00:00:00 2001 From: aycabta Date: Thu, 2 Sep 2021 21:34:53 +0900 Subject: [PATCH] Detect the variable class to show doc --- lib/irb/completion.rb | 15 +++++++++++---- test/irb/test_completion.rb | 6 ++++++ 2 files changed, 17 insertions(+), 4 deletions(-) diff --git a/lib/irb/completion.rb b/lib/irb/completion.rb index d1bb82122..c7c0fb99e 100644 --- a/lib/irb/completion.rb +++ b/lib/irb/completion.rb @@ -315,12 +315,19 @@ def self.retrieve_completion_data(input, bind: IRB.conf[:MAIN_CONTEXT].workspace end else - candidates = eval("methods | private_methods | local_variables | instance_variables | self.class.constants", bind).collect{|m| m.to_s} - candidates |= ReservedWords - if doc_namespace - candidates.find{ |i| i == input } + vars = eval("local_variables | instance_variables", bind).collect{|m| m.to_s} + perfect_match_var = vars.find{|m| m.to_s == input} + if perfect_match_var + eval("#{perfect_match_var}.class.name", bind) + else + candidates = eval("methods | private_methods | local_variables | instance_variables | self.class.constants", bind).collect{|m| m.to_s} + candidates |= ReservedWords + candidates.find{ |i| i == input } + end else + candidates = eval("methods | private_methods | local_variables | instance_variables | self.class.constants", bind).collect{|m| m.to_s} + candidates |= ReservedWords candidates.grep(/^#{Regexp.quote(input)}/) end end diff --git a/test/irb/test_completion.rb b/test/irb/test_completion.rb index ab5776202..260b8a889 100644 --- a/test/irb/test_completion.rb +++ b/test/irb/test_completion.rb @@ -83,5 +83,11 @@ def test_complete_require_relative assert_include candidates, word end end + + def test_complete_variable + str_example = '' + assert_include(IRB::InputCompletor.retrieve_completion_data("str_examp", bind: binding), "str_example") + assert_equal(IRB::InputCompletor.retrieve_completion_data("str_example", bind: binding, doc_namespace: true), "String") + end end end