From 30dc39fff6746878d1d29ea52dab46218f9474b8 Mon Sep 17 00:00:00 2001
From: Jonathan Hodgson <git@jonathanh.co.uk>
Date: Wed, 23 Oct 2019 20:31:35 +0100
Subject: [PATCH] Work on zsh fzf stuff

---
 shells/zsh/STOW/.zshrc              | 24 ++++----
 shells/zsh/oh-my-zsh-custom/fzf.zsh | 86 +++++++++++++++++++++++++++--
 2 files changed, 94 insertions(+), 16 deletions(-)

diff --git a/shells/zsh/STOW/.zshrc b/shells/zsh/STOW/.zshrc
index 6f92c33c..e319ca56 100644
--- a/shells/zsh/STOW/.zshrc
+++ b/shells/zsh/STOW/.zshrc
@@ -108,16 +108,16 @@ TRAPWINCH(){
 #fi
 #
 
-local current="$(project current --path)"
-if [ -n "$current" ]; then
-	echo "yay"
-	local script="/usr/bin/script"
-	echo "'$(ps -ocommand -p $PPID  | grep -v 'COMMAND' | cut -d' ' -f1)'"
-	if [[ ! "$(ps -ocommand -p $PPID | grep -v 'COMMAND' | cut -d' ' -f1 )" == "$script" ]]; then
-		mkdir "$current/shell-logs"
-		/usr/bin/script -f "$current/shell-logs/$(date +"%d-%b-%y_%H-%M-%S")_shell.log"
-	fi
-fi
+#local current="$(project current --path)"
+#if [ -n "$current" ]; then
+#	echo "yay"
+#	local script="/usr/bin/script"
+#	echo "'$(ps -ocommand -p $PPID  | grep -v 'COMMAND' | cut -d' ' -f1)'"
+#	if [[ ! "$(ps -ocommand -p $PPID | grep -v 'COMMAND' | cut -d' ' -f1 )" == "$script" ]]; then
+#		mkdir "$current/shell-logs"
+#		/usr/bin/script -f "$current/shell-logs/$(date +"%d-%b-%y_%H-%M-%S")_shell.log"
+#	fi
+#fi
 
-clear
-tldr --linux -r
+#clear
+#tldr --linux -r
diff --git a/shells/zsh/oh-my-zsh-custom/fzf.zsh b/shells/zsh/oh-my-zsh-custom/fzf.zsh
index b39f1cea..0787cda7 100644
--- a/shells/zsh/oh-my-zsh-custom/fzf.zsh
+++ b/shells/zsh/oh-my-zsh-custom/fzf.zsh
@@ -26,8 +26,8 @@ if [ -e /usr/share/fzf/key-bindings.zsh ]; then
 		zle reset-prompt
 		return $ret
 	}
-	zle     -N   fzf-wordlist-widget
-	bindkey '^W' fzf-wordlist-widget
+	#zle     -N   fzf-wordlist-widget
+	#bindkey '^W' fzf-wordlist-widget
 
 	# CTRL-P to select an IP address from project host
 	__fsel_ip() {
@@ -44,6 +44,84 @@ if [ -e /usr/share/fzf/key-bindings.zsh ]; then
 		zle reset-prompt
 		return $ret
 	}
-	zle     -N   fzf-ip-widget
-	bindkey '^P' fzf-ip-widget
+	#zle     -N   fzf-ip-widget
+	#bindkey '^P' fzf-ip-widget
+
+
+	# I want my tab complete to be based on "current" word I am typing sometimes, before the command
+	custom_tabcomplete(){
+		local tokens cmd prefix trigger tail fzf matches lbuf d_cmds
+		setopt localoptions noshwordsplit noksh_arrays noposixbuiltins
+
+		# http://zsh.sourceforge.net/FAQ/zshfaq03.html
+		# http://zsh.sourceforge.net/Doc/Release/Expansion.html#Parameter-Expansion-Flags
+		tokens=(${(z)LBUFFER})
+		if [ ${#tokens} -lt 1 ]; then
+			zle ${fzf_default_completion:-expand-or-complete}
+			return
+		fi
+		cmd=${tokens[1]}
+		tail=${LBUFFER:$(( ${#LBUFFER} - ${#trigger} ))}
+		local currentProject=$(project current --path)
+		#local newLBuffer="${tokens:1:${#tokens[@]}-1}"
+		local newLBuffer
+		for i in $(seq 1 $((${#tokens[@]} - 1)) ); do
+			newLBuffer="${newLBuffer}${tokens[i]} "
+		done
+		
+
+		# Some of my completions should only work when in a project
+		if [ -n "$currentProject" ]; then
+			if [[ "${LBUFFER[-1]}" == " " ]]; then
+				fzf-completion
+			else
+				case "${tokens[-1]}" in
+					ip)
+						LBUFFER="${newLBuffer}$(project hosts ip --fzf) "
+						local ret=$?
+						zle reset-prompt
+						return $ret
+						;;
+					pf)
+						LBUFFER="${newLBuffer}$(find "$currentProject" -type f | fzf) "
+						local ret=$?
+						zle reset-prompt
+						return $ret
+						;;
+					pd)
+						LBUFFER="${newLBuffer}$(find "$currentProject" -type d | fzf --no-preview) "
+						local ret=$?
+						zle reset-prompt
+						return $ret
+						;;
+					wl)
+						LBUFFER="${newLBuffer}$(__fsel_wordlist)"
+						local ret=$?
+						zle reset-prompt
+						return $ret
+						;;
+					*)
+						# Fall back to fzf completion
+						fzf-completion
+				esac
+			fi
+		else
+			case "${tokens[-1]}" in
+				wl)
+					LBUFFER="${newLBuffer}$(__fsel_wordlist)"
+					local ret=$?
+					zle reset-prompt
+					return $ret
+					;;
+				*)
+					# Fall back to fzf completion
+					fzf-completion
+			esac
+		fi
+	}
+	zle     -N   custom_tabcomplete
+	bindkey '^I' custom_tabcomplete
+
+
+
 fi