From d3e0fe913beedbcca37ba0c26e0dda00086dc43a Mon Sep 17 00:00:00 2001
From: Jesse Luehrs <doy@tozt.net>
Date: Thu, 22 Aug 2019 01:46:36 -0400
Subject: add basic implementation

---
 src/main.rs | 176 +++++++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 175 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/main.rs b/src/main.rs
index e7a11a9..f0a8831 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,3 +1,177 @@
+use std::io::{Read, Write};
+
+const PROJECT_NAME: &str = "ynab";
+
+pub fn api_key() -> std::path::PathBuf {
+    directories::ProjectDirs::from("", "", PROJECT_NAME)
+        .unwrap()
+        .config_dir()
+        .join("api-key")
+}
+
+pub fn read_api_key() -> String {
+    let mut key = String::new();
+    let key_file = api_key();
+    std::fs::File::open(key_file.clone())
+        .unwrap()
+        .read_to_string(&mut key)
+        .unwrap();
+    let key = key.trim();
+    key.to_string()
+}
+
 fn main() {
-    println!("Hello, world!");
+    let key = read_api_key();
+    let mut ynab_config = ynab_api::apis::configuration::Configuration::new();
+    ynab_config.api_key = Some(ynab_api::apis::configuration::ApiKey {
+        prefix: Some("Bearer".to_string()),
+        key: key.to_string(),
+    });
+    let api = ynab_api::apis::client::APIClient::new(ynab_config);
+    let budget_id = api
+        .budgets_api()
+        .get_budgets()
+        .unwrap()
+        .data
+        .budgets
+        .iter()
+        .next()
+        .unwrap()
+        .id
+        .clone();
+    let budget = api
+        .budgets_api()
+        .get_budget_by_id(&budget_id, 0)
+        .unwrap()
+        .data
+        .budget;
+
+    let mut file = std::fs::File::create("accounts.tsv").unwrap();
+    for account in budget.accounts.unwrap() {
+        if account.deleted {
+            continue;
+        }
+        file.write_all(
+            [
+                account.id.as_ref(),
+                account.name.as_ref(),
+                if account.on_budget { "1" } else { "0" },
+                if account.closed { "1" } else { "0" },
+                &format!("{}", account.balance),
+                &format!("{}", account.cleared_balance),
+                &format!("{}", account.uncleared_balance),
+            ]
+            .join("\t")
+            .as_bytes(),
+        )
+        .unwrap();
+        file.write_all(b"\n").unwrap();
+    }
+    file.sync_all().unwrap();
+
+    let mut file = std::fs::File::create("category_groups.tsv").unwrap();
+    for category_group in budget.category_groups.unwrap() {
+        if category_group.deleted {
+            continue;
+        }
+        file.write_all(
+            [
+                category_group.id.as_ref(),
+                category_group.name.as_ref(),
+                if category_group.hidden { "1" } else { "0" },
+            ]
+            .join("\t")
+            .as_bytes(),
+        )
+        .unwrap();
+        file.write_all(b"\n").unwrap();
+    }
+    file.sync_all().unwrap();
+
+    let mut file = std::fs::File::create("categories.tsv").unwrap();
+    for category in budget.categories.unwrap() {
+        if category.deleted {
+            continue;
+        }
+        file.write_all(
+            [
+                category.id.as_ref(),
+                category.category_group_id.as_ref(),
+                category.name.as_ref(),
+                if category.hidden { "1" } else { "0" },
+                &format!("{}", category.budgeted),
+                &format!("{}", category.activity),
+                &format!("{}", category.balance),
+            ]
+            .join("\t")
+            .as_bytes(),
+        )
+        .unwrap();
+        file.write_all(b"\n").unwrap();
+    }
+    file.sync_all().unwrap();
+
+    let mut file = std::fs::File::create("payees.tsv").unwrap();
+    for payee in budget.payees.unwrap() {
+        if payee.deleted {
+            continue;
+        }
+        file.write_all(
+            [
+                payee.id.as_ref(),
+                payee.name.as_ref(),
+                payee
+                    .transfer_account_id
+                    .unwrap_or_else(|| "\\N".to_string())
+                    .as_ref(),
+            ]
+            .join("\t")
+            .as_bytes(),
+        )
+        .unwrap();
+        file.write_all(b"\n").unwrap();
+    }
+    file.sync_all().unwrap();
+
+    let mut file = std::fs::File::create("transactions.tsv").unwrap();
+    for transaction in budget.transactions.unwrap() {
+        if transaction.deleted {
+            continue;
+        }
+        file.write_all(
+            [
+                transaction.id.as_ref(),
+                transaction.date.as_ref(),
+                format!("{}", transaction.amount).as_ref(),
+                transaction
+                    .memo
+                    .unwrap_or_else(|| "\\N".to_string())
+                    .as_ref(),
+                transaction.cleared.as_ref(),
+                if transaction.approved { "1" } else { "0" },
+                transaction
+                    .flag_color
+                    .unwrap_or_else(|| "\\N".to_string())
+                    .as_ref(),
+                transaction.account_id.as_ref(),
+                transaction
+                    .payee_id
+                    .unwrap_or_else(|| "\\N".to_string())
+                    .as_ref(),
+                transaction
+                    .category_id
+                    .unwrap_or_else(|| "\\N".to_string())
+                    .as_ref(),
+                transaction
+                    .transfer_account_id
+                    .unwrap_or_else(|| "\\N".to_string())
+                    .as_ref(),
+            ]
+            .join("\t")
+            .as_bytes(),
+        )
+        .unwrap();
+        file.write_all(b"\n").unwrap();
+    }
+    file.sync_all().unwrap();
 }
-- 
cgit v1.2.3-54-g00ecf