make.ps1

Param(
  [Parameter(Position=0, HelpMessage="The action to take (build, test, install, package, clean).")]
  [string]
  $Command = 'build',

  [Parameter(HelpMessage="The build configuration (Release, Debug).")]
  [string]
  $Config = "Release"
)

$ErrorActionPreference = "Stop"

$target = "lori"
$testPath = "."
$rootDir = Split-Path $script:MyInvocation.MyCommand.Path
$srcDir = Join-Path -Path $rootDir -ChildPath $target

if ($Config -ieq "Release")
{
  $configFlag = ""
  $buildDir = Join-Path -Path $rootDir -ChildPath "build/release"
}
elseif ($Config -ieq "Debug")
{
  $configFlag = "--debug"
  $buildDir = Join-Path -Path $rootDir -ChildPath "build/debug"
}
else
{
  throw "Invalid -Config path '$Config'; must be one of (Debug, Release)."
}

$ponyArgs = "--define libressl  --path $rootDir"

function BuildTest
{
  $testTarget = "$target.exe"

  $testFile = Join-Path -Path $buildDir -ChildPath $testTarget
  $testTimestamp = [DateTime]::MinValue
  if (Test-Path $testFile)
  {
    $testTimestamp = (Get-ChildItem -Path $testFile).LastWriteTimeUtc
  }

  :testFiles foreach ($file in (Get-ChildItem -Path $srcDir -Include "*.pony" -Recurse))
  {
    if ($testTimestamp -lt $file.LastWriteTimeUtc)
    {
      Write-Host "corral fetch"
      $output = (corral fetch)
      $output | ForEach-Object { Write-Host $_ }
      if ($LastExitCode -ne 0) { throw "Error" }

      $testDir = Join-Path -Path $srcDir -ChildPath $testPath
      Write-Host "corral run -- ponyc $configFlag $ponyArgs --output `"$buildDir`" `"$testDir`""
      $output = (corral run -- ponyc $configFlag $ponyArgs --output "$buildDir" "$testDir")
      $output | ForEach-Object { Write-Host $_ }
      if ($LastExitCode -ne 0) { throw "Error" }
      break testFiles
    }
  }

  Write-Output "$testTarget.exe is built" # force function to return a list of outputs
  return $testFile
}

function BuildExamples
{
    $examplesDir = Join-Path -Path $rootDir -ChildPath "examples"

    # Get all subdirectories in examples folder
    $examples = Get-ChildItem -Path $examplesDir -Directory

    foreach ($example in $examples) {
        $exampleName = $example.Name
        $exampleTarget = "$exampleName.exe"
        $exampleFile = Join-Path -Path $buildDir -ChildPath $exampleTarget

        Write-Host "Building example: $exampleName"

        # Check if we need to rebuild
        $needsRebuild = $false
        if (-not (Test-Path $exampleFile)) {
            $needsRebuild = $true
        } else {
            $exampleTimestamp = (Get-ChildItem -Path $exampleFile).LastWriteTimeUtc

            # Check source files in both src and example directories
            Get-ChildItem -Path $srcDir -Include "*.pony" -Recurse | ForEach-Object {
                if ($exampleTimestamp -lt $_.LastWriteTimeUtc) {
                    $needsRebuild = $true
                }
            }

            Get-ChildItem -Path $example.FullName -Include "*.pony" -Recurse | ForEach-Object {
                if ($exampleTimestamp -lt $_.LastWriteTimeUtc) {
                    $needsRebuild = $true
                }
            }
        }

        if ($needsRebuild) {
            Write-Host "corral fetch"
            $output = (corral fetch)
            $output | ForEach-Object { Write-Host $_ }
            if ($LastExitCode -ne 0) { throw "Error during corral fetch" }

            Write-Host "corral run -- ponyc $configFlag $ponyArgs --output `"$buildDir`" `"$($example.FullName)`""
            $output = (corral run -- ponyc $configFlag $ponyArgs --output "$buildDir" "$($example.FullName)")
            $output | ForEach-Object { Write-Host $_ }
            if ($LastExitCode -ne 0) { throw "Error building example $exampleName" }
        } else {
            Write-Host "$exampleTarget is up to date"
        }
    }
}

function BuildStressTests
{
    $stressTestsDir = Join-Path -Path $rootDir -ChildPath "stress-tests"

    # Get all subdirectories in stress-tests folder
    $stressTests = Get-ChildItem -Path $stressTestsDir -Directory

    foreach ($test in $stressTests) {
        $testName = $test.Name
        $testTarget = "$testName.exe"
        $testFile = Join-Path -Path $buildDir -ChildPath $testTarget

        Write-Host "Building stress test: $testName"

        # Check if we need to rebuild
        $needsRebuild = $false
        if (-not (Test-Path $testFile)) {
            $needsRebuild = $true
        } else {
            $testTimestamp = (Get-ChildItem -Path $testFile).LastWriteTimeUtc

            # Check source files in both src and stress-test directories
            Get-ChildItem -Path $srcDir -Include "*.pony" -Recurse | ForEach-Object {
                if ($testTimestamp -lt $_.LastWriteTimeUtc) {
                    $needsRebuild = $true
                }
            }

            Get-ChildItem -Path $test.FullName -Include "*.pony" -Recurse | ForEach-Object {
                if ($testTimestamp -lt $_.LastWriteTimeUtc) {
                    $needsRebuild = $true
                }
            }
        }

        if ($needsRebuild) {
            Write-Host "corral fetch"
            $output = (corral fetch)
            $output | ForEach-Object { Write-Host $_ }
            if ($LastExitCode -ne 0) { throw "Error during corral fetch" }

            Write-Host "corral run -- ponyc $configFlag $ponyArgs --output `"$buildDir`" `"$($test.FullName)`""
            $output = (corral run -- ponyc $configFlag $ponyArgs --output "$buildDir" "$($test.FullName)")
            $output | ForEach-Object { Write-Host $_ }
            if ($LastExitCode -ne 0) { throw "Error building stress test $testName" }
        } else {
            Write-Host "$testTarget is up to date"
        }
    }
}

switch ($Command.ToLower())
{
  "test"
  {
    $testFile = (BuildTest)[-1]
    Write-Host "$testFile"
    $rawOutput = & "$testFile" --sequential 2>&1
    $exitCode = $LastExitCode
    foreach ($line in $rawOutput) { Write-Host $line }
    if ($exitCode -ne 0)
    {
      $stderrLines = @($rawOutput |
        Where-Object { $_ -is [System.Management.Automation.ErrorRecord] } |
        ForEach-Object { $_.ToString() })
      $details = ""
      if ($stderrLines.Count -gt 0)
      {
        $details = "`n" + ($stderrLines -join "`n")
      }
      throw "Test failed with exit code ${exitCode}${details}"
    }
    break
  }

  "examples"
  {
      if (-not (Test-Path $buildDir))
      {
          mkdir "$buildDir"
      }
      BuildExamples
      break
  }

  "stress-tests"
  {
      if (-not (Test-Path $buildDir))
      {
          mkdir "$buildDir"
      }
      BuildStressTests
      break
  }

  default
  {
      throw "Unknown command '$Command'; must be one of (test, examples, stress-tests)."
  }
}