[ROB-2024] assume role support (#468)

Author: Avi-Robusta

poetry.lock

@@ -30,7 +30,7 @@ kubernetes = "^26.1.0"
 prometheus-api-client = "0.5.3"
 numpy = ">=1.26.4,<1.27.0"
 alive-progress = "^3.1.2"
-prometrix = "0.2.2"
+prometrix = "0.2.3"
 slack-sdk = "^3.21.3"
 pandas = "2.2.2"
 requests = "2.32.0"

pyproject.toml

@@ -27,7 +27,7 @@ packaging==24.0 ; python_version >= "3.9" and python_full_version < "3.13"
 pandas==2.2.2 ; python_version >= "3.9" and python_full_version < "3.13"
 pillow==10.3.0 ; python_version >= "3.9" and python_full_version < "3.13"
 prometheus-api-client==0.5.3 ; python_version >= "3.9" and python_full_version < "3.13"
-prometrix==0.2.2 ; python_version >= "3.9" and python_full_version < "3.13"
+prometrix==0.2.3 ; python_version >= "3.9" and python_full_version < "3.13"
 pyasn1-modules==0.3.0 ; python_version >= "3.9" and python_full_version < "3.13"
 pyasn1==0.5.1 ; python_version >= "3.9" and python_full_version < "3.13"
 pydantic==1.10.15 ; python_version >= "3.9" and python_full_version < "3.13"

requirements.txt

@@ -60,6 +60,7 @@ def generate_prometheus_config(
             aws_region=region,
             token=token,
             service_name=service_name,
+            assume_role_arn=settings.eks_assume_role,
             **baseconfig,
         )
     # coralogix config

robusta_krr/core/integrations/prometheus/prometheus_utils.py

@@ -46,6 +46,7 @@ class Config(pd.BaseSettings):
     eks_secret_key: Optional[pd.SecretStr] = pd.Field(None)
     eks_service_name: Optional[str] = pd.Field(None)
     eks_managed_prom_region: Optional[str] = pd.Field(None)
+    eks_assume_role: Optional[str] = pd.Field(None)
     coralogix_token: Optional[pd.SecretStr] = pd.Field(None)
     openshift: bool = pd.Field(False)
 

robusta_krr/core/models/config.py

@@ -183,6 +183,12 @@ def run_strategy(
                     help="Sets the region for eks prometheus connection.",
                     rich_help_panel="Prometheus EKS Settings",
                 ),
+                eks_assume_role: Optional[str] = typer.Option(
+                    None,
+                    "--eks-assume-role",
+                    help="Sets the assumed role for eks prometheus connection. (for cross-account role assumption)",
+                    rich_help_panel="Prometheus EKS Settings",
+                ),
                 coralogix_token: Optional[str] = typer.Option(
                     None,
                     "--coralogix-token",
@@ -343,6 +349,7 @@ def run_strategy(
                         prometheus_label=prometheus_label,
                         eks_managed_prom=eks_managed_prom,
                         eks_managed_prom_region=eks_managed_prom_region,
+                        eks_assume_role=eks_assume_role,
                         eks_managed_prom_profile_name=eks_managed_prom_profile_name,
                         eks_access_key=eks_access_key,
                         eks_secret_key=eks_secret_key,