<reponame>j19sch/pytest-instrument
import logging
import uuid
from datetime import datetime

import pytest

from pytest_instrument.logging_helpers import (
    setup_log_file_handler,
    SessionIdFilter,
    NodeIdFilter,
)


def pytest_addoption(parser):
    parser.addoption(
        "--instrument",
        action="store",
        default=None,
        type=lambda s: [item for item in s.split(",")],
        help="enable pytest-instrument output; comma-separated list; expected valued: json, log",
    )


def pytest_configure(config):
    config.addinivalue_line("markers", "instrument: pytest-instrument mark")


def pytest_sessionstart(session):
    session_id = str(uuid.uuid4())

    session_id_filter = SessionIdFilter(session_id)
    node_id_filter = NodeIdFilter(node_id=None)

    log_handlers = []
    if session.config.getoption("instrument") is not None:
        current_timestamp = datetime.now().strftime("%Y%m%dT%H%M%S")
        if "json" in session.config.getoption("instrument"):
            base_filename = f"{current_timestamp}_{session_id[:8]}"
            log_handler_json = setup_log_file_handler(base_filename, "json")
            log_handler_json.addFilter(session_id_filter)
            log_handler_json.addFilter(node_id_filter)
            log_handlers.append(log_handler_json)
        if "log" in session.config.getoption("instrument"):
            base_filename = f"{current_timestamp}_{session_id[:8]}"
            log_handler_plain = setup_log_file_handler(base_filename, "log")
            log_handlers.append(log_handler_plain)

            # add record with session id to plain log file
            record = {
                "name": "instr.report",
                "node_id": "",
                "levelname": logging.getLevelName(logging.INFO),
                "levelno": logging.INFO,
                "msg": f"session id: {session_id}",
            }

            log_record = logging.makeLogRecord(record)
            log_handler_plain.emit(log_record)
    else:
        log_handlers.append(logging.NullHandler())

    logger = logging.getLogger("instr.log")
    logger.setLevel("DEBUG")
    for handler in log_handlers:
        logger.addHandler(handler)

    session.config.instrument = {
        "session_id": session_id,
        "logger": logger,
        "logfile_handler": log_handlers,
        "node_id_filter": node_id_filter,
    }


def pytest_sessionfinish(session, exitstatus):
    for handler in session.config.instrument["logfile_handler"]:
        handler.close()
        session.config.instrument["logger"].removeHandler(handler)


def pytest_addhooks(pluginmanager):
    from pytest_instrument import hooks

    pluginmanager.add_hookspecs(hooks)


def pytest_runtest_setup(item):
    try:
        labels = list([_.args for _ in item.iter_markers("instrument")][0])
    except IndexError:
        labels = []

    item.config.hook.pytest_instrument_labels(config=item.config, labels=labels)

    try:
        tags = [_.kwargs for _ in item.iter_markers("instrument")][0]
    except IndexError:
        tags = {}

    item.config.hook.pytest_instrument_tags(config=item.config, tags=tags)

    labels = None if not labels else labels
    tags = None if not tags else tags
    labels_and_tags = {"labels": labels, "tags": tags}
    item.user_properties.append(("instrument", labels_and_tags))

    fixtures = item.fixturenames.copy()
    item.config.hook.pytest_instrument_fixtures(fixtures=fixtures)
    fixtures = None if not fixtures else fixtures
    item.user_properties.append(("fixtures", fixtures))

    item.config.instrument["node_id_filter"].node_id = item._nodeid


@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(item, call):
    item.user_properties.append((call.when, {"start": call.start, "stop": call.stop}))

    outcome = yield

    report = outcome.get_result()
    _log_report(report, item.config)


def _log_report(report, config):
    if config.getoption("instrument") is not None:
        labels_and_tags = {}
        for prop in (
            prop for prop in report.user_properties if prop[0] == "instrument"
        ):
            labels_and_tags = prop[1]

        timestamps = {}
        for prop in (prop for prop in report.user_properties if prop[0] == report.when):
            timestamps = prop[1]

        fixtures = []
        for prop in (prop for prop in report.user_properties if prop[0] == "fixtures"):
            fixtures = prop[1]

        record = {
            "name": "instr.report",
            # ToDo: set level based on passed, skipped, failed
            "levelname": logging.getLevelName(logging.INFO),
            "levelno": logging.INFO,
            "msg": f"{report.nodeid} {report.when} {report.outcome}",
            "session_id": config.instrument["session_id"],
            "node_id": report.nodeid,
            "when": report.when,
            "outcome": report.outcome,
            "start": str(timestamps["start"]),
            "stop": str(timestamps["stop"]),
            "duration": f"{report.duration:.12f}",
            "labels": labels_and_tags.get("labels", None),
            "tags": labels_and_tags.get("tags", None),
            "fixtures": fixtures,
        }

        log_record = logging.makeLogRecord(record)
        for handler in config.instrument["logfile_handler"]:
            handler.emit(log_record)

    else:
        pass
