package log

import (
	"fmt"
	"reflect"
	"strings"

	"github.com/aws/aws-sdk-go/aws/awsutil"
	"github.com/golang/glog"
)

type Logger struct {
	name string
}

// New creates a new Logger.
// The name appears in the log lines.
func New(name string) *Logger {
	return &Logger{name: name}
}

// Debugf will print debug messages if debug logging is enabled
func (l *Logger) Debugf(format string, args ...interface{}) {
	debugf(format, l.name, 2, args...)
}

// DebugLevelf will print debug messages if debug logging is enabled
func (l *Logger) DebugLevelf(level int, format string, args ...interface{}) {
	debugf(format, l.name, level, args...)
}

// Infof will print info level messages
func (l *Logger) Infof(format string, args ...interface{}) {
	infof(format, l.name, args...)
}

// Warnf will print warning level messages
func (l *Logger) Warnf(format string, args ...interface{}) {
	warnf(format, l.name, args...)
}

// Errorf will print error level messages
func (l *Logger) Errorf(format string, args ...interface{}) {
	errorf(format, l.name, args...)
}

// Fatalf will print error level messages
func (l *Logger) Fatalf(format string, args ...interface{}) {
	fatalf(format, l.name, args...)
}

// Exitf will print error level messages and exit
func (l *Logger) Exitf(format string, args ...interface{}) {
	exitf(format, l.name, args...)
}

// debugf will print debug messages if debug logging is enabled
func debugf(format, ingressName string, level int, args ...interface{}) {
	if glog.V(2) {
		prefix := fmt.Sprintf("%s: ", ingressName)
		for _, line := range strings.Split(fmt.Sprintf(format, args...), "\n") {
			glog.InfoDepth(level, prefix, line)
		}
	}
}

// infof will print info level messages
func infof(format, ingressName string, args ...interface{}) {
	prefix := fmt.Sprintf("%s: ", ingressName)
	for _, line := range strings.Split(fmt.Sprintf(format, args...), "\n") {
		glog.InfoDepth(2, prefix, line)
	}
}

// warnf will print warning level messages
func warnf(format, ingressName string, args ...interface{}) {
	prefix := fmt.Sprintf("%s: ", ingressName)
	for _, line := range strings.Split(fmt.Sprintf(format, args...), "\n") {
		glog.WarningDepth(2, prefix, line)
	}
}

// errorf will print error level messages
func errorf(format, ingressName string, args ...interface{}) {
	prefix := fmt.Sprintf("%s: ", ingressName)
	for _, line := range strings.Split(fmt.Sprintf(format, args...), "\n") {
		glog.ErrorDepth(2, prefix, line)
	}
}

// fatalf will print error level messages
func fatalf(format, ingressName string, args ...interface{}) {
	prefix := fmt.Sprintf("%s: ", ingressName)
	glog.FatalDepth(2, fmt.Sprintf(prefix+format, args...))
}

// Exitf will print error level messages and exit
func exitf(format, ingressName string, args ...interface{}) {
	prefix := fmt.Sprintf("%s: ", ingressName)
	glog.ExitDepth(2, fmt.Sprintf(prefix+format, args...))
}

// Prettify uses awsutil.Prettify to print structs, but also removes '\n' for better logging.
func Prettify(i interface{}) string {
	return strings.Replace(awsutil.Prettify(i), "\n", "", -1)
}

type stringInt interface {
	String() string
}

// String returns the String function of i or empty if its a nil pointer.
func String(i stringInt) string {
	if reflect.ValueOf(i).Kind() == reflect.Ptr && reflect.ValueOf(i).IsNil() {
		return "<nil>"
	}
	return strings.Replace(i.String(), "\n", "", -1)
}