File: main.go

package info (click to toggle)
golang-github-pelletier-go-toml 1.9.5-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 744 kB
  • sloc: sh: 37; makefile: 26
file content (82 lines) | stat: -rw-r--r-- 1,740 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
// Jsontoml reads JSON and converts to TOML.
//
// Usage:
//   cat file.toml | jsontoml > file.json
//   jsontoml file1.toml > file.json
package main

import (
	"encoding/json"
	"flag"
	"fmt"
	"io"
	"io/ioutil"
	"os"

	"github.com/pelletier/go-toml"
)

func main() {
	flag.Usage = func() {
		fmt.Fprintln(os.Stderr, "jsontoml can be used in two ways:")
		fmt.Fprintln(os.Stderr, "Writing to STDIN and reading from STDOUT:")
		fmt.Fprintln(os.Stderr, "")
		fmt.Fprintln(os.Stderr, "")
		fmt.Fprintln(os.Stderr, "Reading from a file name:")
		fmt.Fprintln(os.Stderr, " 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 {
		file, err := os.Open(files[0])
		if err != nil {
			printError(err, errorOutput)
			return -1
		}
		inputReader = file
		defer file.Close()
	}
	s, err := reader(inputReader)
	if err != nil {
		printError(err, errorOutput)
		return -1
	}
	io.WriteString(output, s)
	return 0
}

func printError(err error, output io.Writer) {
	io.WriteString(output, err.Error()+"\n")
}

func reader(r io.Reader) (string, error) {
	jsonMap := make(map[string]interface{})
	jsonBytes, err := ioutil.ReadAll(r)
	if err != nil {
		return "", err
	}
	err = json.Unmarshal(jsonBytes, &jsonMap)
	if err != nil {
		return "", err
	}

	tree, err := toml.TreeFromMap(jsonMap)
	if err != nil {
		return "", err
	}
	return mapToTOML(tree)
}

func mapToTOML(t *toml.Tree) (string, error) {
	tomlBytes, err := t.ToTomlString()
	if err != nil {
		return "", err
	}
	return string(tomlBytes[:]), nil
}