Commit 667c5cc0 authored by Mike Danese's avatar Mike Danese

add imports of internal apigroup

parent edab99bf
...@@ -97,6 +97,7 @@ func Packages(context *generator.Context, arguments *args.GeneratorArgs) generat ...@@ -97,6 +97,7 @@ func Packages(context *generator.Context, arguments *args.GeneratorArgs) generat
} }
var gv clientgentypes.GroupVersion var gv clientgentypes.GroupVersion
var internalGVPkg string
if isInternal(objectMeta) { if isInternal(objectMeta) {
lastSlash := strings.LastIndex(p.Path, "/") lastSlash := strings.LastIndex(p.Path, "/")
...@@ -104,10 +105,13 @@ func Packages(context *generator.Context, arguments *args.GeneratorArgs) generat ...@@ -104,10 +105,13 @@ func Packages(context *generator.Context, arguments *args.GeneratorArgs) generat
glog.Fatalf("error constructing internal group version for package %q", p.Path) glog.Fatalf("error constructing internal group version for package %q", p.Path)
} }
gv.Group = clientgentypes.Group(p.Path[lastSlash+1:]) gv.Group = clientgentypes.Group(p.Path[lastSlash+1:])
internalGVPkg = p.Path
} else { } else {
parts := strings.Split(p.Path, "/") parts := strings.Split(p.Path, "/")
gv.Group = clientgentypes.Group(parts[len(parts)-2]) gv.Group = clientgentypes.Group(parts[len(parts)-2])
gv.Version = clientgentypes.Version(parts[len(parts)-1]) gv.Version = clientgentypes.Version(parts[len(parts)-1])
internalGVPkg = strings.Join(parts[0:len(parts)-1], "/")
} }
packageList = append(packageList, &generator.DefaultPackage{ packageList = append(packageList, &generator.DefaultPackage{
...@@ -126,6 +130,7 @@ func Packages(context *generator.Context, arguments *args.GeneratorArgs) generat ...@@ -126,6 +130,7 @@ func Packages(context *generator.Context, arguments *args.GeneratorArgs) generat
}, },
outputPackage: arguments.OutputPackagePath, outputPackage: arguments.OutputPackagePath,
groupVersion: gv, groupVersion: gv,
internalGVPkg: internalGVPkg,
typeToGenerate: t, typeToGenerate: t,
imports: generator.NewImportTracker(), imports: generator.NewImportTracker(),
objectMeta: objectMeta, objectMeta: objectMeta,
...@@ -175,6 +180,7 @@ type listerGenerator struct { ...@@ -175,6 +180,7 @@ type listerGenerator struct {
generator.DefaultGen generator.DefaultGen
outputPackage string outputPackage string
groupVersion clientgentypes.GroupVersion groupVersion clientgentypes.GroupVersion
internalGVPkg string
typeToGenerate *types.Type typeToGenerate *types.Type
imports namer.ImportTracker imports namer.ImportTracker
objectMeta *types.Type objectMeta *types.Type
...@@ -206,7 +212,7 @@ func (g *listerGenerator) GenerateType(c *generator.Context, t *types.Type, w io ...@@ -206,7 +212,7 @@ func (g *listerGenerator) GenerateType(c *generator.Context, t *types.Type, w io
glog.V(5).Infof("processing type %v", t) glog.V(5).Infof("processing type %v", t)
m := map[string]interface{}{ m := map[string]interface{}{
"group": g.groupVersion.Group.String(), "Resource": c.Universe.Function(types.Name{Package: g.internalGVPkg, Name: "Resource"}),
"type": t, "type": t,
"objectMeta": g.objectMeta, "objectMeta": g.objectMeta,
} }
...@@ -295,7 +301,7 @@ func (s *$.type|private$Lister) Get(name string) (*$.type|raw$, error) { ...@@ -295,7 +301,7 @@ func (s *$.type|private$Lister) Get(name string) (*$.type|raw$, error) {
return nil, err return nil, err
} }
if !exists { if !exists {
return nil, errors.NewNotFound($.group$.Resource("$.type|lowercaseSingular$"), name) return nil, errors.NewNotFound($.Resource|raw$("$.type|lowercaseSingular$"), name)
} }
return obj.(*$.type|raw$), nil return obj.(*$.type|raw$), nil
} }
...@@ -338,7 +344,7 @@ func (s $.type|private$NamespaceLister) Get(name string) (*$.type|raw$, error) { ...@@ -338,7 +344,7 @@ func (s $.type|private$NamespaceLister) Get(name string) (*$.type|raw$, error) {
return nil, err return nil, err
} }
if !exists { if !exists {
return nil, errors.NewNotFound($.group$.Resource("$.type|lowercaseSingular$"), name) return nil, errors.NewNotFound($.Resource|raw$("$.type|lowercaseSingular$"), name)
} }
return obj.(*$.type|raw$), nil return obj.(*$.type|raw$), nil
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment