Precompile advanced order

Cargo.toml

@@ -57,6 +57,7 @@ useless_conversion = "allow"     # until polkadot is patched
 pallet-alpha-assets = { path = "pallets/alpha-assets", default-features = false }
 node-subtensor-runtime = { path = "runtime", default-features = false }
 pallet-admin-utils = { path = "pallets/admin-utils", default-features = false }
+pallet-limit-orders = { path = "pallets/limit-orders", default-features = false }
 pallet-commitments = { path = "pallets/commitments", default-features = false }
 pallet-registry = { path = "pallets/registry", default-features = false }
 pallet-crowdloan = { path = "pallets/crowdloan", default-features = false }
@@ -72,7 +73,7 @@ subtensor-custom-rpc = { default-features = false, path = "pallets/subtensor/rpc
 subtensor-custom-rpc-runtime-api = { default-features = false, path = "pallets/subtensor/runtime-api" }
 subtensor-precompiles = { default-features = false, path = "precompiles" }
 subtensor-runtime-common = { default-features = false, path = "common" }
-subtensor-swap-interface = { default-features = false, path = "pallets/swap-interface" }
+subtensor-swap-interface = { default-features = false, path = "primitives/swap-interface" }
 subtensor-transaction-fee = { default-features = false, path = "pallets/transaction-fee" }
 subtensor-chain-extensions = { default-features = false, path = "chain-extensions" }
 stp-shield = { git = "https://github.com/opentensor/polkadot-sdk.git", rev = "7cc54bf2d50ae3921d718736dfeb0de9468539c7", default-features = false }

chain-extensions/Cargo.toml

@@ -84,5 +84,6 @@ runtime-benchmarks = [
 	"pallet-subtensor-proxy/runtime-benchmarks",
 	"pallet-subtensor-utility/runtime-benchmarks",
 	"pallet-timestamp/runtime-benchmarks",
-	"subtensor-runtime-common/runtime-benchmarks"
+	"subtensor-runtime-common/runtime-benchmarks",
+	"subtensor-swap-interface/runtime-benchmarks"
 ]

contract-tests/src/contracts/limitOrders.ts

@@ -0,0 +1,263 @@
+export const ILIMITORDERS_ADDRESS =
+  "0x000000000000000000000000000000000000080e";
+
+export const ILimitOrdersABI = [
+  {
+    inputs: [
+      {
+        components: [
+          { internalType: "address", name: "signer", type: "address" },
+          { internalType: "address", name: "hotkey", type: "address" },
+          { internalType: "uint16", name: "netuid", type: "uint16" },
+          { internalType: "uint8", name: "order_type", type: "uint8" },
+          { internalType: "uint64", name: "amount", type: "uint64" },
+          { internalType: "uint64", name: "limit_price", type: "uint64" },
+          { internalType: "uint64", name: "expiry", type: "uint64" },
+          { internalType: "uint32", name: "fee_rate", type: "uint32" },
+          { internalType: "address", name: "fee_recipient", type: "address" },
+          { internalType: "address[]", name: "relayer", type: "address[]" },
+          { internalType: "bool", name: "has_max_slippage", type: "bool" },
+          { internalType: "uint32", name: "max_slippage", type: "uint32" },
+          { internalType: "uint64", name: "chain_id", type: "uint64" },
+          {
+            internalType: "bool",
+            name: "partial_fills_enabled",
+            type: "bool",
+          },
+        ],
+        internalType: "struct OrderInput",
+        name: "order",
+        type: "tuple",
+      },
+    ],
+    name: "cancelOrder",
+    outputs: [],
+    stateMutability: "payable",
+    type: "function",
+  },
+  {
+    inputs: [
+      {
+        components: [
+          { internalType: "address", name: "signer", type: "address" },
+          { internalType: "address", name: "hotkey", type: "address" },
+          { internalType: "uint16", name: "netuid", type: "uint16" },
+          { internalType: "uint8", name: "order_type", type: "uint8" },
+          { internalType: "uint64", name: "amount", type: "uint64" },
+          { internalType: "uint64", name: "limit_price", type: "uint64" },
+          { internalType: "uint64", name: "expiry", type: "uint64" },
+          { internalType: "uint32", name: "fee_rate", type: "uint32" },
+          { internalType: "address", name: "fee_recipient", type: "address" },
+          { internalType: "address[]", name: "relayer", type: "address[]" },
+          { internalType: "bool", name: "has_max_slippage", type: "bool" },
+          { internalType: "uint32", name: "max_slippage", type: "uint32" },
+          { internalType: "uint64", name: "chain_id", type: "uint64" },
+          {
+            internalType: "bool",
+            name: "partial_fills_enabled",
+            type: "bool",
+          },
+        ],
+        internalType: "struct OrderInput",
+        name: "order",
+        type: "tuple",
+      },
+    ],
+    name: "deriveOrderId",
+    outputs: [{ internalType: "bytes32", name: "", type: "bytes32" }],
+    stateMutability: "view",
+    type: "function",
+  },
+  {
+    inputs: [
+      { internalType: "uint16", name: "netuid", type: "uint16" },
+      {
+        components: [
+          {
+            components: [
+              { internalType: "address", name: "signer", type: "address" },
+              { internalType: "address", name: "hotkey", type: "address" },
+              { internalType: "uint16", name: "netuid", type: "uint16" },
+              { internalType: "uint8", name: "order_type", type: "uint8" },
+              { internalType: "uint64", name: "amount", type: "uint64" },
+              {
+                internalType: "uint64",
+                name: "limit_price",
+                type: "uint64",
+              },
+              { internalType: "uint64", name: "expiry", type: "uint64" },
+              { internalType: "uint32", name: "fee_rate", type: "uint32" },
+              {
+                internalType: "address",
+                name: "fee_recipient",
+                type: "address",
+              },
+              {
+                internalType: "address[]",
+                name: "relayer",
+                type: "address[]",
+              },
+              {
+                internalType: "bool",
+                name: "has_max_slippage",
+                type: "bool",
+              },
+              {
+                internalType: "uint32",
+                name: "max_slippage",
+                type: "uint32",
+              },
+              { internalType: "uint64", name: "chain_id", type: "uint64" },
+              {
+                internalType: "bool",
+                name: "partial_fills_enabled",
+                type: "bool",
+              },
+            ],
+            internalType: "struct OrderInput",
+            name: "order",
+            type: "tuple",
+          },
+          { internalType: "bytes", name: "signature", type: "bytes" },
+          {
+            internalType: "bool",
+            name: "has_partial_fill",
+            type: "bool",
+          },
+          { internalType: "uint64", name: "partial_fill", type: "uint64" },
+        ],
+        internalType: "struct SignedOrderInput[]",
+        name: "orders",
+        type: "tuple[]",
+      },
+    ],
+    name: "executeBatchedOrders",
+    outputs: [],
+    stateMutability: "payable",
+    type: "function",
+  },
+  {
+    inputs: [
+      {
+        components: [
+          {
+            components: [
+              { internalType: "address", name: "signer", type: "address" },
+              { internalType: "address", name: "hotkey", type: "address" },
+              { internalType: "uint16", name: "netuid", type: "uint16" },
+              { internalType: "uint8", name: "order_type", type: "uint8" },
+              { internalType: "uint64", name: "amount", type: "uint64" },
+              {
+                internalType: "uint64",
+                name: "limit_price",
+                type: "uint64",
+              },
+              { internalType: "uint64", name: "expiry", type: "uint64" },
+              { internalType: "uint32", name: "fee_rate", type: "uint32" },
+              {
+                internalType: "address",
+                name: "fee_recipient",
+                type: "address",
+              },
+              {
+                internalType: "address[]",
+                name: "relayer",
+                type: "address[]",
+              },
+              {
+                internalType: "bool",
+                name: "has_max_slippage",
+                type: "bool",
+              },
+              {
+                internalType: "uint32",
+                name: "max_slippage",
+                type: "uint32",
+              },
+              { internalType: "uint64", name: "chain_id", type: "uint64" },
+              {
+                internalType: "bool",
+                name: "partial_fills_enabled",
+                type: "bool",
+              },
+            ],
+            internalType: "struct OrderInput",
+            name: "order",
+            type: "tuple",
+          },
+          { internalType: "bytes", name: "signature", type: "bytes" },
+          {
+            internalType: "bool",
+            name: "has_partial_fill",
+            type: "bool",
+          },
+          { internalType: "uint64", name: "partial_fill", type: "uint64" },
+        ],
+        internalType: "struct SignedOrderInput[]",
+        name: "orders",
+        type: "tuple[]",
+      },
+    ],
+    name: "executeOrders",
+    outputs: [],
+    stateMutability: "payable",
+    type: "function",
+  },
+  {
+    inputs: [],
+    name: "getLimitOrdersEnabled",
+    outputs: [{ internalType: "bool", name: "", type: "bool" }],
+    stateMutability: "view",
+    type: "function",
+  },
+  {
+    inputs: [{ internalType: "bytes32", name: "orderId", type: "bytes32" }],
+    name: "getOrderStatus",
+    outputs: [{ internalType: "uint8", name: "", type: "uint8" }],
+    stateMutability: "view",
+    type: "function",
+  },
+] as const;
+
+export type OrderInput = {
+  signer: string;
+  hotkey: string;
+  netuid: number;
+  order_type: number;
+  amount: bigint;
+  limit_price: bigint;
+  expiry: bigint;
+  fee_rate: number;
+  fee_recipient: string;
+  relayer: string[];
+  has_max_slippage: boolean;
+  max_slippage: number;
+  chain_id: bigint;
+  partial_fills_enabled: boolean;
+};
+
+export const FAR_FUTURE = BigInt("18446744073709551615");
+
+export function buildOrderInput(
+  signer: string,
+  hotkey: string,
+  overrides: Partial<OrderInput> = {},
+): OrderInput {
+  return {
+    signer,
+    hotkey,
+    netuid: 1,
+    order_type: 0,
+    amount: BigInt(1_000),
+    limit_price: BigInt(1_000_000_000),
+    expiry: FAR_FUTURE,
+    fee_rate: 0,
+    fee_recipient: signer,
+    relayer: [],
+    has_max_slippage: false,
+    max_slippage: 0,
+    chain_id: BigInt(42),
+    partial_fills_enabled: false,
+    ...overrides,
+  };
+}

contract-tests/test/limitOrders.precompile.test.ts

@@ -0,0 +1,52 @@
+import * as assert from "assert";
+
+import { ethers } from "ethers";
+import { TypedApi } from "polkadot-api";
+import { devnet } from "@polkadot-api/descriptors";
+import {
+  buildOrderInput,
+  ILIMITORDERS_ADDRESS,
+  ILimitOrdersABI,
+} from "../src/contracts/limitOrders";
+import { generateRandomEthersWallet } from "../src/utils";
+import { getDevnetApi } from "../src/substrate";
+import { forceSetBalanceToEthAddress } from "../src/subtensor";
+
+describe("Limit orders precompile E2E smoke", () => {
+  let api: TypedApi<typeof devnet>;
+  let wallet1: ethers.Wallet;
+  let wallet2: ethers.Wallet;
+  let limitOrdersContract: ethers.Contract;
+
+  beforeEach(async () => {
+    api = await getDevnetApi();
+
+    wallet1 = generateRandomEthersWallet();
+    wallet2 = generateRandomEthersWallet();
+    limitOrdersContract = new ethers.Contract(
+      ILIMITORDERS_ADDRESS,
+      ILimitOrdersABI,
+      wallet1,
+    );
+
+    await forceSetBalanceToEthAddress(api, wallet1.address);
+    await forceSetBalanceToEthAddress(api, wallet2.address);
+  });
+
+  it("reads pallet status through the precompile", async () => {
+    const enabled = await limitOrdersContract.getLimitOrdersEnabled();
+    assert.strictEqual(enabled, true);
+  });
+
+  it("derives order ids and cancels orders through the precompile", async () => {
+    const order = buildOrderInput(wallet1.address, wallet2.address);
+
+    const orderId = await limitOrdersContract.deriveOrderId(order);
+    assert.strictEqual(await limitOrdersContract.getOrderStatus(orderId), 0);
+
+    const tx = await limitOrdersContract.cancelOrder(order);
+    await tx.wait();
+
+    assert.strictEqual(await limitOrdersContract.getOrderStatus(orderId), 3);
+  });
+});