Set-PSDebug -Trace 0
Set-StrictMode -Version 3
$ErrorActionPreference = "Stop"


# Create marker file to indicate whether Release or Debug build was installed.
function mark {
    param(
        [Parameter(Mandatory = $true)]
        [string]$installation_dir
    )

    if (-not (Test-Path -Path $installation_dir)) {
        throw "Directory '$installation_dir' does not exist."
    }

    $marker_filepath = Join-Path -Path $installation_dir -ChildPath $ENV:MARKER_FILE
    if (Test-Path -Path $marker_filepath) {
        return
    }
    echo "Marking installation in '$installation_dir' with '$ENV:MARKER_FILE'."
    New-Item -Path $marker_filepath -ItemType File | Out-Null
}

# This function can be deprecated in the future, since installations are marked automatically.
# It's here only to avoid some disruption during the transition period.
function mark_based_on_artifacts {
    param(
        [Parameter(Mandatory = $true)]
        [string]$dependency_name,

        [Parameter(Mandatory = $true)]
        [string]$installation_dir
    )

    if ($dependency_name -eq "hdf5") {
        if ($env:BUILD_CFG -eq "Debug") {
            $artifact = "lib\libhdf5_D.lib"
        }
        else {
            $artifact = "lib\libhdf5.lib"
        }
    }
    elseif ($dependency_name -eq "opencollada") {
        if ($env:BUILD_CFG -eq "Debug") {
            $artifact = "lib\opencollada\OpenCOLLADAFrameworkd.lib"
        }
        else {
            $artifact = "lib\opencollada\OpenCOLLADAFramework.lib"
        }
    }
    elseif ($dependency_name -eq "OpenCASCADE") {
        # New OCCT folder layout was introduced after marker files were added,
        # so installation don't need artifact-based detection.
        return
    }
    elseif ($dependency_name -eq "rocksdb") {
        if ($env:BUILD_CFG -eq "Debug") {
            $artifact = "lib\rocksdb_d.lib"
        }
        else {
            $artifact = "lib\rocksdb.lib"
        }
    }
    else {
        throw "Unexpected dependency name '$dependency_name'."
    }
    $artifact_filepath = Join-Path -Path $installation_dir -ChildPath $artifact
    if (-not (Test-Path -Path $artifact_filepath)) {
        return
    }
    if (-not (Test-Path -Path $installation_dir)) {
        throw "Directory '$installation_dir' does not exist."
    }
    $marker_filepath = Join-Path -Path $installation_dir -ChildPath $ENV:MARKER_FILE
    if (Test-Path -Path $marker_filepath) {
        return
    }
    echo "Found artifact '$artifact' for dependency '$dependency_name' $env:BUILD_CFG."
    & mark $installation_dir
}

# Check if installation exists for the current `BUILD_CFG`.
# Returns exit code 200 if installation exists, 404 otherwise.
# Since we want Release and Debug installation to coexist,
# we add special marker file to indicate which build type was installed.
function check_installation {
    param(
        [Parameter(Mandatory = $true)]
        [string]$dependency_name,
        [Parameter(Mandatory = $true)]
        [string]$installation_dir
    )

    if (-not (Test-Path -Path $installation_dir)) {
        exit 404
    }

    & mark_based_on_artifacts $dependency_name $installation_dir

    $marker_filepath = Join-Path -Path $installation_dir -ChildPath $ENV:MARKER_FILE
    if (-not (Test-Path -Path $marker_filepath)) {
        exit 404
    }
    exit 200
}


# Dependencies Release/Debug configs compatibility:
# - hdf5: incompatible
# - OpenCASCADE: incompatible
# - rocksdb: incompatible
# - opencollada: incompatible
# - zstd: compatible

function setup_build_cfg {
    if (-not $env:BUILD_CFG) {
        throw "Variable 'BUILD_CFG' is not defined."
    }
    if ($env:BUILD_CFG -eq "Debug") {
        $ENV:MARKER_FILE = ".debug_installation"
    }
    else {
        $ENV:MARKER_FILE = ".release_installation"
    }
}

function main {
    & setup_build_cfg
    # Dispatch command.
    $command = $Args[0]
    $command_args = $Args[1..($args.Count - 1)]
    & $command @command_args
}

& main @Args
exit 0