// Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0

package addon

import (
	"errors"
	"fmt"
	"reflect"

	"gopkg.in/yaml.v3"
)

type cfnSection int

const (
	metadataSection cfnSection = iota + 1
	parametersSection
	mappingsSection
	conditionsSection
	transformSection
	resourcesSection
	outputsSection
)

// cfnTemplate represents a parsed YAML AWS CloudFormation template.
type cfnTemplate struct {
	Metadata   yaml.Node `yaml:"Metadata,omitempty"`
	Parameters yaml.Node `yaml:"Parameters,omitempty"`
	Mappings   yaml.Node `yaml:"Mappings,omitempty"`
	Conditions yaml.Node `yaml:"Conditions,omitempty"`
	Transform  yaml.Node `yaml:"Transform,omitempty"`
	Resources  yaml.Node `yaml:"Resources"` // Don't omit as this is the only section that's required by CloudFormation.
	Outputs    yaml.Node `yaml:"Outputs,omitempty"`

	name            string
	templateNameFor map[*yaml.Node]string // Maps a node to the name of the first template where it was defined.
}

func newCFNTemplate(name string) *cfnTemplate {
	return &cfnTemplate{
		name:            name,
		templateNameFor: make(map[*yaml.Node]string),
	}
}

// merge combines non-empty fields of other with t's fields.
func (t *cfnTemplate) merge(other *cfnTemplate) error {
	if err := t.mergeMetadata(other.Metadata); err != nil {
		return wrapKeyAlreadyExistsErr(metadataSection, t, other, err)
	}
	if err := t.mergeParameters(other.Parameters); err != nil {
		return wrapKeyAlreadyExistsErr(parametersSection, t, other, err)
	}
	if err := t.mergeMappings(other.Mappings); err != nil {
		return wrapKeyAlreadyExistsErr(mappingsSection, t, other, err)
	}
	if err := t.mergeConditions(other.Conditions); err != nil {
		return wrapKeyAlreadyExistsErr(conditionsSection, t, other, err)
	}
	if err := t.mergeTransform(other.Transform); err != nil {
		return wrapKeyAlreadyExistsErr(transformSection, t, other, err)
	}
	if err := t.mergeResources(other.Resources); err != nil {
		return wrapKeyAlreadyExistsErr(resourcesSection, t, other, err)
	}
	if err := t.mergeOutputs(other.Outputs); err != nil {
		return wrapKeyAlreadyExistsErr(outputsSection, t, other, err)
	}

	t.assignNewNodesTo(other.name)
	return nil
}

// mergeMetadata updates t's Metadata with additional metadata.
// If the key already exists in Metadata but with a different definition, returns errMetadataAlreadyExists.
func (t *cfnTemplate) mergeMetadata(metadata yaml.Node) error {
	return mergeSingleLevelMaps(&t.Metadata, &metadata)
}

// mergeParameters updates t's Parameters with additional parameters.
// If the parameterLogicalID already exists but with a different value, returns errParameterAlreadyExists.
func (t *cfnTemplate) mergeParameters(params yaml.Node) error {
	return mergeSingleLevelMaps(&t.Parameters, &params)
}

// mergeMappings updates t's Mappings with additional mappings.
// If a mapping already exists with a different value, returns errMappingAlreadyExists.
func (t *cfnTemplate) mergeMappings(mappings yaml.Node) error {
	return mergeTwoLevelMaps(&t.Mappings, &mappings)
}

// mergeConditions updates t's Conditions with additional conditions.
// If a condition already exists with a different value, returns errConditionAlreadyExists.
func (t *cfnTemplate) mergeConditions(conditions yaml.Node) error {
	return mergeSingleLevelMaps(&t.Conditions, &conditions)
}

// mergeTransform adds transform's contents to t's Transform.
func (t *cfnTemplate) mergeTransform(transform yaml.Node) error {
	addToSet(&t.Transform, &transform)
	return nil
}

// mergeResources updates t's Resources with additional resources.
// If a resource already exists with a different value, returns errResourceAlreadyExists.
func (t *cfnTemplate) mergeResources(resources yaml.Node) error {
	return mergeSingleLevelMaps(&t.Resources, &resources)
}

// mergeOutputs updates t's Outputs with additional outputs.
// If an output already exists with a different value, returns errOutputAlreadyExists.
func (t *cfnTemplate) mergeOutputs(outputs yaml.Node) error {
	return mergeSingleLevelMaps(&t.Outputs, &outputs)
}

// assignNewNodesTo associates every new node added to the template t with the tplName.
func (t *cfnTemplate) assignNewNodesTo(tplName string) {
	if t == nil {
		return
	}

	var assign func(node *yaml.Node)
	assign = func(node *yaml.Node) {
		if node == nil {
			return
		}
		if _, ok := t.templateNameFor[node]; ok {
			// node is already associated with a previous template.
			return
		}
		t.templateNameFor[node] = tplName
		for _, c := range node.Content {
			assign(c)
		}
	}

	// Call assign() only on fields that represent CloudFormation sections.
	tpl := reflect.ValueOf(*t)
	for i := 0; i < tpl.NumField(); i += 1 {
		field := tpl.Field(i)
		if !field.CanInterface() {
			// Fields that are not exported will panic if we call Interface(), therefore
			// check the type only against exported cfnTemplate fields.
			continue
		}
		if section, ok := field.Interface().(yaml.Node); ok {
			assign(&section)
		}
	}
}

