package logger

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

// 日志级别
const (
	DEBUG = iota
	INFO
	WARN
	ERROR
	FATAL
)

// 日志级别名称
var levelNames = []string{
	"调试",
	"信息",
	"警告",
	"错误",
	"致命",
}

// 日志级别颜色
var levelColors = []string{
	"\033[36m", // 青色 - 调试
	"\033[32m", // 绿色 - 信息
	"\033[33m", // 黄色 - 警告
	"\033[31m", // 红色 - 错误
	"\033[35m", // 紫色 - 致命
}

// 重置颜色
const resetColor = "\033[0m"

// Logger 日志记录器
type Logger struct {
	level      int
	logger     *log.Logger
	useColors  bool
	timeFormat string
}

// 全局日志记录器
var defaultLogger *Logger

// 初始化默认日志记录器
func init() {
	defaultLogger = NewLogger(INFO, true)
}

// NewLogger 创建新的日志记录器
func NewLogger(level int, useColors bool) *Logger {
	return &Logger{
		level:      level,
		logger:     log.New(os.Stdout, "", 0),
		useColors:  useColors,
		timeFormat: "2006-01-02 15:04:05.000",
	}
}

// SetLevel 设置日志级别
func SetLevel(level int) {
	defaultLogger.level = level
}

// SetUseColors 设置是否使用颜色
func SetUseColors(useColors bool) {
	defaultLogger.useColors = useColors
}

// formatLog 格式化日志消息
func (l *Logger) formatLog(level int, format string, args ...interface{}) string {
	// 格式化消息
	var msg string
	if len(args) > 0 {
		msg = fmt.Sprintf(format, args...)
	} else {
		msg = format
	}

	// 获取当前时间
	now := time.Now().Format(l.timeFormat)

	// 获取级别名称
	levelName := levelNames[level]

	// 构建日志前缀
	prefix := fmt.Sprintf("[%s] [%s] ", now, levelName)

	// 添加颜色
	if l.useColors {
		prefix = levelColors[level] + prefix
		msg = msg + resetColor
	}

	// 处理多行消息，确保每行都有正确的前缀
	lines := strings.Split(msg, "\n")
	for i := 1; i < len(lines); i++ {
		if len(lines[i]) > 0 {
			padding := strings.Repeat(" ", len(now)+3) // +3 for "[] "
			lines[i] = fmt.Sprintf("%s%s", padding, lines[i])
		}
	}

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

// log 记录日志
func (l *Logger) log(level int, format string, args ...interface{}) {
	if level >= l.level {
		l.logger.Println(l.formatLog(level, format, args...))
	}
}

// Debug 调试日志
func (l *Logger) Debug(format string, args ...interface{}) {
	l.log(DEBUG, format, args...)
}

// Info 信息日志
func (l *Logger) Info(format string, args ...interface{}) {
	l.log(INFO, format, args...)
}

// Warn 警告日志
func (l *Logger) Warn(format string, args ...interface{}) {
	l.log(WARN, format, args...)
}

// Error 错误日志
func (l *Logger) Error(format string, args ...interface{}) {
	l.log(ERROR, format, args...)
}

// Fatal 致命日志
func (l *Logger) Fatal(format string, args ...interface{}) {
	l.log(FATAL, format, args...)
	os.Exit(1)
}

// Debug 调试日志
func Debug(format string, args ...interface{}) {
	defaultLogger.Debug(format, args...)
}

// Info 信息日志
func Info(format string, args ...interface{}) {
	defaultLogger.Info(format, args...)
}

// Warn 警告日志
func Warn(format string, args ...interface{}) {
	defaultLogger.Warn(format, args...)
}

// Error 错误日志
func Error(format string, args ...interface{}) {
	defaultLogger.Error(format, args...)
}

// Fatal 致命日志
func Fatal(format string, args ...interface{}) {
	defaultLogger.Fatal(format, args...)
}

// 进度条相关
const (
	progressBarWidth = 50
)

// PrintProgressBar 打印进度条
func PrintProgressBar(current, total int, prefix string) {
	percentage := float64(current) / float64(total)
	filled := int(progressBarWidth * percentage)
	empty := progressBarWidth - filled

	// 构建进度条
	bar := fmt.Sprintf("\r%s [%s%s] %.1f%% (%d/%d)",
		prefix,
		strings.Repeat("█", filled),
		strings.Repeat("░", empty),
		percentage*100,
		current,
		total,
	)

	// 打印进度条（不换行）
	fmt.Print(bar)

	// 如果完成，打印换行
	if current == total {
		fmt.Println()
	}
}