refactor

pr_agent/git_providers/gitlab_provider.py

@@ -1,30 +1,54 @@
+from urllib.parse import urlparse
 import gitlab
 from typing import Optional, Tuple
 
+from pr_agent.config_loader import settings
 
-class GitLabProvider:
-    def __init__(self, merge_request_url: Optional[str] = None, personal_access_token: Optional[str] = None):
-        self.gl = gitlab.Gitlab('https://your.gitlab.com', private_token=personal_access_token)
-        self.project = None
-        self.mr_iid = None
+from .git_provider import FilePatchInfo, GitProvider
+
+
+class GitLabProvider(GitProvider):
+    def __init__(self, merge_request_url: Optional[str] = None):
+        self.gl = gitlab.Gitlab(
+            settings.get("GITLAB.URL"),
+            private_token=settings.get("GITLAB.PERSONAL_ACCESS_TOKEN")
+        )
+
+        self.id_project = None
+        self.id_mr = None
         self.mr = None
-        if merge_request_url:
-            self.set_merge_request(merge_request_url)
+        self.temp_comments = []
+
+        self.set_merge_request(merge_request_url)
+
+    @property
+    def pr(self):
+        return self.mr
 
     def set_merge_request(self, merge_request_url: str):
-        self.project, self.mr_iid = self._parse_merge_request_url(merge_request_url)
+        self.id_project, self.id_mr = self._parse_merge_request_url(merge_request_url)
         self.mr = self._get_merge_request()
 
     def get_diff_files(self) -> list[FilePatchInfo]:
-        diffs = self.mr.diffs.list()
-        diff_files = []
-        for diff in diffs:
-            # GitLab doesn't provide base and head files. Only diffs are available.
-            diff_files.append(FilePatchInfo("", "", diff['diff'], diff['new_path']))
+        diffs = self.mr.changes()['changes']
+        diff_files = [FilePatchInfo("", "", diff['diff'], diff['new_path']) for diff in diffs]
         return diff_files
 
-    def publish_comment(self, mr_comment: str):
-        self.mr.notes.create({'body': mr_comment})
+    def get_files(self):
+        return [change['new_path'] for change in self.mr.changes()['changes']]
+
+    def publish_comment(self, mr_comment: str, is_temporary: bool = False):
+        comment = self.mr.notes.create({'body': mr_comment})
+        print(comment)
+        if is_temporary:
+            self.temp_comments.append(comment)
+
+    def remove_initial_comment(self):
+        try:
+            for comment in self.temp_comments:
+                comment.delete()
+        except Exception as e:
+            logging.exception(f"Failed to remove temp comments, error: {e}")
 
     def get_title(self):
         return self.mr.title
@@ -33,25 +57,44 @@ class GitLabProvider:
         return self.mr.description
 
     def get_languages(self):
-        # GitLab does not have a direct equivalent to get_languages().
-        # An alternative could be to manually parse all the repository files and determine the language from the file extensions.
-        raise NotImplementedError
-
-    def get_main_pr_language(self) -> str:
-        # Similar issue as get_languages().
-        raise NotImplementedError
+        languages = self.gl.projects.get(self.id_project).languages()
+        return languages
 
     def get_pr_branch(self):
         return self.mr.source_branch
 
-    def get_notifications(self):
-        # GitLab doesn't provide a notifications API similar to GitHub's.
-        raise NotImplementedError
+    def get_pr_description(self):
+        return self.mr.description
 
-    @staticmethod
-    def _parse_merge_request_url(merge_request_url: str) -> Tuple[str, int]:
-        # This function will depend on your GitLab setup and URL structure
-        raise NotImplementedError
+    def _parse_merge_request_url(self, merge_request_url: str) -> Tuple[int, int]:
+        parsed_url = urlparse(merge_request_url)
+
+        path_parts = parsed_url.path.strip('/').split('/')
+        print(path_parts)
+        if path_parts[-2] != 'merge_requests':
+            raise ValueError("The provided URL does not appear to be a GitLab merge request URL")
+
+        try:
+            mr_id = int(path_parts[-1])
+        except ValueError as e:
+            raise ValueError("Unable to convert merge request ID to integer") from e
+
+        # Either user or group
+        namespace, project_name = path_parts[:2]
+
+        # Workaround for gitlab API limitation
+        project_ids = [
+            project.get_id() for project in self.gl.projects.list(search=project_name, simple=True)
+            if project.path_with_namespace == f"{namespace}/{project_name}"
+        ]
+
+        if len(project_ids) == 0:
+            raise ValueError(f"Unable to find project with name {project_name}")
+        elif len(project_ids) > 1:
+            raise ValueError(f"Multiple projects found with name {namespace}/{project_name}")
+
+        return project_ids[0], mr_id
 
     def _get_merge_request(self):
-        return self.gl.projects.get(self.project).mergerequests.get(self.mr_iid)
+        mr = self.gl.projects.get(self.id_project).mergerequests.get(self.id_mr)
+        return mr