202 lines
4.3 KiB
Perl
Executable file
202 lines
4.3 KiB
Perl
Executable file
#!/usr/bin/perl
|
|
|
|
=pod
|
|
|
|
=head1 NAME
|
|
|
|
G - shorthand for multi level greps
|
|
|
|
|
|
=head1 SYNOPSIS
|
|
|
|
B<G> [[grep options] string] [[grep options] string] ...
|
|
|
|
|
|
=head1 DESCRIPTION
|
|
|
|
B<G> is a short hand of writing (search for single lines matching expressions):
|
|
|
|
grep --option string | grep --option2 string2
|
|
|
|
or with B<-g> (search full files matching expressions):
|
|
|
|
find . -type f | xargs grep -l string1 | xargs grep -l string1
|
|
|
|
=head1 OPTIONS
|
|
|
|
=over 4
|
|
|
|
=item B<-g>
|
|
|
|
Search current subtree for files that match all expressions - but not
|
|
necessarily on the same line.
|
|
|
|
If a parent dir contains a B<.git> dir B<git grep> is used.
|
|
|
|
|
|
=item B<-->I<X>
|
|
|
|
Apply single letter option X to all following options.
|
|
|
|
E.g. B<--i> will prepend B<-i> to all following options.
|
|
|
|
|
|
=item I<other options>
|
|
|
|
All B<grep> options.
|
|
|
|
=back
|
|
|
|
|
|
=head1 EXAMPLE
|
|
|
|
Grep for lines with Foo but not Bar:
|
|
|
|
G Foo -v Bar
|
|
|
|
|
|
Grep for lines with Foo, bar but not baz. Ignore case for bar and baz:
|
|
|
|
G Foo --i bar -v baz
|
|
|
|
|
|
=head1 AUTHOR
|
|
|
|
Copyright (C) 2017-2023 Ole Tange,
|
|
http://ole.tange.dk and Free Software Foundation, Inc.
|
|
|
|
|
|
=head1 LICENSE
|
|
|
|
Copyright (C) 2012 Free Software Foundation, Inc.
|
|
|
|
This program is free software; you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation; either version 3 of the License, or
|
|
at your option any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
=head1 DEPENDENCIES
|
|
|
|
B<G> uses B<grep> and B<parallel>.
|
|
|
|
|
|
=head1 SEE ALSO
|
|
|
|
B<grep>
|
|
|
|
|
|
=cut
|
|
|
|
my $i = 0;
|
|
my @add_X;
|
|
|
|
# Make groups of grep options: -v foo -v -i bar => [-v foo] [-v -i bar]
|
|
for(@ARGV) {
|
|
if($_ eq "-g") {
|
|
# -g = recursive-and file grep
|
|
$opt::g ||= 1;
|
|
# -g not an option for grep
|
|
next;
|
|
}
|
|
if($_ =~ /^--(.)$/) {
|
|
# --X = add -X to the rest
|
|
$opt::X ||= 1;
|
|
push @add_X, "-$1";
|
|
# --X not an option for grep
|
|
next;
|
|
}
|
|
push @{$cmd[$i]}, $_;
|
|
if(/^[^-]/) {
|
|
# This is not an option
|
|
if(@add_X) { unshift @{$cmd[$i]}, @add_X; }
|
|
$i++;
|
|
}
|
|
}
|
|
|
|
if($opt::g and @cmd) {
|
|
# -g => search files
|
|
sub gitdir {
|
|
# Find .git dir somewhere in parent
|
|
my $dir = shift;
|
|
-d "$dir/.git" and return "$dir/.git";
|
|
if(join(":",stat $dir) eq join(":",stat "$dir/..")) {
|
|
# This is root
|
|
return undef;
|
|
}
|
|
return gitdir("../$dir")
|
|
}
|
|
sub cache_gitdir {
|
|
# cat all files in .git to /dev/null
|
|
# this will (hopefully) move it into disk cache
|
|
# so seeks can be avoided
|
|
my $dir = shift;
|
|
`find "$dir" -type f -print0 | xargs -0 cat >/dev/null`;
|
|
}
|
|
my $a = shift @cmd;
|
|
# -v => Use -L instead of -l
|
|
my $l_or_L = (grep /^-v$/, @$a) ? "L" : "l";
|
|
@$a = (grep { not /^-v$/ } @$a);
|
|
|
|
my $gitdir = gitdir(".");
|
|
if($gitdir) {
|
|
cache_gitdir($gitdir);
|
|
$run = "git grep --threads 30 -$l_or_L ".shell_quote(@$a);
|
|
} else {
|
|
$run = "find . -type f | parallel --lb -Xq grep -$l_or_L ".shell_quote(@$a);
|
|
}
|
|
if(@cmd) {
|
|
$run .= '|' .
|
|
join"|", map { 'xargs -d"\n" grep -'.$l_or_L.' '.
|
|
join(" ", shell_quote(@$_)) } @cmd;
|
|
}
|
|
exec $run;
|
|
} elsif(@cmd) {
|
|
# => search stdin
|
|
exec join"|", map { "grep ".join(" ", shell_quote(@$_)) } @cmd;
|
|
} else {
|
|
# no options => cat
|
|
exec 'cat';
|
|
}
|
|
|
|
sub shell_quote {
|
|
# Input:
|
|
# @strings = strings to be quoted
|
|
# Output:
|
|
# @shell_quoted_strings = string quoted with \ as needed by the shell
|
|
return wantarray ?
|
|
(map { shell_quote_scalar($_) } @_)
|
|
: (join" ",map { shell_quote_scalar($_) } @_);
|
|
}
|
|
|
|
sub shell_quote_scalar {
|
|
# Quote for other shells
|
|
my $a = $_[0];
|
|
if(defined $a) {
|
|
# zsh wants '=' quoted
|
|
# Solaris sh wants ^ quoted.
|
|
# $a =~ s/([\002-\011\013-\032\\\#\?\`\(\)\{\}\[\]\^\*\>\<\~\|\; \"\!\$\&\'\202-\377])/\\$1/g;
|
|
# This is 1% faster than the above
|
|
if(($a =~ s/[\002-\011\013-\032\\\#\?\`\(\)\{\}\[\]\^\*\<\=\>\~\|\; \"\!\$\&\'\202-\377]/\\$&/go)
|
|
+
|
|
# quote newline as '\n'
|
|
($a =~ s/[\n]/'\n'/go)) {
|
|
# A string was replaced
|
|
# No need to test for "" or \0
|
|
} elsif($a eq "") {
|
|
$a = "''";
|
|
} elsif($a eq "\0") {
|
|
$a = "";
|
|
}
|
|
}
|
|
return $a;
|
|
}
|