From f3a512d8915b88d1681c5f892f001c7ea3bb8393 Mon Sep 17 00:00:00 2001
From: Kirill Chernakov <yakiryous@gmail.com>
Date: Tue, 7 Jan 2025 15:55:48 +0400
Subject: [PATCH] feat: enrich `/workflows/{workflow_id}` route with workflow
 metadata, just like in `/wokrflows/{workflow_id}/runs`

---
 keep/api/routes/workflows.py | 30 +++++++++++++++++++++++++-----
 1 file changed, 25 insertions(+), 5 deletions(-)

diff --git a/keep/api/routes/workflows.py b/keep/api/routes/workflows.py
index 7a5912996..e6504df69 100644
--- a/keep/api/routes/workflows.py
+++ b/keep/api/routes/workflows.py
@@ -556,7 +556,6 @@ def get_workflow_by_id(
     tenant_id = authenticated_entity.tenant_id
     # get all workflow
     workflow = get_workflow(tenant_id=tenant_id, workflow_id=workflow_id)
-
     if not workflow:
         logger.warning(
             f"Tenant tried to get workflow {workflow_id} that does not exist",
@@ -564,6 +563,29 @@ def get_workflow_by_id(
         )
         raise HTTPException(404, "Workflow not found")
 
+    installed_providers = get_installed_providers(tenant_id)
+    installed_providers_by_type = {}
+    for installed_provider in installed_providers:
+        if installed_provider.type not in installed_providers_by_type:
+            installed_providers_by_type[installed_provider.type] = {
+                installed_provider.name: installed_provider
+            }
+        else:
+            installed_providers_by_type[installed_provider.type][
+                installed_provider.name
+            ] = installed_provider
+
+    workflowstore = WorkflowStore()
+    try:
+        providers_dto, triggers = workflowstore.get_workflow_meta_data(
+            tenant_id=tenant_id,
+            workflow=workflow,
+            installed_providers_by_type=installed_providers_by_type,
+        )
+    except Exception as e:
+        logger.error(f"Error fetching workflow meta data: {e}")
+        providers_dto, triggers = [], []  # Default in case of failure
+
     try:
         workflow_yaml = yaml.safe_load(workflow.workflow_raw)
         valid_workflow_yaml = {"workflow": workflow_yaml}
@@ -575,13 +597,11 @@ def get_workflow_by_id(
             created_by=workflow.created_by,
             creation_time=workflow.creation_time,
             interval=workflow.interval,
-            providers=[],
-            triggers=[],
+            providers=providers_dto,
+            triggers=triggers,
             workflow_raw=final_workflow_raw,
-            revision=workflow.revision,
             last_updated=workflow.last_updated,
             disabled=workflow.is_disabled,
-            provisioned=workflow.provisioned,
         )
         return workflow_dto
     except yaml.YAMLError: