Zsh/Nvim: python is now mapped to uv run python using Zsh widgets and the compiler script for Nvim

.config/zsh/.zshrc

@@ -2,16 +2,16 @@
 # HELPER FUNCTIONS #
 ####################
 
-precmd_vcs_info() { vcs_info }
+precmd_vcs_info() { vcs_info; }
 
-precmd_functions+=( precmd_vcs_info )
+precmd_functions+=(precmd_vcs_info)
 
 zle-line-init() {
     zle -K viins
     echo -ne "\e[5 q"
 }
 
-preexec() { echo -ne '\e[5 q' ;}
+preexec() { echo -ne '\e[5 q'; }
 
 local copy_widgets=(
     vi-yank vi-yank-eol vi-delete vi-backward-kill-word vi-change-whole-line
@@ -59,23 +59,23 @@ function +vi-git-st() {
     local ahead behind remote
     local -a gitstatus
     remote=${$(git rev-parse --verify ${hook_com[branch]}@{upstream} \
-        --symbolic-full-name 2>/dev/null)/refs\/remotes\/}
+        --symbolic-full-name 2>/dev/null)/refs\/remotes\//}
-    if [[ -n ${remote} ]] ; then
+    if [[ -n ${remote} ]]; then
         # for git prior to 1.7
         # ahead=$(git rev-list origin/${hook_com[branch]}..HEAD | wc -l)
         ahead=$(git rev-list ${hook_com[branch]}@{upstream}..HEAD 2>/dev/null | wc -l)
-        (( $ahead )) && gitstatus+=( " ${c3}+${ahead}${c2}" )
+        (($ahead)) && gitstatus+=(" ${c3}+${ahead}${c2}")
         # for git prior to 1.7
         # behind=$(git rev-list HEAD..origin/${hook_com[branch]} | wc -l)
         behind=$(git rev-list HEAD..${hook_com[branch]}@{upstream} 2>/dev/null | wc -l)
-        (( $behind )) && gitstatus+=( "${c4}-${behind}${c2}" )
+        (($behind)) && gitstatus+=("${c4}-${behind}${c2}")
         hook_com[branch]="${hook_com[branch]} [${remote}${(j:/:)gitstatus}]"
     fi
 }
 
 function +vi-git-stash() {
     local -a stashes
-    if [[ -s ${hook_com[base]}/.git/refs/stash ]] ; then
+    if [[ -s ${hook_com[base]}/.git/refs/stash ]]; then
         stashes=$(git stash list 2>/dev/null | wc -l)
         hook_com[misc]+=" (${stashes} stashed)"
     fi
@@ -113,6 +113,49 @@ uv() {
     fi
 }
 
+find_uv_root() {
+    local target="$1"
+    local dir
+
+    if [[ -n "$target" && -e "$target" ]]; then
+        dir="$(cd "$(dirname "$target")" && pwd)"
+    else
+        dir="$PWD"
+    fi
+
+    while [[ "$dir" != "/" ]]; do
+        if [[ -f "$dir/.python-version" ]] || [[ -f "$dir/pyproject.toml" ]]; then
+            echo "$dir"
+            return 0
+        fi
+
+        dir="$(dirname "$dir")"
+    done
+
+    return 1
+}
+
+uvshim() {
+    local cmd="$1"
+    shift
+
+    local uv_root=""
+    local first_arg="$1"
+
+    uv_root="$(find_uv_root "$first_arg")"
+
+    if [[ -n "$uv_root" ]]; then
+        uv run --project "$uv_root" "$cmd" "$@"
+    else
+        command -- "$cmd" "$@"
+    fi
+}
+
+python() { uvshim python "$@"; }
+python3() { uvshim python3 "$@"; }
+pip() { uvshim pip "$@"; }
+pytest() { uvshim pytest "$@"; }
+
 _find_venv_upwards() {
     local dir="$PWD"
 
@@ -343,7 +386,7 @@ fi
 eval $(thefuck --alias)
 
 # opam
-[[ ! -r "$HOME/.opam/opam-init/init.zsh" ]] || source "$HOME/.opam/opam-init/init.zsh" > /dev/null 2> /dev/null
+[[ ! -r "$HOME/.opam/opam-init/init.zsh" ]] || source "$HOME/.opam/opam-init/init.zsh" >/dev/null 2>/dev/null
 
 ##########################
 # COMMANDS BEFORE PROMPT #

.local/bin/etc/nvim/compiler

@@ -14,43 +14,60 @@ vipercmd="carbon" #"silicon"
 
 cd "$dir" || exit
 
-textype() { \
+has_uv_project() {
+    dir="$PWD"
+    while [ "$dir" != "/" ]; do
+        if [ -f "$dir/.python-version" ] || [ -f "$dir/pyproject.toml" ]; then
+            return 0
+        fi
+        dir=$(dirname "$dir")
+    done
+    return 1
+}
+
+textype() {
     command="pdflatex --shell-escape"
-	( sed 5q "$file" | grep -i -q 'xelatex' ) && command="xelatex"
+    (sed 5q "$file" | grep -i -q 'xelatex') && command="xelatex"
     $command --output-directory="$dir" "$base" &&
         grep -i addbibresource "$file" >/dev/null &&
         biber --input-directory "$dir" "$base" &&
         $command --output-directory="$dir" "$base" &&
         makeglossaries "$basenodir" &&
         $command --output-directory="$dir" "$base"
-	}
+}
 
 case "$file" in
-	*\.[0-9]) refer -PS -e "$file" | groff -mandoc -T pdf > "$base".pdf ;;
+*\.[0-9]) refer -PS -e "$file" | groff -mandoc -T pdf >"$base".pdf ;;
-        *\.bash) bash "$file" ;;
+*\.bash) bash "$file" ;;
-	*\.go) go run "$file" ;;
+*\.go) go run "$file" ;;
-	*\.lua) lua "$file" ;;
+*\.lua) lua "$file" ;;
-	*\.md) pandoc "$file" --pdf-engine=xelatex -o "$base".pdf ;;
+*\.md) pandoc "$file" --pdf-engine=xelatex -o "$base".pdf ;;
-	*\.mom) refer -PS -e "$file" | groff -mom -kept -T pdf > "$base".pdf ;;
+*\.mom) refer -PS -e "$file" | groff -mom -kept -T pdf >"$base".pdf ;;
-	*\.ms) refer -PS -e "$file" | groff -me -ms -kept -T pdf > "$base".pdf ;;
+*\.ms) refer -PS -e "$file" | groff -me -ms -kept -T pdf >"$base".pdf ;;
-	*\.py) python3 "$file" ;;
+*\.py)
-	*\.rmd) echo "require(rmarkdown); render('$file')" | R -q --vanilla ;;
+    if has_uv_project; then
-	*\.sent) setsid sent "$file" 2>/dev/null & ;;
+        uv run python "$file"
-	*config.h) sudo make install ;;
+    else
-        *\.c) if [ -f Makefile ]; then make run; else cc "$file" -o "$base" && "$base"; fi ;;
+        python3 "$file"
-        *\.java) java "$file" ;;
+    fi
-        *\.js) node "$file" ;;
+    ;;
-        *\.m) octave -qW "$file" ;;
+*\.rmd) echo "require(rmarkdown); render('$file')" | R -q --vanilla ;;
-        *\.nim) nim c -r -d:noColors "$file" ;;
+*\.sent) setsid sent "$file" 2>/dev/null & ;;
-        *\.r|*\.R) Rscript "$file" ;;
+*config.h) sudo make install ;;
-        *\.rs) (cd "$dir" && cargo locate-project >/dev/null 2>&1) \
+*\.c) if [ -f Makefile ]; then make run; else cc "$file" -o "$base" && "$base"; fi ;;
-            && (cd "$(dirname "$(cd "$dir" && cargo locate-project --message-format plain)")" && cargo run) \
+*\.java) java "$file" ;;
-            || (rustc "$file" && "./$(basename "${file%.rs}")") ;;
+*\.js) node "$file" ;;
-        *\.sh) dash "$file" ;;
+*\.m) octave -qW "$file" ;;
-        *\.smt2) z3 "$file" ;;
+*\.nim) nim c -r -d:noColors "$file" ;;
-        *\.tex) textype "$file" ;;
+*\.r | *\.R) Rscript "$file" ;;
-        *\.vpr) $vipercmd "$file" ;;
+*\.rs) (cd "$dir" && cargo locate-project >/dev/null 2>&1) &&
-        *\.zsh) zsh "$file" ;;
+    (cd "$(dirname "$(cd "$dir" && cargo locate-project --message-format plain)")" && cargo run) ||
-        *\.ly) lilypond "$file" ;;
+    (rustc "$file" && "./$(basename "${file%.rs}")") ;;
-	*) sed 1q "$file" | grep "^#!/" | sed "s/^#!//" | xargs -r -I % "$file" ;;
+*\.sh) dash "$file" ;;
+*\.smt2) z3 "$file" ;;
+*\.tex) textype "$file" ;;
+*\.vpr) $vipercmd "$file" ;;
+*\.zsh) zsh "$file" ;;
+*\.ly) lilypond "$file" ;;
+*) sed 1q "$file" | grep "^#!/" | sed "s/^#!//" | xargs -r -I % "$file" ;;
 esac