summaryrefslogtreecommitdiffstats
path: root/vendor/github.com/pelletier/go-toml/cmd/tomljson
diff options
context:
space:
mode:
Diffstat (limited to 'vendor/github.com/pelletier/go-toml/cmd/tomljson')
-rw-r--r--vendor/github.com/pelletier/go-toml/cmd/tomljson/main.go67
-rw-r--r--vendor/github.com/pelletier/go-toml/cmd/tomljson/main_test.go82
2 files changed, 149 insertions, 0 deletions
diff --git a/vendor/github.com/pelletier/go-toml/cmd/tomljson/main.go b/vendor/github.com/pelletier/go-toml/cmd/tomljson/main.go
new file mode 100644
index 000000000..7e9dc467e
--- /dev/null
+++ b/vendor/github.com/pelletier/go-toml/cmd/tomljson/main.go
@@ -0,0 +1,67 @@
+package main
+
+import (
+ "encoding/json"
+ "flag"
+ "fmt"
+ "io"
+ "os"
+
+ "github.com/pelletier/go-toml"
+)
+
+func main() {
+ flag.Usage = func() {
+ fmt.Fprintln(os.Stderr, `tomljson can be used in two ways:
+Writing to STDIN and reading from STDOUT:
+ cat file.toml | tomljson > file.json
+
+Reading from a file name:
+ tomljson file.toml
+`)
+ }
+ flag.Parse()
+ os.Exit(processMain(flag.Args(), os.Stdin, os.Stdout, os.Stderr))
+}
+
+func processMain(files []string, defaultInput io.Reader, output io.Writer, errorOutput io.Writer) int {
+ // read from stdin and print to stdout
+ inputReader := defaultInput
+
+ if len(files) > 0 {
+ var err error
+ inputReader, err = os.Open(files[0])
+ if err != nil {
+ printError(err, errorOutput)
+ return -1
+ }
+ }
+ s, err := reader(inputReader)
+ if err != nil {
+ printError(err, errorOutput)
+ return -1
+ }
+ io.WriteString(output, s+"\n")
+ return 0
+}
+
+func printError(err error, output io.Writer) {
+ io.WriteString(output, err.Error()+"\n")
+}
+
+func reader(r io.Reader) (string, error) {
+ tree, err := toml.LoadReader(r)
+ if err != nil {
+ return "", err
+ }
+ return mapToJSON(tree)
+}
+
+func mapToJSON(tree *toml.TomlTree) (string, error) {
+ treeMap := tree.ToMap()
+ bytes, err := json.MarshalIndent(treeMap, "", " ")
+ if err != nil {
+ return "", err
+ }
+ return string(bytes[:]), nil
+}
diff --git a/vendor/github.com/pelletier/go-toml/cmd/tomljson/main_test.go b/vendor/github.com/pelletier/go-toml/cmd/tomljson/main_test.go
new file mode 100644
index 000000000..0b4bdbb11
--- /dev/null
+++ b/vendor/github.com/pelletier/go-toml/cmd/tomljson/main_test.go
@@ -0,0 +1,82 @@
+package main
+
+import (
+ "bytes"
+ "io/ioutil"
+ "os"
+ "strings"
+ "testing"
+)
+
+func expectBufferEquality(t *testing.T, name string, buffer *bytes.Buffer, expected string) {
+ output := buffer.String()
+ if output != expected {
+ t.Errorf("incorrect %s:\n%s\n\nexpected %s:\n%s", name, output, name, expected)
+ t.Log([]rune(output))
+ t.Log([]rune(expected))
+ }
+}
+
+func expectProcessMainResults(t *testing.T, input string, args []string, exitCode int, expectedOutput string, expectedError string) {
+ inputReader := strings.NewReader(input)
+ outputBuffer := new(bytes.Buffer)
+ errorBuffer := new(bytes.Buffer)
+
+ returnCode := processMain(args, inputReader, outputBuffer, errorBuffer)
+
+ expectBufferEquality(t, "output", outputBuffer, expectedOutput)
+ expectBufferEquality(t, "error", errorBuffer, expectedError)
+
+ if returnCode != exitCode {
+ t.Error("incorrect return code:", returnCode, "expected", exitCode)
+ }
+}
+
+func TestProcessMainReadFromStdin(t *testing.T) {
+ input := `
+ [mytoml]
+ a = 42`
+ expectedOutput := `{
+ "mytoml": {
+ "a": 42
+ }
+}
+`
+ expectedError := ``
+ expectedExitCode := 0
+
+ expectProcessMainResults(t, input, []string{}, expectedExitCode, expectedOutput, expectedError)
+}
+
+func TestProcessMainReadFromFile(t *testing.T) {
+ input := `
+ [mytoml]
+ a = 42`
+
+ tmpfile, err := ioutil.TempFile("", "example.toml")
+ if err != nil {
+ t.Fatal(err)
+ }
+ if _, err := tmpfile.Write([]byte(input)); err != nil {
+ t.Fatal(err)
+ }
+
+ defer os.Remove(tmpfile.Name())
+
+ expectedOutput := `{
+ "mytoml": {
+ "a": 42
+ }
+}
+`
+ expectedError := ``
+ expectedExitCode := 0
+
+ expectProcessMainResults(t, ``, []string{tmpfile.Name()}, expectedExitCode, expectedOutput, expectedError)
+}
+
+func TestProcessMainReadFromMissingFile(t *testing.T) {
+ expectedError := `open /this/file/does/not/exist: no such file or directory
+`
+ expectProcessMainResults(t, ``, []string{"/this/file/does/not/exist"}, -1, ``, expectedError)
+}