Use SQLite to store counter

1 files changed, 31 insertions, 3 deletions

diff --git a/src/main/rust/store/mod.rs b/src/main/rust/store/mod.rs
index 42f1809..7c7d9ce 100644
--- a/src/main/rust/store/mod.rs
+++ b/src/main/rust/store/mod.rs
@@ -1,11 +1,23 @@
-pub mod sqlite;
+mod sqlite;
 
-use miette::Result;
+use std::num::TryFromIntError;
+
+use ::sqlite::Error as SQLiteLibraryError;
+use miette::{Diagnostic, Result};
+use thiserror::Error;
 
 pub enum Context {
     SQLite(sqlite::SQLite),
 }
 
+#[derive(Debug, Diagnostic, Error)]
+pub enum ContextError {
+    #[error("SQLite error")]
+    SQLiteError(#[from] SQLiteLibraryError),
+    #[error("Int conversion error")]
+    ConversionError(#[from] TryFromIntError),
+}
+
 impl Default for Context {
     fn default() -> Self {
         Context::SQLite(sqlite::SQLite::default())
@@ -13,7 +25,23 @@ impl Default for Context {
 }
 
 pub trait Store {
-    fn init(&mut self) -> Result<()>;
+    fn init(&mut self) -> Result<(), ContextError>;
+    fn get_counter(&self) -> Result<u8, ContextError>;
+    fn set_counter(&mut self, new_count: u8) -> Result<u8, ContextError>;
+}
+
+impl Context {
+    pub fn unwrap_store(&mut self) -> &mut impl Store {
+        match self {
+            Context::SQLite(store) => store,
+        }
+    }
+
+    pub fn get_store(&self) -> &impl Store {
+        match self {
+            Context::SQLite(store) => store,
+        }
+    }
 }
 
 #[cfg(test)]