bug: add split route to message server in pool manager

proto/osmosis/poolmanager/v1beta1/tx.proto

@@ -12,6 +12,10 @@ service Msg {
       returns (MsgSwapExactAmountInResponse);
   rpc SwapExactAmountOut(MsgSwapExactAmountOut)
       returns (MsgSwapExactAmountOutResponse);
+  rpc SplitRouteSwapExactAmountIn(MsgSplitRouteSwapExactAmountIn)
+      returns (MsgSplitRouteSwapExactAmountInResponse);
+  rpc SplitRouteSwapExactAmountOut(MsgSplitRouteSwapExactAmountOut)
+      returns (MsgSplitRouteSwapExactAmountOutResponse);
 }
 
 // ===================== MsgSwapExactAmountIn

x/poolmanager/msg_server.go

@@ -69,3 +69,53 @@ func (server msgServer) SwapExactAmountOut(goCtx context.Context, msg *types.Msg
 
 	return &types.MsgSwapExactAmountOutResponse{TokenInAmount: tokenInAmount}, nil
 }
+
+func (server msgServer) SplitRouteSwapExactAmountIn(goCtx context.Context, msg *types.MsgSplitRouteSwapExactAmountIn) (*types.MsgSplitRouteSwapExactAmountInResponse, error) {
+	ctx := sdk.UnwrapSDKContext(goCtx)
+
+	sender, err := sdk.AccAddressFromBech32(msg.Sender)
+	if err != nil {
+		return nil, err
+	}
+
+	tokenOutAmount, err := server.keeper.SplitRouteExactAmountIn(ctx, sender, msg.Routes, msg.TokenInDenom, msg.TokenOutMinAmount)
+	if err != nil {
+		return nil, err
+	}
+
+	// Swap event is handled elsewhere
+	ctx.EventManager().EmitEvents(sdk.Events{
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+			sdk.NewAttribute(sdk.AttributeKeySender, msg.Sender),
+		),
+	})
+
+	return &types.MsgSplitRouteSwapExactAmountInResponse{TokenOutAmount: tokenOutAmount}, nil
+}
+
+func (server msgServer) SplitRouteSwapExactAmountOut(goCtx context.Context, msg *types.MsgSplitRouteSwapExactAmountOut) (*types.MsgSplitRouteSwapExactAmountOutResponse, error) {
+	ctx := sdk.UnwrapSDKContext(goCtx)
+
+	sender, err := sdk.AccAddressFromBech32(msg.Sender)
+	if err != nil {
+		return nil, err
+	}
+
+	tokenInAmount, err := server.keeper.SplitRouteExactAmountOut(ctx, sender, msg.Routes, msg.TokenOutDenom, msg.TokenInMaxAmount)
+	if err != nil {
+		return nil, err
+	}
+
+	// Swap event is handled elsewhere
+	ctx.EventManager().EmitEvents(sdk.Events{
+		sdk.NewEvent(
+			sdk.EventTypeMessage,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+			sdk.NewAttribute(sdk.AttributeKeySender, msg.Sender),
+		),
+	})
+
+	return &types.MsgSplitRouteSwapExactAmountOutResponse{TokenInAmount: tokenInAmount}, nil
+}

x/poolmanager/router.go

@@ -159,6 +159,15 @@ func (k Keeper) SplitRouteExactAmountIn(
 		return sdk.Int{}, types.PriceImpactProtectionExactInError{Actual: totalOutAmount, MinAmount: tokenOutMinAmount}
 	}
 
+	ctx.EventManager().EmitEvents(sdk.Events{
+		sdk.NewEvent(
+			types.TypeMsgSplitRouteSwapExactAmountIn,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+			sdk.NewAttribute(sdk.AttributeKeySender, sender.String()),
+			sdk.NewAttribute(types.AttributeKeyTokensOut, totalOutAmount.String()),
+		),
+	})
+
 	return totalOutAmount, nil
 }
 
@@ -436,6 +445,15 @@ func (k Keeper) SplitRouteExactAmountOut(
 		return sdk.Int{}, types.PriceImpactProtectionExactOutError{Actual: totalInAmount, MaxAmount: tokenInMaxAmount}
 	}
 
+	ctx.EventManager().EmitEvents(sdk.Events{
+		sdk.NewEvent(
+			types.TypeMsgSplitRouteSwapExactAmountOut,
+			sdk.NewAttribute(sdk.AttributeKeyModule, types.AttributeValueCategory),
+			sdk.NewAttribute(sdk.AttributeKeySender, sender.String()),
+			sdk.NewAttribute(types.AttributeKeyTokensOut, totalInAmount.String()),
+		),
+	})
+
 	return totalInAmount, nil
 }
 

x/poolmanager/types/events.go

@@ -1,7 +1,10 @@
 package types
 
 const (
-	AttributeValueCategory = ModuleName
-	TypeEvtPoolCreated     = "pool_created"
-	AttributeKeyPoolId     = "pool_id"
+	AttributeValueCategory       = ModuleName
+	TypeEvtPoolCreated           = "pool_created"
+	TypeEvtSplitRouteSwapExactIn = "split_route_swap_exact_in"
+	AttributeKeyTokensIn         = "tokens_in"
+	AttributeKeyTokensOut        = "tokens_out"
+	AttributeKeyPoolId           = "pool_id"
 )