Set-PSDebug -Trace 0

Set-StrictMode -Version 3

$ErrorActionPreference = "Stop"

$cecho = "$PSScriptRoot\cecho.cmd"

# 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

}

. $cecho 0 13 "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

}

. $cecho 0 13 "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 extract_file {

param(

[Parameter(Mandatory = $true)]

[string]$dependency_name,

[Parameter(Mandatory = $true)]

[string]$filename,

[Parameter(Mandatory = $true)]

[string]$destination_dir,

[Parameter(Mandatory = $true)]

[string]$dir_after_extraction

)

if (Test-Path -Path "$dir_after_extraction") {

. $cecho 0 13 "$dependency_name already extracted into '$dir_after_extraction'. Skipping."

return

}

. $cecho 0 13 "Extracting $dependency_name into '$destination_dir' from '$filename'."

7za x "$filename" -o"$destination_dir"

}

function download_file {

param(

[Parameter(Mandatory = $true)]

[string]$dependency_name,

[Parameter(Mandatory = $true)]

[string]$url,

[Parameter(Mandatory = $true)]

[string]$destination_dir,

[Parameter(Mandatory = $true)]

[string]$filename

)

mkdir "$destination_dir" -Force | Out-Null

pushd "$destination_dir"

if (Test-Path -Path "$filename") {

. $cecho 0 13 "$dependency_name already downloaded. Skipping."

return

}

. $cecho 0 13 "Downloading $dependency_name into '$destination_dir'"

Invoke-WebRequest $url -OutFile $filename

}

# Required env variables:

# - DEPS_DIR

function git_clone_and_checkout_revision {

param(

[Parameter(Mandatory = $true)]

[string]$dependency_name,

[Parameter(Mandatory = $true)]

[string]$git_url,

[Parameter(Mandatory = $true)]

[string]$dest_dir,

[Parameter(Mandatory = $false)]

[string]$revision

)

if (Test-Path -Path "$dest_dir") {

. $cecho 0 13 "Cloning $dependency_name is already cloned."

return

}

. $cecho 0 13 "Cloning $dependency_name into '$dest_dir'."

pushd "$env:DEPS_DIR"

git clone $git_url $dest_dir

popd

pushd "$dest_dir"

git fetch

. $cecho 0 13 "Checking out $dependency_name revision $revision."

git reset --hard

git checkout $revision

popd

}

function install_cmake_project {

param(

[Parameter(Mandatory = $true)]

[string]$dependency_name,

[Parameter(Mandatory = $true)]

[string]$build_dir,

[Parameter(Mandatory = $true)]

[string]$configuration

)

pushd "$build_dir"

. $cecho 0 13 "Installing $dependency_name ($configuration). Please be patient, this may take a while."

$command = "cmake --install . --config $configuration"

. $cecho 0 13 "$command"

Invoke-Expression $command

popd

}

function check_boost_vc145_compatibility {

param(

[Parameter(Mandatory = $true)]

[string]$VC_VER,

[Parameter(Mandatory = $true)]

[string]$DEPS_DIR,

[Parameter(Mandatory = $true)]

[string]$BOOST_ROOT

)

$boost_build_path = "$BOOST_ROOT/tools/build"

if ($VC_VER -ne "14.5") {

. $cecho 0 13 "VC_VER is not 14.5, no need to install updated b2."

return

}

$res = Select-String -Path "$boost_build_path/src/engine/build.bat" -Pattern 'vc143, vc145' -Quiet;

if ($res) {

. $cecho 0 13 "vc145 already supported, no need to install updated b2."

return

}

$b2_version = "5.4.2"

$b2_stem = "b2-$b2_version"

$b2_path = "$DEPS_DIR\$b2_stem"

$b2_filename = "$b2_stem.zip"

& download_file "b2" "https://github.com/bfgroup/b2/releases/download/$b2_version/$b2_filename" "$DEPS_DIR" "$b2_filename"

& extract_file "b2" "$b2_filename" "$DEPS_DIR" "$b2_path"

. $cecho 0 13 "Installing b2 with vc145 support..."

Remove-Item -Recurse -Path "$boost_build_path"

Copy-Item -Path "$b2_path" -Destination "$boost_build_path" -Recurse

. $cecho 0 13 "b2 with vc145 support installed."

}

function main {

& setup_build_cfg

# Dispatch command.

$command = $Args[0]

if ($args.Count -gt 1) {

$command_args = $Args[1..($args.Count - 1)]

}

else {

$command_args = @()

}

& $command @command_args

}

& main @Args

exit 0