mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
Colin Adler
961ddad925
Certain distros don't ship with `which` (arch) and `command -v` is built-in to the shell, so this is much more compatible.
81 lines
2.7 KiB
Go
81 lines
2.7 KiB
Go
package provisionersdk
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
// On Windows, VS Code Remote requires a parent process of the
|
|
// executing shell to be named "sshd", otherwise it fails. See:
|
|
// https://github.com/microsoft/vscode-remote-release/issues/5699
|
|
windowsScript = `$ProgressPreference = "SilentlyContinue"
|
|
Invoke-WebRequest -Uri ${ACCESS_URL}bin/coder-windows-${ARCH}.exe -OutFile $env:TEMP\sshd.exe
|
|
Set-MpPreference -DisableRealtimeMonitoring $true -ExclusionPath $env:TEMP\sshd.exe
|
|
$env:CODER_AGENT_AUTH = "${AUTH_TYPE}"
|
|
$env:CODER_AGENT_URL = "${ACCESS_URL}"
|
|
Start-Process -FilePath $env:TEMP\sshd.exe -ArgumentList "agent" -PassThru`
|
|
|
|
linuxScript = `#!/usr/bin/env sh
|
|
set -eux pipefail
|
|
BINARY_LOCATION=$(mktemp -d -t tmp.coderXXXXXX)/coder
|
|
BINARY_URL=${ACCESS_URL}bin/coder-linux-${ARCH}
|
|
if command -v curl >/dev/null 2>&1; then
|
|
curl -fsSL --compressed "${BINARY_URL}" -o "${BINARY_LOCATION}"
|
|
elif command -v wget >/dev/null 2>&1; then
|
|
wget -q "${BINARY_URL}" -O "${BINARY_LOCATION}"
|
|
elif command -v busybox >/dev/null 2>&1; then
|
|
busybox wget -q "${BINARY_URL}" -O "${BINARY_LOCATION}"
|
|
else
|
|
echo "error: no download tool found, please install curl, wget or busybox wget"
|
|
exit 1
|
|
fi
|
|
chmod +x $BINARY_LOCATION
|
|
export CODER_AGENT_AUTH="${AUTH_TYPE}"
|
|
export CODER_AGENT_URL="${ACCESS_URL}"
|
|
exec $BINARY_LOCATION agent`
|
|
|
|
darwinScript = `#!/usr/bin/env sh
|
|
set -eux pipefail
|
|
BINARY_LOCATION=$(mktemp -d -t tmp.coderXXXXXX)/coder
|
|
curl -fsSL --compressed "${ACCESS_URL}bin/coder-darwin-${ARCH}" -o "${BINARY_LOCATION}"
|
|
chmod +x $BINARY_LOCATION
|
|
export CODER_AGENT_AUTH="${AUTH_TYPE}"
|
|
export CODER_AGENT_URL="${ACCESS_URL}"
|
|
exec $BINARY_LOCATION agent`
|
|
|
|
// A mapping of operating-system ($GOOS) to architecture ($GOARCH)
|
|
// to agent install and run script. ${DOWNLOAD_URL} is replaced
|
|
// with strings.ReplaceAll() when being consumed. ${ARCH} is replaced
|
|
// with the architecture when being provided.
|
|
agentScripts = map[string]map[string]string{
|
|
"windows": {
|
|
"amd64": windowsScript,
|
|
"arm64": windowsScript,
|
|
},
|
|
"linux": {
|
|
"amd64": linuxScript,
|
|
"arm64": linuxScript,
|
|
"armv7": linuxScript,
|
|
},
|
|
"darwin": {
|
|
"amd64": darwinScript,
|
|
"arm64": darwinScript,
|
|
},
|
|
}
|
|
)
|
|
|
|
// AgentScriptEnv returns a key-pair of scripts that are consumed
|
|
// by the Coder Terraform Provider. See:
|
|
// https://github.com/coder/terraform-provider-coder/blob/main/internal/provider/provider.go#L97
|
|
func AgentScriptEnv() map[string]string {
|
|
env := map[string]string{}
|
|
for operatingSystem, scripts := range agentScripts {
|
|
for architecture, script := range scripts {
|
|
script := strings.ReplaceAll(script, "${ARCH}", architecture)
|
|
env[fmt.Sprintf("CODER_AGENT_SCRIPT_%s_%s", operatingSystem, architecture)] = script
|
|
}
|
|
}
|
|
return env
|
|
}
|