package sq

import (
	"fmt"
	"log"
	"os"
	"regexp"
	"strings"
	"time"
)

const (
	fmtLogQuery     = `Sql: %s`
	fmtLogArgs      = `Args:  %#v`
	fmtLogError     = `Error: %v`
	fmtLogTimeTaken = `Time:  %0.5fs`
)

var (
	reInvisibleChars = regexp.MustCompile(`[\s\r\n\t]+`)
)

// QueryStatus 表示查询执行后的状态。
type QueryStatus struct {
	Sql   string
	Args  interface{}
	Start time.Time
	End   time.Time
	Err   error
}

// String 返回格式化的日志消息。
func (q *QueryStatus) String() string {
	lines := make([]string, 0, 8)

	if query := q.Sql; query != "" {
		query = reInvisibleChars.ReplaceAllString(query, ` `)
		query = strings.TrimSpace(query)
		lines = append(lines, fmt.Sprintf(fmtLogQuery, query))
	}

	if args, ok := q.Args.([]interface{}); ok && len(args) == 0 {
		q.Args = nil
	}

	if q.Args != nil {
		lines = append(lines, fmt.Sprintf(fmtLogArgs, q.Args))
	}

	if q.Err != nil {
		lines = append(lines, fmt.Sprintf(fmtLogError, q.Err))
	}

	lines = append(lines, fmt.Sprintf(fmtLogTimeTaken, float64(q.End.UnixNano()-q.Start.UnixNano())/float64(1e9)))

	return strings.Join(lines, "\n")
}

// ILogger 表示一个日志收集器.
// 您可以将日志收集器传递给 sq.SetLogger(myCollector)，以使其在执行查询后收集QueryStatus消息。
type ILogger interface {
	Printf(format string, v ...interface{})
}

type defaultLogger struct {
	logging bool
	log     ILogger
}

func (d *defaultLogger) Log(m *QueryStatus, show bool) {
	if d.logging || show {
		d.log.Printf("\n\t%s\n\n", strings.Replace(m.String(), "\n", "\n\t", -1))
	}
}

func (d *defaultLogger) SetLogging(logging bool) {
	d.logging = logging
}

var logger = &defaultLogger{log: log.New(os.Stderr, "", log.LstdFlags)}

// SetLogger 设置日志器。
func SetLogger(l ILogger) {
	logger.log = l
}

// SetLogging 是否打印日志，<print>为true，则打印日志。
func SetLogging(print bool) {
	logger.logging = print
}