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# $FreeBSD$
26
27set -u
28grep=grep
29zcat=zstdcat
30
31endofopts=0
32pattern_file=0
33pattern_found=0
34grep_args=""
35hyphen=0
36silent=0
37
38prg=${0##*/}
39
40# handle being called 'zegrep' or 'zfgrep'
41case ${prg} in
42*egrep)
43	grep_args="-E";;
44*fgrep)
45	grep_args="-F";;
46esac
47
48catargs="-f"
49case ${prg} in
50zstd*)
51	cattool="/usr/bin/zstdcat"
52	catargs="-fq"
53	;;
54bz*)
55	cattool="/usr/bin/bzcat"
56	;;
57z*)
58	cattool="/usr/bin/zcat"
59	;;
60xz*)
61	cattool="/usr/bin/xzcat"
62	;;
63lz*)
64	cattool="/usr/bin/lzcat"
65	;;
66*)
67	echo "Invalid command: ${prg}" >&2
68	exit 1
69	;;
70esac
71
72# skip all options and pass them on to grep taking care of options
73# with arguments, and if -e was supplied
74
75while [ $# -gt 0 -a ${endofopts} -eq 0 ]
76do
77    case $1 in
78    # from GNU grep-2.5.1 -- keep in sync!
79	--)
80	    shift
81	    endofopts=1
82	    ;;
83	--file=*)
84	    pattern_file=1
85	    grep_args="${grep_args} ${1}"
86	    shift
87	    ;;
88	--regexp=*)
89	    pattern="${1#--regexp=}"
90	    pattern_found=1
91	    shift
92	    ;;
93	-h|--no-filename)
94	    silent=1
95	    shift
96	    ;;
97	-V|--version)
98	    exec ${grep} -V
99	    ;;
100	--*)
101	    grep_args="${grep_args} $1"
102	    shift
103	    ;;
104	-*[ABCDXdefm])
105	    if [ $# -lt 2 ]
106		then
107		echo "${prg}: missing argument for $1 flag" >&2
108		exit 1
109	    fi
110	    case $1 in
111		-*e)
112		    pattern="$2"
113		    pattern_found=1
114		    shift 2
115		    continue
116		    ;;
117		-*f)
118		    pattern_file=1
119		    ;;
120		*)
121		    ;;
122	    esac
123	    grep_args="${grep_args} $1 $2"
124	    shift 2
125	    ;;
126	-)
127	    hyphen=1
128	    shift
129	    ;;
130	-r|-R)
131	    echo "${prg}: the ${1} flag is not currently supported" >&2
132	    exit 1
133	    ;;
134	-*)
135	    grep_args="${grep_args} $1"
136	    shift
137	    ;;
138	*)
139	    # pattern to grep for
140	    endofopts=1
141	    ;;
142    esac
143done
144
145# if no -e option was found, take next argument as grep-pattern
146if [ ${pattern_file} -eq 0 -a ${pattern_found} -eq 0 ]
147then
148    if [ $# -ge 1 ]; then
149	pattern="$1"
150	shift
151    elif [ ${hyphen} -gt 0 ]; then
152	pattern="-"
153    else
154	echo "${prg}: missing pattern" >&2
155	exit 1
156    fi
157    pattern_found=1
158fi
159
160# call grep ...
161if [ $# -lt 1 ]
162then
163    # ... on stdin
164    if [ ${pattern_file} -eq 0 ]; then
165	${cattool} ${catargs} - | ${grep} ${grep_args} -- "${pattern}" -
166    else
167	${cattool} ${catargs} - | ${grep} ${grep_args} -- -
168    fi
169    ret=$?
170else
171    # ... on all files given on the command line
172    if [ ${silent} -lt 1 -a $# -gt 1 ]; then
173	grep_args="-H ${grep_args}"
174    fi
175    # Succeed if any file matches.  First assume no match.
176    ret=1
177    for file; do
178	if [ ${pattern_file} -eq 0 ]; then
179	    ${cattool} ${catargs} -- "${file}" |
180		${grep} --label="${file}" ${grep_args} -- "${pattern}" -
181	else
182	    ${cattool} ${catargs} -- "${file}" |
183		${grep} --label="${file}" ${grep_args} -- -
184	fi
185	this_ret=$?
186	# A match (0) overrides a no-match (1).  An error (>=2) overrides all.
187	if [ ${this_ret} -eq 0 -a ${ret} -eq 1 ] || [ ${this_ret} -ge 2 ]; then
188	    ret=${this_ret}
189	fi
190    done
191fi
192
193exit ${ret}
194