godepgraph/graphviz_lang.go at master · kisielk/godepgraph · GitHub

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
package main

import (
	"fmt"
	"go/build"
)

// graphvizPrinter implements graphPrinter for the DOT / GraphViz diagramming language.
type graphvizPrinter struct {
	ids map[string]string
}

func newGraphvizPrinter() *graphvizPrinter {
	p := new(graphvizPrinter)
	p.ids = make(map[string]string)
	return p
}

func (p *graphvizPrinter) writeHeader(hLayout bool) {
	fmt.Println("digraph godep {")
	if hLayout {
		fmt.Println(`rankdir="LR"`)
	}
	fmt.Println(`splines=ortho
nodesep=0.4
ranksep=0.8
node [shape="box",style="rounded,filled"]
edge [arrowsize="0.5"]`)
}

func (p *graphvizPrinter) writeNode(pkgName string, attrs *build.Package) {
	id := p.getId(pkgName)

	var color string
	switch {
	case attrs.Goroot:
		color = "palegreen"
	case len(attrs.CgoFiles) > 0:
		color = "darkgoldenrod1"
	case isVendored(attrs.ImportPath):
		color = "palegoldenrod"
	case hasBuildErrors(attrs):
		color = "red"
	default:
		color = "paleturquoise"
	}

	fmt.Printf("%s [label=\"%s\" color=\"%s\" URL=\"%s\" target=\"_blank\"];\n", id, pkgName, color, pkgDocsURL(pkgName))
}

func (p *graphvizPrinter) writeEdge(u string, v string) {
	uId := p.getId(u)
	vId := p.getId(v)
	fmt.Printf("%s -> %s;\n", uId, vId)
}

func (p *graphvizPrinter) getId(pkgName string) string {
	id, ok := p.ids[pkgName]
	if !ok {
		id = "\"" + pkgName + "\""
		p.ids[pkgName] = id
	}
	return id
}

func (p *graphvizPrinter) writeEnd() {
	fmt.Println("}")
}