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>

1 files changed, 26 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)?
+}