1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
# Show a Tomo dependency graph
_USAGE := "Usage: tomodeps <files...>"
_HELP := "
tomodeps: Show a file dependency graph for Tomo source files.
$_USAGE
"
enum Dependency(File(path:Path), Module(name:Text))
func _get_file_dependencies(file:Path)->{Dependency}:
if not file:is_file():
!! Could not read file: $file
return {:Dependency}
deps := {:Dependency}
if lines := file:by_line():
for line in lines:
if line:matches($/use {..}.tm/):
file_import := Path.without_escaping(line:replace($/use {..}/, "\1")):resolved(relative_to=file)
deps:add(Dependency.File(file_import))
else if line:matches($/use {id}/):
module_name := line:replace($/use {..}/, "\1")
deps:add(Dependency.Module(module_name))
return deps
func _build_dependency_graph(dep:Dependency, dependencies:&{Dependency:{Dependency}}):
return if dependencies:has(dep)
dependencies:set(dep, {:Dependency}) # Placeholder
dep_deps := when dep is File(path):
_get_file_dependencies(path)
is Module(module):
dir := (~/.local/share/tomo/installed/$module)
module_deps := {:Dependency}
visited := {:Path}
unvisited := {f:resolved() for f in dir:files() if f:ends_with(".tm")}
while unvisited.length > 0:
file := unvisited.items[-1]
unvisited:remove(file)
visited:add(file)
for file_dep in _get_file_dependencies(file):
when file_dep is File(f):
if not visited:has(f):
unvisited:add(f)
is Module(m):
module_deps:add(file_dep)
module_deps
dependencies:set(dep, dep_deps)
for dep2 in dep_deps:
_build_dependency_graph(dep2, dependencies)
func get_dependency_graph(dep:Dependency)->{Dependency:{Dependency}}:
graph := {:Dependency:{Dependency}}
_build_dependency_graph(dep, &graph)
return graph
func _printable_name(dep:Dependency)->Text:
when dep is Module(module):
return "$(\x1b)[34;1m$module$(\x1b)[m"
is File(f):
f = f:relative()
if f:exists():
return "$(f.text_content)"
else:
return "$(\x1b)[31;1m$(f.text_content) (not found)$(\x1b)[m"
func _draw_tree(dep:Dependency, dependencies:{Dependency:{Dependency}}, already_printed:&{Dependency}, prefix="", is_last=yes):
if already_printed:has(dep):
say(prefix ++ (if is_last: "└── " else: "├── ") ++ _printable_name(dep) ++ " $\x1b[2m(recursive)$\x1b[m")
return
say(prefix ++ (if is_last: "└── " else: "├── ") ++ _printable_name(dep))
already_printed:add(dep)
child_prefix := prefix ++ (if is_last: " " else: "│ ")
children := dependencies:get(dep) or {:Dependency}
for i,child in children.items:
is_child_last := (i == children.length)
_draw_tree(child, dependencies, already_printed, child_prefix, is_child_last)
func draw_tree(dep:Dependency, dependencies:{Dependency:{Dependency}}):
printed := {:Dependency}
say(_printable_name(dep))
printed:add(dep)
deps := dependencies:get(dep) or {:Dependency}
for i,child in deps.items:
is_child_last := (i == deps.length)
_draw_tree(child, dependencies, already_printed=&printed, is_last=is_child_last)
func main(files:[Text]):
if files.length == 0:
exit("
Please provide at least one file!
$_USAGE
")
for arg in files:
if arg:matches($/{..}.tm/):
path := Path.without_escaping(arg):resolved()
dependencies := get_dependency_graph(File(path))
draw_tree(File(path), dependencies)
else if arg:matches($/{id}/):
dependencies := get_dependency_graph(Module(arg))
draw_tree(Module(arg), dependencies)
else:
say("$\x1b[2mSkipping $arg$\x1b[m")
skip
|
