[keyvault] update build.go files (#25200)

Author: gracewilcox

sdk/security/keyvault/azadmin/backup/build.go

@@ -0,0 +1,7 @@
+//go:generate go run ./testdata/generate/transforms.go
+//go:generate goimports -w .
+
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See License.txt in the project root for license information.
+
+package backup

sdk/security/keyvault/azadmin/backup/testdata/generate/transforms.go

@@ -0,0 +1,36 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See License.txt in the project root for license information.
+
+package main
+
+import (
+	"log"
+	"os"
+	"regexp"
+)
+
+// removing client prefix from types
+func regexReplace(fileName string, regex string, replace string) {
+	file, err := os.ReadFile(fileName)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	r := regexp.MustCompile(regex)
+	file = r.ReplaceAll(file, []byte(replace))
+
+	err = os.WriteFile(fileName, file, 0644)
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func main() {
+	//  modify Restore to use implementation with custom poller handler
+	regexReplace("client.go", `\[PreFullRestoreResponse\], error\) \{\s(?:.+\s)+\}`, "[PreFullRestoreResponse], error) {return client.beginPreFullRestore(ctx, preRestoreOperationParameters, options)}")
+	regexReplace("client.go", `\[FullRestoreResponse\], error\) \{\s(?:.+\s)+\}`, "[FullRestoreResponse], error) {return client.beginFullRestore(ctx, restoreBlobDetails, options)}")
+	regexReplace("client.go", `\[SelectiveKeyRestoreResponse\], error\) \{\s(?:.+\s)+\}`, "[SelectiveKeyRestoreResponse], error) {return client.beginSelectiveKeyRestore(ctx, keyName, restoreBlobDetails, options)}")
+
+	// replace Error with ErrorInfo
+	regexReplace("models.go", `Error \*string`, "Error *ErrorInfo")
+}

sdk/security/keyvault/azadmin/internal/generate/transforms.go

@@ -0,0 +1,7 @@
+//go:generate go run ./testdata/generate/transforms.go
+//go:generate goimports -w .
+
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See License.txt in the project root for license information.
+
+package rbac

sdk/security/keyvault/azadmin/rbac/build.go

@@ -0,0 +1,33 @@
+// Copyright (c) Microsoft Corporation. All rights reserved.
+// Licensed under the MIT License. See License.txt in the project root for license information.
+
+package main
+
+import (
+	"log"
+	"os"
+	"regexp"
+)
+
+// removing client prefix from types
+func regexReplace(fileName string, regex string, replace string) {
+	file, err := os.ReadFile(fileName)
+	if err != nil {
+		log.Fatal(err)
+	}
+
+	r := regexp.MustCompile(regex)
+	file = r.ReplaceAll(file, []byte(replace))
+
+	err = os.WriteFile(fileName, file, 0644)
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func main() {
+	// fix up the scope parameter for fakes
+	regexReplace("fake/server.go", `\, scopeParam\,`, ", rbac.RoleScope(`/`+scopeParam),")
+	regexReplace("fake/server.go", `\(scopeParam\, `, "(rbac.RoleScope(`/`+scopeParam), ")
+	regexReplace("fake/server.go", `(\(\?P<scope\>(.*?)\))`, `?$1?`)
+}

sdk/security/keyvault/azadmin/rbac/testdata/generate/transforms.go

@@ -1,5 +1,4 @@
-//go:generate tsp-client update
-//go:generate go run internal/transforms.go
+//go:generate go run testdata/generate/transforms.go
 //go:generate goimports -w .
 
 // Copyright (c) Microsoft Corporation. All rights reserved.

sdk/security/keyvault/azcertificates/build.go

@@ -35,16 +35,8 @@ func main() {
 	regexReplace("models.go", `\sKID \*string(\s+.*)`, "KID *ID$1")
 	regexReplace("models.go", `\sSID \*string(\s+.*)`, "SID *ID$1")
 
-	// remove the DeletionRecoveryLevel type
-	regexReplace("models.go", "DeletionRecoveryLevel", "string")
-	regexReplace("constants.go", `(?:\/\/.*\s)+type DeletionRecoveryLevel string`, "")
-	regexReplace("constants.go", `(?:\/\/.*\s)+func PossibleDeletionRecovery(?:.+\s)+\}`, "")
-	regexReplace("constants.go", `const \(\n\/\/ DeletionRecoveryLevel(?:.+\s)+\)`, "")
-
 	// replace Error with ErrorInfo
-	regexReplace("models.go", `Error \*KeyVaultErrorError`, `Error *ErrorInfo`)
-	regexReplace("models.go", `type KeyVaultErrorError struct.+\{(?:\s.+\s)+\}`, "")
-	regexReplace("models_serde.go", `(?:\/\/.*\s)+func \(\w \*?KeyVaultErrorError\).*\{\s(?:.+\s)+\}\s`, "")
+	regexReplace("models.go", `Error \*string`, `Error *ErrorInfo`)
 
 	// clean up doc comments
 	regexReplace("models.go", `For valid values\, see JsonWebKeyCurveName\.`, "")

sdk/security/keyvault/azcertificates/internal/transforms.go renamed to sdk/security/keyvault/azcertificates/testdata/generate/transforms.go

@@ -1,5 +1,4 @@
-//go:generate tsp-client update
-//go:generate go run ./internal/transforms.go
+//go:generate go run ./testdata/generate/transforms.go
 //go:generate goimports -w .
 
 // Copyright (c) Microsoft Corporation. All rights reserved.

sdk/security/keyvault/azkeys/build.go

@@ -33,21 +33,12 @@ func main() {
 	// make secret IDs a convenience type so we can add parsing methods
 	regexReplace("models.go", `\sKID \*string(\s+.*)`, "KID *ID$1")
 
-	// change type of KeyOps to KeyOperation
-	regexReplace("models.go", `KeyOps \[\]\*string`, `KeyOps []*KeyOperation`)
-
 	// delete SignatureAlgorithmRSNULL
 	regexReplace("constants.go", `.*(\bSignatureAlgorithmRSNULL\b).*`, "")
 
 	// delete KeyOperationExport
 	regexReplace("constants.go", `.*(\bKeyOperationExport\b).*`, "")
 
-	// delete DeletionRecoveryLevel
-	regexReplace("models.go", `RecoveryLevel \*DeletionRecoveryLevel`, "RecoveryLevel *string")
-	regexReplace("constants.go", `(?:\/\/.*\s)+type DeletionRecoveryLevel string`, "")
-	regexReplace("constants.go", `(?:\/\/.*\s)+func PossibleDeletionRecovery(?:.+\s)+\}`, "")
-	regexReplace("constants.go", `const \(\n\/\/ DeletionRecoveryLevel(?:.+\s)+\)`, "")
-
 	// fix up doc comments
 	regexReplace("models.go", `DeletedKeyBundle`, `DeletedKey`)
 	regexReplace("responses.go", `DeletedKeyBundle`, `DeletedKey`)

sdk/security/keyvault/azkeys/internal/transforms.go renamed to sdk/security/keyvault/azkeys/testdata/generate/transforms.go

@@ -1,6 +1,4 @@
-//go:generate tsp-client update
-//go:generate go run ./internal/transforms.go
-//go:generate rm ./constants.go
+//go:generate go run ./testdata/generate/transforms.go
 //go:generate goimports -w .
 
 // Copyright (c) Microsoft Corporation. All rights reserved.