mirror of
https://github.com/openshift/source-to-image.git
synced 2026-02-05 12:44:54 +01:00
92 lines
2.2 KiB
Go
92 lines
2.2 KiB
Go
package scripts
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/golang/glog"
|
|
"github.com/openshift/source-to-image/pkg/api"
|
|
)
|
|
|
|
// Environment represents a single environment variable definition
|
|
type Environment struct {
|
|
Name string
|
|
Value string
|
|
}
|
|
|
|
// getEnvPath returns prefix/environment path.
|
|
func getEnvPath(config *api.Config, prefix string) (string, error) {
|
|
envPath := filepath.Join(config.WorkingDir, api.Source, prefix, api.Environment)
|
|
if _, err := os.Stat(envPath); os.IsNotExist(err) {
|
|
return "", errors.New("no environment file found in application sources")
|
|
}
|
|
return envPath, nil
|
|
}
|
|
|
|
// GetEnvironment gets the .s2i/environment file located in the sources and
|
|
// parse it into []environment
|
|
func GetEnvironment(config *api.Config) ([]Environment, error) {
|
|
envPath, err := getEnvPath(config, ".s2i")
|
|
if err != nil {
|
|
envPath, err = getEnvPath(config, ".sti")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
glog.Infof("DEPRECATED: Use .s2i/environment instead of .sti/environment")
|
|
}
|
|
|
|
f, err := os.Open(envPath)
|
|
if err != nil {
|
|
return nil, errors.New("unable to read environment file")
|
|
}
|
|
defer f.Close()
|
|
|
|
result := []Environment{}
|
|
|
|
scanner := bufio.NewScanner(f)
|
|
for scanner.Scan() {
|
|
s := scanner.Text()
|
|
// Allow for comments in environment file
|
|
if strings.HasPrefix(s, "#") {
|
|
continue
|
|
}
|
|
parts := strings.SplitN(s, "=", 2)
|
|
if len(parts) != 2 {
|
|
continue
|
|
}
|
|
e := Environment{
|
|
Name: strings.TrimSpace(parts[0]),
|
|
Value: strings.TrimSpace(parts[1]),
|
|
}
|
|
glog.V(1).Infof("Setting '%s' to '%s'", e.Name, e.Value)
|
|
result = append(result, e)
|
|
}
|
|
|
|
return result, scanner.Err()
|
|
}
|
|
|
|
// ConvertEnvironment converts the []Environment to "key=val" strings
|
|
func ConvertEnvironment(env []Environment) (result []string) {
|
|
for _, e := range env {
|
|
result = append(result, fmt.Sprintf("%s=%s", e.Name, e.Value))
|
|
}
|
|
return
|
|
}
|
|
|
|
// ConvertEnvironmentToDocker converts the []Environment into Dockerfile format
|
|
func ConvertEnvironmentToDocker(env []Environment) (result string) {
|
|
for i, e := range env {
|
|
if i == 0 {
|
|
result += fmt.Sprintf("ENV %s=\"%s\"", e.Name, e.Value)
|
|
} else {
|
|
result += fmt.Sprintf(" \\\n\t%s=\"%s\"", e.Name, e.Value)
|
|
}
|
|
}
|
|
result += "\n"
|
|
return
|
|
}
|