-
Notifications
You must be signed in to change notification settings - Fork 0
/
keyword.go
123 lines (101 loc) · 2.94 KB
/
keyword.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
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
116
117
118
119
120
121
122
123
package main
import (
"fmt"
"go/types"
"strings"
. "github.com/dave/jennifer/jen"
"github.com/tutumagi/entitygen/attr"
)
const (
thisKeyword = "a"
attrPackageName = "github.com/tutumagi/entitygen/attr"
setParentFuncName = "SetParent"
// 实体内置的属性 id
buildinIDKey = "eid"
buildinIDSetter = "SetId"
buildinIDGetter = "GetId"
buildinPosKey = "pos"
buildinRotKey = "rot"
// 更新对象
// updateFuncName = "update"
)
// 一些预设的类型或者关键字
var (
// attr.Field
attrField = func() *Statement { return Qual(attrPackageName, "Field") }
// attr.Meta
attrMeta = func() *Statement { return Qual(attrPackageName, "Meta") }
// attr.NewMeta
attrNewMeta = func() *Statement { return Qual(attrPackageName, "NewMeta") }
attrVec3 = func() *Statement { return Qual(attrPackageName, "Vec3") }
)
// 根据 要生成的类型名字,和依赖的 attr 里面的名字生成一些预设的 statement
func aboutThisCode(
structName string,
attrTypeName string,
) (
attrType func() *Statement,
thisFn func() *Statement,
convertThis func() *Statement,
convertAttrType func(string) *Statement,
) {
// 比如 attr.Int32Map attr.StrMap attr.Slice
attrType = func() *Statement { return Qual(attrPackageName, attrTypeName) }
// 比如 a *XXXX (主要用在类方法时的 this 定义)
thisFn = func() *Statement { return Id(thisKeyword).Op("*").Id(structName) }
// 比如 (*attr.Int32Map)(a)
convertThis = func() *Statement { return convertAttrType(thisKeyword) }
// 比如 (*attr.Int32Map)(name)
convertAttrType = func(name string) *Statement { return Parens(Id("*").Add(attrType())).Parens(Id(name)) }
return
}
func EmptyCtor(typName string) string {
return fmt.Sprintf("Empty%s", typName)
}
func NormalCtor(typName string) string {
return fmt.Sprintf("New%s", typName)
}
func CopyCtor(typName string) string {
return fmt.Sprintf("Copy%s", typName)
}
// 通过原始 struct 名字,获取生成的 struct 名字
func StructTypeName(sourceTypeName string) string {
return sourceTypeName
}
// 通过原始 struct 名字,获取生成的 struct 的 meta 变量名字
func StructMetaName(srcName string, isEntity bool) string {
// return strings.ToLower(srcName) + "Meta"
result := srcName + "Meta"
if !isEntity {
return attr.LowerFirst(result)
}
return result
}
// 生成的 Map 名字 KV{Key}{Val}
func MapTypeName(v *types.Map) string {
key := strings.Title(v.Key().String())
if key == "String" {
key = "Str"
}
val := trimHeadStar(getTypString(v.Elem()))
val = strings.Title(val)
if val == "String" {
val = "Str"
}
return fmt.Sprintf("KV%s%s", key, val)
}
func SliceTypeName(v *types.Slice) string {
val := trimHeadStar(getTypString(v.Elem()))
val = strings.Title(val)
if val == "String" {
val = "Str"
}
return fmt.Sprintf("%sSlice", val)
}
// 如果是 *Desk, 则返回 Desk
func trimHeadStar(str string) string {
if strings.HasPrefix(str, "*") {
str = strings.TrimLeft(str, "*")
}
return str
}