package user

import (
	"golang.org/x/net/context"

	"github.com/weaveworks/common/logging"
)

// LogWith returns user and org information from the context as log fields.
func LogWith(ctx context.Context, log logging.Interface) logging.Interface {
	userID, err := ExtractUserID(ctx)
	if err == nil {
		log = log.WithField("userID", userID)
	}

	orgID, err := ExtractOrgID(ctx)
	if err == nil {
		log = log.WithField("orgID", orgID)
	}

	return log
}