Initial import of just-about-working version

src/main.rs

@@ -0,0 +1,141 @@
+use clap::{Parser, Subcommand};
+//use directories::ProjectDirs;
+use std::{
+    error::Error,
+    fs::File,
+    io::{Read, stdin, stdout},
+    path::PathBuf,
+};
+
+use ynabmunger::ynab;
+use ynabmunger::{Transaction, read_transactions};
+
+fn output_csv(transactions: &[Transaction]) -> Result<(), Box<dyn Error>> {
+    let mut writer = csv::Writer::from_writer(stdout());
+
+    writer
+        .write_record(["Date", "Payee", "Memo", "Amount"])
+        .unwrap();
+    for transaction in transactions {
+        let output = transaction.to_record();
+        writer.write_record(output).unwrap();
+    }
+    writer.flush().unwrap();
+    Ok(())
+}
+
+#[derive(Debug)]
+enum Output {
+    Csv,
+    Ynab {
+        token: String,
+        plan: ynab::Lookup,
+        account: ynab::Lookup,
+    },
+}
+
+#[derive(Subcommand)]
+enum Command {
+    Csv {
+        inputs: Vec<PathBuf>,
+    },
+    Ynab {
+        #[arg(short, long, env = "YNAB_API_TOKEN")]
+        token: String,
+
+        #[arg(short, long, help = "Plan name", group = "planref")]
+        plan: Option<String>,
+
+        #[arg(short = 'P', long, help = "Plan ID", group = "planref")]
+        plan_id: Option<String>,
+
+        #[arg(short, long, help = "Account name", group = "accountref")]
+        account: Option<String>,
+
+        #[arg(short = 'A', long, help = "Account ID", group = "accountref")]
+        account_id: Option<String>,
+
+        inputs: Vec<PathBuf>,
+    },
+}
+
+#[derive(Parser)]
+struct Cli {
+    #[arg(short, long, help = "Bank export format", default_value_t = String::from("bulder"))]
+    format: String,
+
+    #[command(subcommand)]
+    command: Command,
+}
+
+fn main() -> Result<(), Box<dyn Error>> {
+    let cli = Cli::parse();
+
+    // Inject config file stuff here
+
+    let inputs: Vec<PathBuf>;
+    let output = match cli.command {
+        Command::Ynab {
+            plan,
+            plan_id,
+            account_id,
+            account,
+            inputs: inp,
+            token,
+        } => {
+            inputs = inp;
+
+            let output_plan = if let Some(plan) = plan {
+                Ok(ynab::Lookup::Name(plan))
+            } else if let Some(plan_id) = plan_id {
+                Ok(ynab::Lookup::Id(plan_id))
+            } else {
+                Err("no plan name or id".to_string())
+            }?;
+
+            let output_account = if let Some(account) = account {
+                Ok(ynab::Lookup::Name(account))
+            } else if let Some(account_id) = account_id {
+                Ok(ynab::Lookup::Id(account_id))
+            } else {
+                Err("no account name or id".to_string())
+            }?;
+
+            Output::Ynab {
+                token,
+                plan: output_plan,
+                account: output_account,
+            }
+        }
+        Command::Csv { inputs: inp } => {
+            inputs = inp;
+            Output::Csv
+        }
+    };
+
+    // Config file reconciliation here?
+
+    let transactions: Vec<Transaction> = if inputs.is_empty() {
+        vec![Box::new(stdin()) as Box<dyn Read>]
+    } else {
+        inputs
+            .into_iter()
+            .map(|p| -> Result<Box<dyn Read>, _> { Ok(Box::new(File::open(p)?)) })
+            .collect::<Result<Vec<_>, std::io::Error>>()?
+    }
+    .iter_mut()
+    .map(|s| read_transactions(s, &cli.format))
+    .collect::<Result<Vec<_>, _>>()?
+    .into_iter()
+    .flatten()
+    .collect();
+
+    match output {
+        Output::Ynab {
+            token,
+            plan,
+            account,
+        } => ynab::Ynab::new(&token, plan, account)?.upload(&transactions),
+        Output::Csv => output_csv(&transactions),
+    }
+}