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
|
use xdg;
use rusqlite::{Result, Connection};
mod database;
mod cli;
use cli::{Cli, Commands};
use crate::database::Database;
fn main() -> Result<()> {
let cli = Cli::new();
match cli.action {
Commands::Init => {
if let Some(datapath) = cli.database_path {
Database::init_database(&datapath)?;
} else {
let datapath = xdg::BaseDirectories::with_prefix("debt")
.expect("xdg file setup failed")
.place_data_file("cache.db")
.expect("xdg directory creaton failed");
Database::init_database(&datapath)?;
}
},
Commands::Register {person, amount, note, payment} => {
let register = Database::new(person, amount as i32, note, !payment);
let datapath = cli.database_path.unwrap();
register.add_register(&datapath)?;
},
Commands::View {agent, filter_note, history, total} => {
let datapath = cli.database_path.expect("database path not found");
let conn = Connection::open(&datapath)?;
match (history, total) {
(true, false) => Database::view_history(conn, agent, filter_note)?,
(false, true) => Database::view_total(conn)?,
(false, false) => Database::view_total(conn)?,
(true, true) => unreachable!()
}
},
Commands::Agent {name, add, delete, update} => {
let register = Database::new(name, 0, None, false);
let datapath = cli.database_path.unwrap();
match (add, delete, update) {
(true, false, None) => register.add_agent(&datapath)?,
(false, true, None) => register.delete_agent(&datapath)?,
(false, false, None) => register.add_agent(&datapath)?,
(false, false, Some(new_name)) => register.update_agent(&datapath, new_name)?,
(_, _, _) => unreachable!()
}
}
};
Ok(())
}
|
