Initial commit
This commit is contained in:
Zhongwei Li
278
scripts/combo-search.sh
Executable file
278
scripts/combo-search.sh
Executable file
@@ -0,0 +1,278 @@
|
||||
#!/usr/bin/env bash
|
||||
#
|
||||
# combo-search.sh - Example combination workflows using CLI tools
|
||||
#
|
||||
# This script demonstrates common patterns for combining fd, rg, ast-grep,
|
||||
# fzf, jq, and yq to solve real-world repository navigation tasks.
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
# Colors for output
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# Helper function to print section headers
|
||||
print_section() {
|
||||
echo -e "\n${BLUE}=== $1 ===${NC}\n"
|
||||
}
|
||||
|
||||
# Helper function to print commands before execution
|
||||
print_command() {
|
||||
echo -e "${YELLOW}$ $1${NC}"
|
||||
}
|
||||
|
||||
# 1. Find Python files with TODO comments, select one, and edit
|
||||
find_and_edit_todos() {
|
||||
print_section "Find and Edit TODOs"
|
||||
print_command "rg -l 'TODO' -t py | fzf --preview 'rg -C 3 --color=always TODO {}' | xargs \$EDITOR"
|
||||
|
||||
local file
|
||||
file=$(rg -l 'TODO' -t py | fzf --preview 'rg -C 3 --color=always TODO {}' || true)
|
||||
|
||||
if [[ -n "$file" ]]; then
|
||||
echo -e "${GREEN}Selected: $file${NC}"
|
||||
# Uncomment to actually open editor:
|
||||
# $EDITOR "$file"
|
||||
else
|
||||
echo -e "${RED}No file selected${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 2. Find function definitions, select one, and show its usages
|
||||
find_function_usages() {
|
||||
print_section "Find Function and Its Usages"
|
||||
|
||||
local func_pattern="def $1(\$\$\$):"
|
||||
|
||||
print_command "ast-grep --pattern '$func_pattern' -l py | fzf"
|
||||
|
||||
local result
|
||||
result=$(ast-grep --pattern "$func_pattern" -l py | fzf --preview 'bat --color=always {1}' || true)
|
||||
|
||||
if [[ -n "$result" ]]; then
|
||||
local func_name
|
||||
func_name=$(echo "$result" | sed -n 's/.*def \([^(]*\)(.*/\1/p')
|
||||
|
||||
echo -e "${GREEN}Function: $func_name${NC}"
|
||||
echo -e "${YELLOW}Usages:${NC}"
|
||||
|
||||
print_command "rg '\\b$func_name\\(' -t py"
|
||||
rg "\b$func_name\(" -t py || echo "No usages found"
|
||||
fi
|
||||
}
|
||||
|
||||
# 3. Find configuration files and edit selected one
|
||||
edit_config_file() {
|
||||
print_section "Find and Edit Configuration File"
|
||||
|
||||
print_command "fd -e json -e yaml -e toml config | fzf --preview 'bat --color=always {}'"
|
||||
|
||||
local config
|
||||
config=$(fd -e json -e yaml -e toml config | fzf --preview 'bat --color=always {}' || true)
|
||||
|
||||
if [[ -n "$config" ]]; then
|
||||
echo -e "${GREEN}Selected: $config${NC}"
|
||||
# Uncomment to actually open editor:
|
||||
# $EDITOR "$config"
|
||||
else
|
||||
echo -e "${RED}No config selected${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 4. Search for imports of a module and show file list
|
||||
find_module_imports() {
|
||||
print_section "Find Module Imports"
|
||||
|
||||
local module="$1"
|
||||
|
||||
print_command "ast-grep --pattern 'from $module import \$\$\$' -l py"
|
||||
print_command "ast-grep --pattern 'import $module' -l py"
|
||||
|
||||
echo -e "${YELLOW}Files importing $module:${NC}"
|
||||
{
|
||||
ast-grep --pattern "from $module import \$\$\$" -l py 2>/dev/null || true
|
||||
ast-grep --pattern "import $module" -l py 2>/dev/null || true
|
||||
} | sort -u
|
||||
}
|
||||
|
||||
# 5. Find recently modified files and select for editing
|
||||
edit_recent_file() {
|
||||
print_section "Find and Edit Recently Modified File"
|
||||
|
||||
print_command "fd -t f --changed-within 7d | fzf --preview 'bat --color=always {}'"
|
||||
|
||||
local file
|
||||
file=$(fd -t f --changed-within 7d | fzf --preview 'bat --color=always {}' || true)
|
||||
|
||||
if [[ -n "$file" ]]; then
|
||||
echo -e "${GREEN}Selected: $file${NC}"
|
||||
# Uncomment to actually open editor:
|
||||
# $EDITOR "$file"
|
||||
else
|
||||
echo -e "${RED}No file selected${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 6. Find JSON files and explore with jq
|
||||
explore_json_files() {
|
||||
print_section "Find and Explore JSON Files"
|
||||
|
||||
print_command "fd -e json | fzf --preview 'jq --color-output . {}'"
|
||||
|
||||
local json_file
|
||||
json_file=$(fd -e json | fzf --preview 'jq --color-output . {}' || true)
|
||||
|
||||
if [[ -n "$json_file" ]]; then
|
||||
echo -e "${GREEN}Selected: $json_file${NC}"
|
||||
echo -e "${YELLOW}Keys:${NC}"
|
||||
jq 'keys' "$json_file"
|
||||
else
|
||||
echo -e "${RED}No JSON file selected${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 7. Find test files with failures
|
||||
find_failing_tests() {
|
||||
print_section "Find Test Files with 'FAIL' or 'ERROR'"
|
||||
|
||||
print_command "rg -l '(FAIL|ERROR)' tests/ -t py | fzf --preview 'rg -C 5 --color=always \"(FAIL|ERROR)\" {}'"
|
||||
|
||||
local test_file
|
||||
test_file=$(rg -l '(FAIL|ERROR)' tests/ -t py 2>/dev/null | \
|
||||
fzf --preview 'rg -C 5 --color=always "(FAIL|ERROR)" {}' || true)
|
||||
|
||||
if [[ -n "$test_file" ]]; then
|
||||
echo -e "${GREEN}Selected: $test_file${NC}"
|
||||
else
|
||||
echo -e "${RED}No failing tests found or none selected${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# 8. Find and count function definitions by file
|
||||
count_functions_per_file() {
|
||||
print_section "Count Function Definitions Per File"
|
||||
|
||||
print_command "ast-grep --pattern 'def \$FUNC(\$\$\$):' -l py | xargs -I {} sh -c 'echo \"{}: \$(ast-grep --pattern \"def \\\$FUNC(\\\$\\\$\\\$):\" {} | wc -l)\"'"
|
||||
|
||||
ast-grep --pattern 'def $FUNC($$$):' -l py 2>/dev/null | \
|
||||
xargs -I {} sh -c 'echo "{}: $(ast-grep --pattern "def \$FUNC(\$\$\$):" {} | wc -l)"' | \
|
||||
sort -t: -k2 -nr | \
|
||||
head -10
|
||||
}
|
||||
|
||||
# 9. Find all TODO/FIXME comments and group by file
|
||||
list_todos_by_file() {
|
||||
print_section "List TODOs/FIXMEs Grouped by File"
|
||||
|
||||
print_command "rg -n '(TODO|FIXME|XXX|HACK):' | sort | fzf --preview 'bat --color=always {1} --highlight-line {2}'"
|
||||
|
||||
rg -n '(TODO|FIXME|XXX|HACK):' 2>/dev/null | \
|
||||
sort | \
|
||||
fzf --preview 'bat --color=always {1} --highlight-line {2}' || true
|
||||
}
|
||||
|
||||
# 10. Interactive refactoring: find pattern and preview replacement
|
||||
interactive_refactor() {
|
||||
print_section "Interactive Refactoring Preview"
|
||||
|
||||
local old_pattern="$1"
|
||||
local new_pattern="$2"
|
||||
|
||||
print_command "ast-grep --pattern '$old_pattern' -l py | fzf -m --preview 'ast-grep --pattern \"$old_pattern\" {}'"
|
||||
|
||||
local files
|
||||
files=$(ast-grep --pattern "$old_pattern" -l py 2>/dev/null | \
|
||||
fzf -m --preview "ast-grep --pattern '$old_pattern' {}" || true)
|
||||
|
||||
if [[ -n "$files" ]]; then
|
||||
echo -e "${GREEN}Selected files for refactoring:${NC}"
|
||||
echo "$files"
|
||||
echo -e "\n${YELLOW}To apply refactoring, run:${NC}"
|
||||
echo -e "ast-grep --pattern '$old_pattern' --rewrite '$new_pattern' -l py --interactive"
|
||||
else
|
||||
echo -e "${RED}No files selected${NC}"
|
||||
fi
|
||||
}
|
||||
|
||||
# Main menu
|
||||
show_menu() {
|
||||
echo -e "\n${BLUE}CLI Ninja - Combination Workflows${NC}\n"
|
||||
echo "1. Find and edit TODOs"
|
||||
echo "2. Find function usages (requires function name)"
|
||||
echo "3. Edit config file"
|
||||
echo "4. Find module imports (requires module name)"
|
||||
echo "5. Edit recently modified file"
|
||||
echo "6. Explore JSON files"
|
||||
echo "7. Find failing tests"
|
||||
echo "8. Count functions per file"
|
||||
echo "9. List TODOs/FIXMEs"
|
||||
echo "10. Interactive refactoring preview (requires old/new patterns)"
|
||||
echo "0. Exit"
|
||||
echo
|
||||
}
|
||||
|
||||
# Main execution
|
||||
main() {
|
||||
if [[ $# -eq 0 ]]; then
|
||||
show_menu
|
||||
read -rp "Select workflow: " choice
|
||||
|
||||
case $choice in
|
||||
1) find_and_edit_todos ;;
|
||||
2)
|
||||
read -rp "Enter function name: " func
|
||||
find_function_usages "$func"
|
||||
;;
|
||||
3) edit_config_file ;;
|
||||
4)
|
||||
read -rp "Enter module name: " module
|
||||
find_module_imports "$module"
|
||||
;;
|
||||
5) edit_recent_file ;;
|
||||
6) explore_json_files ;;
|
||||
7) find_failing_tests ;;
|
||||
8) count_functions_per_file ;;
|
||||
9) list_todos_by_file ;;
|
||||
10)
|
||||
read -rp "Enter old pattern: " old
|
||||
read -rp "Enter new pattern: " new
|
||||
interactive_refactor "$old" "$new"
|
||||
;;
|
||||
0) exit 0 ;;
|
||||
*) echo -e "${RED}Invalid choice${NC}" ;;
|
||||
esac
|
||||
else
|
||||
# Direct command execution
|
||||
case "$1" in
|
||||
todos) find_and_edit_todos ;;
|
||||
function)
|
||||
[[ $# -ge 2 ]] || { echo "Usage: $0 function <name>"; exit 1; }
|
||||
find_function_usages "$2"
|
||||
;;
|
||||
config) edit_config_file ;;
|
||||
imports)
|
||||
[[ $# -ge 2 ]] || { echo "Usage: $0 imports <module>"; exit 1; }
|
||||
find_module_imports "$2"
|
||||
;;
|
||||
recent) edit_recent_file ;;
|
||||
json) explore_json_files ;;
|
||||
failing) find_failing_tests ;;
|
||||
count) count_functions_per_file ;;
|
||||
list-todos) list_todos_by_file ;;
|
||||
refactor)
|
||||
[[ $# -ge 3 ]] || { echo "Usage: $0 refactor <old> <new>"; exit 1; }
|
||||
interactive_refactor "$2" "$3"
|
||||
;;
|
||||
*)
|
||||
echo "Usage: $0 [command] [args]"
|
||||
echo "Run without arguments for interactive menu"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
fi
|
||||
}
|
||||
|
||||
main "$@"
|
||||
Reference in New Issue
Block a user