pkgs/by-name/ts/tskm/src/state.rs


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

use std::path::PathBuf;

use anyhow::Result;
use taskchampion::{storage::AccessMode, Replica, StorageConfig};

pub struct State {
    replica: Replica,
}

impl std::fmt::Debug for State {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "State")
    }
}

impl State {
    fn taskdb_dir() -> PathBuf {
        dirs::data_local_dir().expect("Should exist").join("task")
    }

    fn new(taskdb_dir: PathBuf, access_mode: AccessMode) -> Result<Self> {
        let storage = StorageConfig::OnDisk {
            taskdb_dir,
            create_if_missing: false,
            access_mode,
        }
        .into_storage()?;

        let replica = Replica::new(storage);

        Ok(Self { replica })
    }

    pub fn new_ro() -> Result<Self> {
        Self::new(Self::taskdb_dir(), AccessMode::ReadOnly)
    }
    pub fn new_rw() -> Result<Self> {
        Self::new(Self::taskdb_dir(), AccessMode::ReadWrite)
    }

    #[must_use]
    pub fn replica(&mut self) -> &mut Replica {
        &mut self.replica
    }
}