#!/bin/bash
 
SEARCH_DIR="/mnt/your-partition"
OUTPUT_DIR="/mnt/your-partition/results"
mkdir -p "$OUTPUT_DIR"
 
terms=()
 
echo "=== RG Search Tool ==="
echo "Search directory: $SEARCH_DIR"
echo "Tip: Use + to combine terms (e.g. football+poker matches lines containing BOTH)"
echo ""
 
while true; do
    read -p "Enter search term (or term1+term2 for combo): " term
    terms+=("$term")
 
    while true; do
        echo ""
        echo "Current terms: ${terms[*]}"
        echo "1) Add another term"
        echo "2) Start searching"
        read -p "Choice: " choice
 
        if [[ "$choice" == "1" ]]; then
            break
        elif [[ "$choice" == "2" ]]; then
            break 2
        else
            echo "Invalid choice, enter 1 or 2"
        fi
    done
done
 
echo ""
echo "=== Searching all terms in one pass... ==="
 
# Build pattern - handle + combos as lookaheads so both must appear in any order
build_pattern() {
    local term="$1"
    if [[ "$term" == *"+"* ]]; then
        IFS='+' read -ra parts <<< "$term"
        local pattern=""
        for part in "${parts[@]}"; do
            pattern+="(?=.*${part})"
        done
        echo "${pattern}.*"
    else
        echo "$term"
    fi
}
 
# Run one rg pass per unique pattern (combos cant be merged into one pipe easily)
for term in "${terms[@]}"; do
    clean_name="${term//[^a-zA-Z0-9]/_}"
    output_file="$OUTPUT_DIR/${clean_name}.txt"
    pattern=$(build_pattern "$term")
 
    echo "Searching: $term (pattern: $pattern)"
    rg -iP "$pattern" "$SEARCH_DIR" --no-filename --no-line-number | sort -u > "$output_file"
 
    count=$(wc -l < "$output_file")
    echo "  -> Found $count results -> ${clean_name}.txt"
done
 
echo ""
echo "=== Done! Results saved to $OUTPUT_DIR ==="