X-Git-Url: http://git.shiar.net/perl/file-rewrite.git/blobdiff_plain/46841d08dc662113df30cf49e072b0ba04d4d9aa..be4cabbf32b3220dfe687065b992ca15cdfbf4c8:/lib/File/Rewrite.pm

diff --git a/lib/File/Rewrite.pm b/lib/File/Rewrite.pm
index 784542d..b3bf3d7 100644
--- 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;