AB transformer takes DB urls as command line arguments

1 files changed, 12 insertions, 4 deletions

diff --git a/rust/actualbudget_to_schist_transformer/src/run.rs b/rust/actualbudget_to_schist_transformer/src/run.rs
index 77f0e45..3c88589 100644
--- a/rust/actualbudget_to_schist_transformer/src/run.rs
+++ b/rust/actualbudget_to_schist_transformer/src/run.rs
@@ -1,11 +1,19 @@
-use anyhow::Result;
+use anyhow::{Context, Result};
 use diesel::{Connection, SqliteConnection};
 
 use crate::{actualbudget_state::read_actualbudget_state, schist_state::export_schist_state, transform_state::transform_state};
 
-pub fn run() -> Result<()> {
-    let actualbudget_connection = &mut SqliteConnection::establish("in.sqlite").unwrap();
-    let schist_connection = &mut SqliteConnection::establish("out.sqlite").unwrap();
+pub fn run(in_db_url: &str, out_db_url: &str) -> Result<()> {
+    let actualbudget_connection = &mut SqliteConnection::establish(in_db_url)
+        .with_context(|| format!(
+            "failed to establish connection to input database at {}",
+            in_db_url,
+        ))?;
+    let schist_connection = &mut SqliteConnection::establish(out_db_url)
+        .with_context(|| format!(
+            "failed to establish connection to output database at {}",
+            out_db_url,
+        ))?;
 
     let actualbudget_state = read_actualbudget_state(actualbudget_connection)?;
     let schist_state = transform_state(actualbudget_state)?;