environment.go

package chef

import (
	"fmt"
	"sort"
)

// Environment has a Reader, hey presto
type EnvironmentService struct {
	client *Client
}

type EnvironmentResult map[string]string

// Environment represents the native Go version of the deserialized Environment type
type Environment struct {
	Name               string            `json:"name"`
	Description        string            `json:"description"`
	ChefType           string            `json:"chef_type,omitempty"`
	Attributes         interface{}       `json:"attributes,omitempty"`
	DefaultAttributes  interface{}       `json:"default_attributes,omitempty"`
	OverrideAttributes interface{}       `json:"override_attributes,omitempty"`
	JsonClass          string            `json:"json_class,omitempty"`
	CookbookVersions   map[string]string `json:"cookbook_versions"`
}

type EnvironmentCookbookResult map[string]CookbookVersions

type EnvironmentRecipesResult []string

func strMapToStr(e map[string]string) (out string) {
	keys := make([]string, len(e))
	for k, _ := range e {
		keys = append(keys, k)
	}
	sort.Strings(keys)
	for _, k := range keys {
		if k == "" {
			continue
		}
		out += fmt.Sprintf("%s => %s\n", k, e[k])
	}
	return
}

// String makes EnvironmentResult implement the string result
func (e EnvironmentResult) String() (out string) {
	return strMapToStr(e)
}

// List lists the environments in the Chef server.
//
// Chef API docs: https://docs.chef.io/api_chef_server.html#environments
func (e *EnvironmentService) List() (data *EnvironmentResult, err error) {
	err = e.client.magicRequestDecoder("GET", "environments", nil, &data)
	return
}

// Create an environment in the Chef server.
//
// Chef API docs: https://docs.chef.io/api_chef_server.html#environments
func (e *EnvironmentService) Create(environment *Environment) (data *EnvironmentResult, err error) {
	body, err := JSONReader(environment)
	if err != nil {
		return
	}

	err = e.client.magicRequestDecoder("POST", "environments", body, &data)
	return
}

// Delete an environment from the Chef server.
//
// Chef API docs: https://docs.chef.io/api_chef_server/#delete-9
func (e *EnvironmentService) Delete(name string) (data *Environment, err error) {
	path := fmt.Sprintf("environments/%s", name)
	err = e.client.magicRequestDecoder("DELETE", path, nil, &data)
	return
}

// Get gets an environment from the Chef server.
//
// Chef API docs: https://docs.chef.io/api_chef_server.html#environments-name
func (e *EnvironmentService) Get(name string) (data *Environment, err error) {
	path := fmt.Sprintf("environments/%s", name)
	err = e.client.magicRequestDecoder("GET", path, nil, &data)
	return
}

// Write an environment to the Chef server.
//
// Chef API docs: https://docs.chef.io/api_chef_server.html#environments-name
// TODO: Fix the name restriction. The parms should be name, environment
func (e *EnvironmentService) Put(environment *Environment) (data *Environment, err error) {
	path := fmt.Sprintf("environments/%s", environment.Name)
	body, err := JSONReader(environment)
	if err != nil {
		return
	}

	err = e.client.magicRequestDecoder("PUT", path, body, &data)
	return
}

// Get the versions of a cookbook for this environment from the Chef server.
//
// Chef API docs: https://docs.chef.io/api_chef_server.html#environments-name-cookbooks
func (e *EnvironmentService) ListCookbooks(name string, numVersions string) (data EnvironmentCookbookResult, err error) {
	path := versionParams(fmt.Sprintf("environments/%s/cookbooks", name), numVersions)
	err = e.client.magicRequestDecoder("GET", path, nil, &data)
	return
}

// ListRecipes get the recipes list of recipes available to a given environment.
//
// Chef API docs: https://docs.chef.io/api_chef_server/#get-33
func (e *EnvironmentService) ListRecipes(name string) (data EnvironmentRecipesResult, err error) {
	path := fmt.Sprintf("environments/%s/recipes", name)
	err = e.client.magicRequestDecoder("GET", path, nil, &data)
	return
}

// Get a hash of cookbooks and cookbook versions (including all dependencies) that
// are required by the run_list array. Version constraints may be specified using
// the @ symbol after the cookbook name as a delimiter. Version constraints may also
// be present when the cookbook_versions attributes is specified for an environment
// or when dependencies are specified by a cookbook.
//
// Chef API docs: https://docs.chef.io/api_chef_server.html#cookbooks
// TODO: Write this

// Get a list of cookbooks and cookbook versions that are available to the specified environment.
//
// Chef API docs: https://docs.chef.io/api_chef_server.html#environments-name-cookbooks
// TODO: Write this