fix

fix

fix

debug

linear integration

Author: rushilpatel0

src/codegen/extensions/clients/linear.py

@@ -160,7 +160,11 @@ def register_webhook(self, webhook_url: str, team_id: str, secret: str, enabled:
             "enabled": enabled,
             "secret": secret,
         }}
-
+        
         response = requests.post(self.api_endpoint, headers=self.api_headers, json={"query": mutation, "variables": variables})
+        if response.status_code != 200:
+            return None
+        
         body = response.json()
+        body = body["data"]["webhookCreate"]["webhook"]['id']
         return body

src/codegen/extensions/events/linear.py

@@ -8,9 +8,6 @@
 from codegen.extensions.events.interface import EventHandlerManagerProtocol
 from codegen.extensions.clients.linear import LinearClient
 
-if TYPE_CHECKING:
-    from codegen.extensions.events.app import CodegenApp
-
 
 logger = logging.getLogger(__name__)
 
@@ -20,7 +17,7 @@ class RegisteredWebhookHandler(BaseModel):
     handler_func: Callable
 
 class Linear(EventHandlerManagerProtocol):
-    def __init__(self, app: CodegenApp):
+    def __init__(self, app: modal.App):
         self.app = app
         self.access_token= os.environ["LINEAR_ACCESS_TOKEN"] # move to extensions config.
         self.signing_secret = os.environ["LINEAR_SIGNING_SECRET"]
@@ -36,18 +33,19 @@ def subscribe_handler_to_webhook(self, web_url: str, event_name: str):
             enabled=True, 
             resource_types=[event_name], 
             secret=self.signing_secret)
-        return result.json()
+        return result
 
     def unsubscribe_handler_to_webhook(self, registered_handler: RegisteredWebhookHandler):
         webhook_id = registered_handler.webhook_id
 
         client = LinearClient(access_token=self.access_token)
         if webhook_id:
             print(f"Unsubscribing from webhook {webhook_id}")
-            client.unregister_webhook(webhook_id)
+            result = client.unregister_webhook(webhook_id)
+            return result
         else:
             print("No webhook id found for handler")
-
+            return None
     def unsubscribe_all_handlers(self):
         for handler in self.registered_handlers:
             self.unsubscribe_handler_to_webhook(self.registered_handlers[handler])
@@ -71,8 +69,8 @@ def decorator(func):
 
             self.registered_handlers[func_name] = RegisteredWebhookHandler(handler_func=modal_ready_func)
 
-            self.subscribe_handler_to_webhook(web_url=web_url, event_name=event_name)
-
+            webhook_id = self.subscribe_handler_to_webhook(web_url=web_url, event_name=event_name)
+            self.registered_handlers[func_name].webhook_id = webhook_id
             @functools.wraps(func)
             def wrapper(*args, **kwargs):
                 return func(*args, **kwargs)