// Package parse provides functions for parsing
// CloudFormation templates from JSON and YAML inputs.
package parse

import (
	"fmt"
	"io"
	"os"

	"github.com/aws-cloudformation/rain/cft"
	"github.com/aws-cloudformation/rain/cft/diff"
	"github.com/aws-cloudformation/rain/internal/config"
	"github.com/aws-cloudformation/rain/internal/node"

	"gopkg.in/yaml.v3"
)

// Reader returns a cft.Template parsed from an io.Reader
func Reader(r io.Reader) (cft.Template, error) {
	data, err := io.ReadAll(r)
	if err != nil {
		return cft.Template{}, fmt.Errorf("unable to read input: %s", err)
	}

	return String(string(data))
}

// File returns a cft.Template parsed from a file specified by fileName
func File(fileName string) (cft.Template, error) {
	source, err := os.ReadFile(fileName)
	if err != nil {
		return cft.Template{}, fmt.Errorf("unable to read file: %s", err)
	}

	return String(string(source))
}

// Map returns a cft.Template parsed from a map[string]interface{}
func Map(input map[string]interface{}) (cft.Template, error) {
	var node yaml.Node
	err := node.Encode(input)
	if err != nil {
		return cft.Template{}, err
	}

	return Node(&node)
}

// String returns a cft.Template parsed from a string
func String(input string) (cft.Template, error) {
	var node yaml.Node
	err := yaml.Unmarshal([]byte(input), &node)
	if err != nil {
		return cft.Template{}, fmt.Errorf("invalid YAML: %s", err)
	}

	return Node(&node)
}

// Node returns a cft.Template parse from a *yaml.Node
func Node(node *yaml.Node) (cft.Template, error) {
	err := TransformNode(node)
	return cft.Template{Node: node}, err
}

// Verify confirms that there is no semantic difference between
// the source cft.Template and the string representation in output.
// This can be used to ensure that the parse package hasn't done
// anything unexpected to your template.
func Verify(source cft.Template, output string) error {
	// Check it matches the original
	validate, err := String(output)
	if err != nil {
		return err
	}

	config.Debugf("source: \n%v", node.ToJson(source.Node))
	config.Debugf("validate: \n%v", node.ToJson(validate.Node))

	d := diff.New(source, validate)
	if d.Mode() != diff.Unchanged {
		return fmt.Errorf("semantic difference after formatting:\n%s", d.Format(false))
	}

	return nil
}