mullvad-cli/src/cmds/log.rs


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

use anyhow::Result;
use clap::Subcommand;
use futures::StreamExt;
use mullvad_management_interface::MullvadProxyClient;

#[derive(Subcommand, Debug)]
pub enum Log {
    /// Set the log level for the daemon
    SetLevel {
        /// The log level to set
        level: String,
    },
    /// Follow live updates to the daemon log file. Analogue to running `tail -f` on the daemon log file.
    Listen,
}

impl Log {
    pub async fn handle(self) -> Result<()> {
        match self {
            Log::SetLevel { level } => set_level(level).await,
            Log::Listen => on_listen().await,
        }
    }
}

async fn on_listen() -> std::result::Result<(), anyhow::Error> {
    let mut rpc = MullvadProxyClient::new().await?;
    let log_stream = rpc.log_listen().await?;
    log_stream
        .for_each(|log| async {
            match log {
                Ok(log) => print!("{log}"),
                Err(e) => eprint!("{e}"),
            }
        })
        .await;
    Ok(())
}

async fn set_level(level: String) -> std::result::Result<(), anyhow::Error> {
    let mut rpc = MullvadProxyClient::new().await?;
    rpc.set_log_filter(level).await?;
    Ok(())
}