dearwolf/aider
1
0
Fork 0
mirror of https://github.com/Aider-AI/aider.git synced 2025-05-30 09:14:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

wip

Browse source
This commit is contained in:
Paul Gauthier 2023-05-30 14:15:29 -07:00
parent 2353e54628
commit 313968415a
1 changed files with 21 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

26
aider/repomap.py
View file

@ -13,7 +13,7 @@ from pygments.util import ClassNotFound
from aider import prompts from aider import prompts
# from aider.dump import dump # from .dump import dump
def to_tree(tags): def to_tree(tags):
@ -129,7 +129,7 @@ class RepoMap:
if filename.endswith(".md") or filename.endswith(".json"): if filename.endswith(".md") or filename.endswith(".json"):
tags.append(self.split_path(filename)) tags.append(self.split_path(filename))
continue continue
tags += self.get_tags(filename) tags += self.get_tags(filename, filenames)
if not tags: if not tags:
return return
@ -156,7 +156,15 @@ class RepoMap:
self.TAGS_CACHE[cache_key] = {"mtime": file_mtime, "data": data} self.TAGS_CACHE[cache_key] = {"mtime": file_mtime, "data": data}
return data return data
def get_tags(self, filename): def get_tags(self, filename, files=None):
if not files:
files = set()
external_references = set()
other_files = files - set([filename])
for other_file in other_files:
external_references.update(self.get_name_identifiers(other_file))
data = self.run_ctags(filename) data = self.run_ctags(filename)
tags = [] tags = []
@ -171,6 +179,9 @@ class RepoMap:
name = tag.get("name") name = tag.get("name")
signature = tag.get("signature") signature = tag.get("signature")
if name not in external_references:
continue
last = name last = name
if signature: if signature:
last += " " + signature last += " " + signature
@ -201,8 +212,13 @@ class RepoMap:
cached = set(cached) cached = set(cached)
return cached return cached
with open(fname, "r") as f: try:
content = f.read() with open(fname, "r") as f:
content = f.read()
except UnicodeDecodeError:
self.IDENT_CACHE[fname] = list()
return list()
try: try:
lexer = guess_lexer_for_filename(fname, content) lexer = guess_lexer_for_filename(fname, content)
except ClassNotFound: except ClassNotFound: