ockam_command 0.146.0

End-to-end encryption and mutual authentication for distributed applications.
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

use miette::IntoDiagnostic;
use ockam_api::cli_state::journeys::{JourneyEvent, APPLICATION_EVENT_COMMAND};
use ockam_api::CliState;
use ockam_core::OCKAM_TRACER_NAME;
use ockam_node::Executor;
use opentelemetry::trace::{TraceContextExt, Tracer};
use opentelemetry::{global, Context};
use std::collections::HashMap;
use tracing::error;

/// This function creates a journey event describing the execution of a command
pub fn add_command_event(
    cli_state: CliState,
    command: &str,
    command_arguments: String,
) -> miette::Result<()> {
    let command_name = command.to_string();
    let tracer = global::tracer(OCKAM_TRACER_NAME);
    tracer
        .in_span(command_name.clone(), |_| {
            Executor::execute_future(async move {
                let mut attributes = HashMap::new();
                attributes.insert(
                    APPLICATION_EVENT_COMMAND,
                    sanitize_command_arguments(command_arguments),
                );
                cli_state
                    .add_journey_event(JourneyEvent::ok(command_name), attributes)
                    .await
            })
        })
        .into_diagnostic()??;
    Ok(())
}

/// This function creates a journey event describing the error resulting from the execution of a command
pub fn add_command_error_event(
    cli_state: CliState,
    command_name: &str,
    message: &str,
    command_arguments: String,
) -> miette::Result<()> {
    let message = message.to_string();
    let command = command_name.to_string();
    let tracer = global::tracer(OCKAM_TRACER_NAME);
    tracer
        .in_span(format!("'{}' error", command), |_| {
            Context::current()
                .span()
                .set_status(opentelemetry::trace::Status::error(message.clone()));
            error!("{}", &message);

            Executor::execute_future(async move {
                let mut attributes = HashMap::new();
                attributes.insert(
                    APPLICATION_EVENT_COMMAND,
                    sanitize_command_arguments(command_arguments),
                );
                cli_state
                    .add_journey_error(&command, message, attributes)
                    .await
            })
        })
        .into_diagnostic()??;
    Ok(())
}

/// The ockam project enroll command arguments contain the enrollment ticket which is sensitive
/// information (because it could be potentially reused), so it should be removed from the user event.
pub fn sanitize_command_arguments(command_args: String) -> String {
    if command_args.starts_with("ockam project enroll") {
        "ockam project enroll".to_string()
    } else {
        command_args
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn sanitize_project_enroll() {
        assert_eq!(
            sanitize_command_arguments("ockam project enroll abcdxyz".to_string()),
            "ockam project enroll".to_string()
        );
        assert_eq!(
            sanitize_command_arguments("ockam node create n1".to_string()),
            "ockam node create n1".to_string()
        );
    }
}