import logging
import time
from typing import Any, Dict, Optional

from mcs_node_control.models.node_config import NodeConfig
from tracing.tracer import TracerBackend, TraceSpan
from tracing.utils import swallow_exceptions

logger = logging.getLogger("tracer")
json_logger = logging.getLogger("json_trace")


class TraceparentBackend(TracerBackend):
    """Default backend that logs span lifecycle and mirrors events/status."""
    def __init__(self):
        my_addresses = list(NodeConfig().get_network_addresses())
        logger.info("My addresses: %s", my_addresses)
        json_logger.info("my_addresses", extra={"my_addresses": my_addresses})

    @swallow_exceptions
    def on_span_start(self, span: TraceSpan) -> None:
        logger.info(
            "span_begin name='%s' kind=%s tid=%s sid=%s%s",
            span.name, span.kind, span.trace_id, span.span_id,
            f' psid={span.parent_span_id}' if span.parent_span_id else '',
        )
        json_logger.info("span_begin", extra=span.to_flat_dict())

    @swallow_exceptions
    def on_span_end(self, span: TraceSpan, exc: Optional[BaseException]) -> None:
        duration_ms = (time.time_ns() - span.start_ns) / 1_000_000
        logger.info(
            "span_end name='%s' kind=%s tid=%s sid=%s%s duration_ms=%.3f",
            span.name, span.kind, span.trace_id, span.span_id,
            f' psid={span.parent_span_id}' if span.parent_span_id else '',
            duration_ms,
        )
        json_logger.info("span_end", extra=span.to_flat_dict())

    @swallow_exceptions
    def on_span_event(self, span: TraceSpan, name: str, attrs: Dict[str, Any]) -> None:
        logger.info(
            "span_event name='%s' tid=%s sid=%s%s attrs=%s",
            name, span.trace_id, span.span_id,
            f' psid={span.parent_span_id}' if span.parent_span_id else '',
            attrs,
        )
        json_logger.info("span_event", extra=span.to_flat_dict())