task-072

Co-authored-by: Joe Carstairs <jcarstairs@scottlogic.com>
Reviewed-on: https://git.joeac.net/joeac/schist/pulls/3
Co-authored-by: Joe Carstairs <me@joeac.net>
Co-committed-by: Joe Carstairs <me@joeac.net>

3 files changed, 55 insertions, 0 deletions

diff --git a/schist_desktop_gui/src/dialog/import_from_actualbudget_dialog.rs b/schist_desktop_gui/src/dialog/import_from_actualbudget_dialog.rs
new file mode 100644
index 0000000..10d4ec6
--- /dev/null
+++ b/schist_desktop_gui/src/dialog/import_from_actualbudget_dialog.rs
@@ -0,0 +1,26 @@
+use actualbudget_to_schist_transformer::{
+    actualbudget_state::read_actualbudget_state, schist_state::SchistState,
+    transform_state::transform_state,
+};
+use anyhow::anyhow;
+use diesel::{Connection, SqliteConnection};
+
+pub fn import_from_actualbudget_dialog(
+) -> Option<Result<SchistState, (std::path::PathBuf, anyhow::Error)>> {
+    let path = rfd::FileDialog::new()
+        .add_filter("SQLite database", &["sqlite"])
+        .set_title("Import Actualbudget database")
+        .pick_file();
+
+    path.as_ref()
+        .map(|path| import_from_actualbudget(path).map_err(|err| (path.clone(), err)))
+}
+
+fn import_from_actualbudget(path: &std::path::PathBuf) -> anyhow::Result<SchistState> {
+    let path = path
+        .to_str()
+        .ok_or_else(|| anyhow!("Failed to stringify path: {:?}", path))?;
+    SqliteConnection::establish(path)
+        .map(|mut conn| read_actualbudget_state(&mut conn))?
+        .map(transform_state)?
+}
diff --git a/schist_desktop_gui/src/dialog/new_file_dialog.rs b/schist_desktop_gui/src/dialog/new_file_dialog.rs
new file mode 100644
index 0000000..b950f68
--- /dev/null
+++ b/schist_desktop_gui/src/dialog/new_file_dialog.rs
@@ -0,0 +1,15 @@
+use crate::{database_connection::DatabaseConnectionResult, DatabaseConnection};
+
+pub fn new_file_dialog<P>(starting_directory: Option<P>) -> Option<DatabaseConnectionResult>
+where
+    P: AsRef<std::path::Path>,
+{
+    let mut file_dialog = rfd::FileDialog::new()
+        .add_filter("SQLite database", &["sqlite"])
+        .set_file_name("schist.sqlite")
+        .set_title("New file");
+    if let Some(starting_directory) = starting_directory {
+        file_dialog = file_dialog.set_directory(starting_directory);
+    }
+    file_dialog.save_file().map(DatabaseConnection::establish)
+}
diff --git a/schist_desktop_gui/src/dialog/pick_file_dialog.rs b/schist_desktop_gui/src/dialog/pick_file_dialog.rs
new file mode 100644
index 0000000..53d36c6
--- /dev/null
+++ b/schist_desktop_gui/src/dialog/pick_file_dialog.rs
@@ -0,0 +1,14 @@
+use crate::{database_connection::DatabaseConnectionResult, DatabaseConnection};
+
+pub fn pick_file_dialog<P>(starting_directory: Option<P>) -> Option<DatabaseConnectionResult>
+where
+    P: AsRef<std::path::Path>,
+{
+    let mut file_dialog = rfd::FileDialog::new()
+        .add_filter("SQLite database", &["sqlite"])
+        .set_title("Open file");
+    if let Some(starting_directory) = starting_directory {
+        file_dialog = file_dialog.set_directory(starting_directory);
+    }
+    file_dialog.pick_file().map(DatabaseConnection::establish)
+}