Add a constant-product market maker (ava-labs#1095)

* add barebone cpmm

* start writing add_liquidity_test

* manage reserves and add swap test

* error swap on no liquidity

* rename cpmm -> automated-market-maker, comments

* reverse `ProgramCallErrorCode` order (ava-labs#1093)

* add support for bool type (ava-labs#1094)

---------

Co-authored-by: Richard Pringle <[email protected]>

Cargo.toml

@@ -6,6 +6,7 @@ members = [
     "x/programs/rust/examples/token",
     "x/programs/rust/examples/counter",
     "x/programs/rust/examples/counter-external",
+    "x/programs/rust/examples/automated-market-maker",
     "x/programs/rust/wasmlanche-sdk/tests/test-crate",
     "x/programs/test/programs/*",
     "x/programs/cmd/simulator/test/*",

x/programs/rust/examples/automated-market-maker/Cargo.toml

@@ -0,0 +1,17 @@
+[package]
+name = "automated-market-maker"
+version = "0.1.0"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib", "lib"]
+
+[dependencies]
+wasmlanche-sdk = { workspace = true, features = ["debug"] }
+borsh = { workspace = true }
+
+[dev-dependencies]
+simulator = { workspace = true }
+
+[build-dependencies]
+wasmlanche-sdk = { workspace = true, features = ["build"] }

x/programs/rust/examples/automated-market-maker/build.rs

@@ -0,0 +1,3 @@
+fn main() {
+    wasmlanche_sdk::build::build_wasm_on_test();
+}

x/programs/rust/examples/automated-market-maker/src/lib.rs

@@ -0,0 +1,318 @@
+use wasmlanche_sdk::{public, state_keys, Context, Program};
+
+#[state_keys]
+pub enum StateKeys {
+    // Internal accounting
+    ReserveX,
+    ReserveY,
+
+    // Liquidity token
+    TotalySupply,
+}
+
+#[public]
+pub fn add_liquidity(context: Context<StateKeys>, amount_x: u64, amount_y: u64) -> u64 {
+    let program = context.program();
+    let total_supply = total_supply(program);
+    // tokens    | shares
+    // -------------------------
+    // amount_x  | minted
+    // reserve_x | total_supply
+    let (reserve_x, reserve_y) = reserves(program);
+    let minted = if total_supply == 0 {
+        let minted = amount_x;
+        assert_eq!(minted, amount_y);
+        minted
+    } else {
+        let minted = amount_x * total_supply / reserve_x;
+        assert_eq!(minted, amount_y * total_supply / reserve_y); // make sure that the ratio is good
+        minted
+    };
+
+    program
+        .state()
+        .store(StateKeys::ReserveX, &(reserve_x + amount_x))
+        .unwrap();
+    program
+        .state()
+        .store(StateKeys::ReserveY, &(reserve_y + amount_y))
+        .unwrap();
+    program
+        .state()
+        .store(StateKeys::TotalySupply, &(total_supply + minted))
+        .unwrap();
+
+    minted
+}
+
+#[public]
+pub fn remove_liquidity(context: Context<StateKeys>, shares: u64) -> (u64, u64) {
+    let program = context.program();
+    let total_supply = total_supply(program);
+    let (reserve_x, reserve_y) = reserves(program);
+    let (amount_x, amount_y) = (
+        shares * reserve_x / total_supply,
+        shares * reserve_y / total_supply,
+    );
+
+    program
+        .state()
+        .store(StateKeys::ReserveX, &(reserve_x - amount_x))
+        .unwrap();
+    program
+        .state()
+        .store(StateKeys::ReserveY, &(reserve_y - amount_y))
+        .unwrap();
+    program
+        .state()
+        .store(StateKeys::TotalySupply, &(total_supply - shares))
+        .unwrap();
+
+    (amount_x, amount_y)
+}
+
+#[public]
+pub fn swap(context: Context<StateKeys>, amount_in: u64, x_to_y: bool) -> u64 {
+    let program = context.program();
+    let total_supply = total_supply(program);
+    assert!(total_supply > 0, "no liquidity");
+    // x * y = constant
+    // x' = x + dx
+    // y' = y + dy
+    // (x + dx) * (y + dy) = x * y
+    // y + dy = (x * y) / (x + dx)
+    // dy = ((x * y) / (x + dx)) - y
+    // skip a few steps
+    // -dy = y * dx / (x + dx)
+    let (reserve_x, reserve_y) = reserves(context.program());
+    let (reserve_x, reserve_y, out) = if x_to_y {
+        let dy = (reserve_y * amount_in) / (reserve_x + amount_in);
+        (reserve_x + amount_in, reserve_y - dy, dy)
+    } else {
+        let dx = (reserve_x * amount_in) / (reserve_y + amount_in);
+        (reserve_x - dx, reserve_y + amount_in, dx)
+    };
+
+    program
+        .state()
+        .store(StateKeys::ReserveX, &reserve_x)
+        .unwrap();
+    program
+        .state()
+        .store(StateKeys::ReserveY, &reserve_y)
+        .unwrap();
+
+    out
+}
+
+fn total_supply(program: &Program<StateKeys>) -> u64 {
+    program
+        .state()
+        .get(StateKeys::TotalySupply)
+        .unwrap()
+        .unwrap_or_default()
+}
+
+fn reserves(program: &Program<StateKeys>) -> (u64, u64) {
+    (
+        program
+            .state()
+            .get(StateKeys::ReserveX)
+            .unwrap()
+            .unwrap_or_default(),
+        program
+            .state()
+            .get(StateKeys::ReserveY)
+            .unwrap()
+            .unwrap_or_default(),
+    )
+}
+
+#[cfg(test)]
+mod tests {
+    use simulator::{Endpoint, Key, Step, StepResponseError, TestContext};
+    use wasmlanche_sdk::ExternalCallError;
+
+    const PROGRAM_PATH: &str = env!("PROGRAM_PATH");
+
+    #[test]
+    fn init_state() {
+        let mut simulator = simulator::ClientBuilder::new().try_build().unwrap();
+
+        let owner = "owner";
+
+        let program_id = simulator
+            .run_step(owner, &Step::create_program(PROGRAM_PATH))
+            .unwrap()
+            .id;
+
+        simulator
+            .run_step(owner, &Step::create_key(Key::Ed25519(owner.to_string())))
+            .unwrap();
+
+        let test_context = TestContext::from(program_id);
+
+        let resp_err = simulator
+            .run_step(
+                owner,
+                &Step {
+                    endpoint: Endpoint::Execute,
+                    method: "remove_liquidity".to_string(),
+                    max_units: u64::MAX,
+                    params: vec![test_context.clone().into(), 100000u64.into()],
+                },
+            )
+            .unwrap()
+            .result
+            .response::<(u64, u64)>()
+            .unwrap_err();
+
+        let StepResponseError::ExternalCall(call_err) = resp_err else {
+            panic!("wrong error returned");
+        };
+
+        assert!(matches!(call_err, ExternalCallError::CallPanicked));
+
+        let resp_err = simulator
+            .run_step(
+                owner,
+                &Step {
+                    endpoint: Endpoint::Execute,
+                    method: "swap".to_string(),
+                    max_units: u64::MAX,
+                    params: vec![test_context.into(), 100000u64.into(), true.into()],
+                },
+            )
+            .unwrap()
+            .result
+            .response::<u64>()
+            .unwrap_err();
+
+        let StepResponseError::ExternalCall(call_err) = resp_err else {
+            panic!("wrong error returned");
+        };
+
+        assert!(matches!(call_err, ExternalCallError::CallPanicked));
+    }
+
+    #[test]
+    fn add_liquidity_same_ratio() {
+        let mut simulator = simulator::ClientBuilder::new().try_build().unwrap();
+
+        let owner = "owner";
+
+        let program_id = simulator
+            .run_step(owner, &Step::create_program(PROGRAM_PATH))
+            .unwrap()
+            .id;
+
+        simulator
+            .run_step(owner, &Step::create_key(Key::Ed25519(owner.to_string())))
+            .unwrap();
+
+        let test_context = TestContext::from(program_id);
+
+        let resp = simulator
+            .run_step(
+                owner,
+                &Step {
+                    endpoint: Endpoint::Execute,
+                    method: "add_liquidity".to_string(),
+                    max_units: u64::MAX,
+                    params: vec![test_context.clone().into(), 1000u64.into(), 1000u64.into()],
+                },
+            )
+            .unwrap()
+            .result
+            .response::<u64>()
+            .unwrap();
+
+        assert_eq!(resp, 1000);
+
+        let resp = simulator
+            .run_step(
+                owner,
+                &Step {
+                    endpoint: Endpoint::Execute,
+                    method: "add_liquidity".to_string(),
+                    max_units: u64::MAX,
+                    params: vec![test_context.into(), 1000u64.into(), 1001u64.into()],
+                },
+            )
+            .unwrap()
+            .result
+            .response::<u64>()
+            .unwrap_err();
+
+        let StepResponseError::ExternalCall(call_err) = resp else {
+            panic!("unexpected error");
+        };
+
+        assert!(matches!(call_err, ExternalCallError::CallPanicked));
+    }
+
+    #[test]
+    fn swap_changes_ratio() {
+        let mut simulator = simulator::ClientBuilder::new().try_build().unwrap();
+
+        let owner = "owner";
+
+        let program_id = simulator
+            .run_step(owner, &Step::create_program(PROGRAM_PATH))
+            .unwrap()
+            .id;
+
+        simulator
+            .run_step(owner, &Step::create_key(Key::Ed25519(owner.to_string())))
+            .unwrap();
+
+        let test_context = TestContext::from(program_id);
+
+        let resp = simulator
+            .run_step(
+                owner,
+                &Step {
+                    endpoint: Endpoint::Execute,
+                    method: "add_liquidity".to_string(),
+                    max_units: u64::MAX,
+                    params: vec![test_context.clone().into(), 1000u64.into(), 1000u64.into()],
+                },
+            )
+            .unwrap()
+            .result
+            .response::<u64>()
+            .unwrap();
+
+        assert_eq!(resp, 1000);
+
+        simulator
+            .run_step(
+                owner,
+                &Step {
+                    endpoint: Endpoint::Execute,
+                    method: "swap".to_string(),
+                    max_units: u64::MAX,
+                    params: vec![test_context.clone().into(), 10u64.into(), true.into()],
+                },
+            )
+            .unwrap();
+
+        let (amount_x, amount_y) = simulator
+            .run_step(
+                owner,
+                &Step {
+                    endpoint: Endpoint::Execute,
+                    method: "remove_liquidity".to_string(),
+                    max_units: u64::MAX,
+                    params: vec![test_context.into(), 1000.into()],
+                },
+            )
+            .unwrap()
+            .result
+            .response::<(u64, u64)>()
+            .unwrap();
+
+        assert!(amount_x > 1000);
+        assert!(amount_y < 1000);
+    }
+}

x/programs/rust/wasmlanche-sdk/src/program.rs

@@ -32,7 +32,7 @@ impl BorshDeserialize for DeferDeserialize {
 }
 
 /// An error that is returned from call to public functions.
-#[derive(Error, Debug, BorshSerialize, BorshDeserialize, PartialEq, Eq)]
+#[derive(Error, Debug, BorshSerialize, BorshDeserialize)]
 #[repr(u8)]
 #[non_exhaustive]
 #[borsh(use_discriminant = true)]