Revert "feat(CL): gamm refactor, switch to swaprouter for pool creati…

…on & swaps (merge to main 4) (#3504)" (#3513)

This reverts commit afcf5d7.

app/apptesting/gamm.go

@@ -7,8 +7,6 @@ import (
 	"github.com/osmosis-labs/osmosis/v13/x/gamm/pool-models/balancer"
 	"github.com/osmosis-labs/osmosis/v13/x/gamm/pool-models/stableswap"
 	gammtypes "github.com/osmosis-labs/osmosis/v13/x/gamm/types"
-	swaprouterkeeper "github.com/osmosis-labs/osmosis/v13/x/swaprouter"
-	swaproutertypes "github.com/osmosis-labs/osmosis/v13/x/swaprouter/types"
 )
 
 var DefaultAcctFunds sdk.Coins = sdk.NewCoins(
@@ -133,7 +131,7 @@ func (s *KeeperTestHelper) PrepareBasicStableswapPool() uint64 {
 	}
 
 	msg := stableswap.NewMsgCreateStableswapPool(s.TestAccs[0], params, DefaultStableswapLiquidity, []uint64{}, "")
-	poolId, err := s.App.SwapRouterKeeper.CreatePool(s.Ctx, msg)
+	poolId, err := s.App.GAMMKeeper.CreatePool(s.Ctx, msg)
 	s.NoError(err)
 	return poolId
 }
@@ -148,7 +146,7 @@ func (s *KeeperTestHelper) PrepareImbalancedStableswapPool() uint64 {
 	}
 
 	msg := stableswap.NewMsgCreateStableswapPool(s.TestAccs[0], params, ImbalancedStableswapLiquidity, []uint64{1, 1, 1}, "")
-	poolId, err := s.App.SwapRouterKeeper.CreatePool(s.Ctx, msg)
+	poolId, err := s.App.GAMMKeeper.CreatePool(s.Ctx, msg)
 	s.NoError(err)
 	return poolId
 }
@@ -159,7 +157,7 @@ func (s *KeeperTestHelper) PrepareBalancerPoolWithPoolParams(poolParams balancer
 	s.FundAcc(s.TestAccs[0], DefaultAcctFunds)
 
 	msg := balancer.NewMsgCreateBalancerPool(s.TestAccs[0], poolParams, DefaultPoolAssets, "")
-	poolId, err := s.App.SwapRouterKeeper.CreatePool(s.Ctx, msg)
+	poolId, err := s.App.GAMMKeeper.CreatePool(s.Ctx, msg)
 	s.NoError(err)
 	return poolId
 }
@@ -177,7 +175,7 @@ func (s *KeeperTestHelper) PrepareBalancerPoolWithPoolAsset(assets []balancer.Po
 		SwapFee: sdk.ZeroDec(),
 		ExitFee: sdk.ZeroDec(),
 	}, assets, "")
-	poolId, err := s.App.SwapRouterKeeper.CreatePool(s.Ctx, msg)
+	poolId, err := s.App.GAMMKeeper.CreatePool(s.Ctx, msg)
 	s.NoError(err)
 	return poolId
 }
