summaryrefslogtreecommitdiff
path: root/ar/.local/bin/rgafiles
blob: fe61604669a0a103492e740c0981528cc3a5e6d0 (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
#!/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" "$*")
  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/*' \
    "$*")

  # 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 "$@" | xclip -selection clipboard 2>/dev/null

  openfiles "$files"

  # 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
    # Otherwise, open the files with nvim
    set -- "$(printf "%s\n" "$rga_output")"
    openfiles "$@"
  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