From e3085efc468b59f283534d9de717110a5b1585dd Mon Sep 17 00:00:00 2001
From: Julian Prein <druckdev@protonmail.com>
Date: Fri, 23 Jun 2023 23:11:43 +0200
Subject: [PATCH] zsh:diffcmds(): Handle some edge cases

---
 .config/zsh/zshrc.d/40-functions.zsh | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/.config/zsh/zshrc.d/40-functions.zsh b/.config/zsh/zshrc.d/40-functions.zsh
index d5475c4..3a98517 100644
--- a/.config/zsh/zshrc.d/40-functions.zsh
+++ b/.config/zsh/zshrc.d/40-functions.zsh
@@ -668,6 +668,24 @@ diffcmds() {
 		let i++
 	fi
 
+	# Just execute the command without *diff if there is only one argument
+	if (( i + 1 == # )); then
+		eval "${(q@)${@:1:$((i-1))}//\%\%/${@[$#]}}"
+		return
+	fi
+
+	# Fallback or abort if more than 2 arguments were supplied to `diff`
+	if [[ $cmd = diff ]] && (( # - i > 2 )); then
+		printf >&2 "Too many arguments for diff."
+		if (( $+commands[vimdiff] )); then
+			printf >&2 " Using vimdiff.\n"
+			cmd=vimdiff
+		else
+			printf >&2 "\n"
+			return 1
+		fi
+	fi
+
 	# NOTE: `=()` is necessary since vimdiff is seeking the file. See zshexpn(1)
 	[[ $cmd = vimdiff ]] && ps_sub='=(' || ps_sub='<('