Fixes for generating data.json

- remove rke2/k3s keys so data only has required info for rke1
- decide kdm url based on the tag version

codegen/codegen.go

@@ -1,25 +1,27 @@
 package main
 
 import (
+	"encoding/json"
 	"fmt"
 	"io"
 	"net/http"
 	"os"
+	"strings"
 
+	"github.com/coreos/go-semver/semver"
 	"github.com/rancher/rke/metadata"
 )
 
 const (
-	defaultURL = "https://releases.rancher.com/kontainer-driver-metadata/dev-v2.9/data.json"
-	dataFile   = "data/data.json"
+	defaultDevURL     = "https://releases.rancher.com/kontainer-driver-metadata/dev-v2.9/data.json"
+	defaultReleaseURL = "https://releases.rancher.com/kontainer-driver-metadata/release-v2.9/data.json"
+	dataFile          = "data/data.json"
 )
 
-// Codegen fetch data.json from https://releases.rancher.com/kontainer-driver-metadata/release-v2.7/data.json and generates bindata
+// Codegen fetch data.json from defaultURL or defaultReleaseURL and generates bindata
 func main() {
-	u := os.Getenv(metadata.RancherMetadataURLEnv)
-	if u == "" {
-		u = defaultURL
-	}
+	u := getURL()
+	fmt.Printf("Reading data from %s \n", u)
 	data, err := http.Get(u)
 	if err != nil {
 		panic(fmt.Errorf("failed to fetch data.json from kontainer-driver-metadata repository"))
@@ -31,8 +33,39 @@ func main() {
 		panic(err)
 	}
 
+	var jsonData map[string]interface{}
+	err = json.Unmarshal(b, &jsonData)
+	if err != nil {
+		panic(fmt.Errorf("failed to parse JSON: %v", err))
+	}
+	// rke doesn't need info about rke2 and k3s versions
+	delete(jsonData, "rke2")
+	delete(jsonData, "k3s")
+
+	b, err = json.Marshal(jsonData)
+	if err != nil {
+		panic(fmt.Errorf("failed to marshal json data: %v", err))
+	}
+
 	fmt.Println("Writing data")
 	if err := os.WriteFile(dataFile, b, 0755); err != nil {
 		return
 	}
 }
+
+func getURL() string {
+	u := os.Getenv(metadata.RancherMetadataURLEnv)
+	if u == "" {
+		u = defaultDevURL
+		tag := os.Getenv("TAG")
+		if strings.HasPrefix(tag, "v") {
+			tag = tag[1:]
+		}
+		if v, err := semver.NewVersion(tag); err == nil {
+			if v.PreRelease == "" && v.String() != "" {
+				u = defaultReleaseURL
+			}
+		}
+	}
+	return u
+}