1P Updates: GitSync

content/response_integrations/power_ups/git_sync/core/GitManager.py

@@ -46,16 +46,6 @@
     from soar_sdk.SiemplifyLogger import SiemplifyLogger
 
 
-PUSH_FAILURE_INDICATORS = (
-        "pre-receive hook declined",
-        "not allowed to push",
-        "push rejected",
-        "failed to push",
-        "error: failed to push",
-        "! [rejected]",
-        "! [remote rejected]",
-    )
-
 class Git:
     """GitManager"""
 
@@ -232,26 +222,13 @@ def push(self, force_push=False) -> None:
             equivalent to 'git push --force'. Defaults to False.
 
         """
-        error_content = ""
         try:
-            error_buffer = StringIO()
-            tee_stream = TeeStream(sys.stderr, error_buffer)
-
-            try:
-                porcelain.push(
-                    self.repo,
-                    refspecs=[self.local_branch_ref],
-                    force=force_push,
-                    errstream=tee_stream,
-                    **self.connection_args,
-                )
-            finally:
-                tee_stream.flush()
-                error_content = error_buffer.getvalue().strip()
-                tee_stream.close()
-
-            self._raise_on_push_errors(error_content)
-
+            porcelain.push(
+                self.repo,
+                refspecs=[self.local_branch_ref],
+                force=force_push,
+                **self.connection_args,
+            )
         except porcelain.DivergedBranches:
             self.logger.error("Could not push updates to remote repository!")
             self.logger.warn(
@@ -264,15 +241,6 @@ def push(self, force_push=False) -> None:
                 "Updates will be pushed in the next python script execution",
             )
 
-    def _raise_on_push_errors(self, error_content: str |None) -> None:
-        """Check for push failure indicators and raise exception."""
-        if not error_content:
-            return
-
-        if any(indicator in error_content for indicator in PUSH_FAILURE_INDICATORS):
-            self.logger.error(f"Push operation failed: {error_content}")
-            raise GitSyncException(f"Push operation failed: {error_content}")
-
     def _checkout(self) -> None:
         """Checkout a branch
 
@@ -805,4 +773,4 @@ def _verify_and_decide(self, client, hostname, key) -> None:
         else:
             self.vendor.siemplify_logger.error("Fingerprint verification failed.")
             raise paramiko.ssh_exception.SSHException(
-                f"Host key verification failed for {hostname}")
+                f"Host key verification failed for {hostname}")

content/response_integrations/power_ups/git_sync/core/GitSyncManager.py

@@ -19,6 +19,7 @@
 import uuid
 from typing import TYPE_CHECKING, Any
 from TIPCommon.types import SingleJson
+from TIPCommon.rest.soar_api import install_integration
 
 from jinja2 import Template
 
@@ -87,6 +88,7 @@ def __init__(
             smp_credentials.get("username") if smp_credentials else None,
             smp_credentials.get("password") if smp_credentials else None,
             smp_verify,
+            siemplify_soar=siemplify,
         )
         self.git_server_fingerprint = git_server_fingerprint
         self.git_client = Git(
@@ -306,7 +308,7 @@ def install_connector(self, connector: Connector) -> None:
                 "Please upgrade the connector.",
             )
             connector.raw_data["isUpdateAvailable"] = True
