updated datasource generation to handle certain other code blocks (#14800)

Author: luckyswaminathan

mmv1/api/resource.go

@@ -227,7 +227,7 @@ type Resource struct {
 	ExcludeImport bool `yaml:"exclude_import,omitempty"`
 
 	// If true, resource should be autogenerated as a data source
-	GenerateDatasource bool `yaml:"generate_datasource,omitempty"`
+	Datasource *resource.Datasource `yaml:"datasource,omitempty"`
 
 	// If true, skip sweeper generation for this resource
 	ExcludeSweeper bool `yaml:"exclude_sweeper,omitempty"`
@@ -2078,8 +2078,31 @@ func urlContainsOnlyAllowedKeys(templateURL string, allowedKeys []string) bool {
 	return true
 }
 
-func (r Resource) ShouldGenerateSingularDataSource() bool {
-	return r.GenerateDatasource
+func (r *Resource) ShouldGenerateSingularDataSource() bool {
+
+	if r.Datasource == nil {
+		return false
+	}
+
+	return r.Datasource.Generate
+}
+
+func (r Resource) ShouldDatasourceSetLabels() bool {
+	for _, p := range r.Properties {
+		if p.Name == "labels" && p.Type == "KeyValueLabels" {
+			return true
+		}
+	}
+	return false
+}
+
+func (r Resource) ShouldDatasourceSetAnnotations() bool {
+	for _, p := range r.Properties {
+		if p.Name == "annotations" && p.Type == "KeyValueAnnotations" {
+			return true
+		}
+	}
+	return false
 }
 
 // DatasourceOptionalFields returns a list of fields from the resource's URI

mmv1/api/resource/datasource.go

@@ -0,0 +1,19 @@
+// Copyright 2024 Google Inc.
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package resource
+
+type Datasource struct {
+	// boolean to determine whether the datasource file should be generated
+	Generate bool `yaml:"generate"`
+}

mmv1/products/iap/Client.yaml

@@ -33,7 +33,8 @@ self_link: '{{brand}}/identityAwareProxyClients/{{client_id}}'
 immutable: true
 import_format:
   - '{{brand}}/identityAwareProxyClients/{{client_id}}'
-generate_datasource: true
+datasource:
+  generate: true
 timeouts:
   insert_minutes: 20
   update_minutes: 20

mmv1/templates/terraform/datasource.go.tmpl

@@ -84,6 +84,18 @@ func dataSource{{ $.ResourceName -}}Read(d *schema.ResourceData, meta interface{
 		return err
 	}
 
+  {{if $.ShouldDatasourceSetLabels}}
+	if err := tpgresource.SetDataSourceLabels(d); err != nil {
+		return err
+	}
+  {{end}}
+
+  {{if $.ShouldDatasourceSetAnnotations}}
+	if err := tpgresource.SetDataSourceAnnotations(d); err != nil {
+		return err
+	}
+  {{end}}
+
   if d.Id() == "" {
 		return fmt.Errorf("%s not found", id)
 	}