fileset: explicit undef search to match unconditionally

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

diff --git a/lib/File/Rewrite.pm b/lib/File/Rewrite.pm
index 784542d9f47074854c9789811429148a98f20087..b3bf3d74019bc72ed127ef8015338b2464e674bf 100644 (file)
--- a/lib/File/Rewrite.pm
+++ b/lib/File/Rewrite.pm
@@ -10,15 +10,21 @@ our @EXPORT_OK = qw(fileset);
 
 sub fileset {
        my ($filename, $replace, $search) = @_;
-       $search = $replace if not defined $search;
        my $tmpname = "$filename.$$.tmp";
        my $changes = 0;
 
+       $search = $replace 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) {
                chomp;
-               if ($_ =~ (ref $search ? $search : qr/^\Q$search\E$/)) {
+               if (defined $search and $_ =~ /$search/s) {
                        if (ref $replace eq 'ARRAY') {
                                if (@$replace and $_ eq $replace->[0]) {
                                        shift @$replace;