From 79263e5833aafd72f7cf8f42487147da378c2b80 Mon Sep 17 00:00:00 2001 From: Fazle Arefin Date: Mon, 3 Aug 2026 19:52:33 +1000 Subject: [PATCH] feat(uv-env): create plugin This plugin automatically activates/deactivates a uv-managed virtual environment when you cd into or out of a project directory. It looks for pyproject.toml and uv.lock to detect a uv project, and reads UV_PROJECT_ENVIRONMENT to support custom venv locations. --- plugins/uv-env/README.md | 10 ++++++++++ plugins/uv-env/uv-env.plugin.zsh | 26 ++++++++++++++++++++++++++ 2 files changed, 36 insertions(+) create mode 100644 plugins/uv-env/README.md create mode 100644 plugins/uv-env/uv-env.plugin.zsh diff --git a/plugins/uv-env/README.md b/plugins/uv-env/README.md new file mode 100644 index 000000000..8502e80a9 --- /dev/null +++ b/plugins/uv-env/README.md @@ -0,0 +1,10 @@ +# uv Environment Plugin + +This plugin automatically activates the uv virtual environment when you cd into a project directory, and deactivates it when you cd out. +Note: Script looks for pyproject.toml and uv.lock files to determine if it's a uv project + +To use it, add `uv-env` to the plugins array in your zshrc file: + +```zsh +plugins=(... uv-env) +``` diff --git a/plugins/uv-env/uv-env.plugin.zsh b/plugins/uv-env/uv-env.plugin.zsh new file mode 100644 index 000000000..2c4f8437f --- /dev/null +++ b/plugins/uv-env/uv-env.plugin.zsh @@ -0,0 +1,26 @@ +_toggleUvShell() { + local in_uv_dir=0 + if [[ -f "$PWD/pyproject.toml" && -f "$PWD/uv.lock" ]]; then + in_uv_dir=1 + fi + + if [[ $uv_active -eq 1 ]] && { [[ $in_uv_dir -eq 0 ]] || [[ "$PWD" != "$uv_dir" && "$PWD" != "$uv_dir"/* ]]; }; then + export uv_active=0 + unset uv_dir + (( $+functions[deactivate] )) && deactivate + fi + + if [[ $in_uv_dir -eq 1 ]] && [[ $uv_active -ne 1 ]]; then + venv_dir="${UV_PROJECT_ENVIRONMENT:-$PWD/.venv}" + [[ "$venv_dir" != /* ]] && venv_dir="$PWD/$venv_dir" + + if [[ -f "${venv_dir}/bin/activate" ]]; then + export uv_active=1 + export uv_dir="$PWD" + source "${venv_dir}/bin/activate" + fi + fi +} +autoload -U add-zsh-hook +add-zsh-hook chpwd _toggleUvShell +_toggleUvShell