astree/specs.go at master · knsh14/astree · 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
69
70
71
72
73
74
75
76
77
78
79
80
package astree

import (
	"fmt"
	"go/ast"
	"go/token"
	"io"
)

func importSpec(w io.Writer, fs *token.FileSet, parentPrefix string, prefixes []string, node *ast.ImportSpec) {
	fmt.Fprintf(w, "%s%sImportSpec\n", parentPrefix, prefixes[0])
	fmt.Fprintf(w, "%s%s%s Doc\n", parentPrefix, prefixes[1], getMiddlePrefix())
	if node.Doc != nil {
		tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Doc)
	}
	fmt.Fprintf(w, "%s%s%s Name\n", parentPrefix, prefixes[1], getMiddlePrefix())
	if node.Name != nil {
		tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Name)
	}
	fmt.Fprintf(w, "%s%s%s Path\n", parentPrefix, prefixes[1], getMiddlePrefix())
	tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Path)
	fmt.Fprintf(w, "%s%s%s Comment\n", parentPrefix, prefixes[1], getMiddlePrefix())
	if node.Comment != nil {
		tree(w, fs, parentPrefix+prefixes[1]+tailLine, getTailPrefixes(), node.Comment)
	}
	fmt.Fprintf(w, "%s%s%s EndPos = %s\n", parentPrefix, prefixes[1], getTailPrefix(), fs.Position(node.EndPos))
}

func typeSpec(w io.Writer, fs *token.FileSet, parentPrefix string, prefixes []string, node *ast.TypeSpec) {
	fmt.Fprintf(w, "%s%sTypeSpec\n", parentPrefix, prefixes[0])
	fmt.Fprintf(w, "%s%s%s Doc\n", parentPrefix, prefixes[1], getMiddlePrefix())
	if node.Doc != nil {
		tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Doc)
	}
	fmt.Fprintf(w, "%s%s%s Name\n", parentPrefix, prefixes[1], getMiddlePrefix())
	tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Name)
	fmt.Fprintf(w, "%s%s%s TypeParams\n", parentPrefix, prefixes[1], getMiddlePrefix())
	if node.TypeParams != nil {
		tree(w, fs, parentPrefix+prefixes[1]+tailLine, getTailPrefixes(), node.TypeParams)
	}
	fmt.Fprintf(w, "%s%s%s Assign = %s\n", parentPrefix, prefixes[1], getMiddlePrefix(), fs.Position(node.Assign))
	fmt.Fprintf(w, "%s%s%s Type\n", parentPrefix, prefixes[1], getMiddlePrefix())
	tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Type)
	fmt.Fprintf(w, "%s%s%s Comment\n", parentPrefix, prefixes[1], getTailPrefix())
	if node.Comment != nil {
		tree(w, fs, parentPrefix+prefixes[1]+tailLine, getTailPrefixes(), node.Comment)
	}
}

func valueSpec(w io.Writer, fs *token.FileSet, parentPrefix string, prefixes []string, node *ast.ValueSpec) {
	fmt.Fprintf(w, "%s%sValueSpec\n", parentPrefix, prefixes[0])
	fmt.Fprintf(w, "%s%s%s Doc\n", parentPrefix, prefixes[1], getMiddlePrefix())
	if node.Doc != nil {
		tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Doc)
	}
	fmt.Fprintf(w, "%s%s%s Names (length=%d)\n", parentPrefix, prefixes[1], getMiddlePrefix(), len(node.Names))
	for i := range node.Names {
		if i == len(node.Names)-1 {
			tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Names[i])
		} else {
			tree(w, fs, parentPrefix+prefixes[1]+middleLine, getMiddlePrefixes(), node.Names[i])
		}
	}
	fmt.Fprintf(w, "%s%s%s Type\n", parentPrefix, prefixes[1], getMiddlePrefix())
	if node.Type != nil {
		tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Type)
	}
	fmt.Fprintf(w, "%s%s%s Values (length=%d)\n", parentPrefix, prefixes[1], getMiddlePrefix(), len(node.Values))
	for i := range node.Values {
		if i < len(node.Values)-1 {
			tree(w, fs, parentPrefix+prefixes[1]+middleLine, getMiddlePrefixes(), node.Values[i])
		} else {
			tree(w, fs, parentPrefix+prefixes[1]+middleLine, getTailPrefixes(), node.Values[i])
		}
	}
	fmt.Fprintf(w, "%s%s%s Comment\n", parentPrefix, prefixes[1], getTailPrefix())
	if node.Comment != nil {
		tree(w, fs, parentPrefix+prefixes[1]+tailLine, getTailPrefixes(), node.Comment)
	}
}