package logger

import (
	"bytes"
	"io"
	"os"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewLogger(t *testing.T) {
	// Test creating a logger with stdout
	logger := New("info", "text", "")
	assert.NotNil(t, logger)
	assert.NotNil(t, logger.logger)
	assert.Equal(t, "info", logger.level)
}

func TestNewLoggerWithFile(t *testing.T) {
	// Create a temporary file for testing
	tmpFile, err := os.CreateTemp("", "logger_test_*.log")
	assert.NoError(t, err)
	defer os.Remove(tmpFile.Name())
	err = tmpFile.Close()
	assert.NoError(t, err)

	// Test creating a logger with a file
	logger := New("info", "text", tmpFile.Name())
	assert.NotNil(t, logger)
	assert.NotNil(t, logger.logger)
	assert.Equal(t, "info", logger.level)
}

func TestDebug(t *testing.T) {
	// Capture stdout
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	// Create logger with debug level
	logger := New("debug", "text", "")

	// Log a debug message
	logger.Debug("test debug message")

	// Restore stdout
	w.Close()
	os.Stdout = old

	// Read the captured output
	var buf bytes.Buffer
	io.Copy(&buf, r)
	output := buf.String()

	// Verify the output contains the debug message
	assert.Contains(t, output, "DEBUG:")
	assert.Contains(t, output, "test debug message")
}

func TestDebugWithInfoLevel(t *testing.T) {
	// Capture stdout
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	// Create logger with info level (should not log debug messages)
	logger := New("info", "text", "")

	// Log a debug message
	logger.Debug("test debug message")

	// Restore stdout
	w.Close()
	os.Stdout = old

	// Read the captured output
	var buf bytes.Buffer
	io.Copy(&buf, r)
	output := buf.String()

	// Verify the output does not contain the debug message
	assert.NotContains(t, output, "DEBUG:")
	assert.NotContains(t, output, "test debug message")
}

func TestInfo(t *testing.T) {
	// Capture stdout
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	// Create logger with info level
	logger := New("info", "text", "")

	// Log an info message
	logger.Info("test info message")

	// Restore stdout
	w.Close()
	os.Stdout = old

	// Read the captured output
	var buf bytes.Buffer
	io.Copy(&buf, r)
	output := buf.String()

	// Verify the output contains the info message
	assert.Contains(t, output, "INFO:")
	assert.Contains(t, output, "test info message")
}

func TestInfoWithDebugLevel(t *testing.T) {
	// Capture stdout
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	// Create logger with debug level
	logger := New("debug", "text", "")

	// Log an info message
	logger.Info("test info message")

	// Restore stdout
	w.Close()
	os.Stdout = old

	// Read the captured output
	var buf bytes.Buffer
	io.Copy(&buf, r)
	output := buf.String()

	// Verify the output contains the info message
	assert.Contains(t, output, "INFO:")
	assert.Contains(t, output, "test info message")
}

func TestWarn(t *testing.T) {
	// Capture stdout
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	// Create logger with warn level
	logger := New("warn", "text", "")

	// Log a warning message
	logger.Warn("test warn message")

	// Restore stdout
	w.Close()
	os.Stdout = old

	// Read the captured output
	var buf bytes.Buffer
	io.Copy(&buf, r)
	output := buf.String()

	// Verify the output contains the warning message
	assert.Contains(t, output, "WARN:")
	assert.Contains(t, output, "test warn message")
}

func TestWarnWithInfoLevel(t *testing.T) {
	// Capture stdout
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	// Create logger with info level (should log warnings)
	logger := New("info", "text", "")

	// Log a warning message
	logger.Warn("test warn message")

	// Restore stdout
	w.Close()
	os.Stdout = old

	// Read the captured output
	var buf bytes.Buffer
	io.Copy(&buf, r)
	output := buf.String()

	// Verify the output contains the warning message
	assert.Contains(t, output, "WARN:")
	assert.Contains(t, output, "test warn message")
}

func TestError(t *testing.T) {
	// Capture stdout
	old := os.Stdout
	r, w, _ := os.Pipe()
	os.Stdout = w

	// Create logger
	logger := New("error", "text", "")

	// Log an error message
	logger.Error("test error message")

	// Restore stdout
	w.Close()
	os.Stdout = old

	// Read the captured output
	var buf bytes.Buffer
	io.Copy(&buf, r)
	output := buf.String()

	// Verify the output contains the error message
	assert.Contains(t, output, "ERROR:")
	assert.Contains(t, output, "test error message")
}

func TestErrorWithAllLevels(t *testing.T) {
	// Test that error messages are logged at all levels
	levels := []string{"debug", "info", "warn", "error"}
	for _, level := range levels {
		// Capture stdout
		old := os.Stdout
		r, w, _ := os.Pipe()
		os.Stdout = w

		// Create logger with current level
		logger := New(level, "text", "")

		// Log an error message
		logger.Error("test error message")

		// Restore stdout
		w.Close()
		os.Stdout = old

		// Read the captured output
		var buf bytes.Buffer
		io.Copy(&buf, r)
		output := buf.String()

		// Verify the output contains the error message
		assert.Contains(t, output, "ERROR:")
		assert.Contains(t, output, "test error message")
	}
}