From 1cc32be5f632ff933b3e3f1c7bd9e612a274c095 Mon Sep 17 00:00:00 2001
From: ajatkj <ajatkj@yahoo.co.in>
Date: Wed, 20 Sep 2023 20:21:27 +0530
Subject: [PATCH] feat(poetry-env): create plugin (#11069)

---
 plugins/poetry-env/README.md             | 10 +++++++++
 plugins/poetry-env/poetry-env.plugin.zsh | 27 ++++++++++++++++++++++++
 2 files changed, 37 insertions(+)
 create mode 100644 plugins/poetry-env/README.md
 create mode 100644 plugins/poetry-env/poetry-env.plugin.zsh

diff --git a/plugins/poetry-env/README.md b/plugins/poetry-env/README.md
new file mode 100644
index 000000000..a7d16563e
--- /dev/null
+++ b/plugins/poetry-env/README.md
@@ -0,0 +1,10 @@
+# Poetry Environment Plugin
+
+This plugin automatically changes poetry environment when you cd into or out of the project directory. 
+Note: Script looks for pyproject.toml file to determine poetry if its a poetry environment
+
+To use it, add `poetry-env` to the plugins array in your zshrc file:
+
+```zsh
+plugins=(... poetry-env)
+```
diff --git a/plugins/poetry-env/poetry-env.plugin.zsh b/plugins/poetry-env/poetry-env.plugin.zsh
new file mode 100644
index 000000000..86e5fad4e
--- /dev/null
+++ b/plugins/poetry-env/poetry-env.plugin.zsh
@@ -0,0 +1,27 @@
+# Automatic poetry environment activation/deactivation
+_togglePoetryShell() {
+  # deactivate environment if pyproject.toml doesn't exist and not in a subdir
+  if [[ ! -f "$PWD/pyproject.toml" ]] ; then
+    if [[ "$poetry_active" == 1 ]]; then
+      if [[ "$PWD" != "$poetry_dir"* ]]; then
+        export poetry_active=0
+        deactivate
+        return
+      fi
+    fi
+  fi
+
+  # activate the environment if pyproject.toml exists
+  if [[ "$poetry_active" != 1 ]]; then
+    if [[ -f "$PWD/pyproject.toml" ]]; then
+      if grep -q 'tool.poetry' "$PWD/pyproject.toml"; then
+        export poetry_active=1
+        export poetry_dir="$PWD"
+        source "$(poetry env info --path)/bin/activate"
+      fi
+    fi
+  fi
+}
+autoload -U add-zsh-hook
+add-zsh-hook chpwd _togglePoetryShell
+_togglePoetryShell