PriceImprovement fee policy

crates/autopilot/src/arguments.rs

@@ -349,6 +349,12 @@ pub struct FeePolicy {
     /// - Surplus with cap:
     /// surplus:0.5:0.06
     ///
+    /// - Price improvement without cap:
+    /// price_improvement:0.5:1.0
+    ///
+    /// - Price improvement with cap:
+    /// price_improvement:0.5:0.06
+    ///
     /// - Volume based:
     /// volume:0.1
     #[clap(long, env, default_value = "surplus:0.0:1.0")]
@@ -362,16 +368,23 @@ pub struct FeePolicy {
 }
 
 impl FeePolicy {
-    pub fn to_domain(self) -> domain::fee::Policy {
+    pub fn to_domain_raw(&self) -> domain::fee::PolicyRaw {
         match self.fee_policy_kind {
             FeePolicyKind::Surplus {
                 factor,
                 max_volume_factor,
-            } => domain::fee::Policy::Surplus {
+            } => domain::fee::PolicyRaw::Surplus {
+                factor,
+                max_volume_factor,
+            },
+            FeePolicyKind::PriceImprovement {
+                factor,
+                max_volume_factor,
+            } => domain::fee::PolicyRaw::PriceImprovement {
                 factor,
                 max_volume_factor,
             },
-            FeePolicyKind::Volume { factor } => domain::fee::Policy::Volume { factor },
+            FeePolicyKind::Volume { factor } => domain::fee::PolicyRaw::Volume { factor },
         }
     }
 }
@@ -380,6 +393,9 @@ impl FeePolicy {
 pub enum FeePolicyKind {
     /// How much of the order's surplus should be taken as a protocol fee.
     Surplus { factor: f64, max_volume_factor: f64 },
+    /// How much of the order's price improvement should be taken as a protocol
+    /// fee.
+    PriceImprovement { factor: f64, max_volume_factor: f64 },
     /// How much of the order's volume should be taken as a protocol fee.
     Volume { factor: f64 },
 }

crates/autopilot/src/database/fee_policies.rs

@@ -9,7 +9,7 @@ pub async fn insert_batch(
 ) -> Result<(), sqlx::Error> {
     let mut query_builder = QueryBuilder::new(
         "INSERT INTO fee_policies (auction_id, order_uid, kind, surplus_factor, \
-         max_volume_factor, volume_factor) ",
+         max_volume_factor, volume_factor, quote_sell_amount, quote_buy_amount) ",
     );
 
     query_builder.push_values(fee_policies, |mut b, fee_policy| {
@@ -18,7 +18,9 @@ pub async fn insert_batch(
             .push_bind(fee_policy.kind)
             .push_bind(fee_policy.surplus_factor)
             .push_bind(fee_policy.max_volume_factor)
-            .push_bind(fee_policy.volume_factor);
+            .push_bind(fee_policy.volume_factor)
+            .push_bind(fee_policy.quote_sell_amount)
+            .push_bind(fee_policy.quote_buy_amount);
     });
 
     query_builder.build().execute(ex).await.map(|_| ())
@@ -46,7 +48,7 @@ pub async fn fetch(
 
 #[cfg(test)]
 mod tests {
-    use {super::*, database::byte_array::ByteArray, sqlx::Connection};
+    use {super::*, bigdecimal::BigDecimal, database::byte_array::ByteArray, sqlx::Connection};
 
     #[tokio::test]
     #[ignore]
@@ -66,6 +68,8 @@ mod tests {
             surplus_factor: Some(0.1),
             max_volume_factor: Some(1.0),
             volume_factor: None,
+            quote_sell_amount: None,
+            quote_buy_amount: None,
         };
         // surplus fee policy with caps
         let fee_policy_2 = dto::FeePolicy {
@@ -75,6 +79,8 @@ mod tests {
             surplus_factor: Some(0.2),
             max_volume_factor: Some(0.05),
             volume_factor: None,
+            quote_sell_amount: None,
+            quote_buy_amount: None,
         };
         // volume based fee policy
         let fee_policy_3 = dto::FeePolicy {
@@ -84,19 +90,36 @@ mod tests {
             surplus_factor: None,
             max_volume_factor: None,
             volume_factor: Some(0.06),
+            quote_sell_amount: None,
+            quote_buy_amount: None,
+        };
+        // price improvement fee policy
+        let fee_policy_4 = dto::FeePolicy {
+            auction_id,
+            order_uid,
+            kind: dto::fee_policy::FeePolicyKind::PriceImprovement,
+            surplus_factor: Some(0.3),
+            max_volume_factor: Some(0.07),
+            volume_factor: None,
+            quote_sell_amount: Some(BigDecimal::new(100.into(), 1)),
+            quote_buy_amount: Some(BigDecimal::new(200.into(), 1)),
         };
         insert_batch(
             &mut db,
             vec![
                 fee_policy_1.clone(),
                 fee_policy_2.clone(),
                 fee_policy_3.clone(),
+                fee_policy_4.clone(),
             ],
         )
         .await
         .unwrap();
 
         let output = fetch(&mut db, 1, order_uid).await.unwrap();
-        assert_eq!(output, vec![fee_policy_1, fee_policy_2, fee_policy_3]);
+        assert_eq!(
+            output,
+            vec![fee_policy_1, fee_policy_2, fee_policy_3, fee_policy_4]
+        );
     }
 }

crates/autopilot/src/domain/fee/mod.rs

@@ -4,48 +4,45 @@
 //! we define the way to calculate the protocol fee based on the configuration
 //! parameters.
 
-use crate::{
-    boundary::{self},
-    domain,
+use {
+    crate::{
+        boundary::{self},
+        domain,
+    },
+    primitive_types::U256,
 };
 
 /// Constructs fee policies based on the current configuration.
 #[derive(Debug)]
 pub struct ProtocolFee {
-    policy: Policy,
+    policy: PolicyRaw,
     fee_policy_skip_market_orders: bool,
 }
 
 impl ProtocolFee {
-    pub fn new(policy: Policy, fee_policy_skip_market_orders: bool) -> Self {
+    pub fn new(policy: PolicyRaw, fee_policy_skip_market_orders: bool) -> Self {
         Self {
             policy,
             fee_policy_skip_market_orders,
         }
     }
 
     /// Get policies for order.
-    pub fn get(&self, order: &boundary::Order, quote: Option<&domain::Quote>) -> Vec<Policy> {
+    pub fn get(&self, order: &boundary::Order, quote: &domain::Quote) -> Vec<Policy> {
         match order.metadata.class {
             boundary::OrderClass::Market => {
                 if self.fee_policy_skip_market_orders {
                     vec![]
                 } else {
-                    vec![self.policy]
+                    vec![self.policy.to_domain(quote)]
                 }
             }
             boundary::OrderClass::Liquidity => vec![],
             boundary::OrderClass::Limit => {
                 if !self.fee_policy_skip_market_orders {
-                    return vec![self.policy];
+                    return vec![self.policy.to_domain(quote)];
                 }
 
-                // if the quote is missing, we can't determine if the order is outside the
-                // market price so we protect the user and not charge a fee
-                let Some(quote) = quote else {
-                    return vec![];
-                };
-
                 tracing::debug!(?order.metadata.uid, ?self.policy, ?order.data.sell_amount, ?order.data.buy_amount, ?quote, "checking if order is outside market price");
                 if boundary::is_order_outside_market_price(
                     &order.data.sell_amount,
@@ -55,7 +52,7 @@ impl ProtocolFee {
                     &quote.sell_amount,
                     &quote.fee,
                 ) {
-                    vec![self.policy]
+                    vec![self.policy.to_domain(quote)]
                 } else {
                     vec![]
                 }
@@ -81,6 +78,14 @@ pub enum Policy {
         /// Cap protocol fee with a percentage of the order's volume.
         max_volume_factor: f64,
     },
+    /// A price improvement corresponds to a situation where the order is
+    /// executed at a better price than the top quote. The protocol fee in such
+    /// case is calculated from a cut of this price improvement.
+    PriceImprovement {
+        factor: f64,
+        max_volume_factor: f64,
+        quote: Quote,
+    },
     /// How much of the order's volume should be taken as a protocol fee.
     /// The fee is taken in `sell` token for `sell` orders and in `buy`
     /// token for `buy` orders.
@@ -90,3 +95,48 @@ pub enum Policy {
         factor: f64,
     },
 }
+
+#[derive(Debug)]
+pub enum PolicyRaw {
+    Surplus { factor: f64, max_volume_factor: f64 },
+    PriceImprovement { factor: f64, max_volume_factor: f64 },
+    Volume { factor: f64 },
+}
+
+impl PolicyRaw {
+    pub fn to_domain(&self, quote: &domain::Quote) -> Policy {
+        match self {
+            PolicyRaw::Surplus {
+                factor,
+                max_volume_factor,
+            } => Policy::Surplus {
+                factor: *factor,
+                max_volume_factor: *max_volume_factor,
+            },
+            PolicyRaw::PriceImprovement {
+                factor,
+                max_volume_factor,
+            } => Policy::PriceImprovement {
+                factor: *factor,
+                max_volume_factor: *max_volume_factor,
+                quote: quote.clone().into(),
+            },
+            PolicyRaw::Volume { factor } => Policy::Volume { factor: *factor },
+        }
+    }
+}
+
+#[derive(Debug, Copy, Clone, PartialEq)]
+pub struct Quote {
+    pub sell_amount: U256,
+    pub buy_amount: U256,
+}
+
+impl From<domain::Quote> for Quote {
+    fn from(value: domain::Quote) -> Self {
+        Self {
+            sell_amount: value.sell_amount,
+            buy_amount: value.buy_amount,
+        }
+    }
+}

crates/autopilot/src/infra/persistence/dto/fee_policy.rs

@@ -1,4 +1,8 @@
-use crate::{boundary, domain};
+use {
+    crate::{boundary, domain},
+    bigdecimal::BigDecimal,
+    number::conversions::{big_decimal_to_u256, u256_to_big_decimal},
+};
 
 #[derive(Debug, Clone, PartialEq, sqlx::FromRow)]
 pub struct FeePolicy {
@@ -8,6 +12,8 @@ pub struct FeePolicy {
     pub surplus_factor: Option<f64>,
     pub max_volume_factor: Option<f64>,
     pub volume_factor: Option<f64>,
+    pub quote_sell_amount: Option<BigDecimal>,
+    pub quote_buy_amount: Option<BigDecimal>,
 }
 
 impl FeePolicy {
@@ -27,6 +33,8 @@ impl FeePolicy {
                 surplus_factor: Some(factor),
                 max_volume_factor: Some(max_volume_factor),
                 volume_factor: None,
+                quote_sell_amount: None,
+                quote_buy_amount: None,
             },
             domain::fee::Policy::Volume { factor } => Self {
                 auction_id,
@@ -35,6 +43,22 @@ impl FeePolicy {
                 surplus_factor: None,
                 max_volume_factor: None,
                 volume_factor: Some(factor),
+                quote_sell_amount: None,
+                quote_buy_amount: None,
+            },
+            domain::fee::Policy::PriceImprovement {
+                factor,
+                max_volume_factor,
+                quote,
+            } => Self {
+                auction_id,
+                order_uid: boundary::database::byte_array::ByteArray(order_uid.0),
+                kind: FeePolicyKind::Surplus,
+                surplus_factor: Some(factor),
+                max_volume_factor: Some(max_volume_factor),
+                volume_factor: None,
+                quote_sell_amount: Some(u256_to_big_decimal(&quote.sell_amount)),
+                quote_buy_amount: Some(u256_to_big_decimal(&quote.buy_amount)),
             },
         }
     }
@@ -50,6 +74,20 @@ impl From<FeePolicy> for domain::fee::Policy {
             FeePolicyKind::Volume => domain::fee::Policy::Volume {
                 factor: row.volume_factor.expect("missing volume factor"),
             },
+            FeePolicyKind::PriceImprovement => domain::fee::Policy::PriceImprovement {
+                factor: row.surplus_factor.expect("missing surplus factor"),
+                max_volume_factor: row.max_volume_factor.expect("missing max volume factor"),
+                quote: domain::fee::Quote {
+                    sell_amount: big_decimal_to_u256(
+                        &row.quote_sell_amount.expect("missing sell amount"),
+                    )
+                    .expect("sell amount is not a valid eth::U256"),
+                    buy_amount: big_decimal_to_u256(
+                        &row.quote_buy_amount.expect("missing buy amount"),
+                    )
+                    .expect("buy amount is not a valid eth::U256"),
+                },
+            },
         }
     }
 }
@@ -59,4 +97,5 @@ impl From<FeePolicy> for domain::fee::Policy {
 pub enum FeePolicyKind {
     Surplus,
     Volume,
+    PriceImprovement,
 }