stock-trek 0.7.4

Stock Trek time-series analysis
Documentation
use crate::{
    asset_id::AssetId,
    error::{
        result::{StockTrekError, StockTrekResult},
        value::ValueError,
    },
    exchange_id::ExchangeId,
};
use serde::{Deserialize, Serialize};
use strum::Display;

#[derive(Debug, Display, Clone, Serialize, Deserialize)]
pub enum ScratchValue {
    ExchangeId(ExchangeId),
    AssetId(AssetId),
    Flag(bool),
    Number(f64),
}

impl From<AssetId> for ScratchValue {
    fn from(value: AssetId) -> Self {
        ScratchValue::AssetId(value)
    }
}
impl From<ExchangeId> for ScratchValue {
    fn from(value: ExchangeId) -> Self {
        ScratchValue::ExchangeId(value)
    }
}
impl From<bool> for ScratchValue {
    fn from(value: bool) -> Self {
        ScratchValue::Flag(value)
    }
}
impl From<f64> for ScratchValue {
    fn from(value: f64) -> Self {
        ScratchValue::Number(value)
    }
}

impl TryFrom<ScratchValue> for ExchangeId {
    type Error = StockTrekError;
    fn try_from(value: ScratchValue) -> StockTrekResult<Self> {
        match value {
            ScratchValue::ExchangeId(e) => Ok(e),
            ScratchValue::AssetId(_) => err("ExchangeId", "AssetId"),
            ScratchValue::Flag(_) => err("ExchangeId", "Flag"),
            ScratchValue::Number(_) => err("ExchangeId", "Number"),
        }
    }
}
impl TryFrom<ScratchValue> for AssetId {
    type Error = StockTrekError;
    fn try_from(value: ScratchValue) -> StockTrekResult<Self> {
        match value {
            ScratchValue::ExchangeId(_) => err("AssetId", "ExchangeId"),
            ScratchValue::AssetId(a) => Ok(a),
            ScratchValue::Flag(_) => err("AssetId", "Flag"),
            ScratchValue::Number(_) => err("AssetId", "Number"),
        }
    }
}
impl TryFrom<ScratchValue> for bool {
    type Error = StockTrekError;
    fn try_from(value: ScratchValue) -> StockTrekResult<Self> {
        match value {
            ScratchValue::ExchangeId(_) => err("Flag", "ExchangeId"),
            ScratchValue::AssetId(_) => err("Flag", "AssetId"),
            ScratchValue::Flag(f) => Ok(f),
            ScratchValue::Number(_) => err("Flag", "Number"),
        }
    }
}
impl TryFrom<ScratchValue> for f64 {
    type Error = StockTrekError;
    fn try_from(value: ScratchValue) -> StockTrekResult<Self> {
        match value {
            ScratchValue::ExchangeId(_) => err("Number", "ExchangeId"),
            ScratchValue::AssetId(_) => err("Number", "AssetId"),
            ScratchValue::Flag(_) => err("Number", "Flag"),
            ScratchValue::Number(n) => Ok(n),
        }
    }
}

fn err<T>(expected: impl AsRef<str>, found: impl AsRef<str>) -> StockTrekResult<T> {
    Err(StockTrekError::Value(ValueError::IncorrectType {
        expected: expected.as_ref().to_string(),
        found: found.as_ref().to_string(),
    }))
}