fix(airflow): add catalog classification and env loading

cerencamkiran · cerencamkiran · commit a71aa6c09735 · 2026-04-19T19:25:32.000+03:00
diff --git a/app/integrations/catalog.py b/app/integrations/catalog.py
@@ -8,6 +8,11 @@
 from typing import Any
 
 from app.config import get_tracer_base_url
+from app.integrations.airflow import (
+    DEFAULT_AIRFLOW_BASE_URL,
+    airflow_config_from_env,
+    build_airflow_config,
+)
 from app.integrations.azure_sql import build_azure_sql_config
 from app.integrations.github_mcp import build_github_mcp_config
 from app.integrations.gitlab import DEFAULT_GITLAB_BASE_URL, build_gitlab_config
@@ -79,6 +84,8 @@
     "opensearch": "opensearch",
     "open search": "opensearch",
     "alertmanager": "alertmanager",
+    "airflow": "airflow",
+    "apache airflow": "airflow",
 }
 
 
@@ -558,6 +565,30 @@ def _classify_service_instance(
             }, "rabbitmq"
         return None, None
 
+    if key == "airflow":
+        try:
+            airflow_config = build_airflow_config(
+                {
+                    "base_url": credentials.get("base_url", DEFAULT_AIRFLOW_BASE_URL),
+                    "username": credentials.get("username", ""),
+                    "password": credentials.get("password", ""),
+                    "auth_token": credentials.get("auth_token", ""),
+                    "timeout_seconds": credentials.get("timeout_seconds", 15.0),
+                    "verify_ssl": credentials.get("verify_ssl", True),
+                    "max_results": credentials.get("max_results", 50),
+                }
+            )
+        except Exception:
+            return None, None
+
+        if airflow_config.is_configured:
+            return {
+                **airflow_config.model_dump(),
+                "integration_id": record_id,
+            }, "airflow"
+
+        return None, None
+
     if key == "azure_sql":
         try:
             azure_sql_config = build_azure_sql_config(
@@ -1087,6 +1118,17 @@ def load_env_integrations() -> list[dict[str, Any]]:
             }
         )
 
+    airflow_config = airflow_config_from_env()
+    if airflow_config is not None:
+        integrations.append(
+            {
+                "id": "env-airflow",
+                "service": "airflow",
+                "status": "active",
+                "credentials": airflow_config.model_dump(exclude={"integration_id"}),
+            }
+        )
+
     atlas_pub = os.getenv("MONGODB_ATLAS_PUBLIC_KEY", "").strip()
     atlas_priv = os.getenv("MONGODB_ATLAS_PRIVATE_KEY", "").strip()
     atlas_project = os.getenv("MONGODB_ATLAS_PROJECT_ID", "").strip()
