Risk config

pulse-sdk/src/general.rs

@@ -1,4 +1,4 @@
-use hypersdk::hypercore::Side;
+use hypersdk::{dec, hypercore::Side};
 use rust_decimal::Decimal;
 
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
@@ -30,6 +30,12 @@ pub enum Mode {
     Manual,
 }
 
+#[derive(Debug, Clone)]
+pub enum Allocation {
+    Fixed(Decimal),
+    Percent(Decimal),
+}
+
 #[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
 pub struct Signal {
     pub symbol: String,
@@ -101,3 +107,66 @@ impl std::fmt::Display for ItemState {
         }
     }
 }
+
+impl<'de> serde::Deserialize<'de> for Allocation {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: serde::Deserializer<'de>,
+    {
+        struct AllocationVisitor;
+
+        impl<'de> serde::de::Visitor<'de> for AllocationVisitor {
+            type Value = Allocation;
+
+            fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
+                formatter.write_str("a number or percentage string")
+            }
+
+            fn visit_f64<E>(self, value: f64) -> Result<Self::Value, E>
+            where
+                E: serde::de::Error,
+            {
+                Ok(Allocation::Fixed(Decimal::from_f64_retain(value).unwrap()))
+            }
+
+            fn visit_str<E>(self, value: &str) -> Result<Self::Value, E>
+            where
+                E: serde::de::Error,
+            {
+                if let Some(percent) = value.strip_suffix('%') {
+                    let value = percent.parse::<Decimal>().map_err(E::custom)?;
+
+                    Ok(Allocation::Percent(value))
+                } else {
+                    Err(E::custom("invalid allocation format"))
+                }
+            }
+        }
+
+        deserializer.deserialize_any(AllocationVisitor)
+    }
+}
+
+impl serde::Serialize for Allocation {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: serde::Serializer,
+    {
+        match self {
+            Allocation::Fixed(value) => serde::Serialize::serialize(value, serializer),
+            Allocation::Percent(value) => {
+                let s = format!("{}%", value);
+                serializer.serialize_str(&s)
+            }
+        }
+    }
+}
+
+impl Allocation {
+    pub fn get(&self, full_alloc: Decimal) -> Decimal {
+        match self {
+            Self::Fixed(f) => *f,
+            Self::Percent(p) => (*p / dec!(100.0)) * full_alloc,
+        }
+    }
+}