array.go

package prettyconsole

import (
	"bytes"
	"time"

	"go.uber.org/zap/zapcore"
)

// Test interface conformance
var _ zapcore.ArrayEncoder = (*prettyConsoleEncoder)(nil)

func (e *prettyConsoleEncoder) AppendComplex64(v complex64)   { e.appendComplex(complex128(v), 32) }
func (e *prettyConsoleEncoder) AppendComplex128(v complex128) { e.appendComplex(v, 64) }
func (e *prettyConsoleEncoder) AppendFloat32(v float32)       { e.appendFloat(float64(v), 32) }
func (e *prettyConsoleEncoder) AppendFloat64(v float64)       { e.appendFloat(v, 64) }
func (e *prettyConsoleEncoder) AppendInt(v int)               { e.AppendInt64(int64(v)) }
func (e *prettyConsoleEncoder) AppendInt32(v int32)           { e.AppendInt64(int64(v)) }
func (e *prettyConsoleEncoder) AppendInt16(v int16)           { e.AppendInt64(int64(v)) }
func (e *prettyConsoleEncoder) AppendInt8(v int8)             { e.AppendInt64(int64(v)) }
func (e *prettyConsoleEncoder) AppendUint(v uint)             { e.AppendUint64(uint64(v)) }
func (e *prettyConsoleEncoder) AppendUint32(v uint32)         { e.AppendUint64(uint64(v)) }
func (e *prettyConsoleEncoder) AppendUint16(v uint16)         { e.AppendUint64(uint64(v)) }
func (e *prettyConsoleEncoder) AppendUint8(v uint8)           { e.AppendUint64(uint64(v)) }
func (e *prettyConsoleEncoder) AppendUintptr(v uintptr)       { e.AppendUint64(uint64(v)) }

func (e *prettyConsoleEncoder) AppendBool(b bool) {
	e.addSeparator()
	e.buf.AppendBool(b)

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) AppendByteString(bytes []byte) {
	e.addSeparator()
	e.appendSafeByte(bytes)

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) appendComplex(c complex128, precision int) {
	e.addSeparator()
	// Cast to a platform-independent, fixed-size type.
	r, i := real(c), imag(c)
	// Because we're always in a quoted string, we can use strconv without
	// special-casing NaN and +/-Inf.
	e.buf.AppendFloat(r, precision)
	// If imaginary part is less than 0, minus (-) sign is added by default
	// by AppendFloat.
	if i >= 0 {
		e.buf.AppendByte('+')
	}
	e.buf.AppendFloat(i, precision)
	e.buf.AppendByte('i')

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) appendFloat(f float64, precision int) {
	e.addSeparator()
	e.buf.AppendFloat(f, precision)

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) AppendInt64(i int64) {
	e.addSeparator()
	e.buf.AppendInt(i)

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) AppendString(s string) {
	e.addSeparator()
	e.addSafeString(s)

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) AppendUint64(u uint64) {
	e.addSeparator()
	e.buf.AppendUint(u)

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) AppendDuration(duration time.Duration) {
	e.addSeparator()
	cur := e.buf.Len()
	e.cfg.EncodeDuration(duration, e)
	if cur == e.buf.Len() {
		// User-supplied EncodeDuration is a no-op. Fall back to nanoseconds to keep
		// JSON valid.
		e.buf.AppendInt(int64(duration))
	}

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) AppendTime(t time.Time) {
	e.addSeparator()
	cur := e.buf.Len()
	e.cfg.EncodeTime(t, e)
	if cur == e.buf.Len() {
		// User-supplied EncodeTime is a no-op. Fall back to RFC3339
		e.buf.AppendTime(t, time.RFC3339)
	}

	e.inList = true
	e.listSep = e._listSepComma
}

func (e *prettyConsoleEncoder) AppendArray(marshaler zapcore.ArrayMarshaler) error {
	e.addSeparator()
	enc := e.clone()
	enc.OpenNamespace("")
	enc.colorizeAtLevel("[")
	enc.inList = false
	l := enc.buf.Len()

	if err := marshaler.MarshalLogArray(enc); err != nil {
		return err
	}
	if bytes.ContainsRune(enc.buf.Bytes()[l:], '\n') {
		enc.buf.AppendString(e.cfg.LineEnding)
		for ii := 0; ii < enc.namespaceIndent-1; ii++ {
			enc.buf.AppendByte(' ')
		}
	}
	enc.colorizeAtLevel("]")

	_, _ = e.buf.Write(enc.buf.Bytes())
	putPrettyConsoleEncoder(enc)

	e.inList = true
	e.listSep = e._listSepComma
	return nil
}

func (e *prettyConsoleEncoder) AppendObject(marshaler zapcore.ObjectMarshaler) error {
	e.addSeparator()
	enc := e.clone()
	enc.OpenNamespace("")
	enc.colorizeAtLevel("{")
	enc.inList = false
	enc.keyPrefix = ""
	l := enc.buf.Len()

	if err := marshaler.MarshalLogObject(enc); err != nil {
		return err
	}
	if bytes.ContainsRune(enc.buf.Bytes()[l:], '\n') {
		enc.buf.AppendString(e.cfg.LineEnding)
		for ii := 0; ii < enc.namespaceIndent-1; ii++ {
			enc.buf.AppendByte(' ')
		}
	}
	enc.colorizeAtLevel("}")

	_, _ = e.buf.Write(enc.buf.Bytes())
	putPrettyConsoleEncoder(enc)

	e.inList = true
	e.listSep = e._listSepComma
	return nil
}

func (e *prettyConsoleEncoder) AppendReflected(value interface{}) error {
	e.addSeparator()
	enc := e.clone()
	enc.OpenNamespace("")
	enc.keyPrefix = ""
	enc.inList = false
	l := enc.buf.Len()
	iw := indentingWriter{
		buf:        enc.buf,
		indent:     enc.namespaceIndent,
		lineEnding: []byte(e.cfg.LineEnding),
	}
	if err := e.cfg.NewReflectedEncoder(iw).Encode(value); err != nil {
		return err
	}
	if l-enc.buf.Len() == 0 {
		// User-supplied reflectedEncoder is a no-op. Fall back to dd
		if err := defaultReflectedEncoder(iw).Encode(value); err != nil {
			return err
		}
	}

	_, _ = e.buf.Write(enc.buf.Bytes())
	putPrettyConsoleEncoder(enc)

	e.inList = true
	e.listSep = e._listSepComma
	return nil
}