added check for holmes_slackbot enabled - pre-test

Author: Avi-Robusta

src/robusta/core/sinks/robusta/robusta_sink.py

@@ -42,7 +42,14 @@
 from robusta.integrations.receiver import ActionRequestReceiver
 from robusta.runner.web_api import WebApi
 from robusta.utils.stack_tracer import StackTracer
+from robusta.core.model.env_vars import ROBUSTA_API_ENDPOINT
+from cachetools import TTLCache
 
+RUNNER_GET_HOLMES_SLACKBOT_INFO = f"{ROBUSTA_API_ENDPOINT}/api/holmes/integrations/slack/runner"
+HOLMES_SLACKBOT_CACHE_TTL = int(os.getenv("HOLMES_SLACKBOT_CACHE_TTL", 15 * 60))
+
+# Define the cache with a single slot and the configured TTL
+_holmes_slackbot_cache = TTLCache(maxsize=1, ttl=HOLMES_SLACKBOT_CACHE_TTL)
 
 class RobustaSink(SinkBase, EventHandler):
     services_publish_lock = threading.Lock()
@@ -695,3 +702,28 @@ def __update_job(self, new_job: Job, operation: K8sOperationType):
                 self.__safe_delete_job(job_key)
                 self.__discovery_metrics.on_jobs_updated(1)
                 return
+
+    def is_holmes_slackbot_connected(self) -> bool:
+        if 'status' in _holmes_slackbot_cache:
+            return _holmes_slackbot_cache['status']
+
+        session_token = self.dal.get_session_token()
+        try:
+            message_json = {
+                "session_token": session_token,
+                "account_id": self.account_id,
+            }
+            response = requests.post(
+                RUNNER_GET_HOLMES_SLACKBOT_INFO,
+                data=message_json,
+                headers={"Content-Type": "application/json"}
+            )
+            response.raise_for_status()
+            result = response.json()
+            logging.warning(f"Holmes slackbot result {result}")
+            _holmes_slackbot_cache['status'] = True
+            return True
+        except Exception:
+            logging.info("Failed to get holmes slackbot info")
+            _holmes_slackbot_cache['status'] = False
+            return False

src/robusta/core/sinks/slack/slack_sink.py

@@ -16,7 +16,7 @@ def __init__(self, sink_config: SlackSinkConfigWrapper, registry, is_preview=Fal
         self.slack_channel = slack_sink_params.slack_channel
         self.api_key = slack_sink_params.api_key
         self.slack_sender = slack_module.SlackSender(
-            self.api_key, self.account_id, self.cluster_name, self.signing_key, self.slack_channel, is_preview
+            self.api_key, self.account_id, self.cluster_name, self.signing_key, self.slack_channel, registry, is_preview
         )
         self.registry.subscribe("replace_callback_with_string", self)
 

src/robusta/integrations/slack/sender.py

@@ -48,7 +48,7 @@
 from robusta.core.sinks.sink_base import KeyT
 from robusta.core.sinks.slack.slack_sink_params import SlackSinkParams
 from robusta.core.sinks.slack.preview.slack_sink_preview_params import SlackSinkPreviewParams
-
+from robusta.model.config import Registry
 from robusta.core.sinks.transformer import Transformer
 
 ACTION_TRIGGER_PLAYBOOK = "trigger_playbook"
@@ -62,7 +62,7 @@ class SlackSender:
     verified_api_tokens: Set[str] = set()
     channel_name_to_id = {}
 
-    def __init__(self, slack_token: str, account_id: str, cluster_name: str, signing_key: str, slack_channel: str, is_preview: bool = False):
+    def __init__(self, slack_token: str, account_id: str, cluster_name: str, signing_key: str, slack_channel: str, registry: Registry, is_preview: bool = False):
         """
         Connect to Slack and verify that the Slack token is valid.
         Return True on success, False on failure
@@ -80,6 +80,7 @@ def __init__(self, slack_token: str, account_id: str, cluster_name: str, signing
             timeout=SLACK_REQUEST_TIMEOUT,
             retry_handlers=all_builtin_retry_handlers(),
         )
+        self.registry = registry
         self.signing_key = signing_key
         self.account_id = account_id
         self.cluster_name = cluster_name
@@ -672,6 +673,15 @@ def send_finding_to_slack(
             thread_ts=thread_ts
         )
 
+    def __is_holmes_slackbot_enabled(self) -> bool:
+        robusta_sinks = self.registry.get_sinks().get_robusta_sinks()
+        if not robusta_sinks:
+            logging.debug("No robusta sinks found, holmes not connected to slackbot")
+            return False
+
+        robusta_sink = robusta_sinks[0]
+        return robusta_sink.is_holmes_slackbot_connected()
+
     def __send_finding_to_slack(
         self,
         finding: Finding,
@@ -731,7 +741,8 @@ def __send_finding_to_slack(
 
         blocks.append(DividerBlock())
 
-        holmes_block = self.get_holmes_block(platform_enabled, HOLMES_ENABLED)
+        is_holmes_slackbot_enabled = self.__is_holmes_slackbot_enabled()
+        holmes_block = self.get_holmes_block(platform_enabled, is_holmes_slackbot_enabled)
         if holmes_block:
             blocks.append(holmes_block)