mirror of
https://github.com/gohugoio/hugo.git
synced 2024-11-21 20:46:30 -05:00
tpl/transform: Add template func for TOML/JSON/YAML docs examples conversion
Usage: ```html {{ "title = \"Hello World\"" | transform.Remarshal "json" | safeHTML }} ``` Fixes #4389
This commit is contained in:
parent
2e95ec6844
commit
d382502d6d
4 changed files with 265 additions and 0 deletions
|
@ -465,3 +465,9 @@ func DiffStringSlices(slice1 []string, slice2 []string) []string {
|
||||||
|
|
||||||
return diffStr
|
return diffStr
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// DiffString splits the strings into fields and runs it into DiffStringSlices.
|
||||||
|
// Useful for tests.
|
||||||
|
func DiffStrings(s1, s2 string) []string {
|
||||||
|
return DiffStringSlices(strings.Fields(s1), strings.Fields(s2))
|
||||||
|
}
|
||||||
|
|
|
@ -88,6 +88,13 @@ func init() {
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
ns.AddMethodMapping(ctx.Remarshal,
|
||||||
|
nil,
|
||||||
|
[][2]string{
|
||||||
|
{`{{ "title = \"Hello World\"" | transform.Remarshal "json" | safeHTML }}`, "{\n \"title\": \"Hello World\"\n}\n"},
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
return ns
|
return ns
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
98
tpl/transform/remarshal.go
Normal file
98
tpl/transform/remarshal.go
Normal file
|
@ -0,0 +1,98 @@
|
||||||
|
package transform
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"errors"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gohugoio/hugo/parser"
|
||||||
|
"github.com/spf13/cast"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Remarshal is used in the Hugo documentation to convert configuration
|
||||||
|
// examples from YAML to JSON, TOML (and possibly the other way around).
|
||||||
|
// The is primarily a helper for the Hugo docs site.
|
||||||
|
// It is not a general purpose YAML to TOML converter etc., and may
|
||||||
|
// change without notice if it serves a purpose in the docs.
|
||||||
|
// Format is one of json, yaml or toml.
|
||||||
|
func (ns *Namespace) Remarshal(format string, data interface{}) (string, error) {
|
||||||
|
from, err := cast.ToStringE(data)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
from = strings.TrimSpace(from)
|
||||||
|
format = strings.TrimSpace(strings.ToLower(format))
|
||||||
|
|
||||||
|
if from == "" {
|
||||||
|
return "", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
mark, err := toFormatMark(format)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
fromFormat, err := detectFormat(from)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
var metaHandler func(d []byte) (map[string]interface{}, error)
|
||||||
|
|
||||||
|
switch fromFormat {
|
||||||
|
case "yaml":
|
||||||
|
metaHandler = parser.HandleYAMLMetaData
|
||||||
|
case "toml":
|
||||||
|
metaHandler = parser.HandleTOMLMetaData
|
||||||
|
case "json":
|
||||||
|
metaHandler = parser.HandleJSONMetaData
|
||||||
|
}
|
||||||
|
|
||||||
|
meta, err := metaHandler([]byte(from))
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
var result bytes.Buffer
|
||||||
|
if err := parser.InterfaceToConfig(meta, mark, &result); err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
|
||||||
|
return result.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func toFormatMark(format string) (rune, error) {
|
||||||
|
// TODO(bep) the parser package needs a cleaning.
|
||||||
|
switch format {
|
||||||
|
case "yaml":
|
||||||
|
return rune(parser.YAMLLead[0]), nil
|
||||||
|
case "toml":
|
||||||
|
return rune(parser.TOMLLead[0]), nil
|
||||||
|
case "json":
|
||||||
|
return rune(parser.JSONLead[0]), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0, errors.New("failed to detect target data serialization format")
|
||||||
|
}
|
||||||
|
|
||||||
|
func detectFormat(data string) (string, error) {
|
||||||
|
jsonIdx := strings.Index(data, "{")
|
||||||
|
yamlIdx := strings.Index(data, ":")
|
||||||
|
tomlIdx := strings.Index(data, "=")
|
||||||
|
|
||||||
|
if jsonIdx != -1 && (yamlIdx == -1 || jsonIdx < yamlIdx) && (tomlIdx == -1 || jsonIdx < tomlIdx) {
|
||||||
|
return "json", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if yamlIdx != -1 && (tomlIdx == -1 || yamlIdx < tomlIdx) {
|
||||||
|
return "yaml", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
if tomlIdx != -1 {
|
||||||
|
return "toml", nil
|
||||||
|
}
|
||||||
|
|
||||||
|
return "", errors.New("failed to detect data serialization format")
|
||||||
|
|
||||||
|
}
|
154
tpl/transform/remarshal_test.go
Normal file
154
tpl/transform/remarshal_test.go
Normal file
|
@ -0,0 +1,154 @@
|
||||||
|
// Copyright 2018 The Hugo Authors. All rights reserved.
|
||||||
|
//
|
||||||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
// you may not use this file except in compliance with the License.
|
||||||
|
// You may obtain a copy of the License at
|
||||||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
//
|
||||||
|
// Unless required by applicable law or agreed to in writing, software
|
||||||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
// See the License for the specific language governing permissions and
|
||||||
|
// limitations under the License.
|
||||||
|
|
||||||
|
package transform
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/gohugoio/hugo/helpers"
|
||||||
|
"github.com/spf13/viper"
|
||||||
|
"github.com/stretchr/testify/require"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestRemarshal(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
ns := New(newDeps(viper.New()))
|
||||||
|
assert := require.New(t)
|
||||||
|
|
||||||
|
tomlExample := `title = "Test Metadata"
|
||||||
|
|
||||||
|
[[resources]]
|
||||||
|
src = "**image-4.png"
|
||||||
|
title = "The Fourth Image!"
|
||||||
|
[resources.params]
|
||||||
|
byline = "picasso"
|
||||||
|
|
||||||
|
[[resources]]
|
||||||
|
name = "my-cool-image-:counter"
|
||||||
|
src = "**.png"
|
||||||
|
title = "TOML: The Image #:counter"
|
||||||
|
[resources.params]
|
||||||
|
byline = "bep"
|
||||||
|
`
|
||||||
|
|
||||||
|
yamlExample := `resources:
|
||||||
|
- params:
|
||||||
|
byline: picasso
|
||||||
|
src: '**image-4.png'
|
||||||
|
title: The Fourth Image!
|
||||||
|
- name: my-cool-image-:counter
|
||||||
|
params:
|
||||||
|
byline: bep
|
||||||
|
src: '**.png'
|
||||||
|
title: 'TOML: The Image #:counter'
|
||||||
|
title: Test Metadata
|
||||||
|
`
|
||||||
|
|
||||||
|
jsonExample := `{
|
||||||
|
"resources": [
|
||||||
|
{
|
||||||
|
"params": {
|
||||||
|
"byline": "picasso"
|
||||||
|
},
|
||||||
|
"src": "**image-4.png",
|
||||||
|
"title": "The Fourth Image!"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"name": "my-cool-image-:counter",
|
||||||
|
"params": {
|
||||||
|
"byline": "bep"
|
||||||
|
},
|
||||||
|
"src": "**.png",
|
||||||
|
"title": "TOML: The Image #:counter"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "Test Metadata"
|
||||||
|
}
|
||||||
|
`
|
||||||
|
|
||||||
|
variants := []struct {
|
||||||
|
format string
|
||||||
|
data string
|
||||||
|
}{
|
||||||
|
{"yaml", yamlExample},
|
||||||
|
{"json", jsonExample},
|
||||||
|
{"toml", tomlExample},
|
||||||
|
{"TOML", tomlExample},
|
||||||
|
{"Toml", tomlExample},
|
||||||
|
{" TOML ", tomlExample},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, v1 := range variants {
|
||||||
|
for _, v2 := range variants {
|
||||||
|
// Both from and to may be the same here, but that is fine.
|
||||||
|
fromTo := fmt.Sprintf("%s => %s", v2.format, v1.format)
|
||||||
|
|
||||||
|
converted, err := ns.Remarshal(v1.format, v2.data)
|
||||||
|
assert.NoError(err, fromTo)
|
||||||
|
diff := helpers.DiffStrings(v1.data, converted)
|
||||||
|
if len(diff) > 0 {
|
||||||
|
t.Errorf("[%s] Expected \n%v\ngot\n%v\ndiff:\n%v", fromTo, v1.data, converted, diff)
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestTestRemarshalError(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
ns := New(newDeps(viper.New()))
|
||||||
|
assert := require.New(t)
|
||||||
|
|
||||||
|
_, err := ns.Remarshal("asdf", "asdf")
|
||||||
|
assert.Error(err)
|
||||||
|
|
||||||
|
_, err = ns.Remarshal("json", "asdf")
|
||||||
|
assert.Error(err)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRemarshalDetectFormat(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
assert := require.New(t)
|
||||||
|
|
||||||
|
for i, test := range []struct {
|
||||||
|
data string
|
||||||
|
expect interface{}
|
||||||
|
}{
|
||||||
|
{`foo = "bar"`, "toml"},
|
||||||
|
{` foo = "bar"`, "toml"},
|
||||||
|
{`foo="bar"`, "toml"},
|
||||||
|
{`foo: "bar"`, "yaml"},
|
||||||
|
{`foo:"bar"`, "yaml"},
|
||||||
|
{`{ "foo": "bar"`, "json"},
|
||||||
|
{`asdfasdf`, false},
|
||||||
|
{``, false},
|
||||||
|
} {
|
||||||
|
errMsg := fmt.Sprintf("[%d] %s", i, test.data)
|
||||||
|
|
||||||
|
result, err := detectFormat(test.data)
|
||||||
|
|
||||||
|
if b, ok := test.expect.(bool); ok && !b {
|
||||||
|
assert.Error(err, errMsg)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
assert.NoError(err, errMsg)
|
||||||
|
assert.Equal(test.expect, result, errMsg)
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in a new issue