xiaxinhang/pr-agent
1
0
Fork 0
mirror of https://github.com/qodo-ai/pr-agent.git synced 2025-07-14 01:30:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

static analysis

Browse Source
This commit is contained in:
mrT23
2024-10-02 09:06:30 +03:00
parent c84d84ace2
commit 88c2b90860
2 changed files with 13 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pr_agent/static_analysis/queries/Credits.md
View File

@ -1,5 +1,5 @@
# Credits # Credits
These queries, and some of the static analysis logic, were adopted from the excellent [Aider](https://github.com/paul-gauthier/aider) project. These queries, and some of the logic, were adopted from the excellent [Aider](https://github.com/paul-gauthier/aider/blob/main/aider/queries/README.md) project.

27
pr_agent/static_analysis/src/file_summary.py
View File

@ -8,6 +8,11 @@ from grep_ast.parsers import PARSERS
from tree_sitter_languages import get_language, get_parser from tree_sitter_languages import get_language, get_parser
def filename_to_lang(filename):
file_extension = os.path.splitext(filename)[1]
lang = PARSERS.get(file_extension)
return lang
class FileSummary: class FileSummary:
""" """
This class is used to summarize the content of a file using tree-sitter queries. This class is used to summarize the content of a file using tree-sitter queries.
@ -22,9 +27,7 @@ class FileSummary:
print(f"File {fname_full_path} does not exist") print(f"File {fname_full_path} does not exist")
with open(fname_full_path, "r") as f: with open(fname_full_path, "r") as f:
code = f.read() code = f.read()
if not code.endswith("\n"): self.code = code.rstrip("\n") + "\n"
code += "\n"
self.code = code
self.parent_context = parent_context self.parent_context = parent_context
self.child_context = child_context self.child_context = child_context
self.header_max = header_max self.header_max = header_max
@ -34,7 +37,7 @@ class FileSummary:
summary_str = self.query_processing(query_results) summary_str = self.query_processing(query_results)
return summary_str return summary_str
def render_tree_sitter(self, lines_of_interest: list): def render_file_summary(self, lines_of_interest: list):
code = self.code code = self.code
fname_rel = self.fname_rel fname_rel = self.fname_rel
context = TreeContext( context = TreeContext(
@ -66,28 +69,22 @@ class FileSummary:
def_lines = [q['line'] for q in query_results if q['kind'] == "def"] def_lines = [q['line'] for q in query_results if q['kind'] == "def"]
output += "\n" output += "\n"
output += query_results[0]['fname'] + ":\n" output += query_results[0]['fname'] + ":\n"
output += self.render_tree_sitter(def_lines) output += self.render_file_summary(def_lines)
return output return output
def get_queries_scheme(self, lang) -> str: def get_queries_scheme(self, lang) -> str:
# Load the relevant queries
try: try:
# Load the relevant queries
path = os.path.join(self.main_queries_path, f"tree-sitter-{lang}-tags.scm") path = os.path.join(self.main_queries_path, f"tree-sitter-{lang}-tags.scm")
with open(path, "r") as f: with open(path, "r") as f:
return f.read() return f.read()
except KeyError: except KeyError:
return "" return ""
def filename_to_lang(self, filename):
file_extension = os.path.splitext(filename)[1]
lang = PARSERS.get(file_extension)
return lang
def get_query_results(self): def get_query_results(self):
fname_rel = self.fname_rel fname_rel = self.fname_rel
code = self.code code = self.code
lang = self.filename_to_lang(fname_rel) lang = filename_to_lang(fname_rel)
if not lang: if not lang:
return return
@ -142,10 +139,10 @@ class FileSummary:
# tokens = list(lexer.get_tokens(code)) # tokens = list(lexer.get_tokens(code))
# tokens = [token[1] for token in tokens if token[0] in Token.Name] # tokens = [token[1] for token in tokens if token[0] in Token.Name]
# #
# for token in tokens: # for t in tokens:
# result = dict( # result = dict(
# fname=fname, # fname=fname,
# name=token, # name=t,
# kind="ref", # kind="ref",
# line=-1, # line=-1,
# ) # )