Nathan Baulch
69 lines
1.8 KiB
Go
69 lines
1.8 KiB
Go
// Package testsuite provides helper functions for interoperating with the
|
|
// language-agnostic TOML test suite at github.com/BurntSushi/toml-test.
|
|
package testsuite
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"os"
|
|
|
|
"github.com/pelletier/go-toml/v2"
|
|
)
|
|
|
|
// Marshal is a helper function for calling toml.Marshal
|
|
//
|
|
// Only needed to avoid package import loops.
|
|
func Marshal(v interface{}) ([]byte, error) {
|
|
return toml.Marshal(v)
|
|
}
|
|
|
|
// Unmarshal is a helper function for calling toml.Unmarshal.
|
|
//
|
|
// Only needed to avoid package import loops.
|
|
func Unmarshal(data []byte, v interface{}) error {
|
|
return toml.Unmarshal(data, v)
|
|
}
|
|
|
|
// ValueToTaggedJSON takes a data structure and returns the tagged JSON
|
|
// representation.
|
|
func ValueToTaggedJSON(doc interface{}) ([]byte, error) {
|
|
return json.MarshalIndent(addTag("", doc), "", " ")
|
|
}
|
|
|
|
// DecodeStdin is a helper function for the toml-test binary interface. TOML input
|
|
// is read from STDIN and a resulting tagged JSON representation is written to
|
|
// STDOUT.
|
|
func DecodeStdin() error {
|
|
var decoded map[string]interface{}
|
|
|
|
if err := toml.NewDecoder(os.Stdin).Decode(&decoded); err != nil {
|
|
return fmt.Errorf("Error decoding TOML: %s", err)
|
|
}
|
|
|
|
j := json.NewEncoder(os.Stdout)
|
|
j.SetIndent("", " ")
|
|
if err := j.Encode(addTag("", decoded)); err != nil {
|
|
return fmt.Errorf("Error encoding JSON: %s", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// EncodeStdin is a helper function for the toml-test binary interface. Tagged
|
|
// JSON is read from STDIN and a resulting TOML representation is written to
|
|
// STDOUT.
|
|
func EncodeStdin() error {
|
|
var j interface{}
|
|
err := json.NewDecoder(os.Stdin).Decode(&j)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rm, err := rmTag(j)
|
|
if err != nil {
|
|
return fmt.Errorf("removing tags: %w", err)
|
|
}
|
|
|
|
return toml.NewEncoder(os.Stdout).Encode(rm)
|
|
}
|