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
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
|
mod error;
use crate::error::InternalError;
use std::fs::{self, File};
use std::os::unix::prelude::ExitStatusExt;
use std::path::Path;
use std::process::{Command, ExitStatus, Stdio};
use structopt::StructOpt;
/// This plugin supports the installation, updation and removal of a single unversioned apama project named "project".
/// Installation of multiple parallel projects is not supported.
/// Installing a project will replace the existing project with the new one.
/// Delta update of a project(for eg: updating just the `mon` file definitions in the project) is not supported either.
#[derive(StructOpt)]
struct ApamaCli {
#[structopt(subcommand)]
operation: PluginOp,
}
#[derive(StructOpt)]
pub enum PluginOp {
/// List the one and only apama project if one is installed
List,
/// Install an apama project
Install {
module: String,
#[structopt(short = "v", long = "--module-version")]
version: Option<String>,
#[structopt(long = "--file")]
file_path: String,
},
/// Remove an apama project
Remove {
module: String,
#[structopt(short = "v", long = "--module-version")]
version: Option<String>,
},
/// Prepare a sequences of install/remove commands
Prepare,
/// Finalize a sequences of install/remove commands
Finalize,
}
const APAMA_ENV_EXE: &str = "/opt/softwareag/Apama/bin/apama_env";
const TEDGE_APAMA_PROJECT_DIR: &str = "/etc/tedge/apama/project";
const TMP_APAMA_PROJECT_DIR: &str = "/tmp/tedge_apama_project";
const APAMA_PROJECT_NAME: &str = "project";
fn run(operation: PluginOp) -> Result<ExitStatus, InternalError> {
let success = ExitStatus::from_raw(0);
let tedge_env_exe_path = Path::new(APAMA_ENV_EXE);
if !tedge_env_exe_path.exists() {
return Err(InternalError::ApamaNotInstalled);
}
let tedge_apama_project_path = Path::new(TEDGE_APAMA_PROJECT_DIR);
let tmp_apama_project_path = Path::new(TMP_APAMA_PROJECT_DIR);
let status = match operation {
// Since there can only be a single project named `project`, print its name if installed
PluginOp::List => {
if tedge_apama_project_path.exists() {
println!("{}\t", APAMA_PROJECT_NAME)
}
success
}
PluginOp::Prepare => success,
PluginOp::Finalize => {
// Cleanup any temporary artefacts created by this plugin
if tmp_apama_project_path.exists() {
fs::remove_dir_all(tmp_apama_project_path)?;
}
success
}
PluginOp::Install {
module: _,
version: _,
file_path,
} => {
let archive_path = Path::new(&file_path);
let archive_file = File::open(&archive_path)?;
let mut archive = zip::ZipArchive::new(archive_file)?;
// TODO: Validate the zip to be valid apama project before extraction?
println!("Extracting the archive at {}", file_path);
archive.extract(tmp_apama_project_path)?;
println!("Extraction successful");
// Deleting existing project as the rename API expects the target dir to be empty
if tedge_apama_project_path.exists() {
println!("Removing existing project at {}", TEDGE_APAMA_PROJECT_DIR);
fs::remove_dir_all(tedge_apama_project_path)?;
println!("Removal of existing project successful");
}
println!(
"Installing newly extracted project to {}",
TEDGE_APAMA_PROJECT_DIR
);
fs::create_dir_all(tedge_apama_project_path)?;
fs::rename(tmp_apama_project_path, tedge_apama_project_path)?;
println!("Installation of new project successful");
println!("Restarting apama to load the new project");
run_cmd("service", "apama restart")?;
println!("Restart of apama service successful");
success
}
PluginOp::Remove {
module: _,
version: _,
} => {
if tedge_apama_project_path.exists() {
println!("Stopping apama service");
run_cmd("service", "apama stop")?;
println!("Stopping apama service successful");
println!("Removing existing project at {}", TEDGE_APAMA_PROJECT_DIR);
fs::remove_dir_all(tedge_apama_project_path)?;
println!("Removal of existing project successful");
}
success
}
};
Ok(status)
}
fn run_cmd(cmd: &str, args: &str) -> Result<ExitStatus, InternalError> {
let args: Vec<&str> = args.split_whitespace().collect();
let status = Command::new(cmd)
.args(args)
.stdin(Stdio::null())
.status()
.map_err(|err| InternalError::exec_error(cmd, err))?;
Ok(status)
}
fn main() {
// On usage error, the process exits with a status code of 1
let apama = ApamaCli::from_args();
match run(apama.operation) {
Ok(status) if status.success() => {
std::process::exit(0);
}
Ok(status) => {
if status.code().is_some() {
std::process::exit(2);
} else {
eprintln!("Interrupted by a signal!");
std::process::exit(4);
}
}
Err(err) => {
eprintln!("ERROR: {}", err);
std::process::exit(5);
}
}
}
|
