func AzapiUpdateResourceMigrationV1ToV2()

in internal/services/migration/azapi_update_resource_migration_v1_to_v2.go [13:163]
133 lines of code
5 McCabe index (conditional complexity)

func AzapiUpdateResourceMigrationV1ToV2(ctx context.Context) resource.StateUpgrader {
	return resource.StateUpgrader{
		PriorSchema: &schema.Schema{
			Attributes: map[string]schema.Attribute{
				"id": schema.StringAttribute{
					Computed: true,
				},

				"name": schema.StringAttribute{
					Optional: true,
					Computed: true,
				},

				"parent_id": schema.StringAttribute{
					Optional: true,
					Computed: true,
				},

				"resource_id": schema.StringAttribute{
					Optional: true,
					Computed: true,
				},

				"type": schema.StringAttribute{
					Required: true,
				},

				"body": schema.DynamicAttribute{
					Optional: true,
					Computed: true,
				},

				"ignore_body_changes": schema.ListAttribute{
					ElementType: types.StringType,
					Optional:    true,
				},

				"ignore_casing": schema.BoolAttribute{
					Optional: true,
					Computed: true,
				},

				"ignore_missing_property": schema.BoolAttribute{
					Optional: true,
					Computed: true,
				},

				"response_export_values": schema.ListAttribute{
					ElementType: types.StringType,
					Optional:    true,
				},

				"locks": schema.ListAttribute{
					ElementType: types.StringType,
					Optional:    true,
				},

				"output": schema.DynamicAttribute{
					Computed: true,
				},
			},

			Blocks: map[string]schema.Block{
				"timeouts": timeouts.Block(ctx, timeouts.Opts{
					Create: true,
					Update: true,
					Read:   true,
					Delete: true,
				}),
			},
			Version: 0,
		},
		StateUpgrader: func(ctx context.Context, request resource.UpgradeStateRequest, response *resource.UpgradeStateResponse) {
			type OldModel struct {
				ID                    types.String   `tfsdk:"id"`
				Name                  types.String   `tfsdk:"name"`
				ParentID              types.String   `tfsdk:"parent_id"`
				ResourceID            types.String   `tfsdk:"resource_id"`
				Type                  types.String   `tfsdk:"type"`
				Body                  types.Dynamic  `tfsdk:"body"`
				IgnoreCasing          types.Bool     `tfsdk:"ignore_casing"`
				IgnoreBodyChanges     types.List     `tfsdk:"ignore_body_changes"`
				IgnoreMissingProperty types.Bool     `tfsdk:"ignore_missing_property"`
				ResponseExportValues  types.List     `tfsdk:"response_export_values"`
				Locks                 types.List     `tfsdk:"locks"`
				Output                types.Dynamic  `tfsdk:"output"`
				Timeouts              timeouts.Value `tfsdk:"timeouts"`
			}
			type newModel struct {
				ID                    types.String        `tfsdk:"id"`
				Name                  types.String        `tfsdk:"name"`
				ParentID              types.String        `tfsdk:"parent_id"`
				ResourceID            types.String        `tfsdk:"resource_id"`
				Type                  types.String        `tfsdk:"type"`
				Body                  types.Dynamic       `tfsdk:"body"`
				SensitiveBody         types.Dynamic       `tfsdk:"sensitive_body"`
				IgnoreCasing          types.Bool          `tfsdk:"ignore_casing"`
				IgnoreMissingProperty types.Bool          `tfsdk:"ignore_missing_property"`
				ResponseExportValues  types.Dynamic       `tfsdk:"response_export_values"`
				Locks                 types.List          `tfsdk:"locks"`
				Output                types.Dynamic       `tfsdk:"output"`
				Timeouts              timeouts.Value      `tfsdk:"timeouts"`
				Retry                 retry.RetryValue    `tfsdk:"retry"`
				UpdateHeaders         map[string]string   `tfsdk:"update_headers"`
				UpdateQueryParameters map[string][]string `tfsdk:"update_query_parameters"`
				ReadHeaders           map[string]string   `tfsdk:"read_headers"`
				ReadQueryParameters   map[string][]string `tfsdk:"read_query_parameters"`
			}

			var oldState OldModel
			if response.Diagnostics.Append(request.State.Get(ctx, &oldState)...); response.Diagnostics.HasError() {
				return
			}

			bodyVal, err := migrateToDynamicValue(oldState.Body)
			if err != nil {
				response.Diagnostics.AddError("failed to migrate body to dynamic value", err.Error())
				return
			}

			outputVal, err := migrateToDynamicValue(oldState.Output)
			if err != nil {
				response.Diagnostics.AddError("failed to migrate output to dynamic value", err.Error())
				return
			}

			responseExportValues := types.DynamicNull()
			if !oldState.ResponseExportValues.IsNull() {
				responseExportValues = types.DynamicValue(oldState.ResponseExportValues)
			}

			newState := newModel{
				ID:                    oldState.ID,
				Name:                  oldState.Name,
				ParentID:              oldState.ParentID,
				ResourceID:            oldState.ResourceID,
				Type:                  oldState.Type,
				Body:                  bodyVal,
				Locks:                 oldState.Locks,
				IgnoreCasing:          oldState.IgnoreCasing,
				IgnoreMissingProperty: oldState.IgnoreMissingProperty,
				ResponseExportValues:  responseExportValues,
				Output:                outputVal,
				Timeouts:              oldState.Timeouts,
				Retry:                 retry.NewRetryValueNull(),
			}

			response.Diagnostics.Append(response.State.Set(ctx, newState)...)
		},
	}
}