Port cancel_offer to @marshal

Author: Quexington

chia/_tests/cmds/wallet/test_wallet.py

@@ -42,6 +42,7 @@
 from chia.wallet.wallet_coin_store import GetCoinRecords
 from chia.wallet.wallet_request_types import (
     BalanceResponse,
+    CancelOffer,
     CancelOfferResponse,
     CATAssetIDToName,
     CATAssetIDToNameResponse,
@@ -1180,14 +1181,14 @@ async def get_offer(self, request: GetOffer) -> GetOfferResponse:
 
         async def cancel_offer(
             self,
-            trade_id: bytes32,
+            request: CancelOffer,
             tx_config: TXConfig,
-            fee: uint64 = uint64(0),
-            secure: bool = True,
-            push: bool = True,
+            extra_conditions: tuple[Condition, ...] = tuple(),
             timelock_info: ConditionValidTimes = ConditionValidTimes(),
         ) -> CancelOfferResponse:
-            self.add_to_log("cancel_offer", (trade_id, tx_config, fee, secure, push, timelock_info))
+            self.add_to_log(
+                "cancel_offer", (request.trade_id, tx_config, request.fee, request.secure, request.push, timelock_info)
+            )
             return CancelOfferResponse([STD_UTX], [STD_TX])
 
     inst_rpc_client = CancelOfferRpcClient()

chia/_tests/wallet/rpc/test_wallet_rpc.py

@@ -106,6 +106,7 @@
 from chia.wallet.wallet_protocol import WalletProtocol
 from chia.wallet.wallet_request_types import (
     AddKey,
+    CancelOffer,
     CATAssetIDToName,
     CATGetAssetID,
     CATGetName,
@@ -1622,14 +1623,22 @@ async def test_offer_endpoints(wallet_environments: WalletTestFramework, wallet_
     ).trade_record
     assert TradeStatus(trade_record.status) == TradeStatus.PENDING_CONFIRM
 
-    await env_1.rpc_client.cancel_offer(offer.name(), wallet_environments.tx_config, secure=False)
+    await env_1.rpc_client.cancel_offer(
+        CancelOffer(
+            trade_id=offer.name(),
+            secure=False,
+            push=True,
+        ),
+        tx_config=wallet_environments.tx_config,
+    )
 
     trade_record = (await env_1.rpc_client.get_offer(GetOffer(offer.name(), file_contents=True))).trade_record
     assert trade_record.offer == bytes(offer)
     assert TradeStatus(trade_record.status) == TradeStatus.CANCELLED
 
     failed_cancel_res = await env_1.rpc_client.cancel_offer(
-        offer.name(), wallet_environments.tx_config, fee=uint64(1), secure=True
+        CancelOffer(trade_id=offer.name(), fee=uint64(1), secure=True, push=True),
+        tx_config=wallet_environments.tx_config,
     )
 
     trade_record = (await env_1.rpc_client.get_offer(GetOffer(offer.name()))).trade_record

chia/cmds/wallet_funcs.py

@@ -45,6 +45,7 @@
 from chia.wallet.vc_wallet.vc_store import VCProofs
 from chia.wallet.wallet_coin_store import GetCoinRecords
 from chia.wallet.wallet_request_types import (
+    CancelOffer,
     CATAssetIDToName,
     CATAssetIDToNameResponse,
     CATGetName,
@@ -930,11 +931,8 @@ async def cancel_offer(
 
         cli_confirm(f"Are you sure you wish to cancel offer with ID: {trade_record.trade_id}? (y/n): ")
         res = await wallet_client.cancel_offer(
-            offer_id,
-            CMDTXConfigLoader().to_tx_config(units["chia"], config, fingerprint),
-            secure=secure,
-            fee=fee,
-            push=push,
+            CancelOffer(trade_id=offer_id, secure=secure, fee=fee, push=push),
+            tx_config=CMDTXConfigLoader().to_tx_config(units["chia"], config, fingerprint),
             timelock_info=condition_valid_times,
         )
         if push or not secure:

chia/data_layer/data_layer.py

@@ -69,6 +69,7 @@
 from chia.wallet.transaction_record import TransactionRecord
 from chia.wallet.util.tx_config import DEFAULT_TX_CONFIG
 from chia.wallet.wallet_request_types import (
+    CancelOffer,
     CreateNewDL,
     CreateOfferForIDs,
     DLDeleteMirror,
@@ -1283,9 +1284,7 @@ async def cancel_offer(self, trade_id: bytes32, secure: bool, fee: uint64) -> No
             store_ids = [offered.launcher_id for offered in summary.offered]
 
         await self.wallet_rpc.cancel_offer(
-            trade_id=trade_id,
-            secure=secure,
-            fee=fee,
+            CancelOffer(trade_id=trade_id, secure=secure, fee=fee, push=True),
             # TODO: probably shouldn't be default but due to peculiarities in the RPC, we're using a stop gap.
             # This is not a change in behavior, the default was already implicit.
             tx_config=DEFAULT_TX_CONFIG,

chia/wallet/wallet_request_types.py

@@ -2018,6 +2018,13 @@ def from_json_dict(cls, json_dict: dict[str, Any]) -> Self:
         )
 
 
+@streamable
+@dataclass(frozen=True)
+class CancelOffer(TransactionEndpointRequest):
+    trade_id: bytes32 = field(default_factory=default_raise)
+    secure: bool = field(default_factory=default_raise)
+
+
 @streamable
 @dataclass(frozen=True)
 class CancelOfferResponse(TransactionEndpointResponse):

chia/wallet/wallet_rpc_api.py

@@ -112,6 +112,8 @@
     ApplySignatures,
     ApplySignaturesResponse,
     BalanceResponse,
+    CancelOffer,
+    CancelOfferResponse,
     CATAssetIDToName,
     CATAssetIDToNameResponse,
     CATGetAssetID,
@@ -2412,22 +2414,24 @@ async def get_offers_count(self, request: Empty) -> GetOffersCountResponse:
         )
 
     @tx_endpoint(push=True)
+    @marshal
     async def cancel_offer(
         self,
-        request: dict[str, Any],
+        request: CancelOffer,
         action_scope: WalletActionScope,
         extra_conditions: tuple[Condition, ...] = tuple(),
-    ) -> EndpointResult:
+    ) -> CancelOfferResponse:
         wsm = self.service.wallet_state_manager
-        secure = request["secure"]
-        trade_id = bytes32.from_hexstr(request["trade_id"])
-        fee: uint64 = uint64(request.get("fee", 0))
         async with self.service.wallet_state_manager.lock:
             await wsm.trade_manager.cancel_pending_offers(
-                [trade_id], action_scope, fee=fee, secure=secure, extra_conditions=extra_conditions
+                [request.trade_id],
+                action_scope,
+                fee=request.fee,
+                secure=request.secure,
+                extra_conditions=extra_conditions,
             )
 
-        return {"transactions": None}  # tx_endpoint wrapper will take care of this
+        return CancelOfferResponse([], [])  # tx_endpoint will fill in default values here
 
     @tx_endpoint(push=True, merge_spends=False)
     async def cancel_offers(

chia/wallet/wallet_rpc_client.py

@@ -19,6 +19,7 @@
     AddKeyResponse,
     ApplySignatures,
     ApplySignaturesResponse,
+    CancelOffer,
     CancelOfferResponse,
     CancelOffersResponse,
     CATAssetIDToName,
@@ -715,29 +716,17 @@ async def get_offers_count(self) -> GetOffersCountResponse:
 
     async def cancel_offer(
         self,
-        trade_id: bytes32,
+        request: CancelOffer,
         tx_config: TXConfig,
-        fee: int = 0,
-        secure: bool = True,
         extra_conditions: tuple[Condition, ...] = tuple(),
         timelock_info: ConditionValidTimes = ConditionValidTimes(),
-        push: bool = True,
     ) -> CancelOfferResponse:
-        res = await self.fetch(
-            "cancel_offer",
-            {
-                "trade_id": trade_id.hex(),
-                "secure": secure,
-                "fee": fee,
-                "extra_conditions": conditions_to_json_dicts(extra_conditions),
-                "push": push,
-                **tx_config.to_json_dict(),
-                **timelock_info.to_json_dict(),
-            },
+        return CancelOfferResponse.from_json_dict(
+            await self.fetch(
+                "cancel_offer", request.json_serialize_for_transport(tx_config, extra_conditions, timelock_info)
+            )
         )
 
-        return json_deserialize_with_clvm_streamable(res, CancelOfferResponse)
-
     async def cancel_offers(
         self,
         tx_config: TXConfig,