From 6b9dcff5920a79ab9449969c077e43d5d268da3f Mon Sep 17 00:00:00 2001
From: tombuildsstuff <git@tom.ibuildstuff.eu>
Date: Wed, 29 May 2024 14:11:55 +0200
Subject: [PATCH] `tools/importer-rest-api-specs`: adding a workaround for
 https://github.com/Azure/azure-rest-api-specs/issues/29256

---
 .../workaround_newrelic_29256.go              | 58 +++++++++++++++++++
 .../parser/dataworkarounds/workarounds.go     |  1 +
 2 files changed, 59 insertions(+)
 create mode 100644 tools/importer-rest-api-specs/components/parser/dataworkarounds/workaround_newrelic_29256.go

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 b712c49367f..748cce2db65 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{},