648db22b |
1 | #!/bin/sh |
2 | # |
3 | # Copyright (c) 2003 Thomas Klausner. |
4 | # |
5 | # Redistribution and use in source and binary forms, with or without |
6 | # modification, are permitted provided that the following conditions |
7 | # are met: |
8 | # 1. Redistributions of source code must retain the above copyright |
9 | # notice, this list of conditions and the following disclaimer. |
10 | # 2. Redistributions in binary form must reproduce the above copyright |
11 | # notice, this list of conditions and the following disclaimer in the |
12 | # documentation and/or other materials provided with the distribution. |
13 | # |
14 | # THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR |
15 | # IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES |
16 | # OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. |
17 | # IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, |
18 | # INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT |
19 | # NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
20 | # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
21 | # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
22 | # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF |
23 | # THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
24 | |
25 | grep=${GREP:-grep} |
26 | zcat=${ZCAT:-zstdcat} |
27 | |
28 | endofopts=0 |
29 | pattern_found=0 |
30 | grep_args="" |
31 | hyphen=0 |
32 | silent=0 |
33 | |
34 | prog=${0##*/} |
35 | |
36 | # handle being called 'zegrep' or 'zfgrep' |
37 | case $prog in |
38 | *egrep*) prog=zegrep; grep_args='-E';; |
39 | *fgrep*) prog=zfgrep; grep_args='-F';; |
40 | *) prog=zstdgrep;; |
41 | esac |
42 | |
43 | # skip all options and pass them on to grep taking care of options |
44 | # with arguments, and if -e was supplied |
45 | |
46 | while [ "$#" -gt 0 ] && [ "${endofopts}" -eq 0 ]; do |
47 | case "$1" in |
48 | # from GNU grep-2.5.1 -- keep in sync! |
49 | -[ABCDXdefm]) |
50 | if [ "$#" -lt 2 ]; then |
51 | printf '%s: missing argument for %s flag\n' "${prog}" "$1" >&2 |
52 | exit 1 |
53 | fi |
54 | case "$1" in |
55 | -e) |
56 | pattern="$2" |
57 | pattern_found=1 |
58 | shift 2 |
59 | break |
60 | ;; |
61 | -f) |
62 | pattern_found=2 |
63 | ;; |
64 | *) |
65 | ;; |
66 | esac |
67 | grep_args="${grep_args} $1 $2" |
68 | shift 2 |
69 | ;; |
70 | --) |
71 | shift |
72 | endofopts=1 |
73 | ;; |
74 | -) |
75 | hyphen=1 |
76 | shift |
77 | ;; |
78 | -h) |
79 | silent=1 |
80 | shift |
81 | ;; |
82 | -*) |
83 | grep_args="${grep_args} $1" |
84 | shift |
85 | ;; |
86 | *) |
87 | # pattern to grep for |
88 | endofopts=1 |
89 | ;; |
90 | esac |
91 | done |
92 | |
93 | # if no -e option was found, take next argument as grep-pattern |
94 | if [ "${pattern_found}" -lt 1 ]; then |
95 | if [ "$#" -ge 1 ]; then |
96 | pattern="$1" |
97 | shift |
98 | elif [ "${hyphen}" -gt 0 ]; then |
99 | pattern="-" |
100 | else |
101 | printf '%s: missing pattern\n' "${prog}" >&2 |
102 | exit 1 |
103 | fi |
104 | fi |
105 | |
106 | EXIT_CODE=0 |
107 | # call grep ... |
108 | if [ "$#" -lt 1 ]; then |
109 | # ... on stdin |
110 | set -f # Disable file name generation (globbing). |
111 | # shellcheck disable=SC2086 |
112 | "${zcat}" - | "${grep}" ${grep_args} -- "${pattern}" - |
113 | EXIT_CODE=$? |
114 | set +f |
115 | else |
116 | # ... on all files given on the command line |
117 | if [ "${silent}" -lt 1 ] && [ "$#" -gt 1 ]; then |
118 | grep_args="-H ${grep_args}" |
119 | fi |
120 | set -f |
121 | while [ "$#" -gt 0 ]; do |
122 | # shellcheck disable=SC2086 |
123 | if [ $pattern_found -eq 2 ]; then |
124 | "${zcat}" -- "$1" | "${grep}" --label="${1}" ${grep_args} -- - |
125 | else |
126 | "${zcat}" -- "$1" | "${grep}" --label="${1}" ${grep_args} -- "${pattern}" - |
127 | fi |
128 | [ "$?" -ne 0 ] && EXIT_CODE=1 |
129 | shift |
130 | done |
131 | set +f |
132 | fi |
133 | |
134 | exit "${EXIT_CODE}" |