ifs 0.1.34

util for write / read file
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

use std::{io::Read, path::PathBuf};

pub trait Msg = prost::Message + std::default::Default;

#[derive(Default)]
pub struct Item<T: Msg> {
  pub dir: PathBuf,
  pub name: String,
  t: std::marker::PhantomData<T>,
}

pub fn log_err(err: std::io::Error) {
  if err.kind() != std::io::ErrorKind::NotFound {
    tracing::error!("{}", err);
  }
}

impl<T: Msg> Item<T> {
  pub fn new(dir: impl Into<PathBuf>, name: impl Into<String>) -> Self {
    Self {
      dir: dir.into(),
      name: name.into(),
      t: std::marker::PhantomData,
    }
  }

  pub fn fp(&self) -> PathBuf {
    self.dir.join(&self.name)
    // BIN_HOME.join(&*self.dir).join(&self.name)
  }

  pub fn get(&self) -> Option<T> {
    let fp = self.fp();
    match std::fs::File::open(fp) {
      Ok(mut f) => {
        let mut data = Vec::new();
        if f.read_to_end(&mut data).is_ok() {
          if let Ok(r) = T::decode(&data[..]) {
            return Some(r);
          }
        }
      }
      Err(err) => {
        log_err(err);
      }
    }
    None
  }

  pub fn rm(&self) {
    if let Err(err) = std::fs::remove_file(self.fp()) {
      log_err(err);
    }
  }

  pub fn set(&self, msg: T) {
    use std::fs;
    let fp = self.fp();
    xerr::log!(fs::create_dir_all(fp.parent().unwrap()));
    xerr::log!(fs::write(fp, msg.encode_to_vec()));
  }
}

#[macro_export]
macro_rules! conf {
  ($cls:ident { $($name:ident : $ty:ident),* $(,)? }) => {
    pub struct $cls {
      dir: PathBuf,
    }
    // $(
    //     pub $name: $crate::conf::Item<api::$ty>
    // ),*

    impl<S: Into<std::path::PathBuf>> From<S> for $cls {
      fn from(s: S) -> Self {
        Self { dir: s.into() }
      }
    }

    impl $cls {
      $(
        #[allow(non_snake_case)]
        pub fn $name(&self) -> $crate::conf::Item<api::$ty> {
          $crate::conf::Item::new(&self.dir, stringify!($name))
        }
      )*
    }
  }
}