4b7985384c
* Add test case * Fix tests * Add test dependencies correctly * Feed through error handling * Include test output * Get types to line up * Add remaining test files * Update changelog * Correctly find type paths * Handle transitive objects * Handle required fields * Add feature flag for go * Add required+default test case * Don't `<any>` cast known types. * Add more flags. I realize this should really wait for PR#8400 to merge. * Add plain object to env-helper test This test fails right now. My next problem is fixing it. * Handle plain types * Handle function inputs * Fix the indentation * Handle output types correctly * Remove unnecessary `!` * Add test case * Fix tests * Add test dependencies correctly * Feed through error handling * Include test output * Get types to line up * Add remaining test files * Update changelog * Correctly find type paths * Handle transitive objects * Handle required fields * Add required+default test case * Don't `<any>` cast known types. * Add plain object to env-helper test This test fails right now. My next problem is fixing it. * Handle plain types * Handle function inputs * Fix the indentation * Handle output types correctly * Remove unnecessary `!` * Start on `genPlainObjectDefaultFunc` * Add missing change to fix test * Run tests with merge * Refactor out assign * Merge in next _index.md diff * Change method name to `Defaults` * Handle enums correctly * Another attempt at _index.md * Make module generation deterministic * Add checks for old values * Insert defaults in resources * Fix docs generation Credit to @praneetloke * Progress on adding defaults to Resource arguments * Handle resource argument defaults * Don't create defaults if disableObjectDefaults * Rename test folder * Add test for disable flag * Fix disable test * Update docs * Abstract out nil comparisons * Use reflection to test for empty values * Simplify Ptr and pulumi.Any type handling * Remove unused function * Apply defaults to functions * Update new test with master codegen * Tests + nil check
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package gen
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
|
|
"github.com/pulumi/pulumi/pkg/v3/codegen/schema"
|
|
)
|
|
|
|
// CRDTypes returns a map from each module name to a buffer containing the
|
|
// code for its generated types.
|
|
func CRDTypes(tool string, pkg *schema.Package) (map[string]*bytes.Buffer, error) {
|
|
if err := pkg.ImportLanguages(map[string]schema.Language{"go": Importer}); err != nil {
|
|
return map[string]*bytes.Buffer{}, err
|
|
}
|
|
|
|
var goPkgInfo GoPackageInfo
|
|
if goInfo, ok := pkg.Language["go"].(GoPackageInfo); ok {
|
|
goPkgInfo = goInfo
|
|
}
|
|
packages := generatePackageContextMap(tool, pkg, goPkgInfo)
|
|
|
|
var pkgMods []string
|
|
for mod := range packages {
|
|
pkgMods = append(pkgMods, mod)
|
|
}
|
|
|
|
buffers := map[string]*bytes.Buffer{}
|
|
|
|
for _, mod := range pkgMods {
|
|
pkg := packages[mod]
|
|
buffer := &bytes.Buffer{}
|
|
|
|
for _, r := range pkg.resources {
|
|
importsAndAliases := map[string]string{}
|
|
pkg.getImports(r, importsAndAliases)
|
|
pkg.genHeader(buffer, []string{"context", "reflect"}, importsAndAliases)
|
|
|
|
if err := pkg.genResource(buffer, r, goPkgInfo.GenerateResourceContainerTypes); err != nil {
|
|
return nil, fmt.Errorf("generating resource %s: %w", mod, err)
|
|
}
|
|
}
|
|
|
|
if len(pkg.types) > 0 {
|
|
for _, t := range pkg.types {
|
|
if err := pkg.genType(buffer, t); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
pkg.genTypeRegistrations(buffer, pkg.types)
|
|
}
|
|
|
|
buffers[mod] = buffer
|
|
}
|
|
|
|
return buffers, nil
|
|
}
|