File Coverage

blib/lib/App/UnANSI.pm

Criterion	Covered	Total	%
statement	30	48	62.5
branch	4	14	28.5
condition	2	7	28.5
subroutine	7	12	58.3
pod	5	5	100.0
total	48	86	55.8

package App::UnANSI;
# ABSTRACT: Remove ANSI coloring from output or files

use strict;
use warnings;
use Getopt::Long qw<:config no_ignore_case>;

# Helper function
sub _help {
    my $msg = shift;

    $msg
        and print "Error: $msg\n\n";

    print << "_HELP";
command_that_creates_colored_output | $0 [OPTIONS] FILE1 FILE2...FILEN

Options:

-h | --help     Print this help menu and exit
-v | --version  Print version number and exit
_HELP

    if ($msg) {
        exit 2;
    } else {
        exit 0;
    }
}

sub new_with_options {
    my ( $class, %opts ) = @_;

    my @files;
    GetOptions(
        'help|h'    => sub { _help(); },
        'version|v' => sub {
            my $version = $App::UnANSI::VERSION || 'DEV';
            print "$0 $version\n";
            exit 0;
        },
        '<>' => sub {
            push @files, @_;
        },
    ) or _help();

    return $class->new( 'files' => \@files );
}

sub new {
    my ( $class, %opts ) = @_;

    $opts{'files'} ||= [];

    foreach my $file ( @{ $opts{'files'} } ) {
        -e $file && -r $file
            or _help("$file is not a readable file")
    }

    return bless {%opts}, $class;
}

sub files {
    my $self = shift;
    return @{ $self->{'files'} };
}

sub run {
    my $self  = shift;
    my @files = $self->files;

    if ( !@files ) {
        # Work on STDIN
        while ( my $line =  ) {
            $self->remove_ansi_colors($line);
            print $line;
        }
    } else {
        # Work on files
        foreach my $file (@files) {
            open my $fh, '<', $file
                or die "Cannot open $file: $!\n";

            while ( my $line = <$fh> ) {
                $self->remove_ansi_colors($line);
                print $line;
            }

            close $fh
                or die "Cannot close $file: $!\n";
        }
    }

    return 1;
}

sub remove_ansi_colors {
    $_[1] =~ s{\x1b\[[^m]*m}{}sgmx;
    return 1;
}

1;

__END__