File Coverage

blib/lib/Text/Context/EitherSide.pm

Criterion	Covered	Total	%
statement	52	55	94.5
branch	14	18	77.7
condition	3	3	100.0
subroutine	11	12	91.6
pod	6	6	100.0
total	86	94	91.4

package Text::Context::EitherSide;

use 5.006;
use strict;
use warnings;
use Carp;

our $VERSION = '1.4';

use base 'Exporter';
our @EXPORT_OK = qw(get_context);

use constant DEFAULT_WORDS => 2;

sub get_context {
    my ($n, $string, @words) = @_;
    Text::Context::EitherSide->new($string, context => $n)->as_string(@words);
}

sub new {
    my $class = shift;
    my $text  = shift or carp "No text supplied for context search";
    my %args  = @_;

    return bless {
        n => exists $args{context} ? $args{context} : DEFAULT_WORDS,
        text => $text
    }, $class;
}

sub context {
    my $self = shift;
    $self->{n} = shift if @_;
    return $self->{n};
}

sub as_sparse_list {
    my $self     = shift;
    my @words    = @_;
    my %keywords = map { lc $_ => 1 }
        map { split /\s+/, $_ } @words;    # Decouple phrases

    # First, split the string into words
    my @split_s = split /\s+/, $self->{text};

    # Now, locate keywords and "mark" the indices we want.
    my @marks = (undef) x @split_s;
    my $ok    = 0;
    for (0 .. $#split_s) {
        my $word = lc $split_s[$_];
        for my $subword (split /\W+/, $word) {
            if (exists $keywords{$subword}) {
                $ok++;

                # Mark it and its $n neighbours.
                $marks[$_] = $split_s[$_]
                    for grep { $_ >= 0 and $_ <= $#split_s }
                    $_ - $self->{n} .. $_ + $self->{n};
                last;
            }
        }
    }

    return $ok ? @marks : ();
}

sub as_list {
    my $self   = shift;
    my @sparse = $self->as_sparse_list(@_);
    return () unless @sparse;
    my @ret;
    for (0 .. $#sparse) {
        if (defined $sparse[$_]) {
            push @ret, $sparse[$_];
        } else {
            push @ret, "..." unless @ret and $ret[-1] eq "...";
        }
    }
    return @ret;
}

sub as_string {
    my $self = shift;
    return join " ", $self->as_list(@_);
}

1;
__END__