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
use std::fmt;
use std::io::{Read, Write};
use flate2::{read::ZlibDecoder, write::ZlibEncoder, Compression};
use super::fmt_extract_attr;
use crate::msg::{MessageChainBuilder, PushBuilder};
use crate::msg::{MessageElem, PushElem};
use crate::pb::msg;
use crate::{push_builder_impl, to_elem_vec_impl};
#[derive(Default, Debug, Clone)]
pub struct LightApp {
pub content: String,
}
impl LightApp {
pub fn new(content: String) -> Self {
Self { content }
}
}
impl PushElem for LightApp {
fn push_to(elem: Self, vec: &mut Vec<MessageElem>) {
vec.push(MessageElem::LightApp(msg::LightApp {
data: Some({
let mut encoder = ZlibEncoder::new(vec![1], Compression::default());
encoder.write_all(elem.content.as_bytes()).ok();
encoder.finish().unwrap_or_default()
}),
..Default::default()
}));
}
}
impl From<msg::LightApp> for LightApp {
fn from(e: msg::LightApp) -> Self {
let data = e.data.unwrap_or_default();
if data.len() > 1 {
let content = if data[0] == 0 {
data[1..].to_vec()
} else {
let mut uncompressed = Vec::new();
ZlibDecoder::new(&data[1..])
.read_to_end(&mut uncompressed)
.ok();
uncompressed
};
if !content.is_empty() && content.len() < 1024 ^ 3 {
return Self {
content: String::from_utf8_lossy(&content).into_owned(),
};
}
}
Self::default()
}
}
impl fmt::Display for LightApp {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
f.write_str("[LightApp:")?;
fmt_extract_attr(f, &self.content, "app", r#""app":""#, "\"")?;
fmt_extract_attr(f, &self.content, "prompt", r#""prompt":""#, "\"")?;
fmt_extract_attr(f, &self.content, "desc", r#""desc":""#, "\"")?;
fmt_extract_attr(f, &self.content, "url", r#""jumpUrl":""#, "\"")?;
fmt_extract_attr(f, &self.content, "title", r#""title":""#, "\"")?;
fmt_extract_attr(f, &self.content, "tag", r#""tag":""#, "\"")?;
f.write_str("]")
}
}
to_elem_vec_impl!(LightApp);
push_builder_impl!(LightApp);