package dlog

import (
	"context"
	"log/slog"
)

type contextKey int

var ctxKey contextKey = 1

// FromContext retrieves the slog.Logger from the context.
func FromContext(ctx context.Context) *slog.Logger {
	logger, ok := ctx.Value(ctxKey).(*slog.Logger)
	if !ok {
		return slog.Default()
	}
	return logger
}

// WithContext ...
func WithContext(ctx context.Context, logger *slog.Logger) context.Context {
	return context.WithValue(ctx, ctxKey, logger)
}