logger.go

package logger

import (
	"context"
	"fmt"
	"io"

	"github.com/mgjules/spoty/config"
	"github.com/uptrace/opentelemetry-go-extra/otelzap"
	"go.uber.org/fx"
	"go.uber.org/zap"
)

// Module exported for initialising a new Logger.
var Module = fx.Options(
	fx.Provide(New),
)

// Logger is a simple wrapper around zap.SugaredLogger.
type Logger struct {
	*otelzap.SugaredLogger
}

// New creates a new Logger.
func New(lc fx.Lifecycle, cfg *config.Config) (*Logger, error) {
	var (
		logger *zap.Logger
		err    error
	)

	if cfg.Prod {
		logger, err = zap.NewProduction()
	} else {
		logger, err = zap.NewDevelopment()
	}
	if err != nil {
		return nil, fmt.Errorf("failed to create logger: %w", err)
	}

	lc.Append(fx.Hook{
		OnStop: func(_ context.Context) error {
			logger.Sync() //nolint:errcheck,gosec // see:https://github.com/uber-go/zap/issues/328

			return nil
		},
	})

	otellogger := otelzap.New(logger)

	return &Logger{otellogger.Sugar()}, nil
}

// Writer returns the logger's io.Writer.
func (l *Logger) Writer() io.Writer {
	return zap.NewStdLog(l.Desugar().Logger).Writer()
}