package log

import (
	"context"
	"os"

	"github.com/go-logr/logr"

	"edge-infra.dev/pkg/lib/fog"
)

// Logger creates a new fog logger
func Logger() logr.Logger {
	if os.Getenv("GIN_MODE") == "debug" {
		return fog.New(fog.To(os.Stdout), fog.WithLevel(fog.DEBUG))
	}
	return fog.New()
}

// Get returns a fog logger that was set on the context (or instantiates one if not exists)
func Get(ctx context.Context) logr.Logger {
	log := fog.FromContext(ctx)

	// add the operation id if exists
	opID := fog.OperationID(ctx)
	if opID != "" {
		log = log.WithValues(fog.OperationFields(opID)...)
	}

	return log
}