nu_scripts/modules/virtual_environments/auto-venv/venv_helpers.nu

use path_extensions.nu *

def has-sub [
    folder:    string, 
    subfolder: string
] {
    not (path find-sub $folder $subfolder | is-empty);
}

def get-env [
    name:    string
    default: any
] {
    (if ($name in $env) {
        $env | get $name
    }
    else {
        $default
    })
}

export def venv-is-active [] {
    '__auto_venv' in (overlay list)
}

# Creates a virtual environment under the current directory
export def 'venv-create' [] {
	let venv_path = $env.PWD
	let venv_name = ($env.PWD | path basename) 

	let is_posix = ($nu.os-info.family == 'unix')
	let python_name = if $is_posix {'python3'} else {'py.exe'}
	run-external $python_name "-m" "venv" ".venv" "--clear" "--prompt" $venv_name
	run-external $".venv/bin/($python_name)" "-m" "pip" "install" "-U" "pip" "wheel" "setuptools"

	let trigger_file = ([$env.PWD, $env.AUTO_VENV_TRIGGER] | path join)
	ln -sf $'($env.FILE_PWD)/venvs/python-venv.nu' $trigger_file
}

export def has-entered-venv [
    after: path,
] {

    let target = (path find-sub $after $env.AUTO_VENV_TRIGGER)
    
    (if ($target | is-empty) {
        false
    }
    else {
        # if venv is already active, handle it in "venv swap" hook
        not (venv-is-active)
    })
}

export def has-swapped-venv [
    after: path,
] {

    (if not (venv-is-active) {
        false
    }
    else {
        let target = (path find-sub $after $env.AUTO_VENV_TRIGGER)
            
        (if ($target | is-empty) {
            false
        }
        else {
            (if ('VIRTUAL_ENV' in $env) {
		    $env.VIRTUAL_ENV != $target
		    } else {
		    false # should it default to `false`?
		    })
        })

    })
}

export def has-exited-venv [
    after: path,
] {

    (if not (venv-is-active) {
        false
    } 
    else {
        not (has-sub $after $env.AUTO_VENV_TRIGGER)
    })
}