build script and rules

[perl/file-rewrite.git] / lib / File / Rewrite.pm

diff --git a/lib/File/Rewrite.pm b/lib/File/Rewrite.pm
index bb41513654d7de4fd731fdccdfcb825628df5329..731925929d9536a09ed1e2a998850f648b7ef47e 100644 (file)
--- a/lib/File/Rewrite.pm
+++ b/lib/File/Rewrite.pm
@@ -6,24 +6,52 @@ use warnings;
 use autodie ':all';
 use Exporter 'import';
 
+our $VERSION = '1.00';
 our @EXPORT_OK = qw(fileset);
 
 sub fileset {
-       my ($filename, $replace, $search) = @_;
-       die if defined $replace;
+       my ($filename, $search, $replace) = @_;
        my $tmpname = "$filename.$$.tmp";
        my $changes = 0;
 
+       $replace = $search if @_ < 3;
+
+       # generate a single regex to emulate smart match
+       $search = [$search] unless ref $search eq 'ARRAY';
+       @$search = grep { defined } @$search;
+       $search = @$search ? join('|', map { ref $_ ? $_ : "^\Q$_\E\$" } @$search) : undef;
+
        open my $src,  '<', $filename;
        open my $dest, '>', $tmpname;
        while (readline $src) {
-               if ($_ =~ (ref $search ? $search : qr/^\Q$search\E$/)) {
+               chomp;
+               if (defined $search and $_ =~ /$search/s) {
+                       if (ref $replace eq 'ARRAY') {
+                               if (@$replace and $_ eq $replace->[0]) {
+                                       shift @$replace;
+                               }
+                               else {
+                                       $changes++;
+                                       next;
+                               }
+                       }
+                       elsif (defined $replace and $_ eq $replace) {
+                               undef $replace;
+                       }
+                       else {
                                $changes++;
                                next;
+                       }
                }
                print {$dest} $_, $/;
        }
        close $src;
+       if (defined $replace) {
+               for (ref $replace eq 'ARRAY' ? @$replace : $replace) {
+                       $changes++;
+                       print {$dest} $_, $/;
+               }
+       }
        close $dest;
 
        if ($changes) {
@@ -39,12 +67,12 @@ sub fileset {
 
 =head1 NAME
 
-File::Rewrite
+File::Rewrite - Append and/or strip lines from text files
 
 =head1 SYNOPSIS
 
        use File::Rewrite 'fileset';
-       fileset('/tmp/somefile', undef, qr/match/);
+       fileset('/tmp/somefile', qr/match/ => undef);
 
 =head1 DESCRIPTION