tangetools/G/G
2024-08-06 01:06:11 +02:00

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;
}