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
use crate::config::{Error, LoggerConfig, PersistableConfig, SecureBackend};
use diem_types::chain_id::{self, ChainId};
use serde::{Deserialize, Serialize};
use std::path::{Path, PathBuf};
const DEFAULT_JSON_RPC_ENDPOINT: &str = "https://127.0.0.1:8080";
const DEFAULT_ROTATION_PERIOD_SECS: u64 = 604_800; const DEFAULT_SLEEP_PERIOD_SECS: u64 = 600; const DEFAULT_TXN_EXPIRATION_SECS: u64 = 3600; #[derive(Clone, Debug, Deserialize, PartialEq, Eq, Serialize)]
#[serde(default, deny_unknown_fields)]
pub struct KeyManagerConfig {
pub logger: LoggerConfig,
pub json_rpc_endpoint: String,
pub rotation_period_secs: u64,
pub secure_backend: SecureBackend,
pub sleep_period_secs: u64,
pub txn_expiration_secs: u64,
#[serde(deserialize_with = "chain_id::deserialize_config_chain_id")]
pub chain_id: ChainId,
}
impl Default for KeyManagerConfig {
fn default() -> KeyManagerConfig {
KeyManagerConfig {
json_rpc_endpoint: DEFAULT_JSON_RPC_ENDPOINT.into(),
logger: LoggerConfig::default(),
rotation_period_secs: DEFAULT_ROTATION_PERIOD_SECS,
secure_backend: SecureBackend::InMemoryStorage,
sleep_period_secs: DEFAULT_SLEEP_PERIOD_SECS,
txn_expiration_secs: DEFAULT_TXN_EXPIRATION_SECS,
chain_id: ChainId::test(),
}
}
}
impl KeyManagerConfig {
pub fn load<P: AsRef<Path>>(input_path: P) -> Result<Self, Error> {
Self::load_config(&input_path)
}
pub fn save<P: AsRef<Path>>(&mut self, output_path: P) -> Result<(), Error> {
self.save_config(&output_path)
}
pub fn set_data_dir(&mut self, data_dir: PathBuf) {
if let SecureBackend::OnDiskStorage(backend) = &mut self.secure_backend {
backend.set_data_dir(data_dir);
}
}
}