Mgmt. track2 release tool update (Azure#17972)

* fix: generator preview judge
go through all the *.go file to find api version set sentence and judge if there exists preview API version

* feat: add version-number for refresh-v2

Author: tadelesh

eng/tools/generator/cmd/v2/common/changlogProcessor.go

@@ -5,7 +5,6 @@ package common
 
 import (
 	"encoding/json"
-	"fmt"
 	"io/ioutil"
 	"log"
 	"net/http"
@@ -53,38 +52,38 @@ func GetAllVersionTags(rpName, namespaceName string) ([]string, error) {
 	return tags, nil
 }
 
-func GetCurrentAPIVersion(packagePath string) (string, error) {
-	log.Printf("Get current release API version from '%s' ...", packagePath)
+func ContainsPreviewAPIVersion(packagePath string) (bool, error) {
+	log.Printf("Judge whether contains preview API version from '%s' ...", packagePath)
 
 	files, err := ioutil.ReadDir(packagePath)
 	if err != nil {
-		return "", err
+		return false, err
 	}
 
 	for _, file := range files {
 		if strings.HasSuffix(file.Name(), ".go") {
 			b, err := ioutil.ReadFile(path.Join(packagePath, file.Name()))
 			if err != nil {
-				return "", err
+				return false, err
 			}
 
 			lines := strings.Split(string(b), "\n")
 			for _, line := range lines {
 				if strings.Contains(line, "\"api-version\"") {
 					parts := strings.Split(line, "\"")
-					if len(parts) == 5 {
-						return parts[3], nil
+					if len(parts) == 5 && strings.Contains(parts[3], "preview") {
+						return true, nil
 					}
 				}
 			}
 		}
 	}
 
-	return "", fmt.Errorf("Cannot find API version for current release")
+	return false, nil
 }
 
-func GetPreviousVersionTag(apiVersion string, allReleases []string) string {
-	if strings.Contains(apiVersion, "preview") {
+func GetPreviousVersionTag(isCurrentPreview bool, allReleases []string) string {
+	if isCurrentPreview {
 		// for preview api, always compare with latest release
 		return allReleases[0]
 	} else {

eng/tools/generator/cmd/v2/common/generation.go

@@ -149,15 +149,12 @@ func (ctx GenerateContext) GenerateForSingleRPNamespace(generateParam *GenerateP
 			return nil, err
 		}
 
-		currentAPIVersion, err := GetCurrentAPIVersion(packagePath)
+		isCurrentPreview, err := ContainsPreviewAPIVersion(packagePath)
 		if err != nil {
 			return nil, err
 		}
-		if strings.Contains(currentAPIVersion, "preview") {
-			isCurrentPreview = true
-		}
 
-		previousVersionTag := GetPreviousVersionTag(currentAPIVersion, tags)
+		previousVersionTag := GetPreviousVersionTag(isCurrentPreview, tags)
 
 		oriExports, err = GetExportsFromTag(*ctx.SDKRepo, packagePath, previousVersionTag)
 		if err != nil {

eng/tools/generator/cmd/v2/refresh/refreshCmd.go

@@ -47,6 +47,7 @@ azure-rest-api-specs directory: the directory path of the azure-rest-api-specs w
 }
 
 type Flags struct {
+	VersionNumber       string
 	SwaggerRepo         string
 	SDKRepo             string
 	ReleaseDate         string
@@ -57,6 +58,7 @@ type Flags struct {
 }
 
 func BindFlags(flagSet *pflag.FlagSet) {
+	flagSet.String("version-number", "", "Specify the version number of this release")
 	flagSet.String("sdk-repo", "https://github.com/Azure/azure-sdk-for-go", "Specifies the sdk repo URL for generation")
 	flagSet.String("spec-repo", "https://github.com/Azure/azure-rest-api-specs", "Specifies the swagger repo URL for generation")
 	flagSet.String("release-date", "", "Specifies the release date in changelog")
@@ -68,6 +70,7 @@ func BindFlags(flagSet *pflag.FlagSet) {
 
 func ParseFlags(flagSet *pflag.FlagSet) Flags {
 	return Flags{
+		VersionNumber:       flags.GetString(flagSet, "version-number"),
 		SDKRepo:             flags.GetString(flagSet, "sdk-repo"),
 		SwaggerRepo:         flags.GetString(flagSet, "spec-repo"),
 		ReleaseDate:         flags.GetString(flagSet, "release-date"),
@@ -137,7 +140,7 @@ func (c *commandContext) execute(sdkRepoParam, specRepoParam string) error {
 				RPName:              rpName,
 				NamespaceName:       namespace.Name(),
 				SpecficPackageTitle: "",
-				SpecficVersion:      "",
+				SpecficVersion:      c.flags.VersionNumber,
 				SpecRPName:          specRpName,
 				ReleaseDate:         c.flags.ReleaseDate,
 				SkipGenerateExample: c.flags.SkipGenerateExample,