fix: wip impl webhook registration via codegen

Author: rushilpatel0

src/codegen/extensions/clients/linear.py

@@ -0,0 +1,154 @@
+import json
+import logging
+import requests
+
+from pydantic import BaseModel
+
+logger = logging.getLogger(__name__)
+
+
+# --- TYPES
+
+class LinearUser(BaseModel):
+    id: str
+    name: str
+
+class LinearComment(BaseModel):
+    id: str
+    body: str
+    user: LinearUser | None = None
+
+class LinearIssue(BaseModel):
+    id: str
+    title: str
+    description: str | None = None
+
+
+class LinearClient:
+    api_headers: dict
+    api_endpoint = "https://api.linear.app/graphql"
+
+    def __init__(self, access_token: str):
+        self.access_token = access_token
+        self.api_headers = {
+            "Content-Type": "application/json",
+            "Authorization": self.access_token,
+        }
+
+    def get_issue(self, issue_id: str) -> LinearIssue:
+        query = """
+            query getIssue($issueId: String!) {
+                issue(id: $issueId) {
+                    id
+                    title
+                    description
+                }
+            }
+        """
+        variables = {"issueId": issue_id}
+        response = requests.post(self.api_endpoint, headers=self.api_headers, json={"query": query, "variables": variables})
+        data = response.json()
+        issue_data = data["data"]["issue"]
+        return LinearIssue(
+            id=issue_data["id"],
+            title=issue_data["title"],
+            description=issue_data["description"]
+        )
+        
+
+    def get_issue_comments(self, issue_id: str) -> list[LinearComment]:
+        query = """
+            query getIssueComments($issueId: String!) {
+                issue(id: $issueId) {
+                    comments {
+                    nodes {
+                        id
+                        body
+                        user {
+                            id
+                            name
+                        }
+                    }
+                        
+                    }
+                }
+            }
+        """
+        variables = {"issueId": issue_id}
+        response = requests.post(self.api_endpoint, headers=self.api_headers, json={"query": query, "variables": variables})
+        data = response.json()
+        comments = data["data"]["issue"]["comments"]["nodes"]
+
+        # Parse comments into list of LinearComment objects
+        parsed_comments = []
+        for comment in comments:
+            user = comment.get("user", None)
+            parsed_comment = LinearComment(
+                id=comment["id"],
+                body=comment["body"],
+                user=LinearUser(
+                    id=user.get("id"),
+                    name=user.get("name")
+                ) if user else None
+            )
+            parsed_comments.append(parsed_comment)
+
+        # Convert raw comments to LinearComment objects
+        return parsed_comments
+
+
+    def comment_on_issue(self, issue_id: str, body: str) -> dict:   
+        """issue_id is our internal issue ID"""
+        query = """mutation makeComment($issueId: String!, $body: String!) {
+          commentCreate(input: {issueId: $issueId, body: $body}) {
+            comment {
+              id
+              body
+              url
+              user {
+                id  
+                name
+              }
+            }
+          }
+        }
+        """
+        variables = {"issueId": issue_id, "body": body}
+        response = requests.post(
+            self.api_endpoint,
+            headers=self.api_headers,
+            data=json.dumps({"query": query, "variables": variables}),
+        )
+        data = response.json()
+        try:
+            comment_data = data["data"]["commentCreate"]["comment"]
+
+            return comment_data
+        except:
+            raise Exception(f"Error creating comment\n{data}")
+
+
+    def register_webhook(self, webhook_url: str, team_id: str, secret: str, enabled: bool, resource_types: list[str]):
+        mutation = """
+            mutation createWebhook($input: WebhookCreateInput!) {
+                webhookCreate(input: $input) {
+                    success
+                    webhook {
+                    id
+                    enabled
+                    }
+                }
+            }
+        """
+
+        variables = {"input": {
+            "url": webhook_url,
+            "teamId": team_id,
+            "resourceTypes": resource_types,
+            "enabled": enabled,
+            "secret": secret,
+        }}
+
+        response = requests.post(self.api_endpoint, headers=self.api_headers, json={"query": mutation, "variables": variables})
+        body = response.json()
+        return body

