package logger

import (
	"net/http"
)

// Option func type
type Option func(l *Middleware)

// WithBody triggers request body logging. Body size is limited (default 1k)
func WithBody(l *Middleware) {
	l.logBody = true
}

// MaxBodySize sets size of the logged part of the request body.
func MaxBodySize(maximum int) Option {
	return func(l *Middleware) {
		if maximum >= 0 {
			l.maxBodySize = maximum
		}
	}
}

// Prefix sets log line prefix.
func Prefix(prefix string) Option {
	return func(l *Middleware) {
		l.prefix = prefix
	}
}

// IPfn sets IP masking function. If ipFn is nil then IP address will be logged as is.
func IPfn(ipFn func(ip string) string) Option {
	return func(l *Middleware) {
		l.ipFn = ipFn
	}
}

// UserFn triggers user name logging if userFn is not nil.
func UserFn(userFn func(r *http.Request) (string, error)) Option {
	return func(l *Middleware) {
		l.userFn = userFn
	}
}

// SubjFn triggers subject logging if subjFn is not nil.
func SubjFn(subjFn func(r *http.Request) (string, error)) Option {
	return func(l *Middleware) {
		l.subjFn = subjFn
	}
}

// BodyFn sets a transform applied to the request body before it is logged, e.g. to
// mask secrets. It only runs when body logging is enabled (see WithBody) and the
// body is non-empty; if bodyFn is nil the body is logged unchanged. bodyFn receives
// the body (capped at MaxBodySize) and a truncated flag that is true when the body was longer than
// MaxBodySize and got cut short - a masker can use it to emit a marker instead of
// risking a pass-through of a partial body it cannot parse. The returned string is
// what gets logged, so bodyFn owns the content; the logger still collapses it to a
// single line to keep one log record per request.
func BodyFn(bodyFn func(body string, truncated bool) string) Option {
	return func(l *Middleware) {
		l.bodyFn = bodyFn
	}
}

// ApacheCombined sets format to Apache Combined Log.
// See http://httpd.apache.org/docs/2.2/logs.html#combined
func ApacheCombined(l *Middleware) {
	l.apacheCombined = true
}

// Log sets logging backend.
func Log(log Backend) Option {
	return func(l *Middleware) {
		l.log = log
	}
}