Type-assert gengo CustomArgs blindly

If it does fail, it is catastophic.  Don't bother trying to handle it.
This commit is contained in:
Tim Hockin 2024-01-15 17:00:44 -08:00
parent 0ff7dd1943
commit 875566f997
No known key found for this signature in database
3 changed files with 14 additions and 24 deletions

View File

@ -346,10 +346,8 @@ func GetTargets(context *generator.Context, arguments *args.GeneratorArgs) []gen
klog.Fatalf("Failed loading boilerplate: %v", err)
}
customArgs, ok := arguments.CustomArgs.(*clientgenargs.CustomArgs)
if !ok {
klog.Fatalf("cannot convert arguments.CustomArgs to clientgenargs.CustomArgs")
}
customArgs := arguments.CustomArgs.(*clientgenargs.CustomArgs)
includedTypesOverrides := customArgs.IncludedTypesOverrides
if err := sanitizePackagePaths(context, customArgs); err != nil {

View File

@ -271,17 +271,17 @@ func GetTargets(context *generator.Context, arguments *args.GeneratorArgs) []gen
}
}
customArgs := arguments.CustomArgs.(*conversionargs.CustomArgs)
// Make sure explicit peer-packages are added.
if customArgs, ok := arguments.CustomArgs.(*conversionargs.CustomArgs); ok {
peers := append(customArgs.BasePeerDirs, customArgs.ExtraPeerDirs...)
if expanded, err := context.FindPackages(peers...); err != nil {
klog.Fatalf("cannot find peer packages: %v", err)
} else {
otherPkgs = append(otherPkgs, expanded...)
// for each pkg, add these extras, too
for k := range pkgToPeers {
pkgToPeers[k] = append(pkgToPeers[k], expanded...)
}
peers := append(customArgs.BasePeerDirs, customArgs.ExtraPeerDirs...)
if expanded, err := context.FindPackages(peers...); err != nil {
klog.Fatalf("cannot find peer packages: %v", err)
} else {
otherPkgs = append(otherPkgs, expanded...)
// for each pkg, add these extras, too
for k := range pkgToPeers {
pkgToPeers[k] = append(pkgToPeers[k], expanded...)
}
}
@ -303,11 +303,6 @@ func GetTargets(context *generator.Context, arguments *args.GeneratorArgs) []gen
getManualConversionFunctions(context, p, manualConversions)
}
skipUnsafe := false
if customArgs, ok := arguments.CustomArgs.(*conversionargs.CustomArgs); ok {
skipUnsafe = customArgs.SkipUnsafe
}
// We are generating conversions only for packages that are explicitly
// passed as InputDir.
for _, i := range filteredInputs {
@ -326,7 +321,7 @@ func GetTargets(context *generator.Context, arguments *args.GeneratorArgs) []gen
typesPkg = context.Universe[externalTypes]
unsafeEquality := TypesEqual(memoryEquivalentTypes)
if skipUnsafe {
if customArgs.SkipUnsafe {
unsafeEquality = noEquality{}
}

View File

@ -99,10 +99,7 @@ func GetTargets(context *generator.Context, arguments *args.GeneratorArgs) []gen
klog.Fatalf("Failed loading boilerplate: %v", err)
}
customArgs, ok := arguments.CustomArgs.(*informergenargs.CustomArgs)
if !ok {
klog.Fatalf("Wrong CustomArgs type: %T", arguments.CustomArgs)
}
customArgs := arguments.CustomArgs.(*informergenargs.CustomArgs)
internalVersionOutputDir := arguments.OutputBase
internalVersionOutputPkg := customArgs.OutputPackage