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