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
|
use libangelshark::{AcmRunner, Message, ParallelIterator};
use log::{error, info};
use serde::Deserialize;
use std::convert::Infallible;
use warp::{
body::{content_length_limit, json},
post, reply, Filter, Rejection, Reply,
};
const SIXTEEN_K: u64 = 1024 * 16;
/// Returns a warp filter to handle HTTP POSTs for deprovisioning stations, agents, etc.
pub fn filter(runner: AcmRunner) -> impl Filter<Extract = impl Reply, Error = Rejection> + Clone {
warp::path("deprov")
.and(post())
.and(content_length_limit(SIXTEEN_K))
.and(json())
.and_then(move |entries| remove_entries(entries, runner.to_owned()))
}
/// Queues removal commands for [Entries] on an [AcmRunner]. Gathers any errors encountered and returns those.
async fn remove_entries(entries: Entries, mut runner: AcmRunner) -> Result<impl Reply, Infallible> {
// Construct OSSI messages to carry out removals.
for entry in entries {
match entry {
Entry::StationUser { acm, ext } => {
runner.queue_input(&acm, &Message::new(&format!("list station {}", ext)));
runner.queue_input(&acm, &Message::new(&format!("clear amw all {}", ext)));
runner.queue_input(&acm, &Message::new(&format!("remove station {}", ext)));
}
Entry::AgentLoginId { acm, ext } => {
runner.queue_input(&acm, &Message::new(&format!("list agent-loginID {}", ext)));
runner.queue_input(
&acm,
&Message::new(&format!("remove agent-loginID {}", ext)),
);
}
}
}
// Gather any errors encountered and format them for the client response.
let errors: Vec<String> = runner
.run_cached()
.map(|(acm, output)| match output {
Ok(messages) => messages
.into_iter()
.filter_map(|message| {
if let Some(data) = message
.datas
.and_then(|d| Some(format!("{acm}\t{}", d.get(0)?.join("\t"))))
{
// if there was data (such as in lists), print it out
info!("{data}");
}
Some(format!("ACM {}: {}", acm.clone(), message.error?))
})
.collect(),
Err(error) => vec![format!("ACM {}: {}", acm, error)],
})
.flatten()
.collect();
// Log errors for tracking.
for error in &errors {
error!("{}", error);
}
Ok(reply::json(&errors))
}
/// Collection of [Entry].
type Entries = Vec<Entry>;
/// Very basic [Deserialize] target for deprov inputs. Going from stringly typed to strongly typed.
#[derive(Debug, Deserialize)]
enum Entry {
#[serde(rename(deserialize = "station-user"))]
StationUser { acm: String, ext: String },
#[serde(rename(deserialize = "agent-loginid"))]
AgentLoginId { acm: String, ext: String },
}
|
