import logging

from neo4j import AsyncDriver

from core.edges import EpisodicEdge, EntityEdge, Edge
from core.nodes import EntityNode, EpisodicNode, Node

logger = logging.getLogger(__name__)


def build_episodic_edges(
    entity_nodes: list[EntityNode], episode: EpisodicNode
) -> list[EpisodicEdge]:
    edges: list[EpisodicEdge] = []

    for node in entity_nodes:
        edges.append(
            EpisodicEdge(
                source_node_uuid=episode,
                target_node_uuid=node,
                created_at=episode.created_at,
            )
        )

    return edges