formatted_map.go

package scribe

import (
	"fmt"
	"sort"
	"strings"

	"github.com/paketo-buildpacks/packit"
)

type FormattedMap map[string]interface{}

func (m FormattedMap) String() string {
	var (
		keys    []string
		padding int
	)
	for key := range m {
		if len(key) > padding {
			padding = len(key)
		}
		keys = append(keys, key)
	}

	sort.Strings(keys)

	var builder strings.Builder
	for _, key := range keys {
		value := m[key]
		if value == nil {
			value = "<empty>"
		}

		for i := len(key); i < padding; i++ {
			key = key + " "
		}

		builder.WriteString(fmt.Sprintf("%s -> \"%v\"\n", key, value))
	}

	return strings.TrimSpace(builder.String())
}

func NewFormattedMapFromEnvironment(environment packit.Environment) FormattedMap {
	envMap := FormattedMap{}
	for key, value := range environment {
		parts := strings.SplitN(key, ".", 2)

		switch {
		case parts[1] == "override" || parts[1] == "default":
			envMap[parts[0]] = value
		case parts[1] == "prepend":
			envMap[parts[0]] = strings.Join([]string{value, "$" + parts[0]}, environment[parts[0]+".delim"])
		case parts[1] == "append":
			envMap[parts[0]] = strings.Join([]string{"$" + parts[0], value}, environment[parts[0]+".delim"])
		}
	}

	return envMap
}