File Coverage

blib/lib/PPI/Transform/PackageName.pm

Criterion	Covered	Total	%
statement	48	48	100.0
branch	13	18	72.2
condition	8	19	42.1
subroutine	6	6	100.0
pod	2	2	100.0
total	77	93	82.8

use strict;
use warnings;

package PPI::Transform::PackageName;

# ABSTRACT: Subclass of PPI::Transform specific for modifying package names
our $VERSION = 'v0.0.7'; # VERSION

use base qw(PPI::Transform);

use Carp;

sub new
{
    my $self = shift;
    my $class = ref($self) || $self;
    my %arg = @_;
    if(exists $arg{-all}) {
        croak "-all and other options are contradictory" if exists $arg{-package_name} || exists $arg{-word} || exists $arg{-quote};
        $arg{-package_name} = $arg{-all};
        $arg{-word}         = $arg{-all};
        $arg{-quote}        = $arg{-all};
    }
    return bless {
        _PKG => $arg{-package_name},
        _WORD => $arg{-word},
        _QUOTE => $arg{-quote},
    }, $class;
}

sub document
{
    my ($self, $doc) = @_;
    $doc->prune('PPI::Token::Comment');
    if(defined $self->{_PKG} && defined $self->{_WORD}) {
        my $words = $doc->find('PPI::Token::Word');
        $words ||= [];
        for my $word (@$words) {
            if(defined $self->{_PKG} && $word->statement->class eq 'PPI::Statement::Package') {
                my $content = $word->content;
                $_ = $content;
                $self->{_PKG}->();
                $word->set_content($_) if $_ ne $content;
            } elsif(defined $self->{_WORD}) {
                my $content = $word->content;
                $_ = $content;
                $self->{_WORD}->();
                $word->set_content($_) if $_ ne $content;
            }
        }
    }
    if(defined $self->{_QUOTE}) {
        my $quotes = $doc->find('PPI::Token::Quote');
        $quotes ||= [];
        for my $quote (@$quotes) {
                my $string = $quote->string;
                $_ = $string;
                $self->{_QUOTE}->();
                if($_ ne $string) {
                    my $content = $quote->content;
                    my $index = index($content, $string);
                    substr($quote->{content}, $index, length($string)) = $_;
                }
        }
    }
    return 1;
}

1;

__END__