983501e43b
To maintain the behavior of the coreos-setup-environment that has started to move into cloudinit we need to write out /etc/environment with the public and private addresses, if known. The file is updated so that other contents are not replaced. This behavior is disabled entirely if /etc/environment was written by a write_files entry.
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package initialize
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
|
|
"github.com/coreos/coreos-cloudinit/system"
|
|
)
|
|
|
|
const DefaultSSHKeyName = "coreos-cloudinit"
|
|
|
|
type Environment struct {
|
|
root string
|
|
configRoot string
|
|
workspace string
|
|
netconfType string
|
|
sshKeyName string
|
|
substitutions map[string]string
|
|
}
|
|
|
|
// TODO(jonboulle): this is getting unwieldy, should be able to simplify the interface somehow
|
|
func NewEnvironment(root, configRoot, workspace, netconfType, sshKeyName string, substitutions map[string]string) *Environment {
|
|
if substitutions == nil {
|
|
substitutions = make(map[string]string)
|
|
}
|
|
// If certain values are not in the supplied substitution, fall back to retrieving them from the environment
|
|
for k, v := range map[string]string{
|
|
"$public_ipv4": os.Getenv("COREOS_PUBLIC_IPV4"),
|
|
"$private_ipv4": os.Getenv("COREOS_PRIVATE_IPV4"),
|
|
} {
|
|
if _, ok := substitutions[k]; !ok {
|
|
substitutions[k] = v
|
|
}
|
|
}
|
|
return &Environment{root, configRoot, workspace, netconfType, sshKeyName, substitutions}
|
|
}
|
|
|
|
func (e *Environment) Workspace() string {
|
|
return path.Join(e.root, e.workspace)
|
|
}
|
|
|
|
func (e *Environment) Root() string {
|
|
return e.root
|
|
}
|
|
|
|
func (e *Environment) ConfigRoot() string {
|
|
return e.configRoot
|
|
}
|
|
|
|
func (e *Environment) NetconfType() string {
|
|
return e.netconfType
|
|
}
|
|
|
|
func (e *Environment) SSHKeyName() string {
|
|
return e.sshKeyName
|
|
}
|
|
|
|
func (e *Environment) SetSSHKeyName(name string) {
|
|
e.sshKeyName = name
|
|
}
|
|
|
|
func (e *Environment) Apply(data string) string {
|
|
for key, val := range e.substitutions {
|
|
data = strings.Replace(data, key, val, -1)
|
|
}
|
|
return data
|
|
}
|
|
|
|
func (e *Environment) DefaultEnvironmentFile() *system.EnvFile {
|
|
ef := system.EnvFile{
|
|
File: &system.File{
|
|
Path: "/etc/environment",
|
|
},
|
|
Vars: map[string]string{},
|
|
}
|
|
if ip, ok := e.substitutions["$public_ipv4"]; ok && len(ip) > 0 {
|
|
ef.Vars["COREOS_PUBLIC_IPV4"] = ip
|
|
}
|
|
if ip, ok := e.substitutions["$private_ipv4"]; ok && len(ip) > 0 {
|
|
ef.Vars["COREOS_PRIVATE_IPV4"] = ip
|
|
}
|
|
if len(ef.Vars) == 0 {
|
|
return nil
|
|
} else {
|
|
return &ef
|
|
}
|
|
}
|
|
|
|
// normalizeSvcEnv standardizes the keys of the map (environment variables for a service)
|
|
// by replacing any dashes with underscores and ensuring they are entirely upper case.
|
|
// For example, "some-env" --> "SOME_ENV"
|
|
func normalizeSvcEnv(m map[string]string) map[string]string {
|
|
out := make(map[string]string, len(m))
|
|
for key, val := range m {
|
|
key = strings.ToUpper(key)
|
|
key = strings.Replace(key, "-", "_", -1)
|
|
out[key] = val
|
|
}
|
|
return out
|
|
}
|