initial commit

Author: rohitvinnakota-codecov

webhook_handlers/constants.py

@@ -3,6 +3,7 @@ class GitHubHTTPHeaders:
     DELIVERY_TOKEN = "HTTP_X_GITHUB_DELIVERY"
     SIGNATURE = "HTTP_X_HUB_SIGNATURE"
     SIGNATURE_256 = "HTTP_X_HUB_SIGNATURE_256"
+    HOOK_INSTALLATION_TARGET_ID = "HTTP_X_GITHUB_HOOK_INSTALLATION_TARGET_ID"
 
 
 class GitHubWebhookEvents:

webhook_handlers/views/github.py

@@ -365,6 +365,9 @@ def status(self, request, *args, **kwargs):
         return Response()
 
     def pull_request(self, request, *args, **kwargs):
+        if request.headers.get(GitHubHTTPHeaders.HOOK_INSTALLATION_TARGET_ID):
+            return self.check_codecov_ai_auto_enabled_reviews(request)
+
         repo = self._get_repo(request)
 
         if not repo.active:
@@ -398,6 +401,18 @@ def pull_request(self, request, *args, **kwargs):
 
         return Response()
 
+    def check_codecov_ai_auto_enabled_reviews(self, request):
+        org = Owner.objects.get(
+            service=self.service_name,
+            service_id=request.data["organization"]["id"],
+        )
+        auto_review_enabled = org.yaml.get("ai_pr_review", {}).get("auto_review", False)
+        return Response(
+            data={
+                "auto_review_enabled": auto_review_enabled,
+            }
+        )
+
     def _decide_app_name(self, ghapp: GithubAppInstallation) -> str:
         """Possibly updated the name of a GithubAppInstallation that has been fetched from DB or created.
         Only the real default installation maybe use the name `GITHUB_APP_INSTALLATION_DEFAULT_NAME`
@@ -520,9 +535,11 @@ def _handle_installation_events(
                     AmplitudeEventPublisher().publish(
                         "App Installed",
                         {
-                            "user_ownerid": installer.ownerid
-                            if installer is not None
-                            else owner.ownerid,
+                            "user_ownerid": (
+                                installer.ownerid
+                                if installer is not None
+                                else owner.ownerid
+                            ),
                             "ownerid": owner.ownerid,
                         },
                     )