brk_state 0.0.71

Various states used in the Bitcoin Research Kit
Documentation 
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

use std::{
    collections::BTreeMap,
    fs::{self, File},
    io::{BufReader, BufWriter},
    ops::{Deref, DerefMut},
    path::{Path, PathBuf},
};

use bincode::{Decode, Encode, config, decode_from_std_read, encode_into_std_write};
use brk_core::{Dollars, Height, Result, Sats};
use derive_deref::{Deref, DerefMut};
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug)]
pub struct PriceToAmount {
    pathbuf: PathBuf,
    height: Option<Height>,
    state: State,
}

#[derive(Clone, Default, Debug, Deref, DerefMut, Serialize, Deserialize, Encode, Decode)]
struct State(BTreeMap<Dollars, Sats>);

impl PriceToAmount {
    pub fn forced_import(path: &Path, name: &str) -> Self {
        Self::import(path, name).unwrap_or_else(|_| Self {
            pathbuf: Self::path_(path, name),
            height: None,
            state: State::default(),
        })
    }

    pub fn import(path: &Path, name: &str) -> Result<Self> {
        let path = Self::path_(path, name);
        fs::create_dir_all(&path)?;

        let config = config::standard();
        let file = File::open(Self::path_state_(&path))?;
        let mut reader = BufReader::new(file);
        let state = decode_from_std_read(&mut reader, config)?;

        Ok(Self {
            height: Height::try_from(Self::path_height_(&path).as_path()).ok(),
            pathbuf: path,
            state,
        })
    }

    pub fn reset(&mut self) -> Result<()> {
        self.clear();
        self.height = None;
        fs::remove_dir_all(&self.pathbuf)?;
        fs::create_dir_all(&self.pathbuf)?;
        Ok(())
    }

    pub fn flush(&mut self, height: Height) -> Result<()> {
        self.height = Some(height);
        height.write(&self.path_height())?;

        let config = config::standard();
        let file = File::create(self.path_state()).inspect_err(|_| {
            dbg!(self.path_state());
        })?;
        let mut writer = BufWriter::new(file);
        encode_into_std_write(&self.state, &mut writer, config)?;
        Ok(())
    }

    pub fn height(&self) -> Option<Height> {
        self.height
    }

    fn path_(path: &Path, name: &str) -> PathBuf {
        path.join(format!("{name}_price_to_amount"))
    }

    fn path_state(&self) -> PathBuf {
        Self::path_state_(&self.pathbuf)
    }
    fn path_state_(path: &Path) -> PathBuf {
        path.join("state")
    }

    fn path_height(&self) -> PathBuf {
        Self::path_height_(&self.pathbuf)
    }
    fn path_height_(path: &Path) -> PathBuf {
        path.join("height")
    }
}

impl Deref for PriceToAmount {
    type Target = BTreeMap<Dollars, Sats>;
    fn deref(&self) -> &Self::Target {
        &self.state
    }
}

impl DerefMut for PriceToAmount {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.state
    }
}