iterating over all resources

.gitignore

@@ -41,3 +41,6 @@ vendor/
 *.winfile eol=crlf
 /.vs
 node_modules
+
+#used for schema code generation but is not commited to avoid constant updates
+tools/codegen/open-api-spec.yml

tools/codegen/codespec/api_to_provider_spec_mapper.go

@@ -15,27 +15,37 @@ import (
 	"github.com/mongodb/terraform-provider-mongodbatlas/tools/codegen/openapi"
 )
 
-func ToCodeSpecModel(atlasAdminAPISpecFilePath, configPath string, resourceName SnakeCaseString) (*Model, error) {
+func ToCodeSpecModel(atlasAdminAPISpecFilePath, configPath string, resourceName *string) (*Model, error) {
 	apiSpec, err := openapi.ParseAtlasAdminAPI(atlasAdminAPISpecFilePath)
 	if err != nil {
 		return nil, fmt.Errorf("unable to parse Atlas Admin API: %v", err)
 	}
 
-	config, err := config.ParseGenConfigYAML(configPath)
+	configModel, err := config.ParseGenConfigYAML(configPath)
 	if err != nil {
 		return nil, fmt.Errorf("unable to parse config file: %v", err)
 	}
 
-	// find resource operations, schemas, etc from OAS
-	oasResource, err := getAPISpecResource(apiSpec.Model, config.Resources[resourceName.SnakeCase()], resourceName)
-	if err != nil {
-		return nil, fmt.Errorf("unable to get APISpecResource schema: %v", err)
+	resourceConfigsToIterate := configModel.Resources
+	if resourceName != nil { // only generate a specific resource
+		resourceConfigsToIterate = map[string]config.Resource{
+			*resourceName: configModel.Resources[*resourceName],
+		}
 	}
 
-	// map OAS resource model to CodeSpecModel
-	codeSpecResource := apiSpecResourceToCodeSpecModel(oasResource, config.Resources[resourceName.SnakeCase()], resourceName)
+	var results []Resource
+	for name, resourceConfig := range resourceConfigsToIterate {
+		log.Printf("Generating resource: %s", name)
+		// find resource operations, schemas, etc from OAS
+		oasResource, err := getAPISpecResource(apiSpec.Model, resourceConfig, SnakeCaseString(name))
+		if err != nil {
+			return nil, fmt.Errorf("unable to get APISpecResource schema: %v", err)
+		}
+		// map OAS resource model to CodeSpecModel
+		results = append(results, *apiSpecResourceToCodeSpecModel(oasResource, resourceConfig, SnakeCaseString(name)))
+	}
 
-	return &Model{Resources: []Resource{*codeSpecResource}}, nil
+	return &Model{Resources: results}, nil
 }
 
 func apiSpecResourceToCodeSpecModel(oasResource APISpecResource, resourceConfig config.Resource, name SnakeCaseString) *Resource {

tools/codegen/codespec/api_to_provider_spec_mapper_test.go

@@ -98,7 +98,7 @@ func TestConvertToProviderSpec(t *testing.T) {
 
 	for name, tc := range testCases {
 		t.Run(name, func(t *testing.T) {
-			result, err := codespec.ToCodeSpecModel(tc.inputOpenAPISpecPath, tc.inputConfigPath, codespec.SnakeCaseString(tc.inputResourceName))
+			result, err := codespec.ToCodeSpecModel(tc.inputOpenAPISpecPath, tc.inputConfigPath, &tc.inputResourceName)
 			require.NoError(t, err)
 			assert.Equal(t, tc.expectedResult, result, "Expected result to match the specified structure")
 		})
@@ -269,7 +269,7 @@ func TestConvertToProviderSpec_nested(t *testing.T) {
 
 	for name, tc := range testCases {
 		t.Run(name, func(t *testing.T) {
-			result, err := codespec.ToCodeSpecModel(tc.inputOpenAPISpecPath, tc.inputConfigPath, codespec.SnakeCaseString(tc.inputResourceName))
+			result, err := codespec.ToCodeSpecModel(tc.inputOpenAPISpecPath, tc.inputConfigPath, &tc.inputResourceName)
 			require.NoError(t, err)
 			assert.Equal(t, tc.expectedResult, result, "Expected result to match the specified structure")
 		})

tools/codegen/config.yml

@@ -26,10 +26,10 @@ resources:
             ],
             definition: "stringvalidator.ConflictsWith(path.MatchRoot(\"name\"))"
           }]
-        
+
         prefix_path:
           computability:
             optional: true
             computed: true
 
-      timeouts: ["create", "read", "delete"]
+      timeouts: ["create", "read", "delete"]

tools/codegen/main.go

@@ -13,24 +13,17 @@ import (
 const (
 	atlasAdminAPISpecURL = "https://raw.githubusercontent.com/mongodb/atlas-sdk-go/main/openapi/atlas-api-transformed.yaml"
 	configPath           = "tools/codegen/config.yml"
-	specFilePath         = "open-api-spec.yml"
+	specFilePath         = "tools/codegen/open-api-spec.yml"
 )
 
 func main() {
 	resourceName := getOsArg()
-	if resourceName == nil {
-		log.Fatal("No resource name provided")
-	}
-	if !codespec.ValidateSnakeCase(*resourceName) {
-		log.Fatal("Resource name must be snake case")
-	}
-	log.Printf("Resource name: %s\n", *resourceName)
 
 	if err := openapi.DownloadOpenAPISpec(atlasAdminAPISpecURL, specFilePath); err != nil {
 		log.Fatalf("an error occurred when downloading Atlas Admin API spec: %v", err)
 	}
 
-	model, err := codespec.ToCodeSpecModel(specFilePath, configPath, codespec.SnakeCaseString(*resourceName))
+	model, err := codespec.ToCodeSpecModel(specFilePath, configPath, resourceName)
 	if err != nil {
 		log.Fatalf("an error occurred while generating codespec.Model: %v", err)
 	}