From c0b2d39d8fd5921d8fd53f6f2ff2016710485f22 Mon Sep 17 00:00:00 2001 From: Mohse Morad Date: Sun, 15 Jun 2025 10:22:34 +0300 Subject: [PATCH] Refresh token when connectin to saas --- src/robusta/integrations/receiver.py | 7 ++++--- src/robusta/runner/config_loader.py | 3 +-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/src/robusta/integrations/receiver.py b/src/robusta/integrations/receiver.py index fdfb31bf6..d99a90c18 100644 --- a/src/robusta/integrations/receiver.py +++ b/src/robusta/integrations/receiver.py @@ -76,14 +76,14 @@ class SlackActionsMessage(BaseModel): class ActionRequestReceiver: - def __init__(self, event_handler: PlaybooksEventHandler, auth_token: str): + def __init__(self, event_handler: PlaybooksEventHandler, robusta_sink: "RobustaSink"): self.event_handler = event_handler self.active = True self.account_id = self.event_handler.get_global_config().get("account_id") self.cluster_name = self.event_handler.get_global_config().get("cluster_name") self.auth_provider = AuthProvider() self.healthy = False - self.auth_token = auth_token + self.robusta_sink = robusta_sink self.ws = websocket.WebSocketApp( WEBSOCKET_RELAY_ADDRESS, @@ -287,12 +287,13 @@ def on_error(self, ws, error): def on_open(self, ws): account_id = self.event_handler.get_global_config().get("account_id") cluster_name = self.event_handler.get_global_config().get("cluster_name") + token = self.robusta_sink.dal.get_session_token() open_payload = { "action": "auth", "account_id": account_id, "cluster_name": cluster_name, "version": RUNNER_VERSION, - "token": self.auth_token, + "token": token, } logging.info(f"connecting to server as account_id={account_id}; cluster_name={cluster_name}") ws.send(json.dumps(open_payload)) diff --git a/src/robusta/runner/config_loader.py b/src/robusta/runner/config_loader.py index aae73e9cd..12ce46681 100644 --- a/src/robusta/runner/config_loader.py +++ b/src/robusta/runner/config_loader.py @@ -105,9 +105,8 @@ def __create_receiver(self): return robusta_sink = robusta_sinks[0] - token = robusta_sink.dal.get_session_token() - receiver = ActionRequestReceiver(self.event_handler, token) + receiver = ActionRequestReceiver(self.event_handler, robusta_sink) self.registry.set_receiver(receiver) return receiver