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
113
114
115
116
117
118
119
120
|
use rust_decimal::Decimal;
use serde::{Deserialize, Serialize};
use std::convert::{AsRef, TryFrom};
use std::fs;
use std::hash::{Hash, Hasher};
use std::path::{Path, PathBuf};
use twox_hash::XxHash64;
type Account = String;
type Category = String;
#[derive(Debug)]
#[derive(Hash)]
#[derive(Deserialize, Serialize)]
enum TransactionKind {
Expense,
Income,
//TODO Transfer,
}
#[derive(Debug)]
#[derive(Hash)]
#[derive(Deserialize, Serialize)]
struct Transaction {
description: String,
category: Category,
amount: Decimal,
kind: TransactionKind,
from: Account,
to: Account,
}
#[derive(Debug)]
#[derive(Hash)]
#[derive(Deserialize, Serialize)]
struct Post {
transaction: Transaction,
removed: bool,
}
#[derive(Debug)]
struct Store {
root: PathBuf,
posts: Vec<Post>,
}
impl Store {
//TODO Result
fn open(root: PathBuf) -> Option<Self> {
Some(Self {
posts: Self::open_dir(&root)?,
root,
})
}
//TODO check if hash matches
//TODO Result
//TODO overkill? maybe we can use subfolders later on
fn open_dir(dir: &Path) -> Option<Vec<Post>> {
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 posts = Self::open_dir(&entry.path())?;
res.append(&mut posts);
} else {
res.push(Post::open(&entry.path())?);
}
}
Some(res)
}
fn write(&self) -> std::io::Result<()> {
for post in &self.posts {
let mut path = self.root.clone();
path.push(format!("{}", post.id()));
post.write(&path)?;
}
Ok(())
}
}
impl Post {
fn new(transaction: Transaction) -> Self {
Self {
transaction,
removed: false,
}
}
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()
}
}
fn main() {
// let post = Post::new(Transaction {
// description: "Test".to_string(),
// category: "Mat".to_string(),
// amount: Decimal::try_from(120.4_f64).unwrap(),
// kind: TransactionKind::Expense,
// from: "Kortkonto".to_string(),
// to: "Coop Lambohov".to_string(),
// });
// post.write(&"test").unwrap();
let store = Store::open(PathBuf::from("store")).unwrap();
// println!("{:#?}", store);
store.write().unwrap();
}
|
