package main

import (
	"log"
	"os"
)

// logMessage 记录日志到控制台或文件
func Message(message string, optionalParams ...string) {
	// 设置默认值
	output := "console"            // 默认输出到控制台
	prefix := "[LOG] "             // 默认日志前缀
	flags := log.Ldate | log.Ltime // 默认日志格式

	// 根据传入参数调整
	if len(optionalParams) > 0 && optionalParams[0] != "" {
		output = optionalParams[0] // 日志目标（文件路径或 console）
	}
	if len(optionalParams) > 1 && optionalParams[1] != "" {
		prefix = optionalParams[1] // 日志前缀
	}

	var logger *log.Logger

	// 根据输出目标选择日志记录器
	if output == "console" {
		logger = log.New(os.Stdout, prefix, flags)
	} else {
		// 输出到文件
		file, err := os.OpenFile(output, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
		if err != nil {
			// 文件打开失败时将日志输出到控制台
			log.Printf("[ERROR] 无法打开日志文件 %s: %v\n", output, err)
			logger = log.New(os.Stdout, prefix, flags)
		} else {
			defer file.Close()
			logger = log.New(file, prefix, flags)
		}
	}

	// 输出日志
	logger.Println(message)
}

//func main() {
//	// 示例 1：极简调用，默认输出到控制台
//	Message("This is a console log")
//
//	// 示例 2：输出到指定文件，使用默认前缀
//	Message("This is a file log", "app.log")
//
//	// 示例 3：输出到文件，指定前缀
//	Message("This is another file log", "app.log", "[CUSTOM] ")
//}