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
110
111
112
113
114
115
116
117
118
119
pub const DEFAULT_SPEC: &str = "mainnet";
pub const AVAILABLE_SPECS: &[&str] = &["mainnet", "testnet", "staging", "dev"];
pub const DEFAULT_RPC_PORT: &str = "8114";
pub const DEFAULT_P2P_PORT: &str = "8115";
const START_MARKER: &str = " # {{";
const END_MAKER: &str = "# }}";
const WILDCARD_BRANCH: &str = "# _ => ";
use std::collections::HashMap;
use std::io;
pub struct Template<T>(T);
pub struct TemplateContext<'a> {
spec: &'a str,
kvs: HashMap<&'a str, &'a str>,
}
impl<'a> TemplateContext<'a> {
pub fn new<I>(spec: &'a str, kvs: I) -> Self
where
I: IntoIterator<Item = (&'a str, &'a str)>,
{
Self {
spec,
kvs: kvs.into_iter().collect(),
}
}
pub fn insert(&mut self, key: &'a str, value: &'a str) {
self.kvs.insert(key, value);
}
}
impl<T> Template<T> {
pub fn new(content: T) -> Self {
Template(content)
}
}
fn writeln<W: io::Write>(w: &mut W, s: &str, context: &TemplateContext) -> io::Result<()> {
#[cfg(docker)]
let s = s.replace("127.0.0.1:{rpc_port}", "0.0.0.0:{rpc_port}");
writeln!(
w,
"{}",
context
.kvs
.iter()
.fold(s.replace("\\n", "\n"), |s, (key, value)| s
.replace(format!("{{{}}}", key).as_str(), value))
)
}
#[derive(Debug)]
pub enum TemplateState<'a> {
SearchStartMarker,
MatchBranch(&'a str),
SearchEndMarker,
}
impl<T> Template<T>
where
T: AsRef<str>,
{
pub fn write_to<'c, W: io::Write>(
&self,
w: &mut W,
context: &TemplateContext<'c>,
) -> io::Result<()> {
let spec_branch = format!("# {} => ", context.spec);
let mut state = TemplateState::SearchStartMarker;
for line in self.0.as_ref().lines() {
match state {
TemplateState::SearchStartMarker => {
if line.ends_with(START_MARKER) {
state = TemplateState::MatchBranch(line);
} else {
writeln!(w, "{}", line)?;
}
}
TemplateState::MatchBranch(start_line) => {
if line == END_MAKER {
writeln!(
w,
"{}",
&start_line[..(start_line.len() - START_MARKER.len())],
)?;
state = TemplateState::SearchStartMarker;
} else if line.starts_with(&spec_branch) {
writeln(w, &line[spec_branch.len()..], context)?;
state = TemplateState::SearchEndMarker;
} else if line.starts_with(WILDCARD_BRANCH) {
writeln(w, &line[WILDCARD_BRANCH.len()..], context)?;
state = TemplateState::SearchEndMarker;
}
}
TemplateState::SearchEndMarker => {
if line == END_MAKER {
state = TemplateState::SearchStartMarker;
}
}
}
}
if let TemplateState::MatchBranch(start_line) = state {
writeln!(
w,
"{}",
&start_line[..(start_line.len() - START_MARKER.len())],
)?;
}
Ok(())
}
}