acick_config/
session_config.rs

1use std::time::Duration;
2
3use getset::{CopyGetters, Getters};
4use lazy_static::lazy_static;
5use serde::{Deserialize, Serialize};
6
7use crate::abs_path::AbsPathBuf;
8use crate::DATA_LOCAL_DIR;
9
10static COOKIES_FILE_NAME: &str = "cookies.json";
11
12lazy_static! {
13    static ref DEFAULT_COOKIES_PATH: AbsPathBuf = DATA_LOCAL_DIR.join(COOKIES_FILE_NAME);
14}
15
16const DEFAULT_TIMEOUT: Duration = Duration::from_secs(30);
17const DEFAULT_RETRY_LIMIT: usize = 4;
18const DEFAULT_RETRY_INTERVAL: Duration = Duration::from_secs(2);
19
20#[derive(Serialize, Deserialize, Getters, CopyGetters, Debug, Clone, PartialEq, Eq, Hash)]
21#[serde(default)]
22pub struct SessionConfig {
23    #[serde(with = "humantime_serde")]
24    #[get_copy = "pub"]
25    timeout: Duration,
26    #[serde(skip_serializing_if = "Option::is_none")]
27    cookies_path: Option<AbsPathBuf>,
28    #[get_copy = "pub"]
29    retry_limit: usize,
30    #[serde(with = "humantime_serde")]
31    #[get_copy = "pub"]
32    retry_interval: Duration,
33}
34
35impl SessionConfig {
36    pub fn default_in_dir(base_dir: &AbsPathBuf) -> Self {
37        Self {
38            timeout: DEFAULT_TIMEOUT,
39            cookies_path: Some(base_dir.join(COOKIES_FILE_NAME)),
40            retry_limit: DEFAULT_RETRY_LIMIT,
41            retry_interval: DEFAULT_RETRY_INTERVAL,
42        }
43    }
44
45    pub fn cookies_path(&self) -> &AbsPathBuf {
46        self.cookies_path.as_ref().unwrap_or(&DEFAULT_COOKIES_PATH)
47    }
48}
49
50impl Default for SessionConfig {
51    fn default() -> Self {
52        Self {
53            timeout: DEFAULT_TIMEOUT,
54            cookies_path: None,
55            retry_limit: DEFAULT_RETRY_LIMIT,
56            retry_interval: DEFAULT_RETRY_INTERVAL,
57        }
58    }
59}