src/codegen/extensions/events/app.py

@@ -0,0 +1,99 @@
+import functools
+import os
+from typing import Callable, Optional
+import modal
+import modal.runner
+
+from codegen.extensions.clients.linear import LinearClient
+
+class CodegenApp:
+    def __init__(self, name, modal_api_key, image: modal.Image):
+        self._modal_api_key = modal_api_key
+        self._image = image
+        self._name = name
+
+        self.modal_app = modal.App(name=name, image=image, )
+        # Internal registry for event handlers, keyed by event name.
+        self._registry = {}
+        # Expose a linear attribute that provides the event decorator.
+        self.linear = Linear(self)
+
+    def _register_event_handler(self, event_name, handler):
+        """Registers a handler for a given event."""
+        self._registry.setdefault(event_name, []).append(handler)
+        print(f"Registered handler '{handler}' for event '{event_name}'.")
+
+    def dispatch_event(self, event_name, request):
+        """
+        Dispatches an event to all registered handlers for the event.
+        """
+        handlers = self._registry.get(event_name, [])
+        if not handlers:
+            print(f"No handlers registered for event '{event_name}'.")
+            return
+        for handler in handlers:
+            print(f"Dispatching event '{event_name}' to {handler.__name__}.")
+            handler(request)
+
+
+
+class Linear:
+    def __init__(self, app: CodegenApp):
+        self.app = app
+        self.access_token= os.environ["LINEAR_ACCESS_TOKEN"] # move to extensions config.
+        self.signing_secret = os.environ["LINEAR_SIGNING_SECRET"]
+        self.linear_team_id = os.environ["LINEAR_TEAM_ID"]
+
+
+    def register_handler_to_webhook(self, func_name: str, modal_app: modal.App, event_name):
+        app_name = modal_app.name
+        web_url = modal.Function.from_name(app_name=app_name,name=func_name)
+        client = LinearClient(access_token=self.access_token)
+        client.register_webhook(
+            team_id=self.linear_team_id, 
+            webhook_url=web_url, 
+            enabled=True, 
+            resource_types=[event_name], 
+            secret=self.signing_secret)
+        
+    # def unregister_handler(self):
+        # TODO
+        # pass
+
+
+    def event(self, event_name, register_hook: Optional[Callable]=None):
+        """
+        Decorator for registering an event handler.
+
+        :param event_name: The name of the event to handle.
+        :param register_hook: An optional function to call during registration,
+                              e.g., to make an API call to register the webhook.
+        """
+        def decorator(func):
+            # Register the handler with the app's registry.
+
+            modal_ready_func = apply_decorators(func=func, decorators=[modal.web_endpoint(method="POST"), self.app.modal_app.function()])
+            self.app._register_event_handler(event_name, modal_ready_func)
+
+            # If a custom registration hook is provided, execute it.
+            if callable(register_hook):
+                register_hook(event_name, func, self.app)
+
+            @functools.wraps(func)
+            def wrapper(*args, **kwargs):
+                return func(*args, **kwargs)
+            
+            result = modal.runner.deploy_app(self.app.modal_app)
+            print(result)
+            return wrapper
+            
+        
+        return decorator
+    
+        
+    
+
+def apply_decorators(func, decorators):
+    for decorator in decorators:
+        func = decorator(func)
+    return func

src/codegen/extensions/events/test.ipynb

