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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
use rust_decimal::Decimal;
use serde::{Deserialize, Serialize};
use std::convert::AsRef;
use std::fs;
use std::hash::{Hash, Hasher};
use std::path::{Path, PathBuf};
use std::str::FromStr;
use structopt::StructOpt;
use twox_hash::XxHash64;
type Account = String;
type Category = String;
#[derive(Debug)]
#[derive(Hash)]
#[derive(Deserialize, Serialize)]
#[derive(StructOpt)]
pub enum TransactionKind {
Expense,
Income,
//TODO Transfer,
}
impl FromStr for TransactionKind {
type Err = String;
fn from_str(s: &str) -> Result<Self, Self::Err> {
match s {
"expense" => Ok(TransactionKind::Expense),
"income" => Ok(TransactionKind::Income),
_ => Err(format!("Unknown transaction kind: {:?}", s)),
}
}
}
#[derive(Debug)]
#[derive(Hash)]
#[derive(Deserialize, Serialize)]
pub struct Transaction {
pub description: String,
pub category: Category,
pub amount: Decimal,
pub kind: TransactionKind,
pub from: Account,
pub to: Account,
}
#[derive(Debug)]
pub struct Store {
root: PathBuf,
transactions: Vec<Transaction>,
new_transactions: Vec<Transaction>,
}
impl Store {
//TODO Result
pub fn open(root: PathBuf) -> Option<Self> {
Some(Self {
transactions: Self::open_dir(&root)?,
new_transactions: Vec::new(),
root,
})
}
//TODO check if hash matches
//TODO Result
//TODO overkill? maybe we can use subfolders later on
fn open_dir(dir: &Path) -> Option<Vec<Transaction>> {
let mut res = Vec::new();
for entry in std::fs::read_dir(dir).ok()? {
let entry = entry.ok()?;
if entry.file_type().ok()?.is_dir() {
let mut transactions = Self::open_dir(&entry.path())?;
res.append(&mut transactions);
} else {
res.push(Transaction::open(&entry.path())?);
}
}
Some(res)
}
pub fn push(&mut self, transaction: Transaction) {
self.new_transactions.push(transaction);
}
pub fn write(&self) -> std::io::Result<()> {
for transaction in &self.new_transactions {
let mut path = self.root.clone();
path.push(format!("{}", transaction.id()));
transaction.write(&path)?;
}
Ok(())
}
}
impl Transaction {
fn write<P: AsRef<Path>>(&self, p: &P) -> std::io::Result<()> {
fs::write(p, serde_json::to_string_pretty(self).unwrap()) //TODO control pretty or not
}
//TODO Result
fn open<P: AsRef<Path>>(p: &P) -> Option<Self> {
fs::read_to_string(p).ok().as_ref().and_then(|s| serde_json::from_str(s).ok())
}
fn id(&self) -> u64 {
let mut h = XxHash64::default();
self.hash(&mut h);
h.finish()
}
}
|
