Skip to content

Commit ba76cbf

Browse files
authored
Expose wrapper fields (#2)
1 parent 37479e0 commit ba76cbf

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

filegen/file.go

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ func NewFile(pkgName string, t *types.TypeInfo, db *types.TypeDB) *FileCreate {
3838
// inner [ElementType]
3939
// errorTransformer ErrorTransformer
4040
// }
41-
innerField := jenutils.Type(jen.Id("inner"), t.TypeId.Type)
42-
errTransformerField := jen.Id("errorTransformer").Qual("github.com/CannibalVox/errproxy", "ErrorTransformer")
41+
innerField := jenutils.Type(jen.Id("Inner"), t.TypeId.Type)
42+
errTransformerField := jen.Id("ErrorTransformer").Qual("github.com/CannibalVox/errproxy", "ErrorTransformer")
4343

4444
fileCreate.jen.Type().Id(t.RootType.RootType.WrapperTypeName()).Struct(innerField, errTransformerField)
4545

@@ -124,8 +124,8 @@ func (f *FileCreate) AppendType(t *types.TypeInfo) {
124124
// }
125125
g.Return(structAssign.Id(t.TypeId.WrapperTypeName()).Values(
126126
jen.DictFunc(func(d jen.Dict) {
127-
d[jen.Id("inner")] = innerAssign.Id("inner")
128-
d[jen.Id("errorTransformer")] = jen.Id("errorTransformer")
127+
d[jen.Id("Inner")] = innerAssign.Id("inner")
128+
d[jen.Id("ErrorTransformer")] = jen.Id("errorTransformer")
129129
}),
130130
))
131131
})
@@ -209,7 +209,7 @@ func (f *FileCreate) wrapMethod(t *types.TypeInfo, methodInfo *gotypes.Selection
209209
callLine.Op(":=")
210210
}
211211

212-
callLine.Id(receiverName).Dot("inner").Dot(methodInfo.Obj().Name()).CallFunc(func(g *jen.Group) {
212+
callLine.Id(receiverName).Dot("Inner").Dot(methodInfo.Obj().Name()).CallFunc(func(g *jen.Group) {
213213
for i := 0; i < sig.Params().Len(); i++ {
214214
param := sig.Params().At(i)
215215

@@ -218,9 +218,9 @@ func (f *FileCreate) wrapMethod(t *types.TypeInfo, methodInfo *gotypes.Selection
218218
var paramCodes *jen.Statement
219219
doWrap, paramTypeInfo := f.requiresWrap(param.Type(), types.WrapStatusHard)
220220
if doWrap && paramTypeInfo.TypeId.PointerDepth > 0 {
221-
paramCodes = g.Op("&").Id(paramVal).Dot("inner")
221+
paramCodes = g.Op("&").Id(paramVal).Dot("Inner")
222222
} else if doWrap {
223-
paramCodes = g.Id(paramVal).Dot("inner")
223+
paramCodes = g.Id(paramVal).Dot("Inner")
224224
} else {
225225
paramCodes = g.Id(paramVal)
226226
}
@@ -243,9 +243,9 @@ func (f *FileCreate) wrapMethod(t *types.TypeInfo, methodInfo *gotypes.Selection
243243
// If wrappable type, return WrapSomeType(r0)
244244
// otherwise just return r0
245245
if types.IsError(result.Type()) {
246-
g.Id(receiverName).Dot("errorTransformer").Call(retVar)
246+
g.Id(receiverName).Dot("ErrorTransformer").Call(retVar)
247247
} else if doWrap {
248-
g.Id(typeInfo.TypeId.WrapFuncName()).Call(retVar, jen.Id(receiverName).Dot("errorTransformer"))
248+
g.Id(typeInfo.TypeId.WrapFuncName()).Call(retVar, jen.Id(receiverName).Dot("ErrorTransformer"))
249249
} else {
250250
g.Add(retVar)
251251
}

0 commit comments

Comments
 (0)