OSDN Git Service

Hulk did something
[bytom/vapor.git] / vendor / github.com / pelletier / go-toml / cmd / tomljson / main.go
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 (file)
index 0000000..b2d6fc6
--- /dev/null
@@ -0,0 +1,72 @@
+// Tomljson reads TOML and converts to JSON.
+//
+// Usage:
+//   cat file.toml | tomljson > file.json
+//   tomljson file1.toml > file.json
+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.Tree) (string, error) {
+       treeMap := tree.ToMap()
+       bytes, err := json.MarshalIndent(treeMap, "", "  ")
+       if err != nil {
+               return "", err
+       }
+       return string(bytes[:]), nil
+}