From 971ec8f9d75b75f6cb51c686254ac430d39d3ba4 Mon Sep 17 00:00:00 2001
From: hpal <ahosea15@gmail.com>
Date: Fri, 28 Jun 2024 12:47:27 +0300
Subject: [PATCH] Allow creation of webhooks for project tokens

---
 .../azure_devops/webhooks/webhook_event.py    |  6 ++-
 integrations/azure-devops/bootstrap.py        | 37 ++++++++++++++++++-
 integrations/azure-devops/main.py             | 14 ++++++-
 3 files changed, 52 insertions(+), 5 deletions(-)

diff --git a/integrations/azure-devops/azure_devops/webhooks/webhook_event.py b/integrations/azure-devops/azure_devops/webhooks/webhook_event.py
index 749576c94..dcef6efa3 100644
--- a/integrations/azure-devops/azure_devops/webhooks/webhook_event.py
+++ b/integrations/azure-devops/azure_devops/webhooks/webhook_event.py
@@ -9,10 +9,13 @@ class WebhookEvent(BaseModel):
     consumerId: str = "webHooks"
     consumerActionId: str = "httpRequest"
     consumerInputs: Optional[dict[str, str]] = None
+    publisherInputs: Optional[dict[str, str]] = None
     status: Optional[str] = None
 
-    def set_consumer_url(self, url: str) -> None:
+    def set_webhook_details(self, url: str, project_id: Optional[str] = None) -> None:
         self.consumerInputs = {"url": url}
+        if project_id:
+            self.publisherInputs = {"projectId": project_id}
 
     def get_event_by_subscription(
         self, subscribed_events: list["WebhookEvent"]
@@ -22,6 +25,7 @@ def get_event_by_subscription(
                 subscribed_event.publisherId == self.publisherId
                 and subscribed_event.eventType == self.eventType
                 and subscribed_event.consumerInputs == self.consumerInputs
+                and subscribed_event.publisherInputs == self.publisherInputs
             ):
                 return subscribed_event
 
diff --git a/integrations/azure-devops/bootstrap.py b/integrations/azure-devops/bootstrap.py
index d481597b4..5f06156e5 100644
--- a/integrations/azure-devops/bootstrap.py
+++ b/integrations/azure-devops/bootstrap.py
@@ -1,4 +1,5 @@
 import asyncio
+from typing import List
 from loguru import logger
 from azure_devops.client.azure_devops_client import AzureDevopsClient
 from azure_devops.webhooks.listeners.listener import HookListener
@@ -6,12 +7,15 @@
 from azure_devops.webhooks.listeners.push import PushHookListener
 from azure_devops.webhooks.webhook_event import WebhookEvent
 from azure_devops.webhooks.webhook_event_observer import WebhookEventObserver
+from port_ocean import ocean
+from port_ocean.context import event
+from port_ocean.exceptions.context import EventContextNotFoundError
 
 webhook_event_handler = WebhookEventObserver()
 
 
 async def setup_listeners(
-    app_host: str, azure_devops_client: AzureDevopsClient
+    app_host: str, azure_devops_client: AzureDevopsClient, project_id: str = None
 ) -> None:
     listeners: list[HookListener] = [
         PullRequestHookListener(azure_devops_client),
@@ -20,7 +24,7 @@ async def setup_listeners(
     webhook_events: list[WebhookEvent] = list()
     for listener in listeners:
         for event in listener.webhook_events:
-            event.set_consumer_url(f"{app_host}/integration/webhook")
+            event.set_webhook_details(f"{app_host}/integration/webhook", project_id)
         webhook_event_handler.on(listener.webhook_events, listener.on_hook)
         webhook_events.extend(listener.webhook_events)
     await _upsert_webhooks(azure_devops_client, webhook_events)
@@ -76,3 +80,32 @@ async def _upsert_webhooks(
 
     else:
         logger.info("All relevant subscriptions already exist")
+
+
+def get_all_services() -> List[AzureDevopsClient]:
+    all_tokens_services = []
+
+    logger.info(
+        f"Creating azure devops clients for {len(ocean.integration_config['tokens'])} tokens"
+    )
+    for token in ocean.integration_config["tokens"].items():
+        azure_devops_client = AzureDevopsClient(
+            ocean.integration_config["organization_url"],
+            token,
+        )
+
+        all_tokens_services.append(azure_devops_client)
+
+    return all_tokens_services
+
+
+def get_cached_all_services() -> List[AzureDevopsClient]:
+    try:
+        all_services = event.attributes.get("all_tokens_services")
+        if not all_services:
+            logger.info("Azure Devops clients are not cached, creating them")
+            all_services = get_all_services()
+            event.attributes["all_tokens_services"] = all_services
+        return all_services
+    except EventContextNotFoundError:
+        return get_all_services()
\ No newline at end of file
diff --git a/integrations/azure-devops/main.py b/integrations/azure-devops/main.py
index 0d3a328c7..965fb9f4a 100644
--- a/integrations/azure-devops/main.py
+++ b/integrations/azure-devops/main.py
@@ -22,10 +22,20 @@ async def setup_webhooks() -> None:
     if not ocean.integration_config.get("app_host"):
         logger.warning("No app host provided, skipping webhook creation.")
         return
-
+    
     azure_devops_client = AzureDevopsClient.create_from_ocean_config()
-    await setup_listeners(ocean.integration_config["app_host"], azure_devops_client)
 
+    if ocean.integration_config.get("is_project_admin", False):
+        async for projects in azure_devops_client.generate_projects():
+            for project in projects:
+                await setup_listeners(
+                    ocean.integration_config["app_host"], azure_devops_client, project["id"]
+                )
+    else:
+        await setup_listeners(
+            ocean.integration_config["app_host"], azure_devops_client
+        )
+    
 
 @ocean.on_resync(Kind.PROJECT)
 async def resync_projects(kind: str) -> ASYNC_GENERATOR_RESYNC_TYPE: