func_findinmap.go

package cloudformation

import "encoding/json"

// FindInMap returns a new instance of FindInMapFunc.
func FindInMap(mapName string, topLevelKey Stringable, secondLevelKey Stringable) *StringExpr {
	return FindInMapFunc{
		MapName:        mapName,
		TopLevelKey:    *topLevelKey.String(),
		SecondLevelKey: *secondLevelKey.String(),
	}.String()
}

// FindInMapFunc represents an invocation of the Fn::FindInMap intrinsic.
//
// The intrinsic function Fn::FindInMap returns the value corresponding to
// keys in a two-level map that is declared in the Mappings section.
//
// See http://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/intrinsic-function-reference-findinmap.html
type FindInMapFunc struct {
	MapName        string
	TopLevelKey    StringExpr
	SecondLevelKey StringExpr
}

// MarshalJSON returns a JSON representation of the object
func (f FindInMapFunc) MarshalJSON() ([]byte, error) {
	return json.Marshal(struct {
		FnFindInMap []interface{} `json:"Fn::FindInMap"`
	}{FnFindInMap: []interface{}{f.MapName, f.TopLevelKey, f.SecondLevelKey}})
}

// UnmarshalJSON sets the object from the provided JSON representation
func (f *FindInMapFunc) UnmarshalJSON(buf []byte) error {
	v := struct {
		FnFindInMap [3]json.RawMessage `json:"Fn::FindInMap"`
	}{}
	if err := json.Unmarshal(buf, &v); err != nil {
		return err
	}
	if err := json.Unmarshal(v.FnFindInMap[0], &f.MapName); err != nil {
		return err
	}
	if err := json.Unmarshal(v.FnFindInMap[1], &f.TopLevelKey); err != nil {
		return err
	}
	if err := json.Unmarshal(v.FnFindInMap[2], &f.SecondLevelKey); err != nil {
		return err
	}

	return nil
}

func (f FindInMapFunc) String() *StringExpr {
	return &StringExpr{Func: f}
}

var _ Stringable = FindInMapFunc{} // FindInMapFunc must implement Stringable
var _ StringFunc = FindInMapFunc{} // FindInMapFunc must implement StringFunc