-        if connector.environment not in self.api.get_environment_names():
+        if connector.environment not in self.api.get_environment_names(self._siemplify):
             self.logger.warn(
                 f"Connector is set to non-existing environment {connector.environment}. "
                 f"Using Default Environment instead",
@@ -345,16 +347,16 @@ def install_workflows(self, workflows: list[Workflow]) -> None:
         """
         # Validate all playbook environments exist as environments or environment groups
         environments = (
-            self.api.get_environment_names()
+            self.api.get_environment_names(self._siemplify)
             + self.api.get_environment_group_names()
             + [ALL_ENVIRONMENTS_IDENTIFIER]
         )
         for p in workflows:
             invalid_environments = [x for x in p.environments if x not in environments]
             if invalid_environments:
                 raise Exception(
-                    f"Playbook '{p.name}' is assigned to environment(s) that don't exist: "
-                    f"{', '.join(invalid_environments)}. "
+                    f"Playbook '{p.name}' is assigned to environment(s) that don't "
+                    f"exist: {', '.join(invalid_environments)}. "
                     f"Available environments: {', '.join(environments)}"
                 )
 
@@ -415,7 +417,14 @@ def install_job(self, job: Job) -> None:
             if job_def_id:
                 job.raw_data["jobDefinitionId"] = job_def_id.get("id")
 
-        job_id = next((x for x in self.api.get_jobs() if x["name"] == job.name), None)
+        job_id = next(
+            (
+                x
+                for x in self.api.get_jobs(chronicle_soar=self._siemplify)
+                if x.get("displayName", x.get("name")) == job.name
+            ),
+            None,
+        )
         if job_id:
             job.raw_data["id"] = job_id.get("id")
         self.api.add_job(job.raw_data)
@@ -546,10 +555,12 @@ def install_marketplace_integration(self, integration_name: str) -> bool:
             )
             return False
         try:
-            self.api.install_integration(
-                integration_name,
-                store_integration["version"],
-                store_integration["isCertified"],
+            install_integration(
+                chronicle_soar=self._siemplify,
+                integration_identifier=integration_name,
+                integration_name="",
+                version=store_integration["version"],
+                is_certified=store_integration["isCertified"],
             )
             self.logger.info(f"{integration_name} installed successfully")
             return True
@@ -583,7 +594,7 @@ class WorkflowInstaller:
 
     def __init__(
         self,
-        chronicle_soar: ChronicleSOAR,
+        chronicle_soar: ChronicleSOAR, # type: ignore
         api: SiemplifyApiClient,
         logger: SiemplifyLogger,
         mod_time_cache: Cache[str, int],
@@ -659,6 +670,9 @@ def install_new_workflow(self, workflow: Workflow) -> None:
         self._define_workflow_as_new(workflow)
         self._process_steps(workflow)
         self.api.save_playbook(workflow.raw_data)
+
+        self.refresh_cache_item("playbooks")
+
         self._save_workflow_mod_time_to_context(workflow)
         self.logger.info(f"New workflow '{workflow.name}' was installed successfully")
 
@@ -781,15 +795,18 @@ def _get_local_workflow_mod_time(
         default: int | None = None,
         /,
     ) -> int:
-        playbook: dict[str, Any] = self._installed_playbooks[__workflow_name]
+        playbook = self._installed_playbooks.get(__workflow_name)
+        if not playbook:
+            return default
         return playbook.get("modificationTimeUnixTimeInMs", default)
 
     @property
     def _installed_playbooks(self) -> dict[str, dict[str, Any]]:
         """Currently installed playbooks and blocks"""
         if "playbooks" not in self._cache:
             self._cache["playbooks"] = {
-                x.get("name"): x for x in self.api.get_playbooks()
+                x.get("name"): x
+                for x in self.api.get_playbooks(chronicle_soar=self.chronicle_soar)
             }
         return self._cache.get("playbooks")
 
@@ -877,14 +894,14 @@ def _assign_integration_instance_to_step(
                 instance_id = self.api.get_integration_instance_id_by_name(
                     self.chronicle_soar,
                     step.get("integration"),
-                    environments=environments,
+                    environments=environments, # type: ignore
                     display_name=instance_display_name,
                     consider_404_to_none=True,
                 )
                 self._set_step_parameter_by_name(
                     step,
                     "IntegrationInstance",
-                    instance_id or integration_instances[0].get("identifier"),
+                    instance_id or integration_instances[0].identifier,
                 )
                 self._set_step_parameter_by_name(
                     step,
@@ -905,7 +922,7 @@ def _assign_integration_instance_to_step(
                 self._set_step_parameter_by_name(
                     step,
                     "FallbackIntegrationInstance",
-                    fallback_instance_id or integration_instances[0].get("identifier"),
+                    fallback_instance_id or integration_instances[0].identifier,
                 )
             else:
                 self._set_step_parameter_by_name(
@@ -945,16 +962,18 @@ def _find_integration_instances_for_step(
         """
         cache_key = f"integration_instances_{environment}"
         if cache_key not in self._cache:
-            self._cache[cache_key] = self.api.get_integrations_instances(environment)
+            self._cache[cache_key] = self.api.get_integrations_instances(
+                chronicle_soar=self.chronicle_soar,
+                environment=environment,
+            )
 
         instances = self._cache.get(cache_key)
-        instances.sort(key=lambda x: x.get("instanceName"))
+        instances.sort(key=lambda x: x.instance_name)
 
         return [
             x
             for x in instances
-            if x.get("integrationIdentifier") == integration_name
-            and x.get("isConfigured")
+            if x.integration_identifier == integration_name and x.is_configured
         ]
 
     @staticmethod