// mergeTwoLevelMaps merges the top and second level keys of src node to dst.
// It assumes that both nodes are nested maps. For example, a node can hold:
// Mapping01:  # Top Level is a map.
//    Key01:   # Second Level is also a map.
//      Name: Value01
//    Key02:   # Second Level.
//      Name: Value02
//
// If a second-level key exists in both src and dst but has different values, then returns an errKeyAlreadyExists.
// If a second-level key exists in src but not in dst, it merges the second level key to dst.
// If a top-level key exists in src but not in dst, merges it.
func mergeTwoLevelMaps(dst, src *yaml.Node) error {
	secondLevelHandler := func(key string, dstVal, srcVal *yaml.Node) error {
		if err := mergeSingleLevelMaps(dstVal, srcVal); err != nil {
			var keyExistsErr *errKeyAlreadyExists
			if errors.As(err, &keyExistsErr) {
				keyExistsErr.Key = fmt.Sprintf("%s.%s", key, keyExistsErr.Key)
				return keyExistsErr
			}
			return err
		}
		return nil
	}
	return mergeMapNodes(dst, src, secondLevelHandler)
}

// mergeSingleLevelMaps merges the keys of src node to dst.
// It assumes that both nodes are a map. For example, a node can hold:
// Resources:
//    MyResourceName:
//        ...  # If the contents of "MyResourceName" are not equal in both src and dst then err.
//
// If a key exists in both src and dst but has different values, then returns an errKeyAlreadyExists.
// If a key exists in both src and dst and the values are equal, then do nothing.
// If a key exists in src but not in dst, merges it.
func mergeSingleLevelMaps(dst, src *yaml.Node) error {
	areValuesEqualHandler := func(key string, dstVal, srcVal *yaml.Node) error {
		if !isEqual(dstVal, srcVal) {
			return &errKeyAlreadyExists{
				Key:    key,
				First:  dstVal,
				Second: srcVal,
			}
		}
		return nil
	}
	return mergeMapNodes(dst, src, areValuesEqualHandler)
}

type keyExistsHandler func(key string, dstVal, srcVal *yaml.Node) error

// mergeMapNodes merges the src node to dst.
// It assumes that both nodes have a "mapping" type. See https://yaml.org/spec/1.2/spec.html#id2802432
//
// If a key exists in src but not in dst, then adds the key and value to dst.
// If a key exists in both src and dst, invokes the keyExistsHandler.
func mergeMapNodes(dst, src *yaml.Node, handler keyExistsHandler) error {
	if src.IsZero() {
		return nil
	}

	if dst.IsZero() {
		*dst = *src
		return nil
	}

	dstMap := mappingNode(dst)
	var newContent []*yaml.Node
	for _, srcContent := range mappingContents(src) {
		srcKey := srcContent.keyNode.Value

		dstValueNode, ok := dstMap[srcKey]
		if !ok {
			// The key doesn't exist in dst, we want to retain the two src nodes.
			newContent = append(newContent, srcContent.keyNode, srcContent.valueNode)
			continue
		}

		if err := handler(srcKey, dstValueNode, srcContent.valueNode); err != nil {
			return err
		}
	}
	dst.Content = append(dst.Content, newContent...)
	return nil
}

// addToSet adds a non-zero node to a set. If the node represents a sequence,
// then adds its contents to the set.
//
// If the node or its contents already exist in the set, does nothing.
// Otherwise, appends the node or its contents to the set.
func addToSet(set, node *yaml.Node) {
	if node.IsZero() {
		return
	}
	nodes := []*yaml.Node{node}
	if node.Kind == yaml.SequenceNode {
		// If node is a list, we should add its contents to the set instead.
		nodes = node.Content
	}
	if set.IsZero() {
		*set = yaml.Node{
			Kind:    yaml.SequenceNode,
			Content: nodes,
		}
		return
	}

	var newElements []*yaml.Node
	for _, c := range set.Content {
		for _, n := range nodes {
			if !isEqual(c, n) {
				newElements = append(newElements, n)
			}
		}
	}
	set.Content = append(set.Content, newElements...)
}

// mappingNode transforms a flat "mapping" yaml.Node to a hashmap.
func mappingNode(n *yaml.Node) map[string]*yaml.Node {
	m := make(map[string]*yaml.Node)
	for i := 0; i < len(n.Content); i += 2 {
		m[n.Content[i].Value] = n.Content[i+1]
	}
	return m
}

type mappingContent struct {
	keyNode   *yaml.Node
	valueNode *yaml.Node
}

func mappingContents(mappingNode *yaml.Node) []mappingContent {
	var results []mappingContent
	for i := 0; i < len(mappingNode.Content); i += 2 {
		// The content of a map always come in pairs.
		// The first element represents a key, ex: {Value: "ELBIngressGroup", Kind: ScalarNode, Tag: "!!str", Content: nil}
		// The second element holds the value, ex: {Value: "", Kind: MappingNode, Tag:"!!map", Content:[...]}
		results = append(results, mappingContent{
			keyNode:   mappingNode.Content[i],
			valueNode: mappingNode.Content[i+1],
		})
	}
	return results
}

// isEqual returns true if the first and second nodes are deeply equal in all of their values except stylistic ones.
//
// We ignore the style (ex: single quote vs. double) in which the nodes are defined, the comments associated with
// the nodes, and the indentation and position of the nodes as they're only visual properties and don't matter.
func isEqual(first *yaml.Node, second *yaml.Node) bool {
	if first == nil {
		return second == nil
	}
	if second == nil {
		return false
	}
	if len(first.Content) != len(second.Content) {
		return false
	}
	hasSameContent := true
	for i := 0; i < len(first.Content); i += 1 {
		hasSameContent = hasSameContent && isEqual(first.Content[i], second.Content[i])
	}
	return first.Kind == second.Kind &&
		first.Tag == second.Tag &&
		first.Value == second.Value &&
		first.Anchor == second.Anchor &&
		isEqual(first.Alias, second.Alias) &&
		hasSameContent
}