Add logic to handle single nft claim (#2412)

## Describe your changes and provide context
Previously, `MsgClaimSpecific.Assets.Denom` is unused if the asset type
is CW721. This PR will make the claiming logic happens to only one NFT
if `Denom` is specified for a CW721 claim.

## Testing performed to validate your change
unit test

Author: codchen

precompiles/solo/solo.go

@@ -195,21 +195,25 @@ func (p PrecompileExecutor) ClaimSpecific(ctx sdk.Context, caller common.Address
 			}
 		case asset.IsCW721():
 			allTokens := []string{}
-			startAfter := ""
-			for {
-				res, err := p.wasmViewKeeper.QuerySmartSafe(ctx, contractAddr, CW721TokensQueryPayload(sender, startAfter))
-				if err != nil {
-					return nil, 0, fmt.Errorf("failed to query CW721 contract %s for all tokens: %w", contractAddr.String(), err)
-				}
-				tokens, err := ParseCW721TokensQueryResponse(res)
-				if err != nil {
-					return nil, 0, fmt.Errorf("failed to parse CW20 contract %s balance response: %w", contractAddr.String(), err)
-				}
-				if len(tokens) == 0 {
-					break
+			if token := asset.GetDenom(); token != "" {
+				allTokens = append(allTokens, token)
+			} else {
+				startAfter := ""
+				for {
+					res, err := p.wasmViewKeeper.QuerySmartSafe(ctx, contractAddr, CW721TokensQueryPayload(sender, startAfter))
+					if err != nil {
+						return nil, 0, fmt.Errorf("failed to query CW721 contract %s for all tokens: %w", contractAddr.String(), err)
+					}
+					tokens, err := ParseCW721TokensQueryResponse(res)
+					if err != nil {
+						return nil, 0, fmt.Errorf("failed to parse CW20 contract %s balance response: %w", contractAddr.String(), err)
+					}
+					if len(tokens) == 0 {
+						break
+					}
+					allTokens = append(allTokens, tokens...)
+					startAfter = tokens[len(tokens)-1]
 				}
-				allTokens = append(allTokens, tokens...)
-				startAfter = tokens[len(tokens)-1]
 			}
 			for _, token := range allTokens {
 				_, err := p.wasmKeeper.Execute(ctx, contractAddr, sender, CW721TransferPayload(callerSeiAddr, token), sdk.NewCoins())

precompiles/solo/solo_test.go

@@ -177,7 +177,7 @@ func TestClaimSpecificCW721(t *testing.T) {
 	k.AccountKeeper().SetAccount(origCtx, acc)
 	contractAddr := setupCW721Contract(origCtx, claimeeKey, *wKeeper)
 	ctx, _ := origCtx.CacheContext()
-	ctx = ctx.WithGasMeter(sdk.NewGasMeter(2000000, 1, 1))
+	ctx = ctx.WithGasMeter(sdk.NewGasMeter(3000000, 1, 1))
 	_, remainingGas, err := p.ClaimSpecific(ctx, claimer, &method, []interface{}{signClaimMsg(t, evmtypes.NewMsgClaimSpecific(claimee, claimer, &evmtypes.Asset{AssetType: evmtypes.AssetType_TYPECW721, ContractAddress: contractAddr.String()}), claimee, claimer, acc, claimeeKey)}, false)
 	ctx = ctx.WithGasMeter(sdk.NewInfiniteGasMeterWithMultiplier(ctx))
 	require.NoError(t, err)
@@ -187,6 +187,36 @@ func TestClaimSpecificCW721(t *testing.T) {
 	}
 }
 
+func TestClaimSingleCW721(t *testing.T) {
+	k := &testkeeper.EVMTestApp.EvmKeeper
+	origCtx := testkeeper.EVMTestApp.GetContextForDeliverTx(nil).WithChainID("sei-test").WithBlockTime(time.Now())
+	txConfig := testkeeper.EVMTestApp.GetTxConfig()
+	a := pcommon.MustGetABI(solo.F, "abi.json")
+	method := a.Methods["claimSpecific"]
+	wKeeper := wasmkeeper.NewDefaultPermissionKeeper(testkeeper.EVMTestApp.WasmKeeper)
+	p := solo.NewExecutor(a, k, k.BankKeeper(), k.AccountKeeper(), wKeeper, testkeeper.EVMTestApp.WasmKeeper, txConfig)
+	claimeeKey := testkeeper.MockPrivateKey()
+	claimee, _ := testkeeper.PrivateKeyToAddresses(claimeeKey)
+	claimerKey := testkeeper.MockPrivateKey()
+	_, claimer := testkeeper.PrivateKeyToAddresses(claimerKey)
+	acc := authtypes.NewBaseAccount(claimee, claimeeKey.PubKey(), 10, 0)
+	k.AccountKeeper().SetAccount(origCtx, acc)
+	contractAddr := setupCW721Contract(origCtx, claimeeKey, *wKeeper)
+	ctx, _ := origCtx.CacheContext()
+	ctx = ctx.WithGasMeter(sdk.NewGasMeter(2000000, 1, 1))
+	_, remainingGas, err := p.ClaimSpecific(ctx, claimer, &method, []interface{}{signClaimMsg(t, evmtypes.NewMsgClaimSpecific(claimee, claimer, &evmtypes.Asset{AssetType: evmtypes.AssetType_TYPECW721, ContractAddress: contractAddr.String(), Denom: "5"}), claimee, claimer, acc, claimeeKey)}, false)
+	ctx = ctx.WithGasMeter(sdk.NewInfiniteGasMeterWithMultiplier(ctx))
+	require.NoError(t, err)
+	require.Equal(t, uint64(1838040), remainingGas)
+	for i := 0; i < 15; i++ {
+		if i == 5 {
+			require.Equal(t, k.GetSeiAddressOrDefault(ctx, claimer).String(), queryCW721Owner(ctx, testkeeper.EVMTestApp.WasmKeeper, contractAddr, fmt.Sprintf("%d", i)))
+		} else {
+			require.Equal(t, claimee.String(), queryCW721Owner(ctx, testkeeper.EVMTestApp.WasmKeeper, contractAddr, fmt.Sprintf("%d", i)))
+		}
+	}
+}
+
 func TestClaimSpecificNative(t *testing.T) {
 	k := &testkeeper.EVMTestApp.EvmKeeper
 	origCtx := testkeeper.EVMTestApp.GetContextForDeliverTx(nil).WithChainID("sei-test").WithBlockTime(time.Now())

sei-wasmd/x/wasm/keeper/keeper.go

@@ -1141,7 +1141,10 @@ func (k Keeper) emitCW721OwnerBeforeTransferIfApplicable(ctx sdk.Context, contra
 		if err != nil {
 			return
 		}
-		resBz, err := k.QuerySmart(ctx.WithGasMeter(sdk.NewInfiniteGasMeterWithMultiplier(ctx)), contractAddress, ownerQueryBz)
+		if ctx.IsTracing() && ctx.ChainID() == "pacific-1" && strings.Compare(ctx.ClosestUpgradeName(), "v6.3.0") < 0 {
+			ctx = ctx.WithGasMeter(sdk.NewInfiniteGasMeterWithMultiplier(ctx))
+		}
+		resBz, err := k.QuerySmart(ctx, contractAddress, ownerQueryBz)
 		if err != nil {
 			return
 		}