Add MergeRequestLabelGitlabEvent & PullRequestLabelHandler

PullRequestLabelHandler was renamed from PagurePullRequestLabelHandler
and reacts to MergeRequestLabelGitlabEvent (inherits from MergeRequestGitlabEvent
which now has also source_repo_branch & target_repo_branch attributes).

Creating a Bugzilla bug is now restricted also to set of allowed namespaces and branches.

packit_service/config.py

@@ -68,8 +68,6 @@ def __eq__(self, other: object) -> bool:
 
 
 class ServiceConfig(Config):
-    service_config = None
-
     def __init__(
         self,
         deployment: Deployment = Deployment.stg,
@@ -81,7 +79,9 @@ def __init__(
         fas_password: Optional[str] = "",
         bugzilla_url: str = "",
         bugzilla_api_key: str = "",
-        pr_accepted_labels: List[str] = None,
+        bugz_labels: List[str] = None,
+        bugz_namespaces: List[str] = None,
+        bugz_branches: List[str] = None,
         gitlab_webhook_tokens: List[str] = None,
         enabled_private_namespaces: Union[Set[str], List[str]] = None,
         gitlab_token_secret: str = "",
@@ -108,8 +108,14 @@ def __init__(
 
         self.bugzilla_url = bugzilla_url
         self.bugzilla_api_key = bugzilla_api_key
-        # Labels/Tags to mark a PR as accepted - handler will create a bug & attach patch from PR
-        self.pr_accepted_labels: Set[str] = set(pr_accepted_labels or ["accepted"])
+        # Labels/Tags to mark a MR as accepted - handler will create a bug & attach patch from PR
+        self.bugz_labels: Set[str] = set(bugz_labels or ["accepted"])
+        # Create bugs only for MRs against these namespaces
+        self.bugz_namespaces: Set[str] = set(
+            bugz_namespaces or ["redhat/centos-stream/src"]
+        )
+        # Create bugs only for MRs against one of these branches (regex set)
+        self.bugz_branches: Set[str] = set(bugz_branches or [r"^c8s"])
 
         # List of github users who are allowed to trigger p-s on any repository
         self.admins: Set[str] = set(admins or [])
@@ -139,6 +145,8 @@ def __init__(
         self.koji_logs_url = koji_logs_url
         self.koji_web_url = koji_web_url
 
+    service_config = None
+
     def __repr__(self):
         def hide(token: str) -> str:
             return f"{token[:1]}***{token[-1:]}" if token else ""

packit_service/schema.py

@@ -52,7 +52,9 @@ class ServiceConfigSchema(UserConfigSchema):
     validate_webhooks = fields.Bool(default=False)
     bugzilla_url = fields.String(default="")
     bugzilla_api_key = fields.String(default="")
-    pr_accepted_labels = fields.List(fields.String())
+    bugz_labels = fields.List(fields.String())
+    bugz_namespaces = fields.List(fields.String())
+    bugz_branches = fields.List(fields.String())
     admins = fields.List(fields.String())
     server_name = fields.String()
     gitlab_webhook_tokens = fields.List(fields.String())

packit_service/service/events.py

@@ -8,7 +8,7 @@
 import enum
 import logging
 from datetime import datetime, timezone
-from typing import Dict, List, Optional, Union
+from typing import Dict, List, Optional, Union, Set
 
 from ogr.abstract import GitProject
 from ogr.services.pagure import PagureProject
@@ -504,10 +504,12 @@ def __init__(
         username: str,
         object_id: int,
         object_iid: int,
-        source_repo_name: str,
         source_repo_namespace: str,
+        source_repo_name: str,
+        source_repo_branch: str,
         target_repo_namespace: str,
         target_repo_name: str,
+        target_repo_branch: str,
         project_url: str,
         commit_sha: str,
     ):
@@ -519,10 +521,12 @@ def __init__(
         self.user_login = username
         self.object_id = object_id
         self.identifier = str(object_iid)
-        self.source_repo_name = source_repo_name
         self.source_repo_namespace = source_repo_namespace
+        self.source_repo_name = source_repo_name
+        self.source_repo_branch = source_repo_branch
         self.target_repo_namespace = target_repo_namespace
         self.target_repo_name = target_repo_name
+        self.target_repo_branch = target_repo_branch
         self.project_url = project_url
         self.commit_sha = commit_sha
 
@@ -539,6 +543,31 @@ def get_base_project(self) -> GitProject:
         return fork
 
 
+class MergeRequestLabelGitlabEvent(MergeRequestGitlabEvent):
+    def __init__(
+        self,
+        mr_event: MergeRequestGitlabEvent,
+        labels_added: Set[str],
+        labels_removed: Set[str],
+    ):
+        super().__init__(
+            action=mr_event.action,
+            username=mr_event.user_login,
+            object_id=mr_event.object_id,
+            object_iid=int(mr_event.identifier),
+            source_repo_namespace=mr_event.source_repo_namespace,
+            source_repo_name=mr_event.source_repo_name,
+            source_repo_branch=mr_event.source_repo_branch,
+            target_repo_namespace=mr_event.target_repo_namespace,
+            target_repo_name=mr_event.target_repo_name,
+            target_repo_branch=mr_event.target_repo_branch,
+            project_url=mr_event.project_url,
+            commit_sha=mr_event.commit_sha,
+        )
+        self.labels_added = labels_added
+        self.labels_removed = labels_removed
+
+
 class PullRequestGithubEvent(AddPullRequestDbTrigger, AbstractGithubEvent):
     def __init__(
         self,

packit_service/worker/handlers/__init__.py

@@ -25,6 +25,7 @@
 )
 from packit_service.worker.handlers.forges_handlers import (
     GithubAppInstallationHandler,
+    PullRequestLabelHandler,
 )
 from packit_service.worker.handlers.koji_handlers import (
     KojiBuildHandler,
@@ -43,6 +44,7 @@
     CoprBuildEndHandler.__name__,
     CoprBuildStartHandler.__name__,
     GithubAppInstallationHandler.__name__,
+    PullRequestLabelHandler.__name__,
     ProposeDownstreamHandler.__name__,
     DistGitCommitHandler.__name__,
     TestingFarmHandler.__name__,

packit_service/worker/handlers/abstract.py

@@ -178,7 +178,7 @@ class TaskName(str, enum.Enum):
     propose_downstream = "task.run_propose_downstream_handler"
     koji_build = "task.run_koji_build_handler"
     distgit_commit = "task.run_distgit_commit_handler"
-    pagure_pr_label = "task.run_pagure_pr_label_handler"
+    pr_label = "task.run_pr_label_handler"
     koji_build_report = "task.run_koji_build_report_handler"
 
 

packit_service/worker/handlers/forges_handlers.py

@@ -6,6 +6,7 @@
 TODO: The build and test handlers are independent and should be moved away.
 """
 import logging
+import re
 from typing import Optional
 
 from ogr.abstract import CommitStatus, PullRequest
@@ -20,8 +21,7 @@
 )
 from packit_service.service.events import (
     InstallationEvent,
-    PullRequestLabelPagureEvent,
-    PullRequestLabelAction,
+    MergeRequestLabelGitlabEvent,
 )
 from packit_service.worker.allowlist import Allowlist
 from packit_service.worker.handlers.abstract import (
@@ -97,9 +97,9 @@ def run(self) -> TaskResults:
         return TaskResults(success=True, details={"msg": msg})
 
 
-@reacts_to(event=PullRequestLabelPagureEvent)
-class PagurePullRequestLabelHandler(JobHandler):
-    task_name = TaskName.pagure_pr_label
+@reacts_to(MergeRequestLabelGitlabEvent)
+class PullRequestLabelHandler(JobHandler):
+    task_name = TaskName.pr_label
 
     def __init__(
         self, package_config: PackageConfig, job_config: JobConfig, event: dict
@@ -109,11 +109,12 @@ def __init__(
             job_config=job_config,
             event=event,
         )
-        self.labels = set(event.get("labels"))
-        self.action = PullRequestLabelAction(event.get("action"))
-        self.base_repo_owner = event.get("base_repo_owner")
-        self.base_repo_name = event.get("base_repo_name")
-        self.base_repo_namespace = event.get("base_repo_namespace")
+        self.labels_added = event.get("labels_added")
+        self.labels_removed = event.get("labels_removed")
+
+        self.target_repo_name = event.get("target_repo_name")
+        self.target_repo_namespace = event.get("target_repo_namespace")
+        self.target_repo_branch = event.get("target_repo_branch")
 
         self.pr: PullRequest = self.project.get_pr(self.data.pr_id)
         # lazy properties
@@ -126,8 +127,8 @@ def bz_model(self) -> Optional[BugzillaModel]:
         if self._bz_model is None:
             self._bz_model = BugzillaModel.get_by_pr(
                 pr_id=self.data.pr_id,
-                namespace=self.base_repo_namespace,
-                repo_name=self.base_repo_name,
+                namespace=self.target_repo_namespace,
+                repo_name=self.target_repo_name,
                 project_url=self.data.project_url,
             )
         return self._bz_model
@@ -154,14 +155,14 @@ def _create_bug(self):
         bug_id, bug_url = self.bugzilla.create_bug(
             product="Red Hat Enterprise Linux 8",
             version="CentOS Stream",
-            component=self.base_repo_name,
+            component=self.target_repo_name,
             summary=self.pr.title,
-            description=f"Based on approved CentOS Stream pull-request: {self.pr.url}",
+            description=f"Based on approved Merge Request: {self.pr.url}",
         )
         self._bz_model = BugzillaModel.get_or_create(
             pr_id=self.data.pr_id,
-            namespace=self.base_repo_namespace,
-            repo_name=self.base_repo_name,
+            namespace=self.target_repo_namespace,
+            repo_name=self.target_repo_name,
             project_url=self.data.project_url,
             bug_id=bug_id,
             bug_url=bug_url,
@@ -186,7 +187,7 @@ def _set_status(self):
         """
         if not (self.bz_model and self.bz_model.bug_id and self.bz_model.bug_url):
             raise RuntimeError(
-                "PagurePullRequestLabelHandler._set_status(): bug_id or bug_url not set"
+                "PullRequestLabelHandler._set_status(): bug_id or bug_url not set"
             )
 
         self.status_reporter.set_status(
@@ -198,17 +199,28 @@ def _set_status(self):
 
     def run(self) -> TaskResults:
         logger.debug(
-            f"Handling labels/tags {self.labels} {self.action.value} to Pagure PR "
-            f"{self.base_repo_owner}/{self.base_repo_namespace}/"
-            f"{self.base_repo_name}/{self.data.identifier}"
+            f"Handling labels {self.labels_added} added to MR "
+            f"{self.target_repo_namespace}/{self.target_repo_name}/{self.data.identifier} "
+            f"branch {self.target_repo_branch}"
         )
-        if self.labels.intersection(self.service_config.pr_accepted_labels):
-            if not self.bz_model:
-                self._create_bug()
-            self._attach_patch()
-            self._set_status()
-        else:
+        if not self.labels_added.intersection(self.service_config.bugz_labels):
+            logger.debug(f"We accept only {self.service_config.bugz_labels} labels")
+        if not any(
+            re.match(n, self.target_repo_namespace)
+            for n in self.service_config.bugz_namespaces
+        ):
             logger.debug(
-                f"We accept only {self.service_config.pr_accepted_labels} labels/tags"
+                f"We accept only {self.service_config.bugz_namespaces} namespaces"
             )
+        if not any(
+            re.match(b, self.target_repo_branch)
+            for b in self.service_config.bugz_branches
+        ):
+            logger.debug(f"We accept only {self.service_config.bugz_branches} branches")
+
+        if not self.bz_model:
+            self._create_bug()
+        # We used to attach patch from CentOS Pagure, but don't do it on Gitlab
+        # self._attach_patch()
+        self._set_status()
         return TaskResults(success=True)

packit_service/worker/jobs.py

@@ -34,6 +34,7 @@
     GithubAppInstallationHandler,
     KojiBuildHandler,
     TestingFarmResultsHandler,
+    PullRequestLabelHandler,
 )
 from packit_service.worker.handlers.abstract import (
     JobHandler,
@@ -42,7 +43,6 @@
     MAP_REQUIRED_JOB_TYPE_TO_HANDLER,
     SUPPORTED_EVENTS_FOR_HANDLER,
 )
-from packit_service.worker.handlers.pagure_handlers import PagurePullRequestLabelHandler
 from packit_service.worker.parser import CentosEventParser, Parser
 from packit_service.worker.result import TaskResults
 from packit_service.worker.allowlist import Allowlist
@@ -373,7 +373,7 @@ def process_message(
             TestingFarmResultsHandler,
             CoprBuildStartHandler,
             CoprBuildEndHandler,
-            PagurePullRequestLabelHandler,
+            PullRequestLabelHandler,
         ]
         processing_results = None
 
@@ -385,7 +385,7 @@ def process_message(
             ).apply_async()
         # Label/Tag added event handler is run even when the job is not configured in package
         elif isinstance(event_object, PullRequestLabelPagureEvent):
-            PagurePullRequestLabelHandler.get_signature(
+            PullRequestLabelHandler.get_signature(
                 event=event_object,
                 job=None,
             ).apply_async()