toru/src/state.rs

132 lines
3.7 KiB
Rust
Raw Normal View History

use crate::error;
use crate::tasks;
use crate::tasks::Id;
2022-08-20 04:01:43 +00:00
use std::fs;
use std::path;
use std::io;
use std::io::{Write, Seek};
use std::collections::HashMap;
2022-08-20 04:01:43 +00:00
use serde_with::{serde_as, DisplayFromStr};
2022-08-20 04:01:43 +00:00
pub struct State {
file : fs::File,
pub data : InternalState,
}
#[serde_as]
2022-08-20 04:01:43 +00:00
#[derive(serde::Serialize, serde::Deserialize)]
pub struct InternalState {
pub next_id : Id,
#[serde_as(as = "HashMap<DisplayFromStr, _>")]
pub index : HashMap<String, Vec<Id>>,
2022-08-20 04:01:43 +00:00
}
impl State {
/// This function should be called after creating or checking that the "notes" folder exists.
pub fn load(vault_location : &path::Path) -> Result<Self, error::Error> {
let path = vault_location.join("state.toml");
if path.exists() && path.is_file() {
// Read file before opening (and truncating).
let contents = fs::read_to_string(&path)?;
let file = fs::File::options()
.write(true)
.create(true)
.open(&path)?;
let data = toml::from_str::<InternalState>(&contents)?;
Ok(Self {
file,
data,
})
}
else {
// Calculating the next ID if necessary.
2022-08-20 04:01:43 +00:00
let mut max_id : i128 = -1;
for id in vault_location.join("notes").read_dir()?.filter_map(|p| p.ok()).map(|p| p.path()).filter(|p| p.extension().map(|s| s.to_str()) == Some(Some("toml"))).filter_map(|p| p.file_stem().map(|x| x.to_str().map(|y| y.to_string()))).flatten().filter_map(|p| p.parse::<Id>().ok()) {
if i128::try_from(id).unwrap() > max_id {
max_id = i128::from(id);
}
}
// Calculating out the index.
let tasks = tasks::Task::load_all(vault_location, true)?;
let mut index : HashMap<String, Vec<Id>> = HashMap::with_capacity(tasks.len());
for task in tasks {
match index.get_mut(&task.data.name) {
Some(ids) => {
ids.push(task.data.id);
},
None => {
index.insert(task.data.name.clone(), vec![task.data.id]);
},
}
}
2022-08-20 04:01:43 +00:00
let data = InternalState {
next_id : u64::try_from(max_id + 1).unwrap(),
index,
2022-08-20 04:01:43 +00:00
};
let mut file = fs::File::options()
.write(true)
.create(true)
.open(&path)?;
file.set_len(0)?;
file.seek(io::SeekFrom::Start(0))?;
file.write_all(toml::to_string(&data)?.as_bytes())?;
let task = Self {
file,
data,
};
Ok(task)
}
}
pub fn save(self) -> Result<(), error::Error> {
let Self {
mut file,
data,
} = self;
file.set_len(0)?;
file.seek(io::SeekFrom::Start(0))?;
file.write_all(toml::to_string(&data)?.as_bytes())?;
Ok(())
}
pub fn index_insert(&mut self, name : String, id : Id) {
match self.data.index.get_mut(&name) {
Some(ids) => {
ids.push(id);
},
None => {
self.data.index.insert(name, vec![id]);
}
}
}
pub fn index_remove(&mut self, name : String, id : Id) {
if let Some(mut ids) = self.data.index.remove(&name) {
if let Some(index) = ids.iter().position(|i| i == &id) {
ids.swap_remove(index);
if !ids.is_empty() {
self.data.index.insert(name, ids);
}
}
}
}
2022-08-20 04:01:43 +00:00
}