chore: Add upgrade handler (#3705)

* Add upgrade handler

* Fix tests

* Fix lints

Author: Eric-Warehime

app/app.go

@@ -74,6 +74,7 @@ import (
 	"github.com/cosmos/gaia/v23/app/keepers"
 	"github.com/cosmos/gaia/v23/app/upgrades"
 	v23 "github.com/cosmos/gaia/v23/app/upgrades/v23"
+	upgrade "github.com/cosmos/gaia/v23/app/upgrades/v23_1_1"
 )
 
 var (
@@ -376,6 +377,13 @@ func (app *GaiaApp) Name() string { return app.BaseApp.Name() }
 
 // PreBlocker application updates every pre block
 func (app *GaiaApp) PreBlocker(ctx sdk.Context, _ *abci.RequestFinalizeBlock) (*sdk.ResponsePreBlock, error) {
+	var upgradeHeight int64 = 25283301
+	if ctx.BlockHeight() == upgradeHeight {
+		err := upgrade.AuthzGrantWasmMigrate(ctx, app.AuthzKeeper, *app.GovKeeper)
+		if err != nil {
+			return nil, err
+		}
+	}
 	return app.mm.PreBlock(ctx)
 }
 

app/upgrades/v23_1_1/constants.go

@@ -0,0 +1,10 @@
+package v23_1_1 //nolint:revive
+import "time"
+
+const (
+	IBCWasmMigrateTypeURL = "/ibc.lightclients.wasm.v1.MsgMigrateContract"
+	GranteeAddress        = "cosmos1raa4kyx5ypz75qqk3566c6slx2mw3qzs5ps5du"
+)
+
+// GrantExpiration on Apr 15th 2025, 15:00:00+00:00
+var GrantExpiration = time.Date(2025, time.April, 15, 15, 0, 0, 0, time.UTC)

app/upgrades/v23_1_1/upgrades.go

@@ -0,0 +1,34 @@
+package v23_1_1 //nolint:revive
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/x/authz"
+	authzkeeper "github.com/cosmos/cosmos-sdk/x/authz/keeper"
+	govkeeper "github.com/cosmos/cosmos-sdk/x/gov/keeper"
+)
+
+func AuthzGrantWasmMigrate(ctx sdk.Context, authzKeeper authzkeeper.Keeper, govKeeper govkeeper.Keeper) error {
+	sdkCtx := sdk.UnwrapSDKContext(ctx)
+	grant, err := authz.NewGrant(
+		sdkCtx.BlockTime(),
+		authz.NewGenericAuthorization(IBCWasmMigrateTypeURL),
+		&GrantExpiration,
+	)
+	if err != nil {
+		return err
+	}
+	sdkCtx.Logger().Info("Granting IBC Migrate Code", "granter", govKeeper.GetAuthority(), "grantee",
+		GranteeAddress)
+	resp, err := authzKeeper.Grant(ctx, &authz.MsgGrant{
+		Granter: govKeeper.GetAuthority(),
+		Grantee: GranteeAddress,
+		Grant:   grant,
+	})
+	if err != nil {
+		return err
+	}
+	if resp != nil {
+		sdkCtx.Logger().Info("Authz Keeper Grant", "response", resp.String())
+	}
+	return nil
+}

app/upgrades/v23_1_1/upgrades_test.go

@@ -0,0 +1,46 @@
+package v23_1_1_test
+
+import (
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+
+	tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
+
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	"github.com/cosmos/cosmos-sdk/x/authz"
+
+	"github.com/cosmos/gaia/v23/app/helpers"
+	upgrade "github.com/cosmos/gaia/v23/app/upgrades/v23_1_1"
+)
+
+func TestGrantIBCWasmAuth(t *testing.T) {
+	gaiaApp := helpers.Setup(t)
+	ctx := gaiaApp.NewUncachedContext(true, tmproto.Header{
+		Time: time.Unix(1740829624, 0),
+	})
+
+	err := upgrade.AuthzGrantWasmMigrate(ctx, gaiaApp.AuthzKeeper, *gaiaApp.GovKeeper)
+	require.NoError(t, err)
+
+	granteeAddr, err := sdk.AccAddressFromBech32(upgrade.GranteeAddress)
+	require.NoError(t, err)
+	granterAddr, err := sdk.AccAddressFromBech32(gaiaApp.GovKeeper.GetAuthority())
+	require.NoError(t, err)
+
+	auth, _ := gaiaApp.AuthzKeeper.GetAuthorization(
+		ctx, granteeAddr,
+		granterAddr,
+		upgrade.IBCWasmMigrateTypeURL)
+	require.NotNil(t, auth)
+
+	resp, err := gaiaApp.AuthzKeeper.GranteeGrants(ctx, &authz.QueryGranteeGrantsRequest{
+		Grantee: upgrade.GranteeAddress,
+	})
+	require.NoError(t, err)
+	require.Equal(t, 1, len(resp.Grants))
+	require.Equal(t, granterAddr.String(), resp.Grants[0].Granter)
+	require.Equal(t, granteeAddr.String(), resp.Grants[0].Grantee)
+	require.Equal(t, upgrade.GrantExpiration, *resp.Grants[0].Expiration)
+}