#
# fzy shell integration
#
#     Adapted from fzf (https://github.com/junegunn/fzf).
#
# Copyright (c) 2016 Junegunn Choi
# Copyright (c) 2016 Alexis Sellier
#
if [[ $- != *i* ]]; then
	return
fi

__fzy_fsel () {
	setopt localoptions pipefail 2> /dev/null
	command rg --files -u . * | fzy | while read -r item; do
		echo -n "${(q)item}"
	done
	local ret=$?
	echo
	return $ret
}

fzy-file-widget () {
	LBUFFER="${LBUFFER}$(__fzy_fsel)"
	local ret=$?
	zle redisplay
	if [[ $ret == 0 ]]; then
		zle accept-line "$LBUFFER"
		ret=$?
	fi
	typeset -f zle-line-init >/dev/null && zle zle-line-init
	return $ret
}
zle     -N   fzy-file-widget
bindkey '^P' fzy-file-widget

fzy-history-widget () {
	local selected num
	setopt localoptions noglobsubst pipefail 2> /dev/null
	selected=( $(fc -l -r 1 | fzy -q "${LBUFFER//$/\\$}") )
	local ret=$?
	if [[ -n ${selected} ]]; then
		num=${selected[1]}
		if [[ -n ${num} ]]; then
			zle vi-fetch-history -n ${num}
			zle accept-line "$LBUFFER"
		fi
	fi
	zle redisplay
	typeset -f zle-line-init >/dev/null && zle zle-line-init
	return $ret
}
zle     -N   fzy-history-widget
bindkey '^H' fzy-history-widget