blob: 57b19d4cb01cee77db17c7087d400e587af0a2aa (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
|
#!/bin/sh
# Usage function to display script options
usage() {
echo "Find files using ripgrep and open them in Neovim."
echo ""
echo "Usage: ${0##*/} [-s] [-i] [-l] [-p] [<tag>] <query>"
echo ""
echo "Options:"
echo " -h : Show this message"
echo " -i : Perform a case-insensitive search (default)"
echo " -l : List files associated with the given tag"
echo " -p : Search for files in the specified project directories using the specified tag (default: PROJECT)"
echo " -s : Perform a case-sensitive search"
echo " [<tag>] <query> : Optional tag for project mode, followed by the search query"
echo ""
echo "Examples:"
echo " ${0##*/} -p TODO 'KEYWORD' # Search for 'KEYWORD' in files tagged with 'TODO' in the project directories"
echo " ${0##*/} -l -p 'KEYWORD' # List files associated with the default 'PROJECT' tag and 'KEYWORD'"
echo " ${0##*/} 'KEYWORD' # Open files containing 'KEYWORD' in nvim"
exit 0
}
search_term() {
case_flag="$1"
shift
if ! command -v rga >/dev/null 2>&1; then
echo "Error: 'rga' is not installed." >&2
exit 1
fi
if ! command -v xclip >/dev/null 2>&1; then
echo "Error: 'xclip' is not installed." >&2
exit 1
fi
# Construct the preview command
preview_cmd=$(printf "rga %s --pretty --context 10 '%s' {}" "$case_flag" "$*")
query="$*"
rga_output=$(rga --follow --no-ignore --hidden --text --max-count=1 ${case_flag:+$case_flag} --files-with-matches --no-messages \
--glob '!**/.cache/*' \
--glob '!**/.git/*' \
--glob '!**/.github/*' \
--glob '!**/.next/*' \
--glob '!**/.venv/*' \
--glob '!**/build/*' \
--glob '!**/coverage/*' \
--glob '!**/dist/*' \
--glob '!**/node_modules/*' \
--glob '!**/target/*' \
--glob '!**/vendor/*' \
--glob '!**/venv/*' \
"$query")
# Use fzf to select files
files=$(echo "$rga_output" | fzf-tmux +m --preview="$preview_cmd" --reverse --multi --select-1 --exit-0) || return 1
# Check if files are selected
if [ -z "$files" ]; then
echo "No files selected."
return 0
fi
# copy target to the clipboard
echo "$query" | xclip -selection clipboard 2>/dev/null
# Open files at matching lines
count=$(echo "$files" | wc -l)
if [ "$count" -eq 1 ]; then
# Find the first matching line number in the selected file
line=$(rga --max-count=1 --line-number --no-filename ${case_flag:+$case_flag} --no-messages "$query" "$files" 2>/dev/null | head -1 | cut -d: -f1)
if [ -n "$line" ]; then
${EDITOR:-nvim} "+${line}" "$(realpath "$files")"
else
openfiles "$files"
fi
else
# Multiple files: open at matching lines via quickfix
tmpfile=$(mktemp)
echo "$files" | while IFS= read -r file; do
match=$(rga --max-count=1 --line-number --no-filename ${case_flag:+$case_flag} --no-messages "$query" "$file" 2>/dev/null | head -1)
if [ -n "$match" ]; then
printf '%s:%s\n' "$(realpath "$file")" "$match" >> "$tmpfile"
else
printf '%s:1:\n' "$(realpath "$file")" >> "$tmpfile"
fi
done
if [ "$count" -lt 5 ]; then
${EDITOR:-nvim} -q "$tmpfile" \
-c 'for i in range(2,len(getqflist())) | vsplit | execute "cc ".i | endfor | wincmd t'
else
first_line=$(head -1 "$tmpfile" | cut -d: -f2)
file_args=""
while IFS=: read -r fpath _; do
file_args="$file_args \"$fpath\""
done < "$tmpfile"
eval "${EDITOR:-nvim}" "+${first_line}" "$file_args"
fi
rm -f "$tmpfile"
fi
# print the file names
echo "$rga_output"
}
# Function to list and/or open all files associated with a given project tag
list_or_open_project_files() {
# Use the provided tag or default to "PROJECT"
project_tag="${1:-PROJECT}: $2"
# Define the project paths as a space-separated string
project_paths="$HOME/.dotfiles $HOME/.local/src/suckless $HOME/Public/repos"
# Use rga to find files containing the project tag across all project paths
rga_output=""
for path in $project_paths; do
if [ -d "$path" ]; then
rga_result=$(rga --follow --no-ignore --hidden --text --max-count=1 --files-with-matches --no-messages \
--glob '!**/.cache/*' \
--glob '!**/.git/*' \
--glob '!**/.github/*' \
--glob '!**/.next/*' \
--glob '!**/.venv/*' \
--glob '!**/build/*' \
--glob '!**/coverage/*' \
--glob '!**/dist/*' \
--glob '!**/node_modules/*' \
--glob '!**/target/*' \
--glob '!**/vendor/*' \
--glob '!**/venv/*' \
"$project_tag" "$path")
rga_output="$rga_output $rga_result"
fi
done
# Remove leading/trailing whitespace
rga_output=$(echo "$rga_output" | sed 's/^[[:space:]]*//;s/[[:space:]]*$//')
# Check if any files were found
if [ -z "$rga_output" ]; then
echo "No files found for tag $project_tag."
return 0
fi
# If the script was called in list mode, simply print the files
if [ "$list_mode" -eq 1 ]; then
echo "$rga_output"
else
# Open files at matching lines
file_count=$(echo "$rga_output" | wc -w)
if [ "$file_count" -eq 1 ]; then
line=$(rga --max-count=1 --line-number --no-filename --no-messages "$project_tag" "$rga_output" 2>/dev/null | head -1 | cut -d: -f1)
if [ -n "$line" ]; then
${EDITOR:-nvim} "+${line}" "$(realpath "$rga_output")"
else
openfiles "$rga_output"
fi
else
# Multiple files: open at matching lines via quickfix
tmpfile=$(mktemp)
for file in $rga_output; do
match=$(rga --max-count=1 --line-number --no-filename --no-messages "$project_tag" "$file" 2>/dev/null | head -1)
if [ -n "$match" ]; then
printf '%s:%s\n' "$(realpath "$file")" "$match" >> "$tmpfile"
else
printf '%s:1:\n' "$(realpath "$file")" >> "$tmpfile"
fi
done
if [ "$file_count" -lt 5 ]; then
${EDITOR:-nvim} -q "$tmpfile" \
-c 'for i in range(2,len(getqflist())) | vsplit | execute "cc ".i | endfor | wincmd t'
else
${EDITOR:-nvim} -q "$tmpfile" \
-c 'silent! for i in range(2,len(getqflist())+1) | execute "silent! cc ".i | endfor | silent! cfirst'
fi
rm -f "$tmpfile"
fi
fi
}
# Main function to handle options
case_flag="--ignore-case" # Default to case-insensitive
list_mode=0
project_mode=0
# Parse the options
while getopts "silph" opt; do
case $opt in
s) case_flag="--case-sensitive" ;; # Case-sensitive
i) case_flag="--ignore-case" ;; # Case-insensitive
l) list_mode=1 ;; # List mode
p) project_mode=1 ;; # Project mode
h) usage ;;
*) ;;
esac
done
shift $((OPTIND - 1))
# Handle project mode search
if [ "$project_mode" -eq 1 ]; then
list_or_open_project_files "$1" "$2"
else
# Otherwise, call the common search function
search_term "$case_flag" "$@"
fi
|
