yarl 0.4.0

A simple cli ledger written in rust for keepnig track of all your finances
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

use std::{collections::HashSet, fs};

use chrono::Utc;
use clap::Parser;
use log::info;
use yarl::{cli::{Cli, Command}, database::{self, from_ron, to_ron, Transaction, TransactionKind}, log::{panic_hook, Logger}};

pub static LOGGER: Logger = Logger;

fn main() {
    // set logger & panic handling
    let _ = log::set_logger(&LOGGER).map(|()| log::set_max_level(log::LevelFilter::Info));
    std::panic::set_hook(Box::new(panic_hook));

    // parse cli
    let cli = Cli::parse();

    // open database
    let models = database::models();
    let database = database::open_or_create(cli.path, &models);
    let rw = database.rw_transaction().unwrap();

    // handle cli
    match cli.command {
        Command::Test => info!("hello, world!"),
        Command::Deposit { time, currency, amount, message, tags } => {
            let time = time.unwrap_or_else(|| Utc::now());
            rw.insert(Transaction {
                id: time.timestamp_millis(),
                currency,
                time,
                kind: TransactionKind::Deposit,
                amount,
                message,
                tags: tags.into_boxed_slice(),
            }).expect("failed to insert transaction into ledger");
        },
        Command::Withdraw { time, currency, amount, message, tags } => {
            let time = time.unwrap_or_else(|| Utc::now());
            rw.insert(Transaction {
                id: time.timestamp_millis(),
                currency,
                time,
                kind: TransactionKind::Withdraw,
                amount,
                message,
                tags: tags.into_boxed_slice(),
            }).expect("failed to insert transaction into ledger");
        },
        Command::Balance => {
            info!("getting & displaying balances");
            let balances = database::get_balance(&rw);
            for (key, val) in balances.into_iter() {
                info!("\x1b[33m[{key} balance]:\x1b[0m {}", val);
            }
        },
        Command::Export { path, tags, currency } => {
            info!("exporting ledger to RON");
            let transactions = database::get_sorted_transactions(&rw)
                .into_iter()
                .filter(|taction| tags.iter().all(|tag| taction.tags.contains(tag)))
                .filter(|taction| currency.as_ref().map(|c| taction.currency == *c).unwrap_or(true))
                .rev()
                .collect::<Vec<_>>();

            let ron = to_ron(&transactions);
            fs::write(&path, ron).expect(&format!("failed to write exported RON to {path}"));
        },
        Command::Import { path } => {
            info!("importing RON as a ledger");

            // Get imported data
            let contents = fs::read_to_string(path).expect("failed to open file");
            let transactions = from_ron(&contents);

            // Insert into the database
            for taction in transactions {
                rw.insert(taction).unwrap();
            }
        },
        Command::Tags => {
            info!("searching for tags in the ledger");
            let tags = rw
                .scan()
                .primary::<Transaction>()
                .unwrap()
                .all()
                .map(|taction| taction.unwrap().tags)
                .flatten()
                .fold(HashSet::new(), |mut set, tag| { set.insert(tag); set } );
            info!("\x1b[33m[tags]:\x1b[0m {:?}", tags);
        },
    }

    rw.commit().unwrap();
    info!("successfully completed all actions");
}