package simplelog

import (
	"context"
	"encoding/json"
	"fmt"
	"log/slog"
	"os"
)

// JSONHandler writes each log record to stdout as a JSON document.
type JSONHandler struct {
	attrs handlerAttrs
}

// NewJSONHandler returns a new JSONHandler.
func NewJSONHandler() *JSONHandler {
	return &JSONHandler{}
}

// Handle marshals the record, with its attributes, to one JSON object and
// writes it to stdout.
func (j *JSONHandler) Handle(_ context.Context, record slog.Record) error {
	jsonData, err := json.Marshal(recordToMap(record, j.attrs))
	if err != nil {
		return err
	}

	_, _ = fmt.Fprintln(os.Stdout, string(jsonData))

	return nil
}

// Enabled reports whether the handler processes records at the given
// level; it always returns true.
func (j *JSONHandler) Enabled(_ context.Context, _ slog.Level) bool {
	return true
}

// WithAttrs returns a new handler that also emits attrs, qualified by the
// groups open now. The receiver is not modified.
func (j *JSONHandler) WithAttrs(attrs []slog.Attr) slog.Handler {
	if len(attrs) == 0 {
		return j
	}

	return &JSONHandler{attrs: j.attrs.withAttrs(attrs)}
}

// WithGroup returns a new handler that nests later attributes in an
// object named after the group. An empty name returns the receiver.
func (j *JSONHandler) WithGroup(name string) slog.Handler {
	if name == "" {
		return j
	}

	return &JSONHandler{attrs: j.attrs.withGroup(name)}
}