mirror of
https://codeberg.org/scip/yadu.git
synced 2025-12-16 20:21:00 +01:00
fix #12: unquote eventually quoted keys generated by yaml.Marshall()
This commit is contained in:
37
handler.go
37
handler.go
@@ -17,7 +17,7 @@ import (
|
|||||||
"github.com/fatih/color"
|
"github.com/fatih/color"
|
||||||
)
|
)
|
||||||
|
|
||||||
const VERSION = "0.1.2"
|
const VERSION = "0.1.3"
|
||||||
|
|
||||||
// We use RFC datestring by default
|
// We use RFC datestring by default
|
||||||
const DefaultTimeFormat = "2006-01-02T03:04.05 MST"
|
const DefaultTimeFormat = "2006-01-02T03:04.05 MST"
|
||||||
@@ -38,6 +38,35 @@ type Handler struct {
|
|||||||
replaceAttr func(groups []string, a slog.Attr) slog.Attr
|
replaceAttr func(groups []string, a slog.Attr) slog.Attr
|
||||||
addSource bool
|
addSource bool
|
||||||
indenter *regexp.Regexp
|
indenter *regexp.Regexp
|
||||||
|
|
||||||
|
/*
|
||||||
|
This is being used in Postprocess() to fix
|
||||||
|
https://github.com/go-yaml/yaml/issues/1020 and
|
||||||
|
https://github.com/TLINDEN/yadu/issues/12 respectively.
|
||||||
|
|
||||||
|
yaml.v3 follows the YAML standard and quotes all keys and values
|
||||||
|
matching this regex (see https://yaml.org/type/bool.html):
|
||||||
|
`y|Y|yes|Yes|YES|n|N|no|No|NO|true|True|TRUE|false|False|FALSE|on|On|ON|off|Off|OFF`
|
||||||
|
|
||||||
|
The problem is, that the YAML "standard" does not state wether
|
||||||
|
this applies to values or keys or values&keys and most
|
||||||
|
implementors, as gopkg.in/yaml.v3, do it just for keys and values.
|
||||||
|
|
||||||
|
Therefore if we dump a struct containing a key "Y" it ends up
|
||||||
|
being quoted, while any other keys remain unquoted, which looks
|
||||||
|
pretty ugly, makes evaluating the output harder, especially in
|
||||||
|
game development where you have to dump coordinates, points etc,
|
||||||
|
all containing X,Y with X unquoted and Y quoted.
|
||||||
|
|
||||||
|
To fix this utter nonsence, I just replace all quotes in all
|
||||||
|
keys. Period. This is just a logging module, nobody will and can
|
||||||
|
use its output to postprocess it with some yaml parser, because
|
||||||
|
we not only dump the structs as yaml, we also write a one liner
|
||||||
|
in front of it with the timestamp and the message. So, we don't
|
||||||
|
output valid YAML anyway and we don't give a shit about
|
||||||
|
compliance because of this. AND because this rule is bullshit.
|
||||||
|
*/
|
||||||
|
yamlcleaner *regexp.Regexp
|
||||||
}
|
}
|
||||||
|
|
||||||
// Options are options for the Yadu [log/slog.Handler].
|
// Options are options for the Yadu [log/slog.Handler].
|
||||||
@@ -150,7 +179,9 @@ func (h *Handler) getSource(pc uintptr) string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (h *Handler) Postprocess(yamlstr []byte) string {
|
func (h *Handler) Postprocess(yamlstr []byte) string {
|
||||||
return "\n " + strings.TrimSpace(h.indenter.ReplaceAllString(string(yamlstr), " "))
|
tree := string(yamlstr)
|
||||||
|
clean := h.yamlcleaner.ReplaceAllString(tree, "$1$2:")
|
||||||
|
return "\n " + strings.TrimSpace(h.indenter.ReplaceAllString(clean, " "))
|
||||||
}
|
}
|
||||||
|
|
||||||
// NewHandler returns a [log/slog.Handler] using the receiver's options.
|
// NewHandler returns a [log/slog.Handler] using the receiver's options.
|
||||||
@@ -168,6 +199,7 @@ func NewHandler(out io.Writer, opts *Options) *Handler {
|
|||||||
replaceAttr: opts.ReplaceAttr,
|
replaceAttr: opts.ReplaceAttr,
|
||||||
addSource: opts.AddSource,
|
addSource: opts.AddSource,
|
||||||
indenter: regexp.MustCompile(`(?m)^`),
|
indenter: regexp.MustCompile(`(?m)^`),
|
||||||
|
yamlcleaner: regexp.MustCompile("(?m)^( *)\"([^\"]*)\":"),
|
||||||
}
|
}
|
||||||
|
|
||||||
if opts.Level == nil {
|
if opts.Level == nil {
|
||||||
@@ -269,5 +301,6 @@ func (h *Handler) clone() *Handler {
|
|||||||
replaceAttr: h.replaceAttr,
|
replaceAttr: h.replaceAttr,
|
||||||
addSource: h.addSource,
|
addSource: h.addSource,
|
||||||
indenter: h.indenter,
|
indenter: h.indenter,
|
||||||
|
yamlcleaner: h.yamlcleaner,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -34,6 +34,10 @@ type Enemy struct {
|
|||||||
Ammo []Ammo
|
Ammo []Ammo
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Point struct {
|
||||||
|
y, Y, yes, n, N, no, True, False, on, off int
|
||||||
|
}
|
||||||
|
|
||||||
type Tests struct {
|
type Tests struct {
|
||||||
name string
|
name string
|
||||||
want string
|
want string
|
||||||
@@ -166,6 +170,9 @@ func GetAmmo() *Ammo {
|
|||||||
return &Ammo{Forweapon: "Axe", Range: 50, Impact: 1, Cost: 50}
|
return &Ammo{Forweapon: "Axe", Range: 50, Impact: 1, Cost: 50}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetPoint() *Point {
|
||||||
|
return &Point{}
|
||||||
|
}
|
||||||
func removeTime(_ []string, a slog.Attr) slog.Attr {
|
func removeTime(_ []string, a slog.Attr) slog.Attr {
|
||||||
if a.Key == slog.TimeKey {
|
if a.Key == slog.TimeKey {
|
||||||
return slog.Attr{}
|
return slog.Attr{}
|
||||||
@@ -173,7 +180,7 @@ func removeTime(_ []string, a slog.Attr) slog.Attr {
|
|||||||
return a
|
return a
|
||||||
}
|
}
|
||||||
|
|
||||||
func Test(t *testing.T) {
|
func TestLogger(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
for _, tt := range tests {
|
for _, tt := range tests {
|
||||||
@@ -211,3 +218,22 @@ func Test(t *testing.T) {
|
|||||||
buf.Reset()
|
buf.Reset()
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func TestYamlCleaner(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
var buf bytes.Buffer
|
||||||
|
logger := slog.New(yadu.NewHandler(&buf, &yadu.Options{}))
|
||||||
|
slog.SetDefault(logger)
|
||||||
|
|
||||||
|
logger.Info("got a point", "point", GetPoint())
|
||||||
|
|
||||||
|
got := buf.String()
|
||||||
|
|
||||||
|
bools := []string{"y:", "n:", "true:", "false:"}
|
||||||
|
for _, want := range bools {
|
||||||
|
if !strings.Contains(got, want) {
|
||||||
|
t.Errorf("test TestYamlCleaner failed.\n want: %s:\n got: %s\n", want, got)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user