Handle decoders when setting computed fields in post-create

Author: melinath

mmv1/products/bigqueryanalyticshub/ListingSubscription.yaml

@@ -30,6 +30,7 @@ immutable: true
 import_format:
   - 'projects/{{project}}/locations/{{location}}/subscriptions/{{subscription_id}}'
 custom_code:
+  decoder: 'templates/terraform/decoders/bigqueryanalyticshub_listing_subscription.go.tmpl'
   pre_read: 'templates/terraform/pre_read/bigqueryanalyticshub_listing_subscription.tmpl'
   pre_delete: 'templates/terraform/pre_read/bigqueryanalyticshub_listing_subscription.tmpl'
   post_import: templates/terraform/post_import/analytics_hub_subscription.go.tmpl

mmv1/templates/terraform/decoders/bigqueryanalyticshub_listing_subscription.go.tmpl

@@ -0,0 +1,5 @@
+// The response from Create nests the resource inside a "subscription" key.
+if s, ok := res["subscription"]; ok {
+  return s.(map[string]interface{}), nil
+}
+return res, nil

mmv1/templates/terraform/resource.go.tmpl

@@ -291,13 +291,25 @@ func resource{{ $.ResourceName -}}Create(d *schema.ResourceData, meta interface{
 {{- /* Set output-only resource properties from create API response (as long as Create doesn't use an async operation) */}}
 {{- /* This is necessary so that the ID is built correctly. */}}
 {{- if or (not $.GetAsync) (not (and ($.GetAsync.IsA "OpAsync") ($.GetAsync.Allow "Create"))) }}
+{{- $renderedIdFromName := "false" }}
+{{- $renderedDecoder := "false" }}
 {{- range $prop := $.GettableProperties }}
 {{-  if $.InIdFormat $prop }}
-{{-  if eq $prop.CustomFlatten "templates/terraform/custom_flatten/id_from_name.tmpl" }}
+{{-  if and ($.CustomCode.Decoder) (eq $renderedDecoder "false") }}
+    res, err = resource{{ $.ResourceName -}}Decoder(d, meta, res)
+    if err != nil {
+        return nil, err
+    }
+    if res == nil {
+            return nil, tpgresource.Fake404("decoded", "{{ $.ResourceName }}")
+    }
+{{-  end }}
+{{-  if and (eq $prop.CustomFlatten "templates/terraform/custom_flatten/id_from_name.tmpl") (eq $renderedIdFromName "false") }}
     // Setting `name` field so that `id_from_name` flattener will work properly.
     if err := d.Set("name", flatten{{ if $.NestedQuery -}}Nested{{end}}{{ $.ResourceName -}}Name(res["name"], d, config)); err != nil {
         return fmt.Errorf(`Error setting computed identity field "name": %s`, err)
     }
+    {{- $renderedIdFromName = "true" }}
 {{-  end }}
 {{-    if and $prop.Output (not $prop.IgnoreRead) }}
     if err := d.Set("{{ underscore $prop.Name -}}", flatten{{ if $.NestedQuery -}}Nested{{end}}{{ $.ResourceName -}}{{ camelize $prop.Name "upper"  -}}(res["{{ $prop.ApiName -}}"], d, config)); err != nil {