brk: first commit

_src/parser/datasets/subs/mod.rs

@@ -0,0 +1,90 @@
+use allocative::Allocative;
+
+mod capitalization;
+mod input;
+mod price_paid;
+mod ratio;
+mod realized;
+mod recap;
+mod supply;
+mod unrealized;
+mod utxo;
+
+pub use capitalization::*;
+pub use input::*;
+pub use price_paid::*;
+pub use ratio::*;
+pub use realized::*;
+pub use recap::*;
+use struct_iterable::Iterable;
+pub use supply::*;
+pub use unrealized::*;
+pub use utxo::*;
+
+use crate::{
+    parser::datasets::AnyDataset,
+    structs::{Config, MapPath},
+};
+
+use super::AnyDatasetGroup;
+
+#[derive(Allocative, Iterable)]
+pub struct SubDataset {
+    pub capitalization: CapitalizationDataset,
+    pub input: InputSubDataset,
+    // pub output: OutputSubDataset,
+    pub price_paid: PricePaidSubDataset,
+    pub realized: RealizedSubDataset,
+    pub supply: SupplySubDataset,
+    pub unrealized: UnrealizedSubDataset,
+    pub utxo: UTXOSubDataset,
+}
+
+impl SubDataset {
+    pub fn import(
+        parent_path: &MapPath,
+        name: &Option<String>,
+        config: &Config,
+    ) -> color_eyre::Result<Self> {
+        let s = Self {
+            capitalization: CapitalizationDataset::import(parent_path, name, config)?,
+            input: InputSubDataset::import(parent_path, name, config)?,
+            // output: OutputSubDataset::import(parent_path)?,
+            price_paid: PricePaidSubDataset::import(parent_path, name, config)?,
+            realized: RealizedSubDataset::import(parent_path, name, config)?,
+            supply: SupplySubDataset::import(parent_path, name, config)?,
+            unrealized: UnrealizedSubDataset::import(parent_path, name, config)?,
+            utxo: UTXOSubDataset::import(parent_path, name, config)?,
+        };
+
+        Ok(s)
+    }
+}
+
+impl AnyDatasetGroup for SubDataset {
+    fn as_vec(&self) -> Vec<&(dyn AnyDataset + Send + Sync)> {
+        vec![
+            &self.capitalization,
+            &self.price_paid,
+            &self.realized,
+            &self.supply,
+            &self.unrealized,
+            &self.utxo,
+            &self.input,
+            // &self.output,
+        ]
+    }
+
+    fn as_mut_vec(&mut self) -> Vec<&mut dyn AnyDataset> {
+        vec![
+            &mut self.capitalization,
+            &mut self.price_paid,
+            &mut self.realized,
+            &mut self.supply,
+            &mut self.unrealized,
+            &mut self.utxo,
+            &mut self.input,
+            // &mut self.output,
+        ]
+    }
+}