#!/usr/bin/env bash
set -euo pipefail

# === IMPORTANT =========================================
# Replace this with the result of:
# command -v uv
REAL_UV="/absolute/path/to/the/real/uv"

if [[ ! -x "$REAL_UV" ]]; then
    echo "uv wrapper: real uv executable not found at:" >&2
    echo "  $REAL_UV" >&2
    echo "Update REAL_UV in $0." >&2
    exit 1
fi

find_project_root() {
    local dir="$PWD"
    local candidate=""

    while true; do
        # A lockfile is the strongest indication of the actual uv root.
        # This also behaves more sensibly for uv workspaces.
        if [[ -f "$dir/uv.lock" ]]; then
            printf '%s\n' "$dir"
            return
        fi

        # Remember the nearest project marker in case there is no lockfile yet.
        if [[ -z "$candidate" ]] && {
            [[ -f "$dir/pyproject.toml" ]] ||
            [[ -f "$dir/uv.toml" ]] ||
            [[ -f "$dir/.python-version" ]];
        }; then
            candidate="$dir"
        fi

        if [[ "$dir" == "/" ]]; then
            break
        fi

        dir="$(dirname "$dir")"
    done

    printf '%s\n' "$candidate"
}

project_root="$(find_project_root)"

# Outside a Python project, behave exactly like the real uv command.
if [[ -z "$project_root" ]]; then
    exec "$REAL_UV" "$@"
fi

# macOS normally provides shasum; Linux normally provides sha256sum.
if command -v sha256sum >/dev/null 2>&1; then
    project_hash="$(
        printf '%s' "$project_root" |
        sha256sum |
        cut -c1-12
    )"
elif command -v shasum >/dev/null 2>&1; then
    project_hash="$(
        printf '%s' "$project_root" |
        shasum -a 256 |
        cut -c1-12
    )"
else
    echo "uv wrapper: sha256sum or shasum is required." >&2
    exit 1
fi

project_name="$(
    basename "$project_root" |
    tr -cd '[:alnum:]_.-'
)"

if [[ -z "$project_name" ]]; then
    project_name="project"
fi

env_root="${
    UV_LOCAL_ENV_ROOT:-${
        XDG_DATA_HOME:-$HOME/.local/share
    }/uv-project-envs
}"

mkdir -p "$env_root"
export UV_PROJECT_ENVIRONMENT="$env_root/${project_name}-${project_hash}"
exec "$REAL_UV" "$@"

$ErrorActionPreference = "Stop"

# === IMPORTANT =========================================
# Replace this with the result of:
# (Get-Command uv.exe).Source
$RealUv = "C:\absolute\path\to\the\real\uv.exe"

if (-not (Test-Path -LiteralPath $RealUv -PathType Leaf)) {
    Write-Error "uv wrapper: real uv executable not found at: $RealUv"
    exit 1
}

function Find-UvProjectRoot {
    $dir = (Get-Location).ProviderPath
    $candidate = $null

    while ($true) {
        $lockfile = Join-Path $dir "uv.lock"

        # Prefer the nearest uv.lock. This is also more suitable
        # for projects contained in a uv workspace.
        if (Test-Path -LiteralPath $lockfile -PathType Leaf) {
            return $dir
        }

        $hasProjectMarker = (
            (Test-Path -LiteralPath (Join-Path $dir "pyproject.toml") -PathType Leaf) -or
            (Test-Path -LiteralPath (Join-Path $dir "uv.toml") -PathType Leaf) -or
            (Test-Path -LiteralPath (Join-Path $dir ".python-version") -PathType Leaf)
        )

        if ($null -eq $candidate -and $hasProjectMarker) {
            $candidate = $dir
        }

        $parent = Split-Path -Path $dir -Parent

        if (
            [string]::IsNullOrEmpty($parent) -or
            $parent -eq $dir
        ) {
            break
        }

        $dir = $parent
    }

    return $candidate
}

$projectRoot = Find-UvProjectRoot

# Outside a project, call the real uv normally.
if ([string]::IsNullOrWhiteSpace($projectRoot)) {
    & $RealUv @args
    exit $LASTEXITCODE
}

# Windows paths are case-insensitive, so normalize the path before hashing.
$normalizedRoot = [System.IO.Path]::GetFullPath($projectRoot)
$normalizedRoot = $normalizedRoot.TrimEnd([char]'\')
$normalizedRoot = $normalizedRoot.ToLowerInvariant()

$bytes = [System.Text.Encoding]::UTF8.GetBytes($normalizedRoot)
$sha = [System.Security.Cryptography.SHA256]::Create()

try {
    $hashBytes = $sha.ComputeHash($bytes)
}
finally {
    $sha.Dispose()
}

$projectHash = (
    $hashBytes |
    ForEach-Object { $_.ToString("x2") }
) -join ""

$projectHash = $projectHash.Substring(0, 12)

$projectName = Split-Path -Path $projectRoot -Leaf
$projectName = $projectName -replace '[^A-Za-z0-9_.-]', ''

if ([string]::IsNullOrWhiteSpace($projectName)) {
    $projectName = "project"
}

if ([string]::IsNullOrWhiteSpace($env:UV_LOCAL_ENV_ROOT)) {
    $envRoot = Join-Path $env:LOCALAPPDATA "uv-project-envs"
}
else {
    $envRoot = $env:UV_LOCAL_ENV_ROOT
}

New-Item -ItemType Directory -Force -Path $envRoot |
    Out-Null

$projectEnvironment = Join-Path `
    $envRoot `
    "$projectName-$projectHash"

$previousProjectEnvironment = $env:UV_PROJECT_ENVIRONMENT
$exitCode = 1

try {
    $env:UV_PROJECT_ENVIRONMENT = $projectEnvironment

    & $RealUv @args
    $exitCode = $LASTEXITCODE
}
finally {
    if ($null -eq $previousProjectEnvironment) {
        Remove-Item `
            Env:\UV_PROJECT_ENVIRONMENT `
            -ErrorAction SilentlyContinue
    }
    else {
        $env:UV_PROJECT_ENVIRONMENT = $previousProjectEnvironment
    }
}

exit $exitCode

@echo off
REM You can also use pwsh.exe if PowerShell 7 is installed / preferred
powershell.exe -NoLogo -NoProfile -ExecutionPolicy Bypass -File "%~dp0uv-wrapper.ps1" %*
exit /b %ERRORLEVEL%