package initialize

import (
	"fmt"

	"github.com/coreos/coreos-cloudinit/system"
)

type FleetEnvironment map[string]string

func (fe FleetEnvironment) String() (out string) {
	norm := normalizeSvcEnv(fe)
	out += "[Service]\n"

	for key, val := range norm {
		out += fmt.Sprintf("Environment=\"FLEET_%s=%s\"\n", key, val)
	}

	return
}

// Units generates a Unit file drop-in for fleet, if any fleet options were
// configured in cloud-config
func (fe FleetEnvironment) Units(root string) ([]system.Unit, error) {
	if len(fe) < 1 {
		return nil, nil
	}
	fleet := system.Unit{
		Name:    "fleet.service",
		Runtime: true,
		DropIn:  true,
		Content: fe.String(),
	}
	return []system.Unit{fleet}, nil
}