apple-quant-algorithmic 0.1.0

use thiserror::Error;

#[derive(Debug, Error)]
pub enum OrderIdError {
	#[error("Invalid: {0:?}.")]
	Invalid(OrderId),

	#[error("Unavailable order id.")]
	Unavailable,
}

/// Numerical identifier of different types of orders assigned by the client.
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub enum OrderId {
	Local(LocalOrderId),
	Remote(RemoteOrderId),
}

impl OrderId {
	pub fn err_invalid<T, E: From<OrderIdError>>(self) -> Result<T, E> {
		Err(OrderIdError::Invalid(self).into())
	}
}

impl From<LocalOrderId> for OrderId {
	fn from(value: LocalOrderId) -> Self {
		Self::Local(value)
	}
}

impl From<RemoteOrderId> for OrderId {
	fn from(value: RemoteOrderId) -> Self {
		Self::Remote(value)
	}
}

impl From<&LocalOrderId> for OrderId {
	fn from(value: &LocalOrderId) -> Self {
		Self::Local(*value)
	}
}

impl From<&RemoteOrderId> for OrderId {
	fn from(value: &RemoteOrderId) -> Self {
		Self::Remote(*value)
	}
}

pub struct OrderIdGenerator {
	local_order_id: u64,
}

impl OrderIdGenerator {
	pub(crate) fn new() -> Self {
		Self { local_order_id: 0 }
	}

	pub fn next_local_order_id(&mut self) -> LocalOrderId {
		self.local_order_id += 1;
		LocalOrderId::new_from_u64(self.local_order_id)
	}
}

/// Numerical identifier of an order that will be satisfied locally (eg. stop orders).
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct LocalOrderId(u64);

impl LocalOrderId {
	pub(crate) fn new_from_u64(value: u64) -> Self {
		Self(value)
	}

	pub fn err_invalid<T, E: From<OrderIdError>>(self) -> Result<T, E> {
		Err(OrderIdError::Invalid(OrderId::Local(self)).into())
	}

	pub fn as_order_id(&self) -> OrderId {
		self.into()
	}

	pub fn into_order_id(self) -> OrderId {
		self.into()
	}
}

/// Numerical identifier of an order that will be satisfied remotely (eg. market and limit orders).
#[derive(Debug, Clone, Copy, PartialEq, Eq, PartialOrd, Ord, Hash)]
pub struct RemoteOrderId(u64);

impl RemoteOrderId {
	pub(crate) fn new_from_u64(value: u64) -> Self {
		Self(value)
	}

	pub fn err_invalid<T, E: From<OrderIdError>>(self) -> Result<T, E> {
		Err(OrderIdError::Invalid(OrderId::Remote(self)).into())
	}

	pub fn as_order_id(&self) -> OrderId {
		self.into()
	}

	pub fn into_order_id(self) -> OrderId {
		self.into()
	}
}