/
package.go
66 lines (59 loc) · 1.39 KB
/
package.go
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
package main
import (
"go/build"
"go/parser"
"go/token"
"log"
)
// Package will return
type Package struct {
files []File
objects map[string]Object
gen Generator
}
func (p *Package) init() {
p.objects = map[string]Object{}
p.files = []File{}
p.gen = Generator{}
}
func (p *Package) parse(directory string, text interface{}) {
ctx := build.Default
pkg, err := ctx.ImportDir(directory, 0)
if err != nil {
log.Fatalf("cannot process directory %s: %s", directory, err)
}
var names []string
names = append(names, pkg.GoFiles...)
fs := token.NewFileSet()
for _, name := range names {
parsedFile, err := parser.ParseFile(fs, name, text, parser.ParseComments)
if err != nil {
log.Fatalf("parsing package: %s: %s", name, err)
}
file := File{
pack: p,
name: name,
fileSet: fs,
ast: parsedFile,
}
p.files = append(p.files, file)
file.process()
}
}
func (p *Package) genTopObject() string { // generates top level object with all instances of database
objs := ""
for _, obj := range p.objects {
objs += " " + obj.name + " " + obj.name + "Stored\n"
}
return "var db StoredDB = StoredDB{}\ntype StoredDB struct {\n" + objs + "}"
}
func (p *Package) generate() {
blocks := []string{}
blocks = append(blocks, p.genTopObject())
for _, obj := range p.objects {
if len(obj.fields) > 0 {
blocks = append(blocks, obj.generate())
}
}
p.gen.generate(blocks)
}