Tweaks to complete package names and variables

[p5sagit/Devel-REPL.git] / lib / Devel / REPL / Plugin / LexEnv.pm

diff --git a/lib/Devel/REPL/Plugin/LexEnv.pm b/lib/Devel/REPL/Plugin/LexEnv.pm
index 2d48746..ea6d26b 100644 (file)
--- a/lib/Devel/REPL/Plugin/LexEnv.pm
+++ b/lib/Devel/REPL/Plugin/LexEnv.pm
@@ -4,6 +4,8 @@ use Moose::Role;
 use namespace::clean -except => [ 'meta' ];
 use Lexical::Persistence;
 
+with 'Devel::REPL::Plugin::FindVariable';
+
 has 'lexical_environment' => (
   isa => 'Lexical::Persistence',
   is => 'rw',
@@ -42,4 +44,14 @@ around 'execute' => sub {
   return $self->$orig($wrapped, @rest);
 };
 
+around 'find_variable' => sub {
+  my $orig = shift;
+  my ($self, $name) = @_;
+
+  return \( $self->lexical_environment->get_context('_')->{$name} )
+    if exists $self->lexical_environment->get_context('_')->{$name};
+
+  return $orig->(@_);
+};
+
 1;