go-sylixos/elvish
3
0
Fork 0
mirror of https://github.com/go-sylixos/elvish.git synced 2024-12-01 00:33:05 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use the flag module in tools/imports-graph.elv.

Browse Source
This commit is contained in:
Qi Xiao 2021-12-31 18:49:59 +00:00
parent 5bf688cf4d
commit 32a2c4c73d
1 changed files with 62 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

120
tools/imports-graph.elv
View File

@ -1,73 +1,77 @@
use flag
use re
use str
var prefix = src.elv.sh/
var merge-clusters = $false
var imports-of = [&]
var q = [$prefix''cmd/elvish]
var seen = [&q[0]=$true]
var clusters = [&]
fn keep-if {|p| each {|x| if ($p $x) { put $x }} }
fn get {|x k def| if (has-key $x $k) { put $x[$k] } else { put $def } }
fn get-cluster {|x| put (re:find '^'(re:quote $prefix)'[^/]+/[^/]+' $x)[text] }
while (not-eq $q []) {
var next-q = []
for pkg $q {
var c = (get-cluster $pkg)
set clusters[$c] = [(all (get $clusters $c [])) $pkg]
var @imports = (
go list -json $pkg |
all (get (from-json) Imports []) |
keep-if {|x| str:has-prefix $x $prefix})
set imports-of[$pkg] = $imports
var @new-pkgs = (all $imports | keep-if {|x|
not (has-key $seen $x)
set seen[$x] = $true
})
set @next-q = (all $next-q) (all $new-pkgs)
}
set q = $next-q
}
fn node {|x| put '"'(str:trim-prefix $x $prefix)'"' }
echo 'strict digraph imports {'
echo ' rankdir = LR;'
echo ' node [shape = box, width = 1.5];'
echo ' splines = ortho;'
echo ' nodesep = 0.1;'
if $merge-clusters {
for pkg [(keys $imports-of)] {
for import $imports-of[$pkg] {
var src = (get-cluster $pkg)
var dst = (get-cluster $import)
if (not-eq $src $dst) {
echo ' '(node $src)' -> '(node $dst)';'
fn main {|&merge-clusters=$false|
var imports-of = [&]
var q = [$prefix''cmd/elvish]
var seen = [&q[0]=$true]
var clusters = [&]
while (not-eq $q []) {
var next-q = []
for pkg $q {
var c = (get-cluster $pkg)
set clusters[$c] = [(all (get $clusters $c [])) $pkg]
var @imports = (
go list -json $pkg |
all (get (from-json) Imports []) |
keep-if {|x| str:has-prefix $x $prefix})
set imports-of[$pkg] = $imports
var @new-pkgs = (all $imports | keep-if {|x|
not (has-key $seen $x)
set seen[$x] = $true
})
set @next-q = (all $next-q) (all $new-pkgs)
}
set q = $next-q
}
echo 'strict digraph imports {'
echo ' rankdir = LR;'
echo ' node [shape = box, width = 1.5];'
echo ' splines = ortho;'
echo ' nodesep = 0.1;'
if $merge-clusters {
for pkg [(keys $imports-of)] {
for import $imports-of[$pkg] {
var src = (get-cluster $pkg)
var dst = (get-cluster $import)
if (not-eq $src $dst) {
echo ' '(node $src)' -> '(node $dst)';'
}
}
}
} else {
var cluster-seq = 0
for c [(keys $clusters)] {
var pkgs = $clusters[$c]
if (<= (count $pkgs) 1) { continue }
echo ' subgraph cluster'$cluster-seq' {'
echo ' style = filled;'
echo ' color = lightgrey;'
for pkg $clusters[$c] {
echo ' '(node $pkg)';'
}
echo ' }'
set cluster-seq = (+ $cluster-seq 1)
}
for pkg [(keys $imports-of)] {
for import $imports-of[$pkg] {
echo ' '(node $pkg)' -> '(node $import)';'
}
}
}
} else {
var cluster-seq = 0
for c [(keys $clusters)] {
var pkgs = $clusters[$c]
if (<= (count $pkgs) 1) { continue }
echo ' subgraph cluster'$cluster-seq' {'
echo ' style = filled;'
echo ' color = lightgrey;'
for pkg $clusters[$c] {
echo ' '(node $pkg)';'
}
echo ' }'
set cluster-seq = (+ $cluster-seq 1)
}
for pkg [(keys $imports-of)] {
for import $imports-of[$pkg] {
echo ' '(node $pkg)' -> '(node $import)';'
}
}
echo '}'
}
echo '}'
flag:call $main~ $args