Have LexEnv wrap find_variable to first look at lexical variables. The around doesn...

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

package Devel::REPL::Plugin::Completion;
use Devel::REPL::Plugin;
use Scalar::Util 'weaken';
use PPI;
use namespace::clean -except => [ 'meta' ];

has current_matches => (
  is => 'rw',
  isa => 'ArrayRef',
  lazy => 1,
  default => sub { [] },
);

has match_index => (
  is => 'rw',
  isa => 'Int',
  lazy => 1,
  default => sub { 0 },
);

has no_term_class_warning => (
  isa => "Bool",
  is  => "rw",
  default => 0,
);

sub BEFORE_PLUGIN {
  my ($self) = @_;

  my $weakself = $self;
  weaken($weakself);

  $self->term->Attribs->{attempted_completion_function} = sub {
    $weakself->_completion(@_);
  };
}

sub AFTER_PLUGIN {
  my ($self) = @_;

  warn "Term::ReadLine::Gnu is required for the Completion plugin to work"
    unless $self->term->isa("Term::ReadLine::Gnu") and !$self->no_term_class_warning;
}

sub _completion {
  my ($self, $text, $line, $start, $end) = @_;

  # we're discarding everything after the cursor for completion purposes
  # we can't just use $text because we want all the code before the cursor to
  # matter, not just the current word
  substr($line, $end) = '';

  my $document = PPI::Document->new(\$line);
  return unless defined($document);

  $document->prune('PPI::Token::Whitespace');

  my @matches = $self->complete($text, $document);

  # iterate through the completions
  return $self->term->completion_matches($text, sub {
    my ($text, $state) = @_;

    if (!$state) {
      $self->current_matches(\@matches);
      $self->match_index(0);
    }
    else {
      $self->match_index($self->match_index + 1);
    }

    return $self->current_matches->[$self->match_index];
  });
}

sub complete {
  return ();
}

# recursively find the last element
sub last_ppi_element {
  my ($self, $document, $type) = @_;
  my $last = $document;
  while ($last->can('last_element') && defined($last->last_element)) {
    $last = $last->last_element;
    return $last if $type && $last->isa($type);
  }
  return $last;
}

1;