bmux_cli 0.0.1-alpha.1

Command-line interface for bmux terminal multiplexer
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

use anyhow::{Context, Result};
use bmux_cli_schema::PerfProfileArg;
use bmux_performance_plugin_api::{performance_commands, performance_state};
use bmux_performance_state::{PerformanceRecordingLevel, PerformanceRuntimeSettings};

use super::{
    ConnectionContext, ConnectionPolicyScope, cleanup_stale_pid_file, connect_with_context,
};

const fn performance_level_name(level: PerformanceRecordingLevel) -> &'static str {
    match level {
        PerformanceRecordingLevel::Off => "off",
        PerformanceRecordingLevel::Basic => "basic",
        PerformanceRecordingLevel::Detailed => "detailed",
        PerformanceRecordingLevel::Trace => "trace",
    }
}

const fn profile_to_level(profile: PerfProfileArg) -> PerformanceRecordingLevel {
    match profile {
        PerfProfileArg::Basic => PerformanceRecordingLevel::Basic,
        PerfProfileArg::Detailed => PerformanceRecordingLevel::Detailed,
        PerfProfileArg::Trace => PerformanceRecordingLevel::Trace,
    }
}

fn print_performance_settings(settings: &PerformanceRuntimeSettings) {
    println!(
        "runtime performance recording level: {}",
        performance_level_name(settings.recording_level)
    );
    println!("window ms: {}", settings.window_ms);
    println!("max events/sec: {}", settings.max_events_per_sec);
    println!(
        "max payload bytes/sec: {}",
        settings.max_payload_bytes_per_sec
    );
}

pub(super) async fn run_perf_status(
    json: bool,
    connection_context: ConnectionContext<'_>,
) -> Result<u8> {
    cleanup_stale_pid_file().await?;
    let mut client = connect_with_context(
        ConnectionPolicyScope::Normal,
        "bmux-cli-perf-status",
        connection_context,
    )
    .await?;
    let settings = performance_state::client::get_settings(&mut client)
        .await?
        .into();

    if json {
        println!(
            "{}",
            serde_json::to_string_pretty(&settings)
                .context("failed encoding performance status json")?
        );
        return Ok(0);
    }

    print_performance_settings(&settings);
    Ok(0)
}

pub(super) async fn run_perf_on(
    profile: PerfProfileArg,
    json: bool,
    connection_context: ConnectionContext<'_>,
) -> Result<u8> {
    cleanup_stale_pid_file().await?;
    let mut client = connect_with_context(
        ConnectionPolicyScope::Normal,
        "bmux-cli-perf-on",
        connection_context,
    )
    .await?;
    let mut settings: PerformanceRuntimeSettings =
        performance_state::client::get_settings(&mut client)
            .await?
            .into();
    settings.recording_level = profile_to_level(profile);
    let updated: PerformanceRuntimeSettings =
        performance_commands::client::set_settings(&mut client, settings.into())
            .await?
            .into();

    if json {
        println!(
            "{}",
            serde_json::to_string_pretty(&updated)
                .context("failed encoding performance settings json")?
        );
        return Ok(0);
    }

    println!(
        "runtime performance telemetry enabled ({})",
        performance_level_name(updated.recording_level)
    );
    print_performance_settings(&updated);
    Ok(0)
}

pub(super) async fn run_perf_off(
    json: bool,
    connection_context: ConnectionContext<'_>,
) -> Result<u8> {
    cleanup_stale_pid_file().await?;
    let mut client = connect_with_context(
        ConnectionPolicyScope::Normal,
        "bmux-cli-perf-off",
        connection_context,
    )
    .await?;
    let mut settings: PerformanceRuntimeSettings =
        performance_state::client::get_settings(&mut client)
            .await?
            .into();
    settings.recording_level = PerformanceRecordingLevel::Off;
    let updated: PerformanceRuntimeSettings =
        performance_commands::client::set_settings(&mut client, settings.into())
            .await?
            .into();

    if json {
        println!(
            "{}",
            serde_json::to_string_pretty(&updated)
                .context("failed encoding performance settings json")?
        );
        return Ok(0);
    }

    println!("runtime performance telemetry disabled");
    print_performance_settings(&updated);
    Ok(0)
}