import logging
import logzero
import pytest
import tempfile
import subprocess
import os
import sys
import textwrap
from logzero import logger as log
from slic.utils.logcfg import add_log_Level, logcfg, setup_import_logging

# Set for all the tests
@pytest.fixture(scope="module", autouse=True)
def setup_custom_levels():
    add_log_Level(log, "LONG", logging.INFO + 5, func_name="long")
    add_log_Level(log, "ENLARGE", logging.INFO + 6, func_name="enlarge")
    logcfg("LONG")
    setup_import_logging()

# Test simple custom log

@pytest.mark.parametrize("levelname, logfunc, message", [
    ("LONG", lambda msg: log.long(msg), "This is a LONG message"),
    ("ENLARGE", lambda msg: log.enlarge(msg), "Please ENLARGE this!"),
])

def test_custom_log_outputs(levelname, logfunc, message, capsys):
    # Remove all handlers to avoid duplicate logs
    for h in log.handlers[:]:
        log.removeHandler(h)

    # Create a new handler that logs to stderr
    handler = logging.StreamHandler(sys.stderr)
    formatter = logzero.LogFormatter()
    handler.setFormatter(formatter)
    log.addHandler(handler)

    logcfg(levelname)

    # Call and capture
    logfunc(message)
    captured = capsys.readouterr().err

    print("Captured stderr:", repr(captured))

    assert message in captured, f"Expected message '{message}' not found in log"
    assert levelname in captured, f"Expected level '{levelname}' not present in log"


# Test subprocess: top-level import logs only once

def test_import_logging_once_per_module():
    code = textwrap.dedent("""
        from slic.utils.logcfg import *
        import math
        import io
        import random
    """)
 
    with tempfile.NamedTemporaryFile("w", suffix=".py", delete=False) as tmp:
        tmp.write(code)
        tmp_path = tmp.name

    env = os.environ.copy()
    root_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
    env["PYTHONPATH"] = root_path + os.pathsep + env.get("PYTHONPATH", "")

    result = subprocess.run([sys.executable, tmp_path], capture_output=True, text=True, env=env)
    os.remove(tmp_path)

    assert result.returncode == 0, f"Script failed:\n{result.stderr}"

    stderr = result.stderr
    print(stderr)
    lines = stderr.splitlines()
    for mod in ["math", "io", "random"]:
        count = sum(1 for line in lines if f"importing: {mod}" in line)
        assert count == 1, f"Expected 1 import log for '{mod}', found {count}"