From df5df5e774a4e5fe2add3c922a42d891c75480cc Mon Sep 17 00:00:00 2001
From: Vladimir Bauer <vbauerster@gmail.com>
Date: Fri, 15 Nov 2024 20:49:55 +0500
Subject: [PATCH] kamp-grep: utilize getopts

---
 README.md         |  4 ++--
 scripts/kamp-grep | 47 ++++++++++++++++++++++++++++-------------------
 2 files changed, 30 insertions(+), 21 deletions(-)

diff --git a/README.md b/README.md
index 9b84f76..d689b06 100644
--- a/README.md
+++ b/README.md
@@ -58,8 +58,8 @@ alias global popup tmux-terminal-popup
 map global normal -docstring 'files'          <c-f> ':connect popup kamp-files<ret>'
 map global normal -docstring 'git ls-files'   <c-l> ':connect popup kamp-files -b git<ret>'
 map global normal -docstring 'buffers'        <c-b> ':connect popup kamp-buffers<ret>'
-map global normal -docstring 'grep selection' <c-g> ':connect popup kamp-grep ''query=%val{selection}<a-!>''<ret>'
-map global normal -docstring 'grep limit by filetype' <c-y> ':connect popup kamp-grep -t %opt{filetype}<ret>'
+map global normal -docstring 'grep selection' <c-g> ':connect popup kamp-grep -q ''%val{selection}<a-!>''<ret>'
+map global normal -docstring 'grep by filetype' <c-y> ':connect popup kamp-grep -- -t %opt{filetype}<ret>'
 ```
 
 ## Shell integration
diff --git a/scripts/kamp-grep b/scripts/kamp-grep
index c5c4774..c201bf1 100755
--- a/scripts/kamp-grep
+++ b/scripts/kamp-grep
@@ -2,10 +2,6 @@
 #
 # Open files by content.
 #
-# Usage:
-#
-# kamp-grep [<query=...>] [rg_options]
-
 # requires:
 # – fzf (https://github.com/junegunn/fzf)
   # https://github.com/junegunn/fzf/blob/master/CHANGELOG.md#0190
@@ -19,32 +15,45 @@ set -euf
 # define SHELL so --preview arguments do not error if current SHELL is not POSIX
 SHELL=/bin/sh
 
-query="$(kamp get opt kamp_grep_query)"
+usage() {
+    printf "Usage: %s: [-q query] [rg_options]\n" "$(basename "$0")" >&2
+    exit 2
+}
 
-if [ $# -ne 0 ]; then
-  case "$1" in
-    query=*)
-      query="${1#*=}"
-      shift
-    ;;
-  esac
-fi
+qflag=
+while getopts 'q:h' OPTION; do
+    case $OPTION in
+    q)
+        qflag=1
+        query="$OPTARG"
+        ;;
+    h|?)
+        usage
+        ;;
+    esac
+done
+shift $((OPTIND - 1))
 
-rg_cmd="rg $*"
+rg_cmd="rg --color always --column $*"
+
+if [ ! "$qflag" ]; then
+    query="$(kamp get opt kamp_grep_query)"
+fi
 
 if [ -z "$query" ]; then
-  export FZF_DEFAULT_COMMAND="$rg_cmd --files"
+    export FZF_DEFAULT_COMMAND="rg --files $*"
 else
-  export FZF_DEFAULT_COMMAND="$rg_cmd --color always --column -- '$query'"
+    pattern=$(printf %s "$query" | sed 's/"/\\"/')
+    export FZF_DEFAULT_COMMAND="$rg_cmd -- \"$pattern\""
 fi
 
 fzf \
   --phony \
-  --query "${query:-}" \
+  --query "$query" \
   --delimiter ':' \
   --ansi \
-  --bind "change:reload($rg_cmd --color always --column -- {q} || true)" \
-  --bind 'enter:execute-silent(kamp send set global kamp_grep_query "{q}")+become(kamp edit {1} +{2}:{3})' \
+  --bind "change:reload($rg_cmd -- {q} || true)" \
+  --bind 'enter:execute-silent(kamp send set global kamp_grep_query {q})+become(kamp edit {1} +{2}:{3})' \
   --preview '
       highlight_line={2}
       line_range_begin=$((highlight_line - FZF_PREVIEW_LINES / 2))