feat(mapping): enrich nodes and edges with service and container_kind

Author: akhundMurad

pacta/mapping/enricher.py

@@ -72,12 +72,23 @@ def _enrich_node(self, node: IRNode, model: ArchitectureModel) -> IRNode:
         if container_id is not None:
             context_id = model.get_context_for_container(container_id)
 
+        # v2: derive service (top-level ancestor) and container_kind
+        service: str | None = None
+        container_kind: str | None = None
+        if container_id is not None:
+            service = container_id.split(".")[0]
+            container = model.get_container(container_id)
+            if container is not None and container.kind is not None:
+                container_kind = container.kind.value
+
         # Return enriched node if anything changed, otherwise return original
         if (
             node.container == container_id
             and node.layer == layer_id
             and node.context == context_id
             and node.tags == tags
+            and node.service == service
+            and node.container_kind == container_kind
         ):
             return node
 
@@ -87,6 +98,8 @@ def _enrich_node(self, node: IRNode, model: ArchitectureModel) -> IRNode:
             layer=layer_id,
             context=context_id,
             tags=tags,
+            service=service,
+            container_kind=container_kind,
         )
 
     def _match_container(self, node: IRNode, model: ArchitectureModel) -> str | None:
@@ -160,10 +173,14 @@ def _enrich_edge(self, edge: IREdge, node_lookup: dict[str, IRNode]) -> IREdge:
         src_container = src_node.container if src_node else None
         src_layer = src_node.layer if src_node else None
         src_context = src_node.context if src_node else None
+        src_service = src_node.service if src_node else None
+        src_container_kind = src_node.container_kind if src_node else None
 
         dst_container = dst_node.container if dst_node else None
         dst_layer = dst_node.layer if dst_node else None
         dst_context = dst_node.context if dst_node else None
+        dst_service = dst_node.service if dst_node else None
+        dst_container_kind = dst_node.container_kind if dst_node else None
 
         # Return enriched edge if anything changed
         if (
@@ -173,6 +190,10 @@ def _enrich_edge(self, edge: IREdge, node_lookup: dict[str, IRNode]) -> IREdge:
             and edge.dst_container == dst_container
             and edge.dst_layer == dst_layer
             and edge.dst_context == dst_context
+            and edge.src_service == src_service
+            and edge.dst_service == dst_service
+            and edge.src_container_kind == src_container_kind
+            and edge.dst_container_kind == dst_container_kind
         ):
             return edge
 
@@ -184,6 +205,10 @@ def _enrich_edge(self, edge: IREdge, node_lookup: dict[str, IRNode]) -> IREdge:
             dst_container=dst_container,
             dst_layer=dst_layer,
             dst_context=dst_context,
+            src_service=src_service,
+            dst_service=dst_service,
+            src_container_kind=src_container_kind,
+            dst_container_kind=dst_container_kind,
         )
 
     def _normalize_path(self, path: str) -> str: