From ffb6d0b2331ad5dfbdeb4eb7d68336f339289774 Mon Sep 17 00:00:00 2001
From: Alexander Barton <alex@barton.de>
Date: Wed, 6 Feb 2019 23:42:42 +0100
Subject: [PATCH] virtualenv: Add support for "pipenv"

---
 plugins/virtualenv/virtualenv.zshrc | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/plugins/virtualenv/virtualenv.zshrc b/plugins/virtualenv/virtualenv.zshrc
index 436c667..43614af 100644
--- a/plugins/virtualenv/virtualenv.zshrc
+++ b/plugins/virtualenv/virtualenv.zshrc
@@ -5,6 +5,18 @@
 [[ -z "$AXZSH_PLUGIN_CHECK" ]] || return 92
 
 function activate() {
+	local d r
+
+	# Make sure no "virtual environment" is already active!
+	if [[ -n "$VIRTUAL_ENV" ]]; then
+		echo "Oops, looks like a virtual environment is already active!" >&2
+		return 1
+	fi
+
+	if [[ -r Pipfile ]]; then
+		pipenv run "$SHELL"; r=$?
+		return $r
+	fi
 	for d (
 		./bin
 		./env/bin
@@ -20,6 +32,11 @@ function activate() {
 		echo "Failed to read script \"$script\"!" >&2
 		return 1
 	done
+
 	echo "No virtual environment found!" >&2
 	return 1
 }
+
+if [[ -n "$PIPENV_ACTIVE" ]]; then
+	alias deactivate=exit
+fi
-- 
2.39.2