From b716c9a24889f951fbbc1311ca56cf491e00e99b Mon Sep 17 00:00:00 2001
From: omriyoffe-panw <oyoffe@paloaltonetworks.com>
Date: Mon, 4 Nov 2024 18:48:00 +0200
Subject: [PATCH] fix(arm): add middleware function for platform integration
 for Arm definitions (#6811)

* create definitions
---
 checkov/arm/utils.py | 26 ++++++++++++++++++++++++--
 1 file changed, 24 insertions(+), 2 deletions(-)

diff --git a/checkov/arm/utils.py b/checkov/arm/utils.py
index b6b35da6f74..c04bff61e9f 100644
--- a/checkov/arm/utils.py
+++ b/checkov/arm/utils.py
@@ -1,8 +1,11 @@
 from __future__ import annotations
 
+import logging
 import os
 from enum import Enum
 from typing import Iterable, Callable, Any
+from collections.abc import Collection
+from pathlib import Path
 
 from checkov.arm.parser.parser import parse
 from checkov.common.runners.base_runner import filter_ignored_paths
@@ -40,11 +43,30 @@ def get_scannable_file_paths(root_folder: str | None = None, excluded_paths: lis
     return file_paths
 
 
-# RunnerFilter argument added to support the platform integration
+def create_definitions(
+    root_folder: str,
+    _: Collection[Path] | None = None,
+    runner_filter: RunnerFilter | None = None,
+) -> tuple[dict[str, dict[str, Any]], dict[str, list[tuple[int, str]]]]:
+    definitions: dict[str, dict[str, Any]] = {}
+    definitions_raw: dict[str, list[tuple[int, str]]] = {}
+    parsing_errors: list[str] = []
+    runner_filter = runner_filter or RunnerFilter()
+
+    if root_folder:
+        file_paths = get_scannable_file_paths(root_folder, runner_filter.excluded_paths)
+        filepath_fn = lambda f: f"/{os.path.relpath(f, os.path.commonprefix((root_folder, f)))}"
+        definitions, definitions_raw, parsing_errors = get_files_definitions(files=file_paths, filepath_fn=filepath_fn)
+
+    if parsing_errors:
+        logging.warning(f"[arm] found errors while parsing definitions: {parsing_errors}")
+
+    return definitions, definitions_raw
+
+
 def get_files_definitions(
         files: Iterable[str],
         filepath_fn: Callable[[str], str] | None = None,
-        _: RunnerFilter | None = None,
 ) -> tuple[dict[str, dict[str, Any]], dict[str, list[tuple[int, str]]], list[str]]:
     """Parses ARM files into its definitions and raw data"""