func()

in cft/comments.go [17:84]

• 56 lines of code
• 14 McCabe index (conditional complexity)

func (t Template) AddComments(comments []*Comment) error {
	node := t.Node

	if node.Kind == yaml.DocumentNode {
		node = node.Content[0]
	}

	for _, comment := range comments {
		var n *yaml.Node
		var err error

		if len(comment.Path) == 0 {
			n, err = s11n.GetPath(node, comment.Path)
			if err != nil {
				return err
			}

			if len(n.Content) == 0 {
				return fmt.Errorf("Unable to set head at node root")
			}

			n.Content[0].HeadComment = comment.Value

			continue
		}

		path, last := comment.Path[0:len(comment.Path)-1], comment.Path[len(comment.Path)-1]

		n, err = s11n.GetPath(node, path)
		if err != nil {
			return err
		}

		switch v := last.(type) {
		case string:
			key, value := s11n.GetMapValue(n, v)
			if value == nil {
				return fmt.Errorf("Can't find map key: '%s'", v)
			}

			switch value.Kind {
			case yaml.MappingNode, yaml.SequenceNode:
				if len(value.Content) == 0 {
					value.LineComment = comment.Value
				} else {
					key.LineComment = comment.Value
				}
			default:
				value.LineComment = comment.Value
			}
		case int:
			n, err = s11n.GetPath(node, comment.Path)
			if err != nil {
				return err
			}

			if n.Kind == yaml.ScalarNode {
				n.LineComment = comment.Value
			} else {
				n.HeadComment = comment.Value
			}
		default:
			return fmt.Errorf("Unexpected path element '%#v'", v)
		}
	}

	return nil
}