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
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
// SPDX-FileCopyrightText: 2020 Serokell <https://serokell.io/>
//
// SPDX-License-Identifier: MPL-2.0
use super::data;
use tokio::process::Command;
fn deploy_path_to_activate_path_str(
deploy_path: &std::path::Path,
) -> Result<String, Box<dyn std::error::Error>> {
Ok(format!(
"{}/activate",
deploy_path
.parent()
.ok_or("Deploy path too short")?
.to_str()
.ok_or("Deploy path is not valid utf8")?
.to_owned()
))
}
#[test]
fn test_activate_path_generation() {
match deploy_path_to_activate_path_str(&std::path::PathBuf::from(
"/blah/blah/deploy-rs/bin/deploy",
)) {
Err(_) => panic!(""),
Ok(x) => assert_eq!(x, "/blah/blah/deploy-rs/bin/activate".to_string()),
}
}
fn build_activate_command(
activate_path_str: String,
sudo: &Option<String>,
profile_path: &str,
closure: &str,
activate_cmd: &Option<String>,
bootstrap_cmd: &Option<String>,
auto_rollback: bool,
) -> Result<String, Box<dyn std::error::Error>> {
let mut self_activate_command =
format!("{} '{}' '{}'", activate_path_str, profile_path, closure);
if let Some(sudo_cmd) = &sudo {
self_activate_command = format!("{} {}", sudo_cmd, self_activate_command);
}
if let Some(ref bootstrap_cmd) = bootstrap_cmd {
self_activate_command = format!(
"{} --bootstrap-cmd '{}'",
self_activate_command, bootstrap_cmd
);
}
if let Some(ref activate_cmd) = activate_cmd {
self_activate_command = format!(
"{} --activate-cmd '{}'",
self_activate_command, activate_cmd
);
}
if auto_rollback {
self_activate_command = format!("{} --auto-rollback", self_activate_command);
}
Ok(self_activate_command)
}
#[test]
fn test_activation_command_builder() {
let activate_path_str = "/blah/bin/activate".to_string();
let sudo = Some("sudo -u test".to_string());
let profile_path = "/blah/profiles/test";
let closure = "/blah/etc";
let activate_cmd = Some("$THING/bin/aaaaaaa".to_string());
let bootstrap_cmd = None;
let auto_rollback = true;
match build_activate_command(
activate_path_str,
&sudo,
profile_path,
closure,
&activate_cmd,
&bootstrap_cmd,
auto_rollback,
) {
Err(_) => panic!(""),
Ok(x) => assert_eq!(x, "sudo -u test /blah/bin/activate '/blah/profiles/test' '/blah/etc' --activate-cmd '$THING/bin/aaaaaaa' --auto-rollback".to_string()),
}
}
pub async fn deploy_profile(
profile: &data::Profile,
profile_name: &str,
node: &data::Node,
node_name: &str,
merged_settings: &data::GenericSettings,
deploy_data: &super::DeployData<'_>,
auto_rollback: bool,
) -> Result<(), Box<dyn std::error::Error>> {
info!(
"Activating profile `{}` for node `{}`",
profile_name, node_name
);
let activate_path_str = deploy_path_to_activate_path_str(&deploy_data.current_exe)?;
let self_activate_command = build_activate_command(
activate_path_str,
&deploy_data.sudo,
&deploy_data.profile_path,
&profile.profile_settings.path,
&profile.profile_settings.activate,
&profile.profile_settings.bootstrap,
auto_rollback,
)?;
let mut c = Command::new("ssh");
let mut ssh_command = c.arg(format!(
"ssh://{}@{}",
deploy_data.ssh_user, node.node_settings.hostname
));
for ssh_opt in &merged_settings.ssh_opts {
ssh_command = ssh_command.arg(ssh_opt);
}
ssh_command.arg(self_activate_command).spawn()?.await?;
Ok(())
}
|