From e92e6e3f651ea7836edc4b1022e72f62420dcc3f Mon Sep 17 00:00:00 2001
From: Daniel Donenfeld <dadonenf@microsoft.com>
Date: Fri, 15 Feb 2019 17:58:14 -0800
Subject: [PATCH] Add API functions file

---
 asm/scripts/API-Functions.ps1 | 92 +++++++++++++++++++++++++++++++++++
 1 file changed, 92 insertions(+)
 create mode 100644 asm/scripts/API-Functions.ps1

diff --git a/asm/scripts/API-Functions.ps1 b/asm/scripts/API-Functions.ps1
new file mode 100644
index 0000000..a948beb
--- /dev/null
+++ b/asm/scripts/API-Functions.ps1
@@ -0,0 +1,92 @@
+function Get-AppVeyorBuild {
+    param()
+
+    if (-not ($env:APPVEYOR_API_TOKEN)) {
+        throw "missing api token for AppVeyor."
+    }
+
+    Invoke-RestMethod -Uri "https://ci.appveyor.com/api/projects/$env:APPVEYOR_ACCOUNT_NAME/$env:APPVEYOR_PROJECT_SLUG" -Method GET -Headers @{
+        "Authorization" = "Bearer $env:APPVEYOR_API_TOKEN"
+        "Content-type"  = "application/json"
+    }
+}
+
+function Get-TravisBuild {
+    param(
+        [int]$limit    = 10,
+        [int]$maxLimit = 100
+    )
+    if (-not ($env:TRAVIS_API_TOKEN)) {
+        throw "missing api token for Travis-CI."
+    }
+    
+    $builds = Invoke-RestMethod -Uri "https://api.travis-ci.org/builds?limit=$($limit)" -Method Get -Headers @{
+        "Authorization" = "token $env:TRAVIS_API_TOKEN"
+        "Travis-API-Version" = "3" 
+    } 
+    $currentBuild = $builds.builds | Where-Object {$_.commit.sha -eq $env:APPVEYOR_REPO_COMMIT}
+    if (!$currentBuild) {
+        if($limit+10 -le $maxLimit) {
+            return Get-TravisBuild -limit $limit+10 -maxLimit $maxLimit
+        } else {
+            throw "Could not get information about Travis build with sha $REPO_COMMIT"
+        }
+    } 
+    return $currentBuild
+}
+
+function appveyorFinished {
+    param()
+    $buildData = Get-AppVeyorBuild
+    $lastJob = ($buildData.build.jobs | Select-Object -Last 1).jobId
+
+    if ($lastJob -ne $env:APPVEYOR_JOB_ID) {
+        return $false
+    }
+
+    [datetime]$stop = ([datetime]::Now).AddMinutes($env:TIMEOUT_MINS)
+
+    do {
+
+        $allSuccess = $true
+        (Get-AppVeyorBuild).build.jobs | Where-Object {$_.jobId -ne $env:APPVEYOR_JOB_ID} | Foreach-Object `
+            { 
+                $job = $_
+                switch ($job.status) {
+                    "failed" { throw "AppVeyor's Job ($($job.jobId)) failed." }
+                    "success" { continue }
+                    Default { $allSuccess = $false }
+                }
+            }
+        if ($allSuccess) { return $true }
+        Start-sleep 5
+    } while (([datetime]::Now) -lt $stop)
+
+    throw "Test jobs were not finished in $env:TIMEOUT_MINS minutes"
+}
+
+function travisFinished {
+    param()
+
+    [datetime]$stop = ([datetime]::Now).AddMinutes($env:TIMEOUT_MINS)
+
+    do {
+        $builds = Get-TravisBuild
+        $currentBuild = $builds.builds | Where-Object {$_.commit.sha -eq $env:APPVEYOR_REPO_COMMIT}
+        if (!$currentBuild) {
+            throw "Could not get information about Travis build with sha $REPO_COMMIT"
+        }    
+        switch -regex ($currentBuild.state) {
+            "^passed$" {
+                return $true
+            }
+            "^(errored|failed|canceled)" {
+                throw "Travis Job ($($builds.builds.id)) failed"
+            }
+        }
+
+        Start-sleep 5
+    } while (([datetime]::Now) -lt $stop)
+
+    throw "Travis build did not finished in $env:TIMEOUT_MINS minutes"
+}
\ No newline at end of file