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

wip

Browse source
This commit is contained in:
Paul Gauthier 2023-05-24 22:04:43 -07:00
parent 43b79b0c2d
commit a1b5206345
1 changed files with 45 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

49
aider/repomap.py
View file

@ -3,13 +3,14 @@ import json
import sys
import subprocess
import tiktoken
from collections import defaultdict
from aider import prompts
from aider import prompts, utils
# Global cache for tags
TAGS_CACHE = {}
# from aider.dump import dump
from aider.dump import dump
def to_tree(tags):
@ -170,6 +171,46 @@ class RepoMap:
if __name__ == "__main__":
fnames = sys.argv[1:]
rm = RepoMap()
res = rm.get_tags_map(sys.argv[1:])
print(res)
# res = rm.get_tags_map(fnames)
# print(res)
defines = defaultdict(set)
references = defaultdict(set)
for fname in fnames:
data = rm.run_ctags(fname)
for tag in data:
ident = tag["name"]
defines[ident].add(fname)
dump("def", fname, ident)
idents = utils.get_name_identifiers(fname)
for ident in idents:
dump("ref", fname, ident)
references[ident].add(fname)
idents = set(defines.keys()).intersection(set(references.keys()))
import graphviz
dot = graphviz.Digraph()
for fname in fnames:
dot.node(fname)
for ident in idents:
dump(ident)
for refs in references[ident]:
defs = defines[ident]
if len(defs) != 1:
continue
defs = list(defs)[0]
if refs == defs:
continue
dot.edge(refs, defs, label=ident)
print(f"{refs} -{ident}-> {defs}")
dot.render("tmp", format="png", view=True)