bing_client 0.1.2

Async reverse client for Bing copilot
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

use serde_json::Value;

use crate::{
    types::{bot_easy_resp_type::Image, chat_msg_type::EasyMsg},
    BingClient,
};

use super::{
    draw_image::poll_images,
    process_bot_resp::{process_source_msg, process_suggested_responses},
};

pub async fn process_chat_msgs(
    json: &Value,
    client: &BingClient,
) -> Result<Vec<EasyMsg>, anyhow::Error> {
    let messages = match json["messages"].as_array() {
        Some(messages) => messages,
        None => return Ok(Vec::new()),
    };

    let mut easy_msgs = Vec::new();
    for msg in messages {
        if !(msg["scores"].is_array()
            || msg["suggestedResponses"].is_array()
            || (msg["contentType"].is_string() && !msg["adaptiveCards"].is_null()))
        {
            continue;
        }

        let (author, text) = match (msg["author"].as_str(), msg["text"].as_str()) {
            (Some(author), Some(text)) => (author, text),
            _ => continue,
        };

        let sources = msg["sourceAttributions"]
            .as_array()
            .map_or_else(Vec::new, |s| {
                process_source_msg(s.to_vec()).unwrap_or_else(Vec::new)
            });

        let suggest_replys = msg["suggestedResponses"]
            .as_array()
            .map_or_else(Vec::new, |s| {
                process_suggested_responses(s.to_vec()).unwrap_or_else(Vec::new)
            });

        let images = match author {
            "user" => msg["imageUrl"]
                .as_str()
                .or_else(|| msg["originalImageUrl"].as_str())
                .map(|url| {
                    vec![Image {
                        name: "user_image_attachment.jpg".to_string(),
                        url: url.to_owned(),
                    }]
                })
                .unwrap_or_else(Vec::new),
            "bot" => {
                if msg["contentType"].as_str() != Some("IMAGE") {
                    Vec::new()
                } else if let Value::Array(a) = &msg["adaptiveCards"] {
                    let mut images_rst = Vec::new();
                    for b in a {
                        if let Value::Array(c) = &b["body"] {
                            for d in c {
                                if let Value::String(url) = &d["persistentUrl"] {
                                    match poll_images(
                                        url.to_string(),
                                        client.gen_header().unwrap(),
                                        false,
                                    )
                                    .await
                                    {
                                        Ok(images) => images_rst = images,
                                        Err(e) => easy_msgs.push(EasyMsg {
                                            author: author.to_owned(),
                                            text: e.to_string(),
                                            images: Vec::with_capacity(0),
                                            sources: Vec::with_capacity(0),
                                            suggest_replys: Vec::with_capacity(0),
                                        }),
                                    }
                                }
                            }
                        }
                    }
                    images_rst
                } else {
                    Vec::new()
                }
            }
            _ => Vec::new(),
        };

        easy_msgs.push(EasyMsg {
            author: author.to_owned(),
            text: text.to_owned(),
            images,
            sources,
            suggest_replys,
        });
    }

    Ok(easy_msgs)
}