@@ -0,0 +1,84 @@
+{
+ "cells": [
+  {
+   "cell_type": "code",
+   "execution_count": 4,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from codegen.extensions.events.app import CodegenApp\n",
+    "import modal"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": 5,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "image = modal.Image.debian_slim(python_version=\"3.13\").apt_install(\"git\").pip_install(\"fastapi[standard]\", \"codegen>=0.5.30\")\n",
+    "\n",
+    "app = CodegenApp(modal_api_key=\"\", image=image)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [
+    {
+     "name": "stdout",
+     "output_type": "stream",
+     "text": [
+      "Registered handler 'test' for event 'Issue'.\n"
+     ]
+    },
+    {
+     "ename": "TypeError",
+     "evalue": "<lambda>() takes 1 positional argument but 3 were given",
+     "output_type": "error",
+     "traceback": [
+      "\u001b[0;31m---------------------------------------------------------------------------\u001b[0m",
+      "\u001b[0;31mTypeError\u001b[0m                                 Traceback (most recent call last)",
+      "Cell \u001b[0;32mIn[12], line 1\u001b[0m\n\u001b[0;32m----> 1\u001b[0m \u001b[38;5;129;43m@app\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mlinear\u001b[49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mevent\u001b[49m\u001b[43m(\u001b[49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mIssue\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mregister_hook\u001b[49m\u001b[38;5;241;43m=\u001b[39;49m\u001b[38;5;28;43;01mlambda\u001b[39;49;00m\u001b[43m \u001b[49m\u001b[43mx\u001b[49m\u001b[43m:\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43mprint\u001b[39;49m\u001b[43m(\u001b[49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[38;5;124;43mtest statement\u001b[39;49m\u001b[38;5;124;43m\"\u001b[39;49m\u001b[43m)\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m      2\u001b[0m \u001b[38;5;28;43;01mdef\u001b[39;49;00m\u001b[38;5;250;43m \u001b[39;49m\u001b[38;5;21;43mtest\u001b[39;49m\u001b[43m(\u001b[49m\u001b[43m)\u001b[49m\u001b[43m:\u001b[49m\n\u001b[1;32m      3\u001b[0m \u001b[43m    \u001b[49m\u001b[38;5;28;43;01mpass\u001b[39;49;00m\n",
+      "File \u001b[0;32m~/dev/codegen-sdk/src/codegen/extensions/events/app.py:50\u001b[0m, in \u001b[0;36mLinear.event.<locals>.decorator\u001b[0;34m(func)\u001b[0m\n\u001b[1;32m     48\u001b[0m \u001b[38;5;66;03m# If a custom registration hook is provided, execute it.\u001b[39;00m\n\u001b[1;32m     49\u001b[0m \u001b[38;5;28;01mif\u001b[39;00m \u001b[38;5;28mcallable\u001b[39m(register_hook):\n\u001b[0;32m---> 50\u001b[0m     \u001b[43mregister_hook\u001b[49m\u001b[43m(\u001b[49m\u001b[43mevent_name\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[43mfunc\u001b[49m\u001b[43m,\u001b[49m\u001b[43m \u001b[49m\u001b[38;5;28;43mself\u001b[39;49m\u001b[38;5;241;43m.\u001b[39;49m\u001b[43mapp\u001b[49m\u001b[43m)\u001b[49m\n\u001b[1;32m     52\u001b[0m \u001b[38;5;129m@functools\u001b[39m\u001b[38;5;241m.\u001b[39mwraps(func)\n\u001b[1;32m     53\u001b[0m \u001b[38;5;28;01mdef\u001b[39;00m\u001b[38;5;250m \u001b[39m\u001b[38;5;21mwrapper\u001b[39m(\u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs):\n\u001b[1;32m     54\u001b[0m     \u001b[38;5;28;01mreturn\u001b[39;00m func(\u001b[38;5;241m*\u001b[39margs, \u001b[38;5;241m*\u001b[39m\u001b[38;5;241m*\u001b[39mkwargs)\n",
+      "\u001b[0;31mTypeError\u001b[0m: <lambda>() takes 1 positional argument but 3 were given"
+     ]
+    }
+   ],
+   "source": [
+    "@app.linear.event(\"Issue\", register_hook=lambda x,y,z: print(\"test statement\"))\n",
+    "def test():\n",
+    "    pass"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.13.0"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}