1 files changed, 68 insertions, 5 deletions
diff --git a/src/app/store/mod.rs b/src/app/store/mod.rs
index 70df5c2..cd4b859 100644
--- a/src/app/store/mod.rs
+++ b/src/app/store/mod.rs
@@ -44,6 +44,7 @@ impl Default for AppContext {
     fn default() -> Self {
         AppContext {
             context: Context::default(),
+            // these defaults get overwritten when #init_context is run
             accounts: vec![],
             symbols: vec![],
             events: vec![],
@@ -62,17 +63,79 @@ pub trait Store {
 }
 
 impl AppContext {
-    pub fn unwrap_store(&mut self) -> &mut impl Store {
-        match &mut self.context {
-            Context::SQLite(store) => store,
-        }
+    /// Returns a mutable version of the generic implementation of the store
+    /// along with references to the objects in context. Both store and context
+    /// need to be updated in tandem, so a tuple must be returned to be able to
+    /// produce mutable references in the same lifetime
+    fn unwrap_store(
+        &mut self,
+    ) -> (
+        &mut impl Store,
+        &mut Vec<FiduciaryAccount>,
+        &mut Vec<EquitySymbol>,
+        &mut Vec<EquityChangeEvent>,
+    ) {
+        (
+            match &mut self.context {
+                Context::SQLite(store) => store,
+            },
+            &mut self.accounts,
+            &mut self.symbols,
+            &mut self.events,
+        )
     }
 
-    pub fn get_store(&self) -> &impl Store {
+    fn get_store(&self) -> &impl Store {
         match &self.context {
             Context::SQLite(store) => store,
         }
     }
+
+    pub fn init_context(&mut self) -> Result<(), ContextError> {
+        self.unwrap_store().0.init()?;
+
+        self.accounts = self.get_store().get_accounts()?;
+        self.symbols = self.get_store().get_symbols()?;
+        self.events = self.get_store().get_events()?;
+
+        Ok(())
+    }
+
+    pub fn get_accounts(&self) -> &Vec<FiduciaryAccount> {
+        &self.accounts
+    }
+
+    pub fn get_symbols(&self) -> &Vec<EquitySymbol> {
+        &self.symbols
+    }
+
+    pub fn get_events(&self) -> &Vec<EquityChangeEvent> {
+        &self.events
+    }
+
+    pub fn add_account(&mut self, account: FiduciaryAccount) -> Result<(), ContextError> {
+        let (store, accounts, ..) = self.unwrap_store();
+        let acc = accounts.push_mut(account);
+
+        store.add_account(acc)?;
+        Ok(())
+    }
+
+    pub fn add_symbol(&mut self, symbol: EquitySymbol) -> Result<(), ContextError> {
+        let (store, _, symbols, ..) = self.unwrap_store();
+        let sym = symbols.push_mut(symbol);
+
+        store.add_symbol(sym)?;
+        Ok(())
+    }
+
+    pub fn add_event(&mut self, event: EquityChangeEvent) -> Result<(), ContextError> {
+        let (store, .., events) = self.unwrap_store();
+        let ev = events.push_mut(event);
+
+        store.add_event(ev)?;
+        Ok(())
+    }
 }
 
 #[cfg(test)]