From d842ac5921ddab03fb902e346d559b3fc8d54f7a Mon Sep 17 00:00:00 2001
From: Troy <troy.burmeister@gmail.com>
Date: Thu, 26 Sep 2024 17:26:52 -0600
Subject: [PATCH] add cancel stale order

---
 src/synthetix/perps/perps.py | 50 ++++++++++++++++++++++++++++++++++--
 1 file changed, 48 insertions(+), 2 deletions(-)

diff --git a/src/synthetix/perps/perps.py b/src/synthetix/perps/perps.py
index a9aa934..a3d5307 100644
--- a/src/synthetix/perps/perps.py
+++ b/src/synthetix/perps/perps.py
@@ -233,10 +233,10 @@ def _prepare_oracle_call(self, market_names: [str] = []):
         to, data, _ = make_pyth_fulfillment_request(
             self.snx,
             self.snx.contracts["pyth_erc7412_wrapper"]["PythERC7412Wrapper"]["address"],
-            1, # update type
+            1,  # update type
             feed_ids,
             price_update_data,
-            30, # staleness tolerance
+            30,  # staleness tolerance
             0,
         )
         value = len(market_names)
@@ -1800,6 +1800,52 @@ def settle_order(
 
         raise Exception("Failed to settle order after maximum attempts")
 
+    def cancel_stale_order(
+        self,
+        account_id: int = None,
+        market_id: int = None,
+        market_name: str = None,
+        submit: bool = False,
+    ):
+        """
+        Cancels and order for the specified account and market.
+
+        :param int | None account_id: The id of the account to cancel the order for. If not provided, uses the default account.
+        :param int | None market_id: The id of the market to cancel the order in.
+        :param str | None market_name: The name of the market to cancel the order in.
+        :param bool submit: If True, submit the transaction to the blockchain. If False, return the transaction parameters.
+        :return: If submit is True, returns the transaction hash. Otherwise, returns the transaction parameters.
+        :rtype: str | dict
+        """
+        market_id, market_name = self._resolve_market(market_id, market_name)
+        if account_id is None:
+            account_id = self.default_account_id
+
+        # Prepare the transaction
+        tx_params = write_erc7412(
+            self.snx,
+            self.market_proxy,
+            "cancelStaleOrder",
+            [
+                account_id,
+                market_id,
+            ],
+        )
+
+        if submit:
+            try:
+                tx_hash = self.snx.execute_transaction(tx_params)
+                self.logger.info(
+                    f"Cancelling stale order for account {account_id} in market {market_name}"
+                )
+                self.logger.info(f"cancelStaleOrder tx: {tx_hash}")
+                return tx_hash
+            except Exception as e:
+                self.logger.error(f"Failed to cancel order: {str(e)}")
+                raise
+        else:
+            return tx_params
+
     def pay_debt(
         self,
         amount: int = None,