rust_ledger 0.4.0

CLI accounting tool
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

mod account;
mod balance;
mod csv;
mod register;
mod version;

use crate::error::{Error, Result};
use crate::model::ledger::Group;
use pargs::Pargs;
use std::env;

pub fn run() -> Result<()> {
    let args: Vec<String> = env::args().collect();

    // define expected args for pargs
    let expected_command_args: Vec<String> = vec![
        String::from("account"),
        String::from("balance"),
        String::from("register"),
        String::from("csv"),
        String::from("version"),
    ];
    let expected_flag_args: Vec<String> = vec![];
    let expected_option_args: Vec<String> = vec![
        String::from("-f"),
        String::from("-o"),
        String::from("-g"),
        String::from("-s"),
    ];

    let Pargs {
        command_args,
        option_args,
        ..
    } = Pargs::parse(
        args,
        expected_command_args,
        expected_flag_args,
        expected_option_args,
    )?;

    let ledger_file_env = match std::env::var("RLEDGER_FILE") {
        Ok(p) => p,
        Err(err) => format!("{}", err),
    };

    let ledger_file = match option_args.get("-f") {
        Some(value) => value.as_str(),
        None => ledger_file_env.as_str(),
    };

    let options_arg = match option_args.get("-o") {
        Some(value) => String::from(value),
        None => String::from(""),
    };

    let group_arg = match option_args.get("-g") {
        Some(value) => match value.as_str() {
            "month" => Group::Month,
            "year" => Group::Year,
            _ => panic!("that group command was not recognized."),
        },
        None => Group::None,
    };

    let offset_arg = match option_args.get("-s") {
        Some(value) => String::from(value),
        None => String::from(""),
    };

    match &command_args.len() {
        0 => Err(Error::InvalidArg(String::from("please enter a command."))),
        _ => match &command_args[0][..] {
            "account" => account::account(ledger_file),
            "balance" => balance::balance(ledger_file),
            "register" => register::register(ledger_file, &options_arg, group_arg),
            "csv" => csv::csv(ledger_file, &options_arg, &offset_arg),
            "version" => version::version(),
            _ => panic!("command not found."),
        },
    }
}