mirror of
https://github.com/wader/fq.git
synced 2024-12-24 13:52:02 +03:00
9b81d4d3ab
Preparation to make decoder use less memory and API more type safe. Now each scalar type has it's own struct type so it can store different things and enables to have a scalar interface. Also own types will enable experimenting with decode DLS designs like using chained methods that are type aware.
70 lines
1.4 KiB
Go
70 lines
1.4 KiB
Go
package toml
|
|
|
|
import (
|
|
"bytes"
|
|
"embed"
|
|
|
|
"github.com/BurntSushi/toml"
|
|
"github.com/wader/fq/format"
|
|
"github.com/wader/fq/internal/gojqex"
|
|
"github.com/wader/fq/pkg/bitio"
|
|
"github.com/wader/fq/pkg/decode"
|
|
"github.com/wader/fq/pkg/interp"
|
|
"github.com/wader/fq/pkg/scalar"
|
|
)
|
|
|
|
//go:embed toml.jq
|
|
var tomlFS embed.FS
|
|
|
|
func init() {
|
|
interp.RegisterFormat(decode.Format{
|
|
Name: format.TOML,
|
|
Description: "Tom's Obvious, Minimal Language",
|
|
ProbeOrder: format.ProbeOrderTextFuzzy,
|
|
Groups: []string{format.PROBE},
|
|
DecodeFn: decodeTOML,
|
|
Functions: []string{"_todisplay"},
|
|
})
|
|
interp.RegisterFS(tomlFS)
|
|
interp.RegisterFunc0("totoml", toTOML)
|
|
}
|
|
|
|
func decodeTOML(d *decode.D, _ any) any {
|
|
br := d.RawLen(d.Len())
|
|
var r any
|
|
|
|
if _, err := toml.NewDecoder(bitio.NewIOReader(br)).Decode(&r); err != nil {
|
|
d.Fatalf("%s", err)
|
|
}
|
|
var s scalar.Any
|
|
s.Actual = gojqex.Normalize(r)
|
|
|
|
// TODO: better way to handle that an empty file is valid toml and parsed as an object
|
|
switch v := s.Actual.(type) {
|
|
case map[string]any:
|
|
if len(v) == 0 {
|
|
d.Fatalf("root object has no values")
|
|
}
|
|
case []any:
|
|
default:
|
|
d.Fatalf("root not object or array")
|
|
}
|
|
|
|
d.Value.V = &s
|
|
d.Value.Range.Len = d.Len()
|
|
|
|
return nil
|
|
}
|
|
|
|
func toTOML(_ *interp.Interp, c any) any {
|
|
if c == nil {
|
|
return gojqex.FuncTypeError{Name: "totoml", V: c}
|
|
}
|
|
|
|
b := &bytes.Buffer{}
|
|
if err := toml.NewEncoder(b).Encode(gojqex.Normalize(c)); err != nil {
|
|
return err
|
|
}
|
|
return b.String()
|
|
}
|