86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
|
package provisioning
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"io/ioutil"
|
||
|
|
"log"
|
||
|
|
"path/filepath"
|
||
|
|
)
|
||
|
|
|
||
|
|
// ProvisionerConfigTerraform targets a Terraform cluster by reading the config
|
||
|
|
// from a file, as output by 'terraform output provisioning'. Malformed inputs
|
||
|
|
// will log.Fatal so that we halt the test run.
|
||
|
|
func ProvisionerConfigTerraform(config ProvisionerConfig) *ProvisioningTargets {
|
||
|
|
configFile, err := filepath.Abs(config.TerraformConfig)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("could not find -provision.terraform file: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
file, err := ioutil.ReadFile(configFile)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("could not read -provision.terraform file: %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
targets := &ProvisioningTargets{}
|
||
|
|
err = json.Unmarshal(file, &targets)
|
||
|
|
if err != nil {
|
||
|
|
log.Fatalf("decoding error: %v\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
for _, server := range targets.Servers {
|
||
|
|
canonicalize(server, config)
|
||
|
|
}
|
||
|
|
for _, client := range targets.Clients {
|
||
|
|
canonicalize(client, config)
|
||
|
|
}
|
||
|
|
return targets
|
||
|
|
}
|
||
|
|
|
||
|
|
func canonicalize(target *ProvisioningTarget, config ProvisionerConfig) {
|
||
|
|
|
||
|
|
// allow the '-nomad.*' command line flags to override
|
||
|
|
// the values we get from 'terraform output provisioning'
|
||
|
|
if config.NomadVersion != "" {
|
||
|
|
target.Deployment.NomadVersion = config.NomadVersion
|
||
|
|
}
|
||
|
|
if config.NomadSha != "" {
|
||
|
|
target.Deployment.NomadSha = config.NomadSha
|
||
|
|
}
|
||
|
|
if config.NomadLocalBinary != "" {
|
||
|
|
target.Deployment.NomadLocalBinary = config.NomadLocalBinary
|
||
|
|
}
|
||
|
|
|
||
|
|
if target.Deployment.RemoteBinaryPath == "" {
|
||
|
|
log.Fatal("bad runner config for 'remote_binary_path': missing value")
|
||
|
|
}
|
||
|
|
key, ok := target.Runner["key"].(string)
|
||
|
|
if !ok {
|
||
|
|
log.Fatalf("bad runner config for 'key': %v", target.Runner)
|
||
|
|
}
|
||
|
|
user, ok := target.Runner["user"].(string)
|
||
|
|
if !ok {
|
||
|
|
log.Fatalf("bad runner config for 'user': %v", target.Runner)
|
||
|
|
}
|
||
|
|
hostname, ok := target.Runner["host"].(string)
|
||
|
|
if !ok {
|
||
|
|
log.Fatalf("bad runner config for 'host': %v", target.Runner)
|
||
|
|
}
|
||
|
|
port, ok := target.Runner["port"].(float64)
|
||
|
|
if !ok {
|
||
|
|
log.Fatalf("bad runner config for 'port': %v", target.Runner)
|
||
|
|
}
|
||
|
|
|
||
|
|
runner := &SSHRunner{
|
||
|
|
Key: key,
|
||
|
|
User: user,
|
||
|
|
Host: hostname,
|
||
|
|
Port: int(port),
|
||
|
|
}
|
||
|
|
if target.Deployment.Platform == "windows_amd64" {
|
||
|
|
runner.copyMethod = copyWindows
|
||
|
|
} else {
|
||
|
|
runner.copyMethod = copyLinux
|
||
|
|
}
|
||
|
|
target.runner = runner
|
||
|
|
}
|