File Coverage

blib/lib/UTF8BOM.pm

Criterion	Covered	Total	%
statement	17	62	27.4
branch	2	30	6.6
condition	0	34	0.0
subroutine	6	13	46.1
pod	7	7	100.0
total	32	146	21.9

package UTF8BOM;
use strict;

use IO::File;
use IO::Dir;
use File::Spec;

our $VERSION = '1.02';

my $BOM = "\x{ef}\x{bb}\x{bf}";

sub check_bom {
    use bytes;
    my($class, $str) = @_;
    return ($BOM eq substr($str, 0, length($BOM)))
        ? 1 : 0;
}

sub remove_from_str {
    my($class, $str) = @_;
    return $class->check_bom($str)
        ? return substr($str, length($BOM))
        : $str;
}

sub insert_into_str {
    my($class, $str) = @_;
    return $class->check_bom($str)
        ? $str
        : $BOM.$str;
}

sub remove_from_file {
    my($class, $file) = @_;
    my $fh = IO::File->new($file)
        or $class->_croak(qq/Couldn't open file "$file"./);
    my @lines = $fh->getlines;
    $fh->close;
    my $str = $class->remove_from_str( join("", @lines) );
    $fh = IO::File->new($file, "w")
        or $class->_croak(qq/Couldn't open file "$file"./);
    $fh->print($str);
    $fh->close;
}

sub insert_into_file {
    my($class, $file) = @_;
    my $fh = IO::File->new($file)
        or $class->_croak(qq/Couldn't open file "$file"./);
    my @lines = $fh->getlines;
    $fh->close;
    my $str = $class->insert_into_str( join("", @lines) );
    $fh = IO::File->new($file, "w")
        or $class->_croak(qq/Couldn't open file "$file"./);
    $fh->print($str);
    $fh->close;
}

sub remove_from_files {
    my($class, %options) = @_;
    my $dir = $options{dir} || File::Spec->curdir;
    my $recursive = $options{recursive} || 0;
    my $dh = IO::Dir->new($dir)
        or $class->_croak(qq/Couldn't open directory "$dir"./);
    while( defined( my $file = $dh->read ) ) {
        next if($file eq '.' || $file eq '..');
        my $path = File::Spec->catfile($dir, $file);
        if (-e $path && -f $path) {
            $class->remove_from_file($path);
        } elsif (-e $path && -d $path && $recursive) {
            $class->remove_from_files(
                dir       => $path,
                recursive => 1,
            );
        }
    }
    $dh->close;
}

sub insert_into_files {
    my($class, %options) = @_;
    my $dir = $options{dir} || File::Spec->curdir;
    my $recursive = $options{recursive} || 0;
    my $dh = IO::Dir->new($dir)
        or $class->_croak(qq/Coundn't open directory "$dir"./);
    while( defined( my $file = $dh->read ) ) {
        next if($file eq '.' || $file eq '..');
        my $path = File::Spec->catfile($dir, $file);
        if (-e $path && -f $path) {
            $class->insert_into_file($path);
        } elsif (-e $path && -d $path && $recursive) {
            $class->insert_into_files(
                dir       => $path,
                recursive => 1,
            );
        }
    }
    $dh->close;
}

sub _croak {
    my($self, $msg) = @_;
    require Carp; Carp::croak($msg);
}

1;
__END__