-
Notifications
You must be signed in to change notification settings - Fork 0
/
method.go
78 lines (64 loc) · 1.98 KB
/
method.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
package method
import (
"fmt"
"go/token"
"strings"
"github.com/ricdeau/enki"
"github.com/ricdeau/strcase"
)
// Method methods info.
type Method struct {
StructName string `json:"structName"`
FieldName string `json:"fieldName"`
FieldType string `json:"fieldType"`
receiverVar string
receiver string
param string
}
// New new method info.
func New(structName string, fieldName string, fieldType string) *Method {
receiverVar := strings.ToLower(structName[:1])
return &Method{
StructName: structName,
FieldName: fieldName,
FieldType: fieldType,
receiverVar: receiverVar,
receiver: fmt.Sprintf("%s *%s", receiverVar, structName),
param: normalizeParamName(fieldName),
}
}
// Getter get block for method.
func (m *Method) Getter() enki.Block {
return enki.M(m.GetterName()).Receiver(m.receiver).Returns(m.FieldType).Body(
enki.Stmt().Line("return @1.@2", m.receiverVar, m.FieldName),
)
}
// GetterName name of the get method.
func (m *Method) GetterName() string {
return "Get" + strcase.GoExported(m.FieldName)
}
// GetterDefinition get method definition for interface.
func (m *Method) GetterDefinition() enki.Function {
return enki.Def("Get" + strcase.GoExported(m.FieldName)).Returns(m.FieldType)
}
// Setter set method block.
func (m *Method) Setter() enki.Block {
return enki.M(m.SetterName()).Receiver(m.receiver).Params(m.param + " " + m.FieldType).Body(
enki.Stmt().Line("@1.@2 = @3", m.receiverVar, m.FieldName, m.param),
)
}
// SetterName name of the set method.
func (m *Method) SetterName() string {
return "Set" + strcase.GoExported(m.FieldName)
}
// SetterDefinition set method definition for interface.
func (m *Method) SetterDefinition() enki.Function {
return enki.Def("Set" + strcase.GoExported(m.FieldName)).Params(m.param + " " + m.FieldType)
}
func normalizeParamName(fieldName string) string {
paramName := strcase.GoUnexported(fieldName)
if token.Lookup(paramName).IsKeyword() {
paramName = "_" + paramName
}
return paramName
}