Implementations for SubClasses

Author: Gauravpadam

ogr/services/base.py

@@ -45,6 +45,12 @@ class BaseGitProject(GitProject):
     @property
     def full_repo_name(self) -> str:
         return f"{self.namespace}/{self.repo}"
+    
+    def can_close_issue(self, username: str) -> bool:
+        return username in self.who_can_close_issue()
+    
+    def can_merge_pr(self, username: str) -> bool:
+        return username in self.who_can_merge_pr()
 
 
 class BasePullRequest(PullRequest):
@@ -80,7 +86,11 @@ def search(
     def get_statuses(self) -> list[CommitFlag]:
         commit = self.get_all_commits()[-1]
         return self.target_project.get_commit_statuses(commit)
-
+    
+    def can_close(self, username):
+        return self.status == "open" and username in self.who_can_close()
+    def can_merge(self, username) -> bool:
+        return self.status == "open" and username in self.who_can_merge()
 
 class BaseGitUser(GitUser):
     pass
@@ -97,7 +107,7 @@ def get_comments(
         return filter_comments(all_comments, filter_regex, reverse, author)
 
     def can_close(self, username: str) -> bool:
-        return username == self.author or username in self.project.who_can_close_issue()
+        return username == self.author or username in self.who_can_close()
 
 
 class BaseCommitFlag(CommitFlag):

ogr/services/github/issue.py

@@ -166,6 +166,12 @@ def comment(self, body: str) -> IssueComment:
     def close(self) -> "Issue":
         self._raw_issue.edit(state="closed")
         return self
+    
+    def who_can_close(self) -> set[str]:
+        return self.project.who_can_close_issue()
+    
+    def can_close(self, username):
+        return username == self.author or username in self.who_can_close()
 
     def add_label(self, *labels: str) -> None:
         for label in labels:

ogr/services/github/project.py

@@ -239,6 +239,9 @@ def who_can_close_issue(self) -> set[str]:
 
     def who_can_merge_pr(self) -> set[str]:
         return self.__get_collaborators()
+    
+    def can_close_issue(self, username: str) -> bool:
+        return username in self.who_can_close_issue()
 
     def can_merge_pr(self, username) -> bool:
         return (

ogr/services/github/pull_request.py

@@ -253,10 +253,34 @@ def close(self) -> "PullRequest":
     def merge(self) -> "PullRequest":
         self._raw_pr.merge()
         return self
-
+        
     def add_label(self, *labels: str) -> None:
         for label in labels:
             self._raw_pr.add_to_labels(label)
 
     def get_comment(self, comment_id: int) -> PRComment:
         return GithubPRComment(self._raw_pr.get_issue_comment(comment_id))
+        
+    def who_can_close(self) -> set[str]:
+        people_who_can_close: set[str] = set()
+
+        people_who_can_close.add(self.author)
+        project = self._target_project
+        people_who_can_close.update(project.__get_collaborators())
+
+        return people_who_can_close
+        
+    def can_close(self, username):
+        return username == self.author or username in self.who_can_close()
+    
+    def who_can_merge(self) -> set[str]:
+        people_who_can_merge: set[str] = set()
+        project = self._target_project
+        people_who_can_merge.update(project.__get_collaborators())
+
+        return people_who_can_merge
+    
+    def can_merge(self, username):
+        return username in self.who_can_merge()
+    
+

ogr/services/gitlab/issue.py

@@ -183,3 +183,6 @@ def add_assignee(self, *assignees: str) -> None:
 
     def get_comment(self, comment_id: int) -> IssueComment:
         return GitlabIssueComment(self._raw_issue.notes.get(comment_id))
+    
+    def who_can_close(self) -> set[str]:
+        return self.project.who_can_close_issue()

ogr/services/gitlab/project.py

@@ -184,6 +184,9 @@ def who_can_merge_pr(self) -> set[str]:
             ),
         )
 
+    def can_close_issue(self, username):
+        return username in self.who_can_close_issue()
+
     def can_merge_pr(self, username) -> bool:
         return username in self.who_can_merge_pr()
 

ogr/services/gitlab/pull_request.py

@@ -299,3 +299,25 @@ def add_label(self, *labels: str) -> None:
 
     def get_comment(self, comment_id: int) -> PRComment:
         return GitlabPRComment(self._raw_pr.notes.get(comment_id))
+    
+    def who_can_close(self) -> set[str]:
+        people_who_can_close: set[str] = set()
+        people_who_can_close.add(self.author)
+
+        project = self._target_project
+        people_who_can_close.update(set(project.get_owners()))
+
+        return people_who_can_close
+    
+    def can_close(self, username):
+        return username in self.who_can_close()
+    
+    def who_can_merge(self) -> set[str]:
+        people_who_can_merge: set[str] = set()
+        project = self._target_project
+        people_who_can_merge.update(set(project.get_owners()))
+
+        return people_who_can_merge
+    
+    def can_merge(self, username):
+        return username in self.who_can_merge()

ogr/services/pagure/issue.py

@@ -241,3 +241,9 @@ def get_comment(self, comment_id: int) -> IssueComment:
                 method="GET",
             ),
         )
+    
+    def who_can_close_issue(self) -> set[str]:
+        return self.project.who_can_close_issue()
+    
+    
+

ogr/services/pagure/project.py

@@ -235,7 +235,10 @@ def which_groups_can_merge_pr(self) -> set[str]:
         groups.update(project["access_groups"]["admin"])
         groups.update(project["access_groups"]["commit"])
         return groups
-
+    
+    def can_close_issue(self, username):
+        return username in self.who_can_close_issue()
+        
     def can_merge_pr(self, username) -> bool:
         accounts_that_can_merge_pr = self.who_can_merge_pr()
 

ogr/services/pagure/pull_request.py

@@ -394,3 +394,20 @@ def get_comment(self, comment_id: int) -> PRComment:
             f"No comment with id#{comment_id} in PR#{self.id} found.",
             response_code=404,
         )
+    
+    def who_can_close(self) -> set[str]:
+        people_who_can_close: set[str] = set()
+        people_who_can_close.add(self.author)
+
+        project: ogr_pagure.PagureProject = self.target_project
+        people_who_can_close.update(project.get_owners())
+        return people_who_can_close
+
+    def can_close(self, username):
+        return username in self.who_can_close()
+    
+    def who_can_merge(self) -> set[str]:
+        return self._target_project.who_can_merge_pr()
+    
+    def can_merge(self, username):
+        return username in self.who_can_merge()