#!/usr/bin/env bash
 
VENV_BASE_DIR="${HOME}/.local/share/pyproject/virtualenvs"
 
# Check if venv base directory exists
if [ ! -d "$VENV_BASE_DIR" ]; then
    echo "Error: Poetry virtualenvs directory not found at $VENV_BASE_DIR" >&2
    exit 1
fi
 
# Check if fzf is installed
if ! command -v fzf >/dev/null 2>&1; then
    echo "Error: fzf is not installed. Please install it first." >&2
    exit 1
fi
 
# Helper function to extract project name from virtualenv path
get_project_name() {
    local venv_name="$1"
    basename "$venv_name" | sed -E 's/-[^-]*-py[0-9.]+$//'
}
 
# Helper function to list and format virtualenvs
list_virtualenvs() {
    local venv_dir
    while IFS= read -r venv_dir; do
        # Skip if not a directory
        [ ! -d "$venv_dir" ] && continue
        # Skip if no activation script exists
        [ ! -f "$venv_dir/bin/activate" ] && continue
 
        # Get and output project name
        get_project_name "$venv_dir"
    done < <(find "$VENV_BASE_DIR" -mindepth 1 -maxdepth 1 -type d)
}
 
# Helper function to find virtualenv path from project name
find_venv_path() {
    local project_name="$1"
    local venv_dir
 
    while IFS= read -r venv_dir; do
        [ ! -d "$venv_dir" ] && continue
        if [ "$(get_project_name "$venv_dir")" = "$project_name" ]; then
            echo "$venv_dir"
            return 0
        fi
    done < <(find "$VENV_BASE_DIR" -mindepth 1 -maxdepth 1 -type d)
 
    return 1
}
 
# Get list of projects and select with fzf
local selected_project
selected_project=$(list_virtualenvs | sort | fzf --prompt="Select virtualenv: ")
 
# Exit if nothing selected
[ -z "$selected_project" ] && exit 0
 
# Find corresponding virtualenv path
local venv_path
venv_path=$(find_venv_path "$selected_project")
 
if [ -z "$venv_path" ]; then
    echo "Error: Could not find virtualenv for $selected_project" >&2
    exit 1
fi
 
# Check if activation script exists
if [ ! -f "$venv_path/bin/activate" ]; then
    echo "Error: Activation script not found in $venv_path" >&2
    exit 1
fi 
 
# Activate virtualenv and start new shell
. "$venv_path/bin/activate"
exec "$SHELL"