feat: add drs version split between testnets and mainnet (#1243)

Author: artemijspavlovs

cmd/relayer/setup/setup.go

@@ -312,7 +312,7 @@ func installRelayerDependencies(
 		return err
 	}
 
-	drsInfo, err := firebaseutils.GetLatestDrsVersionCommit(drsVersion)
+	drsInfo, err := firebaseutils.GetLatestDrsVersionCommit(drsVersion, hd.Environment)
 	if err != nil {
 		pterm.Error.Println("failed to retrieve latest DRS version: ", err)
 		return err
@@ -326,6 +326,10 @@ func installRelayerDependencies(
 		raCommit = drsInfo.WasmCommit
 	}
 
+	if raCommit == "UNRELEASED" {
+		return fmt.Errorf("rollapp does not support drs version: %s", drsVersion)
+	}
+
 	rbi := dependencies.NewRollappBinaryInfo(
 		raResp.Rollapp.GenesisInfo.Bech32Prefix,
 		raCommit,

cmd/rollapp/drs/update/update.go

@@ -62,7 +62,10 @@ func Cmd() *cobra.Command {
 				pterm.Info.Println("Installing the latest version", err)
 			}
 
-			drsInfo, err := firebaseutils.GetLatestDrsVersionCommit(drsVersion)
+			drsInfo, err := firebaseutils.GetLatestDrsVersionCommit(
+				drsVersion,
+				rollerData.HubData.Environment,
+			)
 			if err != nil {
 				pterm.Error.Println("Failed to get the latest commit:", err)
 				return
@@ -76,6 +79,11 @@ func Cmd() *cobra.Command {
 				raCommit = drsInfo.WasmCommit
 			}
 
+			if raCommit == "UNRELEASED" {
+				pterm.Error.Println("rollapp does not support drs version: " + drsVersion)
+				return
+			}
+
 			// if doesn't match, take latest as the reference
 			// download the latest, build into ~/.roller/tmp
 			if raCommit[:6] == cv {

cmd/rollapp/init/init.go

@@ -188,7 +188,7 @@ func Cmd() *cobra.Command {
 				}
 
 				if !shouldSkipBinaryInstallation {
-					_, _, err = dependencies.InstallBinaries(true, raRespMock)
+					_, _, err = dependencies.InstallBinaries(true, raRespMock, env)
 					if err != nil {
 						pterm.Error.Println("failed to install binaries: ", err)
 						return
@@ -241,7 +241,7 @@ func Cmd() *cobra.Command {
 			}
 
 			start := time.Now()
-			builtDeps, _, err := dependencies.InstallBinaries(false, *raResponse)
+			builtDeps, _, err := dependencies.InstallBinaries(false, *raResponse, env)
 			if err != nil {
 				pterm.Error.Println("failed to install binaries: ", err)
 				return

utils/dependencies/dependencies.go

@@ -26,7 +26,7 @@ import (
 	"github.com/dymensionxyz/roller/utils/rollapp"
 )
 
-func InstallBinaries(withMockDA bool, raResp rollapp.ShowRollappResponse) (
+func InstallBinaries(withMockDA bool, raResp rollapp.ShowRollappResponse, env string) (
 	map[string]types.Dependency,
 	map[string]types.Dependency,
 	error,
@@ -68,7 +68,7 @@ func InstallBinaries(withMockDA bool, raResp rollapp.ShowRollappResponse) (
 
 		drsVersion = strconv.Itoa(as.RollappParams.Params.DrsVersion)
 		pterm.Info.Println("RollApp drs version from the genesis file : ", drsVersion)
-		drsInfo, err := firebaseutils.GetLatestDrsVersionCommit(drsVersion)
+		drsInfo, err := firebaseutils.GetLatestDrsVersionCommit(drsVersion, env)
 		if err != nil {
 			return nil, nil, err
 		}
@@ -80,6 +80,10 @@ func InstallBinaries(withMockDA bool, raResp rollapp.ShowRollappResponse) (
 			raCommit = drsInfo.WasmCommit
 		}
 
+		if raCommit == "UNRELEASED" {
+			return nil, nil, errors.New("rollapp does not support drs version: " + drsVersion)
+		}
+
 		pterm.Info.Println(
 			"Latest RollApp binary commit for the current DRS version: ",
 			raCommit[:6],

utils/firebase/drs.go

@@ -6,7 +6,6 @@ import (
 
 	"cloud.google.com/go/firestore"
 	firebase "firebase.google.com/go"
-	"google.golang.org/api/iterator"
 	"google.golang.org/api/option"
 )
 
@@ -20,7 +19,9 @@ type DrsVersionInfo struct {
 // GetLatestDrsVersionCommit
 // Fetch DRS version information using the nested collection path
 // Path format: versions/{version}/revisions/{revision}
-func GetLatestDrsVersionCommit(drsVersion string) (*DrsVersionInfo, error) {
+// Path format: /testnets/versions/{version}/revisions/{revision}/version-info
+// Path format: /mainnet/versions/{version}/revisions/{revision}/version-info
+func GetLatestDrsVersionCommit(drsVersion string, env string) (*DrsVersionInfo, error) {
 	ctx := context.Background()
 	conf := &firebase.Config{ProjectID: "drs-metadata"}
 	app, err := firebase.NewApp(ctx, conf, option.WithoutAuthentication())
@@ -34,27 +35,41 @@ func GetLatestDrsVersionCommit(drsVersion string) (*DrsVersionInfo, error) {
 	}
 	defer client.Close()
 
-	// Fetch DRS version information using the nested collection path
-	// Path format: versions/{version}/revisions/{revision}
-	drsDoc := client.Collection("versions").
-		Doc(drsVersion).
-		Collection("revisions").
-		OrderBy("timestamp", firestore.Desc).
-		Limit(1).
-		Documents(ctx)
-
-	doc, err := drsDoc.Next()
-	if err == iterator.Done {
-		return nil, err
+	var rootCollection string
+	if env == "mainnet" {
+		rootCollection = "mainnet"
+	} else {
+		rootCollection = "testnets"
 	}
-	if err != nil {
-		return nil, err
+
+	// Get the path to revisions
+	versionsDoc := client.Collection(rootCollection).Doc("versions")
+	drsVersionColl := versionsDoc.Collection(drsVersion)
+	revisionsDoc := drsVersionColl.Doc("revisions")
+
+	fmt.Println(revisionsDoc.Path)
+
+	// Find the highest revision number by trying incrementing numbers
+	var latestDoc *firestore.DocumentSnapshot
+	for i := 1; ; i++ {
+		doc, err := drsVersionColl.Doc("revisions").
+			Collection(fmt.Sprintf("%d", i)).
+			Doc("version-info").
+			Get(ctx)
+		if err != nil {
+			// If we get an error, we've gone past the last valid revision
+			break
+		}
+		latestDoc = doc
 	}
 
-	var drsInfo DrsVersionInfo
-	if err := doc.DataTo(&drsInfo); err != nil {
-		return nil, fmt.Errorf("failed to parse DRS version info: %v", err)
+	if latestDoc == nil {
+		return nil, fmt.Errorf("no version-info documents found for version %s", drsVersion)
 	}
 
-	return &drsInfo, nil
+	var info DrsVersionInfo
+	if err := latestDoc.DataTo(&info); err != nil {
+		return nil, fmt.Errorf("failed to parse DRS version info: %v", err)
+	}
+	return &info, nil
 }