diff --git a/pr_agent/git_providers/azuredevops_provider.py b/pr_agent/git_providers/azuredevops_provider.py index 5c2c126b..75fa6001 100644 --- a/pr_agent/git_providers/azuredevops_provider.py +++ b/pr_agent/git_providers/azuredevops_provider.py @@ -2,11 +2,9 @@ import os from typing import Optional, Tuple from urllib.parse import urlparse -from pr_agent.algo.pr_processing import find_line_number_of_relevant_line_in_file - from ..log import get_logger from ..algo.language_handler import is_valid_file -from ..algo.utils import clip_tokens, load_large_diff +from ..algo.utils import clip_tokens, find_line_number_of_relevant_line_in_file, load_large_diff from ..config_loader import get_settings from .git_provider import GitProvider from pr_agent.algo.types import EDIT_TYPE, FilePatchInfo @@ -169,58 +167,6 @@ class AzureDevopsProvider(GitProvider): return False return True - - def publish_inline_comment(self, body: str, relevant_file: str, relevant_line_in_file: str): - self.publish_inline_comments([self.create_inline_comment(body, relevant_file, relevant_line_in_file)]) - - def create_inline_comment(self, body: str, relevant_file: str, relevant_line_in_file: str, - absolute_position: int = None): - position, absolute_position = find_line_number_of_relevant_line_in_file(self.get_diff_files(), - relevant_file.strip('`'), - relevant_line_in_file, - absolute_position) - if position == -1: - if get_settings().config.verbosity_level >= 2: - get_logger().info(f"Could not find position for {relevant_file} {relevant_line_in_file}") - subject_type = "FILE" - else: - subject_type = "LINE" - path = relevant_file.strip() - return dict(body=body, path=path, position=position, absolute_position=absolute_position) if subject_type == "LINE" else {} - - def publish_inline_comments(self, comments: list[dict], disable_fallback: bool = False): - try: - for comment in comments: - comment_obj = Comment(content=comment["body"], comment_type=1) - thread = CommentThread(comments=[comment_obj], - thread_context={ - "filePath": comment["path"], - "rightFileStart": { - "line": comment["absolute_position"], - "offset": 1, # TODO - }, - "rightFileEnd": { - "line": comment["absolute_position"], - "offset": 1, - }, - }, - status="active") - self.azure_devops_client.create_thread( - comment_thread=thread, - project=self.workspace_slug, - repository_id=self.repo_slug, - pull_request_id=self.pr_num - ) - if get_settings().config.verbosity_level >= 2: - get_logger().info( - f"Published code suggestion on {self.pr_num} at {comment_obj['path']}" - ) - return True - except Exception as e: - if get_settings().config.verbosity_level >= 2: - get_logger().error(f"Failed to publish code suggestion, error: {e}") - return False - def set_pr(self, pr_url: str): self.workspace_slug, self.repo_slug, self.pr_num = self._parse_pr_url(pr_url) self.pr = self._get_pr() @@ -363,9 +309,9 @@ class AzureDevopsProvider(GitProvider): print(f"Error: {str(e)}") return [] - def publish_comment(self, pr_comment: str, is_temporary: bool = False): + def publish_comment(self, pr_comment: str, is_temporary: bool = False, thread_context=None): comment = Comment(content=pr_comment) - thread = CommentThread(comments=[comment]) + thread = CommentThread(comments=[comment], thread_context=thread_context, status=1) thread_response = self.azure_devops_client.create_thread( comment_thread=thread, project=self.workspace_slug, @@ -400,6 +346,51 @@ class AzureDevopsProvider(GitProvider): except Exception as e: get_logger().exception(f"Failed to remove temp comments, error: {e}") + def publish_inline_comment(self, body: str, relevant_file: str, relevant_line_in_file: str): + self.publish_inline_comments([self.create_inline_comment(body, relevant_file, relevant_line_in_file)]) + + + def create_inline_comment(self, body: str, relevant_file: str, relevant_line_in_file: str, + absolute_position: int = None): + position, absolute_position = find_line_number_of_relevant_line_in_file(self.get_diff_files(), + relevant_file.strip('`'), + relevant_line_in_file, + absolute_position) + if position == -1: + if get_settings().config.verbosity_level >= 2: + get_logger().info(f"Could not find position for {relevant_file} {relevant_line_in_file}") + subject_type = "FILE" + else: + subject_type = "LINE" + path = relevant_file.strip() + return dict(body=body, path=path, position=position, absolute_position=absolute_position) if subject_type == "LINE" else {} + + def publish_inline_comments(self, comments: list[dict], disable_fallback: bool = False): + overall_sucess = True + for comment in comments: + try: + self.publish_comment(comment["body"], + thread_context={ + "filePath": comment["path"], + "rightFileStart": { + "line": comment["absolute_position"], + "offset": comment["position"], + }, + "rightFileEnd": { + "line": comment["absolute_position"], + "offset": comment["position"], + }, + }) + if get_settings().config.verbosity_level >= 2: + get_logger().info( + f"Published code suggestion on {self.pr_num} at {comment['path']}" + ) + except Exception as e: + if get_settings().config.verbosity_level >= 2: + get_logger().error(f"Failed to publish code suggestion, error: {e}") + overall_sucess = False + return overall_sucess + def get_title(self): return self.pr.title