arm explicit reference edges implementation

checkov/arm/graph_builder/local_graph.py

@@ -7,6 +7,7 @@
 from checkov.arm.graph_builder.graph_components.blocks import ArmBlock
 from checkov.arm.utils import ArmElements
 from checkov.common.graph.graph_builder import CustomAttributes
+from checkov.common.graph.graph_builder.graph_components.edge import Edge
 from checkov.common.graph.graph_builder.local_graph import LocalGraph
 from checkov.common.util.consts import START_LINE, END_LINE
 from checkov.common.util.data_structures_utils import pickle_deepcopy
@@ -21,6 +22,8 @@ def __init__(self, definitions: dict[str, dict[str, Any]]) -> None:
         self.vertices: list[ArmBlock] = []
         self.definitions = definitions
         self.vertices_by_path_and_id: dict[tuple[str, str], int] = {}
+        self.vertices_by_name: dict[str, int] = {}
+
 
     def build_graph(self, render_variables: bool = False) -> None:
         self._create_vertices()
@@ -38,6 +41,10 @@ def _create_vertices(self) -> None:
             self.vertices_by_block_type[vertex.block_type].append(i)
             self.vertices_block_name_map[vertex.block_type][vertex.name].append(i)
             self.vertices_by_path_and_id[(vertex.path, vertex.id)] = i
+            self.vertices_by_name[vertex.name] = i
+
+            self.in_edges[i] = []
+            self.out_edges[i] = []
 
     def _create_parameter_vertices(self, file_path: str, parameters: dict[str, dict[str, Any]] | None) -> None:
         if not parameters:
@@ -85,13 +92,46 @@ def _create_resource_vertices(self, file_path: str, resources: list[dict[str, An
                     path=file_path,
                     block_type=BlockType.RESOURCE,
                     attributes=attributes,
-                    id=f"{resource_type}.{resource_name}",
+                    id=f"{resource_type}.{resource_name}"
                 )
             )
 
     def _create_edges(self) -> None:
-        # no edges yet
-        pass
+        for origin_vertex_index, vertex in enumerate(self.vertices):
+            if 'dependsOn' not in vertex.attributes:
+                continue
+            for dep in vertex.attributes['dependsOn']:
+                if 'resourceId' in dep:
+                    # Extract name from resourceId function
+                    processed_dep = dep.split(',')[-1].split(')')[0]
+                    if '(' in processed_dep:
+                        processed_dep = processed_dep+')'
+
+                    if 'variables' in processed_dep:
+                        # TODO: Render resource name from variables
+                        pass
+                    if 'parameters' in processed_dep:
+                        # TODO: Render resource name from parameters
+                        pass
+                else:
+                    processed_dep = dep.split('/')[-1]
+                # Check if the processed dependency exists in the map
+                if processed_dep in self.vertices_by_name:
+                    self._create_edge(processed_dep, origin_vertex_index, dep)
+                else:
+                    # Dependency not found
+                    logging.debug(f"[ArmLocalGraph] resource dependency {processed_dep} defined in {dep} for resource"
+                                  f" {vertex.name} not found")
+                    continue
+
+    def _create_edge(self, element_name: str, origin_vertex_index: int, label: str) -> None:
+        dest_vertex_index = self.vertices_by_name.get(element_name)
+        if origin_vertex_index == dest_vertex_index:
+            return
+        edge = Edge(origin_vertex_index, dest_vertex_index, label)
+        self.edges.append(edge)
+        self.out_edges[origin_vertex_index].append(edge)
+        self.in_edges[dest_vertex_index].append(edge)
 
     def update_vertices_configs(self) -> None:
         # not used