diff --git a/tools/importer-rest-api-specs/components/parser/dataworkarounds/workaround_newrelic_29256.go b/tools/importer-rest-api-specs/components/parser/dataworkarounds/workaround_newrelic_29256.go new file mode 100644 index 00000000000..94901181fe0 --- /dev/null +++ b/tools/importer-rest-api-specs/components/parser/dataworkarounds/workaround_newrelic_29256.go @@ -0,0 +1,58 @@ +package dataworkarounds + +import ( + "fmt" + + "github.com/hashicorp/pandora/tools/data-api-sdk/v1/models" + importerModels "github.com/hashicorp/pandora/tools/importer-rest-api-specs/models" +) + +var _ workaround = workaroundNewRelic29256{} + +// workaroundNewRelic29256 works around the Swagger defining Identity as SystemAssignedAndUserAssigned +// when the API only supports SystemAssigned. +// Swagger Issue: https://github.com/Azure/azure-rest-api-specs/issues/29256 +type workaroundNewRelic29256 struct { +} + +func (workaroundNewRelic29256) IsApplicable(apiDefinition *importerModels.AzureApiDefinition) bool { + serviceMatches := apiDefinition.ServiceName == "NewRelic" + apiVersionMatches := apiDefinition.ApiVersion == "2022-07-01" || apiDefinition.ApiVersion == "2024-03-01" + return serviceMatches && apiVersionMatches +} + +func (workaroundNewRelic29256) Name() string { + return "NewRelic / 29256" +} + +func (workaroundNewRelic29256) Process(apiDefinition importerModels.AzureApiDefinition) (*importerModels.AzureApiDefinition, error) { + // NewRelicMonitorResource,NewRelicMonitorResourceUpdate + resource, ok := apiDefinition.Resources["Monitors"] + if !ok { + return nil, fmt.Errorf("couldn't find API Resource `Monitors`") + } + + modelsToPatch := []string{ + "NewRelicMonitorResource", + "NewRelicMonitorResourceUpdate", + } + + for _, modelName := range modelsToPatch { + model, ok := resource.Models[modelName] + if !ok { + return nil, fmt.Errorf("couldn't find Model `%s`", modelName) + } + field, ok := model.Fields["Identity"] + if !ok { + return nil, fmt.Errorf("expected the Model `%s` to have a field `Identity` but it didn't exist", modelName) + } + field.ObjectDefinition = models.SDKObjectDefinition{ + Type: models.SystemAssignedIdentitySDKObjectDefinitionType, + } + model.Fields["Identity"] = field + resource.Models[modelName] = model + } + + apiDefinition.Resources["Monitors"] = resource + return &apiDefinition, nil +} diff --git a/tools/importer-rest-api-specs/components/parser/dataworkarounds/workarounds.go b/tools/importer-rest-api-specs/components/parser/dataworkarounds/workarounds.go index 4c3b703f7ff..f11badc627a 100644 --- a/tools/importer-rest-api-specs/components/parser/dataworkarounds/workarounds.go +++ b/tools/importer-rest-api-specs/components/parser/dataworkarounds/workarounds.go @@ -25,6 +25,7 @@ var workarounds = []workaround{ workaroundLoadTest20961{}, workaroundRedis22407{}, workaroundMachineLearning25142{}, + workaroundNewRelic29256{}, workaroundOperationalinsights27524{}, workaroundRecoveryServicesSiteRecovery26680{}, workaroundStreamAnalytics27577{},