daemon_console 0.3.3

A flexible console for daemon applications.
Documentation 
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

use daemon_console::{
    AppAction, TerminalApp, events::DaemonConsoleEvent, get_info, get_warn, logger::LogLevel,
    utils::get_local_timestring,
};

fn handle_user_input_event(
    raw: &str,
    timestamp: i64,
    action_tx: &tokio::sync::mpsc::UnboundedSender<AppAction>,
) {
    if raw.trim() == "test" {
        let _ = action_tx.send(AppAction::Info("ok".to_string()));
    } else if raw.trim() == "hello" {
        let _ = action_tx.send(AppAction::Info("Hello there!".to_string()));
    } else if raw.trim().starts_with("echo ") {
        let echo_content = raw.trim().strip_prefix("echo ").unwrap_or("");
        let _ = action_tx.send(AppAction::Info(format!("Echo: {}", echo_content)));
    }

    let _ = action_tx.send(AppAction::Info(format!(
        "event 'CommandPromptInput': raw='{}', local_time={}",
        raw,
        get_local_timestring(timestamp)
    )));
}

fn handle_terminal_log_event(
    level: LogLevel,
    message: &str,
    module_name: &Option<String>,
    timestamp: i64,
    action_tx: &tokio::sync::mpsc::UnboundedSender<AppAction>,
) {
    let _ = action_tx.send(AppAction::Debug(format!(
        "event 'TerminalLog': level={:?}, message='{}', module_name='{:?}', timestamp={}",
        level, message, module_name, timestamp
    )));
}

fn handle_subprocess_log_event(
    pid: u32,
    message: &str,
    timestamp: i64,
    action_tx: &tokio::sync::mpsc::UnboundedSender<AppAction>,
) {
    let _ = action_tx.send(AppAction::Warn(format!(
        "event 'SubprocessLog': pid={}, message='{}', timestamp={}",
        pid, message, timestamp
    )));
}

fn start_event_listener(
    mut event_rx: tokio::sync::broadcast::Receiver<DaemonConsoleEvent>,
    action_tx: tokio::sync::mpsc::UnboundedSender<AppAction>,
) {
    tokio::spawn(async move {
        while let Ok(event) = event_rx.recv().await {
            match event {
                DaemonConsoleEvent::UserConsoleInput { raw, timestamp } => {
                    handle_user_input_event(&raw, timestamp, &action_tx);
                }
                DaemonConsoleEvent::TerminalLog {
                    level,
                    message,
                    module_name,
                    timestamp,
                } => {
                    handle_terminal_log_event(level, &message, &module_name, timestamp, &action_tx);
                }
                DaemonConsoleEvent::SubprocessLog {
                    pid,
                    message,
                    timestamp,
                } => {
                    handle_subprocess_log_event(pid, &message, timestamp, &action_tx);
                }
            }
        }
    });
}

#[tokio::main]
async fn main() {
    let mut app = TerminalApp::new();

    app.set_unknown_command_handler(|_| {
        get_warn!("The command system disabled for developing the event system.")
    });

    let action_tx = app
        .get_action_sender()
        .expect("Failed to get action sender");
    let event_rx = app
        .subscribe_events()
        .expect("Failed to subscribe to events");

    start_event_listener(event_rx, action_tx);

    app.info("This message used for debugging 'TerminalLog' event.");

    let _ = app
        .run(&get_info!("App demo starting...", "Demo"), "")
        .await;
}