130 lines
3.4 KiB
PowerShell
130 lines
3.4 KiB
PowerShell
|
|
# Set the runner name
|
|
if ($null -ne $env:RUNNER_NAME) {
|
|
$RUNNER_NAME = $env:RUNNER_NAME
|
|
} else {
|
|
if ($null -ne $env:RUNNER_NAME_PREFIX) {
|
|
$RUNNER_NAME = $env:RUNNER_NAME_PREFIX
|
|
} else {
|
|
$RUNNER_NAME = "windows-runner"
|
|
}
|
|
|
|
$RUNNER_NAME += "-" + (((New-Guid).Guid).replace("-", "")).substring(0, 8)
|
|
}
|
|
|
|
# Set GitHub host if not set
|
|
if ($null -ne $env:GITHUB_HOST) {
|
|
$GITHUB_HOST = $env:GITHUB_HOST
|
|
} else {
|
|
$GITHUB_HOST = 'github.com'
|
|
}
|
|
|
|
# Set the api url
|
|
if ('github.com' -eq $GITHUB_HOST) {
|
|
$URI = "https://api.$GITHUB_HOST"
|
|
} else {
|
|
$URI = "https://$GITHUB_HOST/api/v3"
|
|
}
|
|
|
|
# Set the api to get the access token from
|
|
switch ($env:RUNNER_SCOPE) {
|
|
org {
|
|
if ($null -eq $env:ORG_NAME) {
|
|
Write-Error "ORG_NAME required for organisational runners"
|
|
exit 1
|
|
}
|
|
Write-Host "Setting up GitHub Self Hosted Runner for organisation: $env:ORG_NAME"
|
|
$TOKEN_URL = "$URI/orgs/$env:ORG_NAME/actions/runners/registration-token"
|
|
$CONFIG_URL = "https://$GITHUB_HOST/$env:ORG_NAME"
|
|
}
|
|
|
|
enterprise {
|
|
if ($null -eq $env:ENTERPRISE_NAME) {
|
|
Write-Error "ENTERPRISE_NAME required for enterprise runners"
|
|
exit
|
|
}
|
|
Write-Host "Setting up GitHub Self Hosted Runner for enterprise: $env:ENTERPRISE_NAME"
|
|
$TOKEN_URL = "$URI/enterprises/$env:ENTERPRISE_NAME/actions/runners/registration-token"
|
|
$CONFIG_URL = "https://$GITHUB_HOST/enterprises/$env:ENTERPRISE_NAME"
|
|
}
|
|
|
|
default {
|
|
if ($null -eq $env:REPO_URL) {
|
|
Write-Error "REPO_URL required for repository runners"
|
|
exit
|
|
}
|
|
if ($null -ne $env:RUNNER_TOKEN) {
|
|
$RUNNER_TOKEN = $env:RUNNER_TOKEN
|
|
} elseif ($null -ne $env:ACCESS_TOKEN) {
|
|
$PATTERN = "https://(?:[^/]+/)?([^/]+)/([^/]+)"
|
|
if ($env:REPO_URL -match $PATTERN) {
|
|
|
|
$OWNER = $Matches[1]
|
|
$REPO = $Matches[2]
|
|
|
|
$TOKEN_URL = "$URI/repos/$OWNER/$REPO/actions/runners/registration-token"
|
|
} else {
|
|
Write-Error "URL format not recognized: $env:REPO_URL"
|
|
}
|
|
} else {
|
|
Write-Error "ACCESS_TOKEN or RUNNER_TOKEN required for repository runners"
|
|
exit
|
|
}
|
|
Write-Host "Setting up GitHub Self Hosted Runner for repository: $env:REPO_URL"
|
|
|
|
$CONFIG_URL = $env:REPO_URL
|
|
}
|
|
}
|
|
|
|
if ($null -ne $TOKEN_URL) {
|
|
$HEADERS = @{
|
|
'Accept' = 'application/vnd.github.v3+json';
|
|
'Authorization' = "token $env:ACCESS_TOKEN";
|
|
'Content-Length' = '0';
|
|
}
|
|
|
|
try {
|
|
Write-Host "Obtaining the token for the runner"
|
|
$RUNNER_TOKEN = ((Invoke-WebRequest -Uri $TOKEN_URL -Method "POST" -Headers $HEADERS).Content | ConvertFrom-Json).token
|
|
}
|
|
catch {
|
|
Write-Error "Cannot obtain the token => $_.Exception.Message"
|
|
exit
|
|
}
|
|
}
|
|
|
|
# Set the labels if given
|
|
if ($null -ne $env:LABELS) {
|
|
$LABELS = $env:LABELS
|
|
} else {
|
|
$LABELS = 'default'
|
|
}
|
|
|
|
# Set the labels if given
|
|
if ($null -ne $env:RUNNER_GROUP) {
|
|
$RUNNER_GROUP = $env:RUNNER_GROUP
|
|
} else {
|
|
$RUNNER_GROUP = 'Default'
|
|
}
|
|
|
|
$EXTRA_ARGS=''
|
|
|
|
# Disable auto update if set
|
|
if ($null -ne $env:DISABLE_AUTO_UPDATE) {
|
|
Write-Host "Auto updating is disabled"
|
|
$EXTRA_ARGS += " --disableupdate"
|
|
}
|
|
|
|
try {
|
|
Write-Host "Configuring runner: $RUNNER_NAME"
|
|
./config.cmd --unattended --replace --url $CONFIG_URL --token $RUNNER_TOKEN --name $RUNNER_NAME --labels $LABELS --runnergroup $RUNNER_GROUP $EXTRA_ARGS
|
|
|
|
# Remove access token for security reasons
|
|
$env:ACCESS_TOKEN=$null
|
|
|
|
./run.cmd
|
|
} catch {
|
|
Write-Error $_.Exception.Message
|
|
} finally {
|
|
./config.cmd remove --unattended --token $RUNNER_TOKEN
|
|
} |