diff --git a/pkg/tools/gen/genkcl_test.go b/pkg/tools/gen/genkcl_test.go index 5bd9773d..c83695e2 100644 --- a/pkg/tools/gen/genkcl_test.go +++ b/pkg/tools/gen/genkcl_test.go @@ -2,6 +2,7 @@ package gen import ( "bytes" + "fmt" "log" "os" "path/filepath" @@ -142,11 +143,14 @@ func TestGenKclFromJsonSchema(t *testing.T) { for _, testcase := range cases { t.Run(testcase.name, func(t *testing.T) { var buf bytes.Buffer - err := GenKcl(&buf, testcase.input, nil, &GenKclOptions{}) + err := GenKcl(&buf, testcase.input, nil, &GenKclOptions{ + Mode: ModeJsonSchema, + }) if err != nil { t.Fatal(err) } result := buf.Bytes() + fmt.Printf("result: %v\n", string(result)) assert2.Equal(t, testcase.expect, string(bytes.ReplaceAll(result, []byte("\r\n"), []byte("\n")))) }) } diff --git a/pkg/tools/gen/genkcl_value.go b/pkg/tools/gen/genkcl_value.go index 12f89847..5ecd34ef 100644 --- a/pkg/tools/gen/genkcl_value.go +++ b/pkg/tools/gen/genkcl_value.go @@ -52,6 +52,7 @@ type printer struct { writer io.Writer listInline bool configInline bool + escape bool } func (p *printer) writeIndent() { @@ -243,7 +244,7 @@ func (p *printer) walkValue(v any) error { p.writeConfigEnd() case reflect.String: value := val.String() - if isStringEscaped(value) { + if isStringEscaped(value) && p.escape { if value[len(value)-1] == '"' { // if the string ends with '"' then we need to add a space after the closing triple quote p.write(fmt.Sprintf(`r"""%s """`, value)) diff --git a/pkg/tools/gen/template.go b/pkg/tools/gen/template.go index 4352a136..48c25559 100644 --- a/pkg/tools/gen/template.go +++ b/pkg/tools/gen/template.go @@ -29,10 +29,11 @@ var ( ) var funcs = template.FuncMap{ - "formatType": formatType, - "formatValue": formatValue, - "formatName": formatName, - "indentLines": indentLines, + "formatType": formatType, + "formatValue": formatValue, + "formatValueWithEscape": formatValueWithEscape, + "formatName": formatName, + "indentLines": indentLines, "isKclData": func(v interface{}) bool { _, ok := v.([]data) return ok @@ -92,12 +93,13 @@ func formatType(t typeInterface) string { return typAny } -func formatValue(v interface{}) string { +func formatValueWithEscape(v interface{}, escape bool) string { var buf bytes.Buffer p := &printer{ listInline: true, configInline: true, writer: &buf, + escape: escape, } err := p.walkValue(v) if err != nil { @@ -107,6 +109,10 @@ func formatValue(v interface{}) string { } } +func formatValue(v interface{}) string { + return formatValueWithEscape(v, true) +} + var kclKeywords = map[string]struct{}{ "True": {}, "False": {}, diff --git a/pkg/tools/gen/templates/kcl/document.gotmpl b/pkg/tools/gen/templates/kcl/document.gotmpl index f9b44b9b..a217cf20 100644 --- a/pkg/tools/gen/templates/kcl/document.gotmpl +++ b/pkg/tools/gen/templates/kcl/document.gotmpl @@ -10,7 +10,7 @@ ---------- {{- range .Properties }} {{ formatName .Name }} : {{ formatType .Type }}, {{ if .Required }}required{{ else }}optional{{ end }} - {{- if .HasDefault }}, default is {{ formatValue .DefaultValue }}{{ end }} + {{- if .HasDefault }}, default is {{ formatValueWithEscape .DefaultValue false}}{{ end }} {{- if .Description }}{{ "\n" }}{{ indentLines .Description " " }}{{ end }} {{- end -}} diff --git a/pkg/tools/gen/testdata/jsonschema/document_quota/expect.k b/pkg/tools/gen/testdata/jsonschema/document_quota/expect.k new file mode 100644 index 00000000..3ac61848 --- /dev/null +++ b/pkg/tools/gen/testdata/jsonschema/document_quota/expect.k @@ -0,0 +1,16 @@ +""" +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. +""" + +schema Input: + r""" + Input + + Attributes + ---------- + foo : str, required, default is "more \"quotes\"" + """ + + foo: str = r"""more "quotes" """ + diff --git a/pkg/tools/gen/testdata/jsonschema/document_quota/input.json b/pkg/tools/gen/testdata/jsonschema/document_quota/input.json new file mode 100644 index 00000000..e6a35815 --- /dev/null +++ b/pkg/tools/gen/testdata/jsonschema/document_quota/input.json @@ -0,0 +1,12 @@ +{ + "type": "object", + "properties": { + "foo": { + "type": "string", + "default": "more \"quotes\"" + } + }, + "required": [ + "foo" + ] +} \ No newline at end of file diff --git a/pkg/tools/gen/testdata/jsonschema/document_quota_0/expect.k b/pkg/tools/gen/testdata/jsonschema/document_quota_0/expect.k new file mode 100644 index 00000000..986ecc77 --- /dev/null +++ b/pkg/tools/gen/testdata/jsonschema/document_quota_0/expect.k @@ -0,0 +1,24 @@ +""" +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. +""" + +schema Input: + r""" + Input + + Attributes + ---------- + foo : str, required, default is "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Quisque pellentesque ligula consequat, aliquam elit vitae, dapibus velit. Donec posuere venenatis diam, et viverra mi facilisis nec. Mauris eget velit eu risus ornare accumsan. Suspendisse rutrum vehicula tellus. Morbi lobortis lorem eget odio consequat dapibus. Etiam metus arcu, tincidunt id enim eget, mollis tempor purus. Sed finibus ante id odio dignissim mollis. Fusce auctor sit amet risus nec dictum. Interdum et malesuada fames ac ante ipsum primis in faucibus. Donec ornare orci nec fermentum accumsan. Vivamus fermentum dolor non ligula consequat, vitae pellentesque quam accumsan. Ut lobortis facilisis pharetra. Nullam tristique orci in nisi tempor, ut consequat metus molestie.\n\nSuspendisse justo quam, fermentum id ultrices sed, fringilla non eros. Ut leo risus, efficitur id posuere sit amet, sagittis vitae risus. Nullam mattis arcu quis ligula tempus ullamcorper. Donec vel nibh vel sem fermentum efficitur id a velit. Maecenas congue iaculis arcu et consectetur. Maecenas gravida nisl vitae eros dignissim, rutrum malesuada felis pulvinar. Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia curae;\n\nCurabitur vitae augue et sapien pulvinar scelerisque. Proin eleifend varius feugiat. Nullam sagittis ante at ligula pharetra, vitae molestie turpis aliquet. Donec vitae odio nec tortor accumsan condimentum. Phasellus suscipit semper tortor, ac sodales orci varius non. Praesent iaculis ultrices dui in fringilla. Sed tristique libero sed sapien sagittis eleifend.\n\nCurabitur id fermentum ex, ac aliquam ligula. Duis tellus neque, luctus eu tristique eget, viverra vitae magna. Maecenas facilisis mauris quam, eu sodales elit vulputate quis. Suspendisse nulla ipsum, auctor ut efficitur sed, rhoncus ut nisi. Donec viverra libero a rhoncus bibendum. Nullam sit amet metus nisi. Nunc venenatis eu ante quis egestas.\n\nUt sollicitudin pellentesque sem in consectetur. Nunc imperdiet lacus in venenatis consectetur. Vestibulum sed turpis tempor, mattis velit eu, hendrerit quam. Morbi ac pretium orci, ut mattis felis. Phasellus id faucibus orci, lobortis bibendum eros. Nunc nec libero consectetur, elementum tortor vitae, cursus orci. Curabitur nisl velit, auctor eu rutrum id, suscipit cursus est." + """ + + foo: str = r"""Lorem ipsum dolor sit amet, consectetur adipiscing elit. Quisque pellentesque ligula consequat, aliquam elit vitae, dapibus velit. Donec posuere venenatis diam, et viverra mi facilisis nec. Mauris eget velit eu risus ornare accumsan. Suspendisse rutrum vehicula tellus. Morbi lobortis lorem eget odio consequat dapibus. Etiam metus arcu, tincidunt id enim eget, mollis tempor purus. Sed finibus ante id odio dignissim mollis. Fusce auctor sit amet risus nec dictum. Interdum et malesuada fames ac ante ipsum primis in faucibus. Donec ornare orci nec fermentum accumsan. Vivamus fermentum dolor non ligula consequat, vitae pellentesque quam accumsan. Ut lobortis facilisis pharetra. Nullam tristique orci in nisi tempor, ut consequat metus molestie. + +Suspendisse justo quam, fermentum id ultrices sed, fringilla non eros. Ut leo risus, efficitur id posuere sit amet, sagittis vitae risus. Nullam mattis arcu quis ligula tempus ullamcorper. Donec vel nibh vel sem fermentum efficitur id a velit. Maecenas congue iaculis arcu et consectetur. Maecenas gravida nisl vitae eros dignissim, rutrum malesuada felis pulvinar. Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia curae; + +Curabitur vitae augue et sapien pulvinar scelerisque. Proin eleifend varius feugiat. Nullam sagittis ante at ligula pharetra, vitae molestie turpis aliquet. Donec vitae odio nec tortor accumsan condimentum. Phasellus suscipit semper tortor, ac sodales orci varius non. Praesent iaculis ultrices dui in fringilla. Sed tristique libero sed sapien sagittis eleifend. + +Curabitur id fermentum ex, ac aliquam ligula. Duis tellus neque, luctus eu tristique eget, viverra vitae magna. Maecenas facilisis mauris quam, eu sodales elit vulputate quis. Suspendisse nulla ipsum, auctor ut efficitur sed, rhoncus ut nisi. Donec viverra libero a rhoncus bibendum. Nullam sit amet metus nisi. Nunc venenatis eu ante quis egestas. + +Ut sollicitudin pellentesque sem in consectetur. Nunc imperdiet lacus in venenatis consectetur. Vestibulum sed turpis tempor, mattis velit eu, hendrerit quam. Morbi ac pretium orci, ut mattis felis. Phasellus id faucibus orci, lobortis bibendum eros. Nunc nec libero consectetur, elementum tortor vitae, cursus orci. Curabitur nisl velit, auctor eu rutrum id, suscipit cursus est.""" + diff --git a/pkg/tools/gen/testdata/jsonschema/document_quota_0/input.json b/pkg/tools/gen/testdata/jsonschema/document_quota_0/input.json new file mode 100644 index 00000000..02cf26cd --- /dev/null +++ b/pkg/tools/gen/testdata/jsonschema/document_quota_0/input.json @@ -0,0 +1,12 @@ +{ + "type": "object", + "properties": { + "foo": { + "type": "string", + "default": "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Quisque pellentesque ligula consequat, aliquam elit vitae, dapibus velit. Donec posuere venenatis diam, et viverra mi facilisis nec. Mauris eget velit eu risus ornare accumsan. Suspendisse rutrum vehicula tellus. Morbi lobortis lorem eget odio consequat dapibus. Etiam metus arcu, tincidunt id enim eget, mollis tempor purus. Sed finibus ante id odio dignissim mollis. Fusce auctor sit amet risus nec dictum. Interdum et malesuada fames ac ante ipsum primis in faucibus. Donec ornare orci nec fermentum accumsan. Vivamus fermentum dolor non ligula consequat, vitae pellentesque quam accumsan. Ut lobortis facilisis pharetra. Nullam tristique orci in nisi tempor, ut consequat metus molestie.\n\nSuspendisse justo quam, fermentum id ultrices sed, fringilla non eros. Ut leo risus, efficitur id posuere sit amet, sagittis vitae risus. Nullam mattis arcu quis ligula tempus ullamcorper. Donec vel nibh vel sem fermentum efficitur id a velit. Maecenas congue iaculis arcu et consectetur. Maecenas gravida nisl vitae eros dignissim, rutrum malesuada felis pulvinar. Vestibulum ante ipsum primis in faucibus orci luctus et ultrices posuere cubilia curae;\n\nCurabitur vitae augue et sapien pulvinar scelerisque. Proin eleifend varius feugiat. Nullam sagittis ante at ligula pharetra, vitae molestie turpis aliquet. Donec vitae odio nec tortor accumsan condimentum. Phasellus suscipit semper tortor, ac sodales orci varius non. Praesent iaculis ultrices dui in fringilla. Sed tristique libero sed sapien sagittis eleifend.\n\nCurabitur id fermentum ex, ac aliquam ligula. Duis tellus neque, luctus eu tristique eget, viverra vitae magna. Maecenas facilisis mauris quam, eu sodales elit vulputate quis. Suspendisse nulla ipsum, auctor ut efficitur sed, rhoncus ut nisi. Donec viverra libero a rhoncus bibendum. Nullam sit amet metus nisi. Nunc venenatis eu ante quis egestas.\n\nUt sollicitudin pellentesque sem in consectetur. Nunc imperdiet lacus in venenatis consectetur. Vestibulum sed turpis tempor, mattis velit eu, hendrerit quam. Morbi ac pretium orci, ut mattis felis. Phasellus id faucibus orci, lobortis bibendum eros. Nunc nec libero consectetur, elementum tortor vitae, cursus orci. Curabitur nisl velit, auctor eu rutrum id, suscipit cursus est." + } + }, + "required": [ + "foo" + ] +} \ No newline at end of file