fix: yaml stream format convension in the kcl generator (#310)

Signed-off-by: peefy <xpf6677@163.com>

pkg/kcl/api.go

@@ -444,7 +444,7 @@ func ExecResultToKCLResult(o *Option, resp *gpyrpc.ExecProgram_Result, logger io
 		return &result, nil
 	}
 
-	documents, err := splitDocuments(resp.YamlResult)
+	documents, err := SplitDocuments(resp.YamlResult)
 	if err != nil {
 		return &result, nil
 	}
@@ -486,10 +486,10 @@ func run(pathList []string, opts ...Option) (*KCLResultList, error) {
 	return runWithHooks(pathList, DefaultHooks, opts...)
 }
 
-// splitDocuments returns a slice of all documents contained in a YAML string. Multiple documents can be divided by the
+// SplitDocuments returns a slice of all documents contained in a YAML string. Multiple documents can be divided by the
 // YAML document separator (---). It allows for white space and comments to be after the separator on the same line,
 // but will return an error if anything else is on the line.
-func splitDocuments(s string) ([]string, error) {
+func SplitDocuments(s string) ([]string, error) {
 	docs := make([]string, 0)
 	if len(s) > 0 {
 		// The YAML document separator is any line that starts with ---

pkg/tools/gen/genkcl_yaml.go

@@ -2,9 +2,10 @@ package gen
 
 import (
 	"bytes"
-	"github.com/goccy/go-yaml"
 	"io"
-	"strings"
+
+	"github.com/goccy/go-yaml"
+	"kcl-lang.io/kcl-go/pkg/kcl"
 )
 
 func (k *kclGenerator) genKclFromYaml(w io.Writer, filename string, src interface{}) error {
@@ -58,7 +59,11 @@ func convertKclFromYamlString(byteData []byte) ([]data, error) {
 	byteData = bytes.ReplaceAll(byteData, []byte("\r\n"), []byte("\n"))
 	var result []data
 	// split yaml with ‘---’
-	for _, item := range strings.Split(string(byteData), "---") {
+	items, err := kcl.SplitDocuments(string(byteData))
+	if err != nil {
+		return nil, err
+	}
+	for _, item := range items {
 		yamlData := &yaml.MapSlice{}
 		if err := yaml.UnmarshalWithOptions([]byte(item), yamlData, yaml.UseOrderedMap()); err != nil {
 			return nil, err

pkg/tools/gen/testdata/yaml/comment/expect.k

@@ -0,0 +1,8 @@
+"""
+This file was generated by the KCL auto-gen tool. DO NOT EDIT.
+Editing this file might prove futile when you re-run the KCL auto-gen generate command.
+"""
+
+{
+    "cluster.name" = "my-application"
+}

pkg/tools/gen/testdata/yaml/comment/input.yaml

@@ -0,0 +1,2 @@
+# ---------------------------------- Cluster -----------------------------------
+cluster.name: my-application