Rewrite fully second db table into result

src/pages/repetitions.rs

@@ -6,10 +6,13 @@ use crate::data_provider::card_sets::update_deck;
 use crate::data_provider::card_stats::{add_stat_list, load_stats_of_deck};
 use crate::navigation::Page::{History, PreviousPage, Repetition, RepetitionSettings};
 use crate::navigation::{NavigatedPage, Page, RootMessage};
+use crate::notification::{NotificationData, NotificationRequest};
 use crate::pages::history::HistoryState;
 use crate::pages::repetition::RepetitionState;
 use crate::pages::repetition_settings::RepetitionSettingsState;
 use crate::pages::repetitions::RepetitionsMessage::*;
+use crate::push_error;
+use crate::push_error_task;
 use crate::styling::*;
 use chrono::{Days, Local, Utc};
 use hashbrown::{HashMap, HashSet};
@@ -27,7 +30,6 @@ use iced_core::svg::Handle;
 use parking_lot::Mutex;
 use std::ops::Deref;
 use std::sync::Arc;
-use crate::notification::NotificationRequest;
 
 #[derive(Clone)]
 pub struct RepetitionsState {
@@ -51,7 +53,9 @@ impl NavigatedPage<RepetitionsMessage> for RepetitionsState {
 
             let selected_set = self.selected_deck_mut().unwrap();
             if selected_set.append_mode == AppendMode::Full {
-                Self::append_all_words(&mut clone.lock(), selected_set);
+                if let Err(_err) = Self::append_all_words(&mut clone.lock(), selected_set){
+                    return None;
+                }
                 selected_set.existing_words_indices = selected_set.available_words_indices.clone();
             } else {
                 if selected_set.available_words_indices.is_empty()
@@ -61,10 +65,13 @@ impl NavigatedPage<RepetitionsMessage> for RepetitionsState {
                 }
             }
 
-            Some(Repetition(RepetitionState::new(
-                selected_set.general_settings.clone(),
-                clone,
-            )))
+            if let Ok(repetition_state) =
+                RepetitionState::new(selected_set.general_settings.clone(), clone)
+            {
+                return Some(Repetition(repetition_state));
+            }
+
+            None
         } else if let GoToSettings = message {
             Some(RepetitionSettings(RepetitionSettingsState::new(
                 self.selected_deck().unwrap().index.unwrap(),
@@ -83,10 +90,9 @@ impl NavigatedPage<RepetitionsMessage> for RepetitionsState {
                     self.decks.remove(index);
                     self.clear_selection();
                 } else {
-                    self.decks[index].general_settings =
-                        state.decks[index].clone();
+                    self.decks[index].general_settings = state.decks[index].clone();
                     drop(state);
-                    self.fill_deck_word_indices();
+                    let _ = self.fill_deck_word_indices();
                 }
             } else {
                 drop(state);
@@ -139,16 +145,20 @@ impl NavigatedPage<RepetitionsMessage> for RepetitionsState {
             SelectSet(index) => {
                 self.select_deck(index);
                 let deck = self.selected_deck().unwrap();
-                if !deck.indices_ready {
-                    self.fill_deck_word_indices();
+                if !deck.indices_ready
+                    && let Err(err) = self.fill_deck_word_indices()
+                {
+                    return push_error_task!(err);
                 }
             }
             Save => {
                 let mut deck = self.selected_deck().unwrap().clone();
                 let mut state = self.state.lock();
                 {
-                    if let Err(e) = update_deck(&mut deck.general_settings, &state.connection){
-                        return Task::done(RootMessage::PushNotification(NotificationRequest::Error(e)))
+                    if let Err(e) = update_deck(&mut deck.general_settings, &state.connection) {
+                        return Task::done(RootMessage::PushNotification(
+                            NotificationRequest::Error(e),
+                        ));
                     }
                     if let Some(index) = deck.index {
                         state.decks[index] = deck.general_settings.clone();
@@ -212,7 +222,11 @@ impl NavigatedPage<RepetitionsMessage> for RepetitionsState {
 
                 let deck = self.selected_deck().unwrap();
                 let mut state = self.state.lock();
-                Self::append_words(&mut state, &deck.general_settings, adding.into_iter());
+                if let Err(err) =
+                    Self::append_words(&mut state, &deck.general_settings, adding.into_iter())
+                {
+                    return push_error_task!(err);
+                }
             }
         }
         Task::none()
@@ -548,7 +562,7 @@ impl RepetitionsState {
 }
 
 impl RepetitionsState {
-    fn append_all_words(state: &mut AppState, deck: &DeckViewData) {
+    fn append_all_words(state: &mut AppState, deck: &DeckViewData) -> Result<(), NotificationData> {
         let existing = &deck.existing_words_indices;
         let mut created_set = HashSet::with_capacity(existing.len());
         existing.iter().for_each(|c| {
@@ -559,13 +573,14 @@ impl RepetitionsState {
             .iter()
             .filter(|i| !created_set.contains(i))
             .cloned();
-        Self::append_words(state, deck, required);
+        Self::append_words(state, deck, required)?;
+        Ok(())
     }
     fn append_words(
         state: &mut AppState,
         deck: &DeckSettings,
         indices: impl Iterator<Item = usize>,
-    ) {
+    ) -> Result<(), NotificationData> {
         let words = &state.dictionary;
         let stats = &mut indices
             .map(|i| {
@@ -581,8 +596,9 @@ impl RepetitionsState {
             .collect::<Vec<_>>();
 
         if !stats.is_empty() {
-            add_stat_list(stats, &mut state.connection);
+            add_stat_list(stats, &mut state.connection)?;
         }
+        Ok(())
     }
 
     fn select_deck(&mut self, index: usize) {
@@ -602,10 +618,10 @@ impl RepetitionsState {
         self.selected_deck_index = None;
     }
 
-    fn fill_deck_word_indices(&mut self) {
+    fn fill_deck_word_indices(&mut self) -> Result<(), NotificationData> {
         let state = &self.state.lock();
         let mut deck = self.selected_deck().unwrap().clone();
-        let added: Vec<_> = load_stats_of_deck(&deck, &state.connection)
+        let added: Vec<_> = load_stats_of_deck(&deck, &state.connection)?
             .iter()
             .map(|c| self.word_id_index_map[&c.word_id])
             .collect();
@@ -613,6 +629,7 @@ impl RepetitionsState {
         deck.existing_words_indices = added;
         deck.available_words_indices = total;
         self.decks[self.selected_deck_index.unwrap()] = deck;
+        Ok(())
     }
 }
 #[derive(Clone)]