色んなデータを柔軟に絞り込みしたい

各種のデータをパイプで処理するとき、適当にワードで絞り込んでからpecoでインクリメンタルサーチしたいことがよくある。これを簡単にできるようにしたい。

想定される用途はこんな感じ。

$ ls | grepeco hoge

パイプで与えられたデータを、ターミナルから入力された文字列で絞り込む。

パイプからの入力とターミナルからの入力を区別する

ここで、パイプで与えられたデータとはシェルスクリプトの標準入力、ターミナルから入力された文字列とは引数である。

シェルスクリプト内において、この二つは以下のように得ることができる。

# 標準入力を得る
cat -
# 引数を得る
echo "$1" # 1つめ
echo "$*" # すべて（1つの文字列として）
for i in "$@" # すべて（配列として）
do
  echo "arg: $i"
done

この場合、引数を配列として扱うのであれば、複数の検索語による、複数の検索結果からさらに絞り込むことができる。

コード

以上を踏まえて、コードは以下のようになる。

grepeco.sh

#!/bin/bash
# 引数を単一の文字列として扱う
# 変数を初期化
pipes=()
prompt="select stdin from pipe"
optm=""
usage () {
  echo "Usage: grepeco [-1h] [-p prompt] <args>" 1>&2
  echo "  -1: find only one" 1>&2
  echo "  -p: change peco prompt" 1>&2
  echo "  -h: view help" 1>&2
}
while getopts "1p:h" OPT
do
  case $OPT in
    1)
      optm="m1" # grepのオプションを変更
      ;;
    p)
      prompt="$OPTARG" # pecoのプロンプトを変更
      ;;
    h)
      usage
      exit 0
      ;;
    ?)
      echo "ERROR: Undifined option" 1>&2
      usage
      exit 1
      ;;
  esac
done
shift $(($OPTIND - 1))
if [ -p /dev/stdin ];then # パイプ入力があるか判断
  while read pipe
  do
    pipes+=("$pipe")
  done < <(if [ -n "$1" ]; then # 引数があるか判断
    cat - | grep -i"$optm" "$*"
  else
    cat -
  fi)
else
  echo "ERROR: No stdin from pipe" 1>&2
  exit 1
fi
case  "${#pipes[@]}" in # pipesの要素数による分岐
  0 )
    echo "ERROR: No hit." 1>&2
    exit 1
    ;;
  1)
    echo "${pipes[*]}"
    ;;
  *)
    for i in "${pipes[@]}"
    do
      echo "$i"
    done | peco --prompt "$prompt"
    ;;
esac

grepeco.sh

#!/bin/bash
# 引数を配列として扱う
-  done < <(if [ -n "$1" ]; then
-    cat - | grep -i"$optm" "$*"
-  else
-    cat -
-  fi)

+  done < <(if [ -n "$1" ]; then
+    stdin="$(cat -)" # ループで標準出力がリセットされるため変数に格納
+    for arg in "$@"
+    do
+      echo "$stdin" | grep -i "$arg"
+    done | sort -u # 重複する結果を削除
+  else
+    cat -
+  fi)

if文の結果をwhile文に渡すということを思いついてコードがシンプルになった。

参考