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
105
106
107
108
109
use crate::{commands::KmsCommand, config::KmsConfig};
use abscissa_core::{
application::{self, AppCell},
trace, Application, FrameworkError, StandardPaths,
};
pub static APPLICATION: AppCell<KmsApplication> = AppCell::new();
pub fn app_reader() -> application::lock::Reader<KmsApplication> {
APPLICATION.read()
}
pub fn app_writer() -> application::lock::Writer<KmsApplication> {
APPLICATION.write()
}
pub fn app_config() -> abscissa_core::config::Reader<KmsApplication> {
abscissa_core::config::Reader::new(&APPLICATION)
}
#[derive(Debug)]
pub struct KmsApplication {
config: Option<KmsConfig>,
state: application::State<Self>,
}
impl Default for KmsApplication {
fn default() -> Self {
Self {
config: None,
state: application::State::default(),
}
}
}
impl Application for KmsApplication {
type Cmd = KmsCommand;
type Cfg = KmsConfig;
type Paths = StandardPaths;
fn config(&self) -> &KmsConfig {
self.config.as_ref().expect("not configured yet")
}
fn state(&self) -> &application::State<Self> {
&self.state
}
fn state_mut(&mut self) -> &mut application::State<Self> {
&mut self.state
}
fn register_components(&mut self, command: &Self::Cmd) -> Result<(), FrameworkError> {
#[allow(unused_mut)]
let mut components = self.framework_components(command)?;
#[cfg(feature = "tx-signer")]
components.push(Box::new(abscissa_tokio::TokioComponent::new()?));
self.state.components.register(components)
}
fn after_config(&mut self, config: Self::Cfg) -> Result<(), FrameworkError> {
self.state.components.after_config(&config)?;
self.config = Some(config);
Ok(())
}
fn tracing_config(&self, command: &KmsCommand) -> trace::Config {
if command.verbose() {
trace::Config::verbose()
} else {
trace::Config::default()
}
}
}