pidcat 0.2.1

An adb logcat wrapper and filters
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
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

use anyhow::Result;
use async_stream::stream;
use async_trait::async_trait;
use futures::{Sink, Stream, StreamExt};
use regex::Regex;
use std::collections::HashMap;
use std::error::Error;
use std::fmt::{Display, Formatter};
use std::pin::Pin;
use tokio::io::AsyncBufReadExt;
use tokio::process::{Child, Command};

#[derive(Debug, Clone)]
pub(crate) struct Log {
    pub date: String,
    pub time: String,
    pub pid: String,
    pub tid: String,
    pub level: String,
    pub tag: String,
    pub message: String,
    pub buffer: String,
}

impl Display for Log {
    fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "{} {} {} {} {} {} {}",
            self.date, self.time, self.pid, self.tid, self.level, self.tag, self.message
        )
    }
}

type LogStream = Pin<Box<dyn Stream<Item = Result<Log, Box<dyn Error + Send>>> + Send>>;

#[async_trait]
trait Source {
    async fn source(&self) -> LogStream;
}

struct AdbSource;

impl AdbSource {
    async fn spawn_adb_logcat(&self) -> Child {
        let mut command = Command::new("adb");
        command.stdout(std::process::Stdio::piped());
        command.arg("logcat");
        command.arg("-D");
        command.arg("-v").arg("long");
        command.arg("-b").arg("all");
        command.spawn().expect("Failed to execute adb logcat")
    }
}

#[async_trait]
impl Source for AdbSource {
    async fn source(&self) -> LogStream {
        let mut logcat = self.spawn_adb_logcat().await;
        let mut reader = tokio::io::BufReader::new(logcat.stdout.take().unwrap());
        let re =
            Regex::new(r"\[ (\d{2}-\d{2})\s(\d{2}:\d{2}:\d{2}\.\d{3})\s+(\d+):\s+(\d+)\s+(.*) ]")
                .unwrap();

        let s = stream! {
            let mut line = String::new();
            let mut buffer = String::new();
            let mut map = HashMap::new();

            while let Ok(bytes_read) = reader.read_line(&mut line).await {
                if bytes_read == 0 {
                    break;
                }

                if line.starts_with("---------") {
                    let spl = line.trim().split_whitespace().collect::<Vec<&str>>();
                    buffer = spl[3].to_string();
                    map.insert("buffer", buffer.clone());
                } else if line.trim().is_empty() {
                    if map.contains_key("message") {
                        let log = Log {
                            tag: map.remove("tag").unwrap(),
                            date: map.remove("date").unwrap(),
                            time: map.remove("time").unwrap(),
                            pid: map.remove("pid").unwrap(),
                            tid: map.remove("tid").unwrap(),
                            level: map.remove("level").unwrap(),
                            message: map.remove("message").unwrap(),
                            buffer: map.get("buffer").unwrap().clone(),
                        };
                        yield Ok(log);
                    }
                } else {
                    if let Some(captures) = re.captures(&line) {
                        map.insert("date", captures.get(1).unwrap().as_str().to_string());
                        map.insert("time", captures.get(2).unwrap().as_str().to_string());
                        map.insert("pid", captures.get(3).unwrap().as_str().to_string());
                        map.insert("tid", captures.get(4).unwrap().as_str().to_string());
                        let content = captures.get(5).unwrap().as_str().to_string();
                        let content = content.splitn(2, '/').collect::<Vec<&str>>();
                        map.insert("level", content[0].to_string());
                        map.insert("tag", content[1].to_string());
                    } else {
                        if let Some(msg) = map.get_mut("message") {
                            msg.push('\n');
                            msg.push_str(line.trim_end());
                        } else {
                            map.insert("message", line.trim_end().to_string());
                        }
                    }
                }
                line.clear();
            }
        };
        return Box::pin(s);
    }
}

#[tokio::main]
async fn main() {
    let adb = AdbSource;

    tokio::spawn(async move {
        let mut logs = adb.source().await;
        while let Some(r) = logs.next().await {
            match r {
                Ok(log) => {
                    println!("{}", log);
                }
                Err(_) => {}
            }
        }
    })
    .await
    .expect("TODO: panic message");
}