Add gron i/o

Author: JFryy

README.md

@@ -49,6 +49,7 @@ Note: these unsupported formats are on a roadmap for inclusion.
 | INI         | ✅ Supported   | ✅ Supported   |
 | HCL         | ✅ Supported   | ✅ Supported   |
 | TF          | ✅ Supported   | ✅ Supported   |
+| GRON        | ✅ Supported   | ✅ Supported   |
 | CSV         | ✅ Supported   | ❌ Not Supported |
 | Protobuf    | ❌ Not Supported | ❌ Not Supported |
 | HTML        | ❌ Not Supported | ❌ Not Supported |

cli/qq.go

@@ -1,15 +1,15 @@
 package cli
 
 import (
-    "fmt"
-    "io"
-    "os"
-    "github.com/spf13/cobra"
-    "github.com/JFryy/qq/codec"
-    "path/filepath"
-    "strings"
-    "github.com/JFryy/qq/internal/tui"
-    "github.com/itchyny/gojq"
+	"fmt"
+	"github.com/JFryy/qq/codec"
+	"github.com/JFryy/qq/internal/tui"
+	"github.com/itchyny/gojq"
+	"github.com/spf13/cobra"
+	"io"
+	"os"
+	"path/filepath"
+	"strings"
 )
 
 func CreateRootCmd() *cobra.Command {
@@ -145,7 +145,6 @@ func handleCommand(args []string, inputtype string, outputtype string, rawInput
 	os.Exit(0)
 }
 
-
 func isTerminal(f *os.File) bool {
 	info, err := f.Stat()
 	if err != nil {
@@ -154,7 +153,6 @@ func isTerminal(f *os.File) bool {
 	return (info.Mode() & os.ModeCharDevice) != 0
 }
 
-
 func isFile(path string) bool {
 	info, err := os.Stat(path)
 	if err != nil {
@@ -174,7 +172,6 @@ func inferFileType(fName string) codec.EncodingType {
 	return codec.JSON
 }
 
-
 func executeQuery(query *gojq.Query, data interface{}, fileType codec.EncodingType, rawOut bool) {
 	iter := query.Run(data)
 	for {

codec/codec.go

@@ -21,10 +21,11 @@ const (
 	CSV
 	XML
 	INI
+	GRON
 )
 
 func (e EncodingType) String() string {
-	return [...]string{"json", "yaml", "yml", "toml", "hcl", "tf", "csv", "xml", "ini"}[e]
+	return [...]string{"json", "yaml", "yml", "toml", "hcl", "tf", "csv", "xml", "ini", "gron"}[e]
 }
 
 type Encoding struct {
@@ -57,6 +58,7 @@ var SupportedFileTypes = []Encoding{
 	{CSV, csvUnmarshal, jsonMarshalIndent},
 	{XML, xmlUnmarshal, xmlMarshal},
 	{INI, iniUnmarshal, iniMarshal},
+	{GRON, gronUnmarshal, gronMarshal},
 }
 
 func Unmarshal(input []byte, inputFileType EncodingType) (interface{}, error) {

codec/codec_test.go

@@ -11,7 +11,6 @@ require (
 	github.com/hashicorp/hcl/v2 v2.21.0
 	github.com/mattn/go-isatty v0.0.20
 	github.com/mitchellh/mapstructure v1.5.0
-	github.com/stretchr/testify v1.9.0
 	github.com/tmccombs/hcl2json v0.6.3
 	github.com/zclconf/go-cty v1.14.4
 	gopkg.in/ini.v1 v1.67.0
@@ -20,18 +19,16 @@ require (
 require (
 	github.com/agext/levenshtein v1.2.3 // indirect
 	github.com/apparentlymart/go-textseg/v15 v15.0.0 // indirect
-	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/dlclark/regexp2 v1.11.0 // indirect
 	github.com/fatih/color v1.17.0 // indirect
 	github.com/google/go-cmp v0.6.0 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
 	github.com/mitchellh/go-wordwrap v1.0.1 // indirect
-	github.com/pmezard/go-difflib v1.0.0 // indirect
+	github.com/stretchr/testify v1.9.0 // indirect
 	golang.org/x/mod v0.18.0 // indirect
 	golang.org/x/sync v0.7.0 // indirect
 	golang.org/x/sys v0.21.0 // indirect
 	golang.org/x/text v0.16.0 // indirect
 	golang.org/x/tools v0.22.0 // indirect
 	golang.org/x/xerrors v0.0.0-20231012003039-104605ab7028 // indirect
-	gopkg.in/yaml.v3 v3.0.1 // indirect
 )

codec/go.mod

@@ -71,7 +71,6 @@ golang.org/x/tools v0.22.0 h1:gqSGLZqv+AI9lIQzniJ0nZDRG5GBPsSi+DRNHWNz6yA=
 golang.org/x/tools v0.22.0/go.mod h1:aCwcsjqvq7Yqt6TNyX7QMU2enbQ/Gt0bo6krSeEri+c=
 golang.org/x/xerrors v0.0.0-20231012003039-104605ab7028 h1:+cNy6SZtPcJQH3LJVLOSmiC7MMxXNOb3PU/VUEz+EhU=
 golang.org/x/xerrors v0.0.0-20231012003039-104605ab7028/go.mod h1:NDW/Ps6MPRej6fsCIbMTohpP40sJ/P/vI1MoTEGwX90=
-gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405 h1:yhCVgyC4o1eVCa2tZl7eS0r+SDo693bJlVdllGtEeKM=
 gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
 gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA=
 gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k=

codec/go.sum

@@ -0,0 +1,105 @@
+package codec
+
+import (
+	"bytes"
+	"fmt"
+	"github.com/goccy/go-json"
+	"reflect"
+	"strconv"
+	"strings"
+)
+
+func gronUnmarshal(data []byte, v interface{}) error {
+	d := make(map[string]interface{})
+
+	lines := strings.Split(string(data), "\n")
+	for _, line := range lines {
+		if len(line) == 0 {
+			continue
+		}
+		parts := strings.SplitN(line, " = ", 2)
+		if len(parts) != 2 {
+			return nil
+		}
+
+		key := strings.TrimSpace(parts[0])
+		value := strings.Trim(parts[1], `";`)
+
+		setValueJSON(d, key, value)
+	}
+
+	*v.(*interface{}) = d
+	return nil
+
+}
+
+func gronMarshal(v interface{}) ([]byte, error) {
+	var buf bytes.Buffer
+	traverseJSON("", v, &buf)
+	return buf.Bytes(), nil
+}
+
+func traverseJSON(prefix string, v interface{}, buf *bytes.Buffer) {
+	rv := reflect.ValueOf(v)
+	switch rv.Kind() {
+	case reflect.Map:
+		for _, key := range rv.MapKeys() {
+			strKey := fmt.Sprintf("%v", key)
+			traverseJSON(addPrefix(prefix, strKey), rv.MapIndex(key).Interface(), buf)
+		}
+	case reflect.Slice:
+		for i := 0; i < rv.Len(); i++ {
+			traverseJSON(fmt.Sprintf("%s[%d]", prefix, i), rv.Index(i).Interface(), buf)
+		}
+	default:
+		buf.WriteString(fmt.Sprintf("%s = %s;\n", prefix, formatJSONValue(v)))
+	}
+}
+
+func addPrefix(prefix, name string) string {
+	if prefix == "" {
+		return name
+	}
+	return prefix + "." + name
+}
+
+func formatJSONValue(v interface{}) string {
+	switch val := v.(type) {
+	case string:
+		return fmt.Sprintf("%q", val)
+	case bool:
+		return strconv.FormatBool(val)
+	case float64:
+		return strconv.FormatFloat(val, 'f', -1, 64)
+	default:
+		if v == nil {
+			return "null"
+		}
+		data, _ := json.Marshal(v)
+		return string(data)
+	}
+}
+
+func setValueJSON(data map[string]interface{}, key, value string) {
+	parts := strings.Split(key, ".")
+	var m = data
+	for i, part := range parts {
+		if i == len(parts)-1 {
+			if strings.Contains(part, "[") && strings.Contains(part, "]") {
+				k := strings.Split(part, "[")[0]
+				if _, ok := m[k]; !ok {
+					m[k] = make([]interface{}, 0)
+				}
+
+				m[k] = append(m[k].([]interface{}), value)
+			} else {
+				m[part] = value
+			}
+		} else {
+			if _, ok := m[part]; !ok {
+				m[part] = make(map[string]interface{})
+			}
+			m = m[part].(map[string]interface{})
+		}
+	}
+}

codec/gron.go

@@ -7,6 +7,8 @@ require (
 	github.com/JFryy/qq/codec v0.0.0-20240626034628-3e21f7f0cea6
 )
 
+//replace github.com/JFryy/qq/codec => ./codec
+
 require (
 	github.com/BurntSushi/toml v1.4.0 // indirect
 	github.com/JFryy/qq/internal/tui v0.0.0-20240626033059-fa396add9f28 // indirect

go.mod

@@ -0,0 +1,8 @@
+example.name = "John";
+example.age = 30;
+example.address.city = "New York";
+example.address.zip_code = "10001";
+example.skills[0] = "Go";
+example.skills[1] = "JavaScript";
+example.contacts.email = "john@example.com";
+example.contacts.phone = "+1234567890";

tests/test.gron

@@ -60,9 +60,9 @@ done
 previous_ext="json"
 for file in ${extensions}; do
     print "" "============================================"
-    print "" "Executing: cat tests/test.xml | jq . | bin/qq -o $previous_ext"
+    print "" "Executing: cat $file | jq . | bin/qq -o $previous_ext"
     print "" "============================================"
-    bin/qq tests/test.xml | jq . | bin/qq -o $previous_ext
+    bin/qq $file | jq . | bin/qq -o $previous_ext
     print "green" "============================================"
     print "green" "Success."
     print "green" "============================================"