@@ -207,15 +205,15 @@ func (s *KeeperTestHelper) RunBasicSwap(poolId uint64) {
 	swapIn := sdk.NewCoins(sdk.NewCoin(denoms[0], sdk.NewInt(1000)))
 	s.FundAcc(s.TestAccs[0], swapIn)
 
-	msg := swaproutertypes.MsgSwapExactAmountIn{
+	msg := gammtypes.MsgSwapExactAmountIn{
 		Sender:            s.TestAccs[0].String(),
-		Routes:            []swaproutertypes.SwapAmountInRoute{{PoolId: poolId, TokenOutDenom: denoms[1]}},
+		Routes:            []gammtypes.SwapAmountInRoute{{PoolId: poolId, TokenOutDenom: denoms[1]}},
 		TokenIn:           swapIn[0],
 		TokenOutMinAmount: sdk.ZeroInt(),
 	}
 
-	swaprouterMsgServer := swaprouterkeeper.NewMsgServerImpl(s.App.SwapRouterKeeper)
-	_, err = swaprouterMsgServer.SwapExactAmountIn(sdk.WrapSDKContext(s.Ctx), &msg)
+	gammMsgServer := gammkeeper.NewMsgServerImpl(s.App.GAMMKeeper)
+	_, err = gammMsgServer.SwapExactAmountIn(sdk.WrapSDKContext(s.Ctx), &msg)
 	s.Require().NoError(err)
 }
 

app/apptesting/test_suite.go

@@ -257,7 +257,7 @@ func (s *KeeperTestHelper) SetupGammPoolsWithBondDenomMultiplier(multipliers []s
 	// TODO: use sdk crypto instead of tendermint to generate address
 	acc1 := sdk.AccAddress(ed25519.GenPrivKey().PubKey().Address().Bytes())
 
-	params := s.App.SwapRouterKeeper.GetParams(s.Ctx)
+	params := s.App.GAMMKeeper.GetParams(s.Ctx)
 
 	pools := []gammtypes.PoolI{}
 	for index, multiplier := range multipliers {
@@ -291,7 +291,7 @@ func (s *KeeperTestHelper) SetupGammPoolsWithBondDenomMultiplier(multipliers []s
 		}
 		msg := balancer.NewMsgCreateBalancerPool(acc1, poolParams, poolAssets, defaultFutureGovernor)
 
-		poolId, err := s.App.SwapRouterKeeper.CreatePool(s.Ctx, msg)
+		poolId, err := s.App.GAMMKeeper.CreatePool(s.Ctx, msg)
 		s.Require().NoError(err)
 
 		pool, err := s.App.GAMMKeeper.GetPoolAndPoke(s.Ctx, poolId)
@@ -313,18 +313,13 @@ func (s *KeeperTestHelper) SwapAndSetSpotPrice(poolId uint64, fromAsset sdk.Coin
 	coins := sdk.Coins{sdk.NewInt64Coin(fromAsset.Denom, 100000000000000)}
 	s.FundAcc(acc1, coins)
 
-	pool, err := s.App.GAMMKeeper.GetPool(s.Ctx, poolId)
-	s.Require().NoError(err)
-	swapFee := pool.GetSwapFee(s.Ctx)
-
-	_, err = s.App.GAMMKeeper.SwapExactAmountOut(
+	_, err := s.App.GAMMKeeper.SwapExactAmountOutLegacy(
 		s.Ctx,
 		acc1,
-		pool,
+		poolId,
 		fromAsset.Denom,
 		fromAsset.Amount,
 		sdk.NewCoin(toAsset.Denom, toAsset.Amount.Quo(sdk.NewInt(4))),
-		swapFee,
 	)
 	s.Require().NoError(err)
 

app/keepers/keepers.go

@@ -245,6 +245,7 @@ func (appKeepers *AppKeepers) InitNormalKeepers(
 
 	gammKeeper := gammkeeper.NewKeeper(
 		appCodec, appKeepers.keys[gammtypes.StoreKey],
+		appKeepers.GetSubspace(gammtypes.ModuleName),
 		appKeepers.AccountKeeper, appKeepers.BankKeeper, appKeepers.DistrKeeper)
 	appKeepers.GAMMKeeper = &gammKeeper
 
@@ -258,7 +259,7 @@ func (appKeepers *AppKeepers) InitNormalKeepers(
 		appKeepers.keys[swaproutertypes.StoreKey],
 		appKeepers.GetSubspace(swaproutertypes.ModuleName),
 		appKeepers.GAMMKeeper,
-		nil, // TODO: propagate CL keeper once it is merged.
+		nil, // TODO: set CL keeper once it is merged.
 		appKeepers.BankKeeper,
 		appKeepers.AccountKeeper,
 		appKeepers.DistrKeeper,
@@ -277,7 +278,7 @@ func (appKeepers *AppKeepers) InitNormalKeepers(
 		appKeepers.AccountKeeper,
 		appKeepers.BankKeeper,
 		appKeepers.keys[txfeestypes.StoreKey],
-		appKeepers.SwapRouterKeeper,
+		appKeepers.GAMMKeeper,
 		appKeepers.GAMMKeeper,
 		txfeestypes.FeeCollectorName,
 		txfeestypes.NonNativeFeeCollectorName,
@@ -317,10 +318,10 @@ func (appKeepers *AppKeepers) InitNormalKeepers(
 		appKeepers.BankKeeper,
 		appKeepers.IncentivesKeeper,
 		appKeepers.DistrKeeper,
-		appKeepers.SwapRouterKeeper,
+		appKeepers.GAMMKeeper,
 	)
 	appKeepers.PoolIncentivesKeeper = &poolIncentivesKeeper
-	appKeepers.SwapRouterKeeper.SetPoolIncentivesKeeper(appKeepers.PoolIncentivesKeeper)
+	appKeepers.GAMMKeeper.SetPoolIncentivesKeeper(appKeepers.PoolIncentivesKeeper)
 
 	tokenFactoryKeeper := tokenfactorykeeper.NewKeeper(
 		appKeepers.keys[tokenfactorytypes.StoreKey],
@@ -551,14 +552,6 @@ func (appKeepers *AppKeepers) SetupHooks() {
 		),
 	)
 
-	appKeepers.SwapRouterKeeper.SetPoolCreationListeners(
-		swaproutertypes.NewPoolCreationListeners(
-			// insert gamm hooks receivers here
-			appKeepers.PoolIncentivesKeeper.Hooks(),
-			appKeepers.TwapKeeper.PoolCreationListeners(),
-		),
-	)
-
 	appKeepers.LockupKeeper.SetHooks(
 		lockuptypes.NewMultiLockupHooks(
 			// insert lockup hooks receivers here

app/modules.go

@@ -63,8 +63,6 @@ import (
 	poolincentivestypes "github.com/osmosis-labs/osmosis/v13/x/pool-incentives/types"
 	superfluid "github.com/osmosis-labs/osmosis/v13/x/superfluid"
 	superfluidtypes "github.com/osmosis-labs/osmosis/v13/x/superfluid/types"
-	swaproutermodule "github.com/osmosis-labs/osmosis/v13/x/swaprouter/module"
-	swaproutertypes "github.com/osmosis-labs/osmosis/v13/x/swaprouter/types"
 	"github.com/osmosis-labs/osmosis/v13/x/tokenfactory"
 	tokenfactorytypes "github.com/osmosis-labs/osmosis/v13/x/tokenfactory/types"
 	"github.com/osmosis-labs/osmosis/v13/x/twap/twapmodule"
@@ -150,7 +148,6 @@ func appModules(
 			app.EpochsKeeper,
 		),
 		tokenfactory.NewAppModule(*app.TokenFactoryKeeper, app.AccountKeeper, app.BankKeeper),
-		swaproutermodule.NewAppModule(*app.SwapRouterKeeper, app.GAMMKeeper),
 		valsetprefmodule.NewAppModule(appCodec, *app.ValidatorSetPreferenceKeeper),
 		ibc_hooks.NewAppModule(app.AccountKeeper),
 	}
@@ -212,7 +209,6 @@ func OrderInitGenesis(allModuleNames []string) []string {
 		icatypes.ModuleName,
 		gammtypes.ModuleName,
 		twaptypes.ModuleName,
-		swaproutertypes.ModuleName,
 		txfeestypes.ModuleName,
 		genutiltypes.ModuleName,
 		evidencetypes.ModuleName,

app/upgrades/v12/upgrades.go

@@ -62,10 +62,8 @@ func CreateUpgradeHandler(
 				// Change: Removed authz messages
 				sdk.MsgTypeURL(&gammtypes.MsgJoinPool{}),
 				sdk.MsgTypeURL(&gammtypes.MsgExitPool{}),
-				// N.B.: messsages were moved to another module.
-				// Leaving for historic reasons.
-				// sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountIn{}),
-				// sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountOut{}),
+				sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountIn{}),
+				sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountOut{}),
 				sdk.MsgTypeURL(&gammtypes.MsgJoinSwapExternAmountIn{}),
 				sdk.MsgTypeURL(&gammtypes.MsgJoinSwapShareAmountOut{}),
 				sdk.MsgTypeURL(&gammtypes.MsgExitSwapExternAmountOut{}),
@@ -77,7 +75,7 @@ func CreateUpgradeHandler(
 		keepers.ICAHostKeeper.SetParams(ctx, hostParams)
 
 		// Initialize TWAP state
-		latestPoolId := keepers.SwapRouterKeeper.GetNextPoolId(ctx) - 1
+		latestPoolId := keepers.GAMMKeeper.GetNextPoolId(ctx) - 1
 		err := keepers.TwapKeeper.MigrateExistingPools(ctx, latestPoolId)
 		if err != nil {
 			return nil, err

app/upgrades/v13/constants.go

@@ -4,8 +4,8 @@ import (
 	store "github.com/cosmos/cosmos-sdk/store/types"
 
 	"github.com/osmosis-labs/osmosis/v13/app/upgrades"
-	swaproutertypes "github.com/osmosis-labs/osmosis/v13/x/swaprouter/types"
-	valsetpreftypes "github.com/osmosis-labs/osmosis/v13/x/valset-pref/types"
+	valsetpreftypes "github.com/osmosis-labs/osmosis/v13/x/swaprouter/types"
+	swaproutertypes "github.com/osmosis-labs/osmosis/v13/x/valset-pref/types"
 )
 
 // UpgradeName defines the on-chain upgrade name for the Osmosis v13 upgrade.

app/upgrades/v13/upgrade_test.go

@@ -2,11 +2,8 @@ package v13_test
 
 import (
 	"fmt"
-	"reflect"
 	"testing"
 
-	gamm "github.com/osmosis-labs/osmosis/v13/x/gamm/keeper"
-
 	ibcratelimittypes "github.com/osmosis-labs/osmosis/v13/x/ibc-rate-limit/types"
 
 	"github.com/cosmos/cosmos-sdk/store/prefix"
@@ -15,7 +12,6 @@ import (
 	abci "github.com/tendermint/tendermint/abci/types"
 
 	"github.com/osmosis-labs/osmosis/v13/app/apptesting"
-	v13 "github.com/osmosis-labs/osmosis/v13/app/upgrades/v13"
 	ibc_hooks "github.com/osmosis-labs/osmosis/v13/x/ibc-hooks"
 )
 
@@ -105,44 +101,3 @@ func (suite *UpgradeTestSuite) TestUpgrade() {
 		})
 	}
 }
-
-func (suite *UpgradeTestSuite) TestMigrateNextPoolIdAndCreatePool() {
-	suite.SetupTest() // reset
-
-	const (
-		expectedNextPoolId uint64 = 3
-	)
-
-	var (
-		gammKeeperType = reflect.TypeOf(&gamm.Keeper{})
-	)
-
-	ctx := suite.Ctx
-	gammKeeper := suite.App.GAMMKeeper
-	swaprouterKeeper := suite.App.SwapRouterKeeper
-
-	// Set next pool id to given constant, because creating pools doesn't
-	// increment id on current version
-	gammKeeper.SetNextPoolId(ctx, expectedNextPoolId)
-	nextPoolId := gammKeeper.GetNextPoolId(ctx)
-	suite.Require().Equal(expectedNextPoolId, nextPoolId)
-
-	// system under test.
-	v13.MigrateNextPoolId(ctx, gammKeeper, swaprouterKeeper)
-
-	// validate swaprouter's next pool id.
-	actualNextPoolId := swaprouterKeeper.GetNextPoolId(ctx)
-	suite.Require().Equal(expectedNextPoolId, actualNextPoolId)
-
-	// create a pool after migration.
-	actualCreatedPoolId := suite.PrepareBalancerPool()
-	suite.Require().Equal(expectedNextPoolId, actualCreatedPoolId)
-
-	// validate that module route mapping has been created for each pool id.
-	for poolId := uint64(1); poolId < expectedNextPoolId; poolId++ {
-		swapModule, err := swaprouterKeeper.GetSwapModule(ctx, poolId)
-		suite.Require().NoError(err)
-
-		suite.Require().Equal(gammKeeperType, reflect.TypeOf(swapModule))
-	}
-}

app/upgrades/v13/upgrades.go

@@ -15,11 +15,8 @@ import (
 
 	"github.com/osmosis-labs/osmosis/v13/app/keepers"
 	"github.com/osmosis-labs/osmosis/v13/app/upgrades"
-	gammkeeper "github.com/osmosis-labs/osmosis/v13/x/gamm/keeper"
 	ibcratelimittypes "github.com/osmosis-labs/osmosis/v13/x/ibc-rate-limit/types"
 	lockuptypes "github.com/osmosis-labs/osmosis/v13/x/lockup/types"
-	"github.com/osmosis-labs/osmosis/v13/x/swaprouter"
-	swaproutertypes "github.com/osmosis-labs/osmosis/v13/x/swaprouter/types"
 )
 
 //go:embed rate_limiter.wasm
@@ -75,32 +72,9 @@ func CreateUpgradeHandler(
 ) upgradetypes.UpgradeHandler {
 	return func(ctx sdk.Context, plan upgradetypes.Plan, fromVM module.VersionMap) (module.VersionMap, error) {
 		keepers.LockupKeeper.SetParams(ctx, lockuptypes.DefaultParams())
-
-		keepers.SwapRouterKeeper.SetParams(ctx, swaproutertypes.DefaultParams())
-
-		// N.B: pool id in gamm is to be deprecated in the future
-		// Instead,it is moved to swaprouter.
-		migrateNextPoolId(ctx, keepers.GAMMKeeper, keepers.SwapRouterKeeper)
-
-		//  N.B.: this is done to avoid initializing genesis for swaprouter module.
-		// Otherwise, it would overwrite migrations with InitGenesis().
-		// See RunMigrations() for details.
-		fromVM[swaproutertypes.ModuleName] = 0
-
 		if err := setupRateLimiting(ctx, keepers); err != nil {
 			return nil, err
 		}
 		return mm.RunMigrations(ctx, configurator, fromVM)
 	}
 }
-
-func migrateNextPoolId(ctx sdk.Context, gammKeeper *gammkeeper.Keeper, swaprouterKeeper *swaprouter.Keeper) {
-	// N.B: pool id in gamm is to be deprecated in the future
-	// Instead,it is moved to swaprouter.
-	nextPoolId := gammKeeper.GetNextPoolId(ctx)
-	swaprouterKeeper.SetNextPoolId(ctx, nextPoolId)
-
-	for poolId := uint64(1); poolId < nextPoolId; poolId++ {
-		swaprouterKeeper.SetModuleRoute(ctx, poolId, swaproutertypes.Balancer)
-	}
-}

app/upgrades/v9/upgrades.go

@@ -67,10 +67,8 @@ func CreateUpgradeHandler(
 				sdk.MsgTypeURL(&authz.MsgRevoke{}),
 				sdk.MsgTypeURL(&gammtypes.MsgJoinPool{}),
 				sdk.MsgTypeURL(&gammtypes.MsgExitPool{}),
-				// N.B.: messsages were moved to another module.
-				// Leaving for historic reasons.
-				// sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountIn{}),
-				// sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountOut{}),
+				sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountIn{}),
+				sdk.MsgTypeURL(&gammtypes.MsgSwapExactAmountOut{}),
 				sdk.MsgTypeURL(&gammtypes.MsgJoinSwapExternAmountIn{}),
 				sdk.MsgTypeURL(&gammtypes.MsgJoinSwapShareAmountOut{}),
 				sdk.MsgTypeURL(&gammtypes.MsgExitSwapExternAmountOut{}),

cmd/osmosisd/cmd/balances_from_state_export.go

@@ -57,7 +57,7 @@ func newDerivedAccount(address string) DerivedAccount {
 }
 
 // underlyingCoins returns liquidity pool's underlying coin balances.
-func underlyingCoins(originCoins sdk.Coins, pools map[string]gammtypes.TraditionalAmmInterface) sdk.Coins {
+func underlyingCoins(originCoins sdk.Coins, pools map[string]gammtypes.PoolI) sdk.Coins {
 	balances := sdk.Coins{}
 	convertAgain := false
 	for _, coin := range originCoins {
@@ -85,7 +85,7 @@ func underlyingCoins(originCoins sdk.Coins, pools map[string]gammtypes.Tradition
 // TODO: Make a separate type for this.
 func underlyingCoinsForSelectPools(
 	originCoins sdk.Coins,
-	pools map[string]gammtypes.TraditionalAmmInterface,
+	pools map[string]gammtypes.PoolI,
 	selectPoolIDs []uint64,
 ) map[uint64]sdk.Coins {
 	balancesByPool := make(map[uint64]sdk.Coins)
@@ -263,9 +263,9 @@ Example:
 			}
 
 			// collect gamm pools
-			pools := make(map[string]gammtypes.TraditionalAmmInterface)
+			pools := make(map[string]gammtypes.PoolI)
 			for _, any := range gammGenesis.Pools {
-				var pool gammtypes.TraditionalAmmInterface
+				var pool gammtypes.PoolI
 				err := clientCtx.InterfaceRegistry.UnpackAny(any, &pool)
 				if err != nil {
 					panic(err)

osmoutils/store_helper_test.go

@@ -811,7 +811,7 @@ func (s *TestSuite) TestMustGet() {
 	}
 }
 
-// TestGetIfFound tests that GetIfFound returns a boolean indicating
+// TestGetIfFound tests that GetIfFound returns a boolean indicating 
 // whether value exists for the given key and error
 func (s *TestSuite) TestGetIfFound() {
 	tests := map[string]struct {
@@ -893,7 +893,7 @@ func (s *TestSuite) TestGetIfFound() {
 				s.Require().Equal(found, tc.expectFound)
 				if tc.expectErr {
 					s.Require().Error(err)
-				}
+				} 
 				// make sure found by key & Unmarshal successfully
 				if !tc.expectErr && tc.expectFound {
 					s.Require().Equal(expectedValue.String(), tc.actualResultProto.String())