mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-07-04 21:00:40 +08:00
b2d952cafa
2. Lint 3. Inject GitHub app installation ID into GitHub provider using the settings mechanism.
83 lines
2.7 KiB
Python
83 lines
2.7 KiB
Python
from abc import ABC, abstractmethod
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class FilePatchInfo:
|
|
base_file: str
|
|
head_file: str
|
|
patch: str
|
|
filename: str
|
|
tokens: int = -1
|
|
|
|
|
|
class GitProvider(ABC):
|
|
@abstractmethod
|
|
def get_diff_files(self) -> list[FilePatchInfo]:
|
|
pass
|
|
|
|
@abstractmethod
|
|
def publish_comment(self, pr_comment: str, is_temporary: bool = False):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def remove_initial_comment(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_languages(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_pr_branch(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_user_id(self):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_pr_description(self):
|
|
pass
|
|
|
|
|
|
def get_main_pr_language(languages, files) -> str:
|
|
"""
|
|
Get the main language of the commit. Return an empty string if cannot determine.
|
|
"""
|
|
main_language_str = ""
|
|
try:
|
|
top_language = max(languages, key=languages.get).lower()
|
|
|
|
# validate that the specific commit uses the main language
|
|
extension_list = []
|
|
for file in files:
|
|
extension_list.append(file.filename.rsplit('.')[-1])
|
|
|
|
# get the most common extension
|
|
most_common_extension = max(set(extension_list), key=extension_list.count)
|
|
|
|
# look for a match. TBD: add more languages, do this systematically
|
|
if most_common_extension == 'py' and top_language == 'python' or \
|
|
most_common_extension == 'js' and top_language == 'javascript' or \
|
|
most_common_extension == 'ts' and top_language == 'typescript' or \
|
|
most_common_extension == 'go' and top_language == 'go' or \
|
|
most_common_extension == 'java' and top_language == 'java' or \
|
|
most_common_extension == 'c' and top_language == 'c' or \
|
|
most_common_extension == 'cpp' and top_language == 'c++' or \
|
|
most_common_extension == 'cs' and top_language == 'c#' or \
|
|
most_common_extension == 'swift' and top_language == 'swift' or \
|
|
most_common_extension == 'php' and top_language == 'php' or \
|
|
most_common_extension == 'rb' and top_language == 'ruby' or \
|
|
most_common_extension == 'rs' and top_language == 'rust' or \
|
|
most_common_extension == 'scala' and top_language == 'scala' or \
|
|
most_common_extension == 'kt' and top_language == 'kotlin' or \
|
|
most_common_extension == 'pl' and top_language == 'perl' or \
|
|
most_common_extension == 'swift' and top_language == 'swift':
|
|
main_language_str = top_language
|
|
|
|
except Exception:
|
|
pass
|
|
|
|
return main_language_str
|