#!/bin/sh
set -ef

# Requirements for this script:
#
# 1. When the external search program (e.g. `rga`, `rg`) is not installed, 
#    exit with status 127. `kio-filenamesearch` (called by the Dolphin search)
#    will then use the its (slower) fallback search code.
#
# 2. STDOUT should only contain paths separated by '\0'. Paths can be either
#    absolute or relative to the current directory.
#
# 3. If search is successful, either the exit code should be 0, or STDERR should
#    not contain any output. Because `rg` returns 1 when no matches are found,
#    and returns 2 when there are broken symlinks or permission errors, we can't
#    rely solely on the exit code being 0 to determine success. So we also considers
#    a non-zero exit code AND an empty STDERR to be a success.
#
# 4. So, when the search fails (other than the "not installed" case in point 1),
#    in addition for exiting with a non-zero status, you should also print an error
#    message to STDERR. The message will be displayed to the user in Dolphin.

case $1 in
    --check)
        if rga --version 2>/dev/null | grep -q 'ripgrep-all'; then
            exit 0
        elif rg --version 2>/dev/null | grep -q 'ripgrep'; then
            exit 0
        else
            exit 127
        fi
        ;;

    --run)
        # These options are required for the search to work.
        REQUIRED_OPTIONS="--files-with-matches --null --no-messages"
        
        # You can modify these options.
        # -j2:             Limit search to 2 threads. Remove this to get maximum performance on SSD, but it will hurt performance badly on HDD.
        # --line-buffered: Show each result in Dolphin as soon as it is found. Removing this might speed up the search a tiny bit, but
        #                  the first result might appear later.
        # -L:              Follow symlinks.
        # -i:              Make search case insensitive
        # -s:              Make search case sensitive
        # -S:              Do "smart" care matching (a lower case search string does a case
        #                  insensitive search, including upper case makes the search case sensitive).
        # --hidden:        Don't skip hidden files and directories.
        # --no-ignore:     Don't respect .gitignore/.ignore settings.
        OPTIONS="-j2 --line-buffered -L -i --hidden --no-ignore"

	# If the working drectory is root, exclude the directories "/dev", "/proc" and "/sys".
	PWD=`pwd`
	case $PWD in
	    ("/dev"|"/proc"|"/sys")
	        echo "Invalid directory:" $PWD 1>&2
	        exit 1;;
            ("/")
                EXCL="--glob !/dev/ --glob !/sys/ --glob !/proc/"
		;;
        esac
        
        if rga --version 2>/dev/null | grep -q 'ripgrep-all'; then
            exec rga $OPTIONS $EXCL $REQUIRED_OPTIONS -e "$2" .
        elif rg --version 2>/dev/null | grep -q 'ripgrep'; then
            exec rg $OPTIONS $EXCL $REQUIRED_OPTIONS -e "$2" .
        else
            exit 127
        fi
        ;;

    *)
        echo "Invalid argument: $1" >&2
        exit 1
        ;;
esac