feat: extract reponse time

src/main.rs

@@ -7,7 +7,7 @@ use chrono::{self, Timelike};
 use dotenv;
 use ipaddress;
 use itertools::{self, Itertools};
-use lettre;
+use lettre::{self, message};
 use reqwest;
 use serde_json::{self, json};
 use zip;
@@ -170,6 +170,14 @@ fn main() -> anyhow::Result<()> {
         std::fs::create_dir(format!("./evaluations/{formatted_day_before}")).expect("Failed to create directory")
     }
 
+    // Create the response time folder
+    if !std::fs::exists(format!("./evaluations/{formatted_day_before}/response_time.csv")).unwrap() {
+        let mut response_time_file = std::fs::File::create_new(format!("./evaluations/{formatted_day_before}/response_time.csv")).expect("Failed to response_time.csv");
+        let header = "NOME;ID_TALK;TEMPO DE RESPOSTA;DATA";
+        response_time_file.write_all(header.as_bytes()).expect("Failed to write header to file");
+        drop(response_time_file)
+    }
+
     // Read system prompt
     let prompt = std::fs::read_to_string("PROMPT.txt").unwrap();
 
@@ -190,6 +198,163 @@ fn main() -> anyhow::Result<()> {
 
     println!("IDS {:?}", talk_ids);
 
+    // Calculate the response time in seconds
+    let response_time = talk_ids
+        .iter()
+        .cloned()
+        .map(|talk_id| {
+            let talk_id_get_request = client
+                .get(format!("https://{}/api/talk_histories", PIPERUN_API_URL))
+                .bearer_auth(&access_token)
+                .header("Content-Type", "application/json")
+                .header("Accept", "application/json")
+                .query(&[
+                    ("talk_id", talk_id),
+                    ("type", "a".to_string()),
+                    ("only_view", "1".to_string()),
+                ]);
+
+            let talk_id_get_result = talk_id_get_request.send();
+
+            return talk_id_get_result;
+        }).filter_map_ok(|result| {
+            let json = result.json::<serde_json::Value>().expect("Failed to deserialize response to JSON").to_owned();
+            let talk_histories = &json["talk_histories"];
+            let data = &talk_histories["data"];
+
+            // Filter chats that have very few messages
+            let talk_lenght = talk_histories.as_array().expect("Wrong message type received from talk histories").len();
+            if  talk_lenght < MINIMUM_NUMBER_OF_MESSAGES_TO_EVALUATE {return None;}
+
+            // Filter chats that have less that specified ammount of talks with support agent form the last queue transfer
+            let found = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().enumerate().find(|(pos, message_object)|{
+                let message  = message_object["message"].as_str().expect("Failed to decode message as string");
+                let found = message.find("Atendimento transferido para a fila [NovaNet -> Atendimento -> Suporte]");
+                found.is_some()
+            });
+
+            match found {
+                None => {return None;},
+                Some(pos) => {
+                    let pos_found = pos.0;
+                    if pos_found < MINIMUM_NUMBER_OF_MESSAGES_WITH_AGENT_TO_EVALUATE {return None;}
+                }
+            };
+
+            // Filter chats that pass by multiple agents
+            let number_of_agents_except_pipebot = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter()
+                .map(|message| &message["user"]["name"])
+                .filter(|name| name.as_str().unwrap() != "PipeBot".to_string())
+                .unique()
+                .count();
+            if number_of_agents_except_pipebot > 1 {return None;}
+
+            // Filter stop notification active
+            let found_stop_notification = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().enumerate().find(|(pos, message_object)|{
+                let message  = message_object["message"].as_str().expect("Failed to decode message as string");
+                let found = message.find("ATENÇÃO CLIENTE COM NOTIFICAÇÃO DE PARADA ATIVA");
+                found.is_some()
+            });
+
+            if found_stop_notification.is_some() {return None;}
+
+            // Filter lack of dialogue
+            let lack_of_dialogue = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{
+                let message  = message_object["message"].as_str().expect("Failed to decode message as string");
+                let found = message.to_uppercase().find(&"Esse atendimento está sendo encerrado devido a inexistência de diálogo".to_uppercase());
+                found.is_some()
+            });
+
+            if lack_of_dialogue.is_some() {return None;}
+
+            // Filter is portal access
+            let portal_link_offer = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{
+                let message  = message_object["message"].as_str().expect("Failed to decode message as string");
+                let found = message.to_uppercase().find(&"https://novanet.portal.7az.com.br/login".to_uppercase());
+                found.is_some()
+            });
+
+            if portal_link_offer.is_some() {return None;}
+
+            // Filter no comercial agent available
+            let comercial_agents_unavailable = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter().find(|message_object|{
+                let message  = message_object["message"].as_str().expect("Failed to decode message as string");
+                let found = message.to_uppercase().find(&"Transbordo automático para a fila [NovaNet -> Atendimento -> Suporte], pois não havia agentes disponíveis na fila [NovaNet -> Atendimento -> Comercial]".to_uppercase());
+                found.is_some()
+            });
+
+            if comercial_agents_unavailable.is_some() {return None;}
+
+            // Filter Bot finished  chats
+            if json["agent"]["user"]["name"].as_str().unwrap_or("unknown_user") == "PipeBot" {return None;}
+
+            return Some(json);
+        })
+        // .clone()
+        .map(|messages| {
+            let json = messages.unwrap();
+            let talk_histories = &json["talk_histories"];
+
+            // dbg!(&talk_histories);
+
+            // talk_histories.as_array().unwrap().into_iter().enumerate().for_each(|(pos, message_obj)|{println!("{}: {}", pos, message_obj["message"])});
+            
+            // find the bot transfer message
+            let bot_transfer_message = talk_histories
+            .as_array().expect("Wrong message type received from talk histories").into_iter()
+            .enumerate()
+            .filter(|(pos, message_object)| {
+                let user_name  = message_object["user"]["name"].as_str().expect("Failed to decode message as string");
+                user_name == "PipeBot".to_string()
+            }).find(|(pos, message_object)|{
+                let message  = message_object["message"].as_str().expect("Failed to decode message as string");
+                // let found = message.find("Atendimento transferido para a fila [NovaNet -> Atendimento -> Suporte]");
+                let found = message.find("Atendimento entregue da fila de espera para o agente");
+                found.is_some()
+            });
+
+            // Find first agent message sent after the last bot message
+            let (pos, transfer_message) = bot_transfer_message.expect("Failed to get the transfer bot message position");
+            
+            let msg = talk_histories.as_array().expect("Wrong message type received from talk histories").into_iter()
+            .take(pos)
+            .rev()
+            .filter(|message| {
+                message["type"] == "out".to_string() && message["user"]["name"] != "PipeBot".to_string()
+            })
+            .take(1).collect_vec();
+            
+            let agent_first_message = msg[0];
+            
+            // Calculate time difference between bot message and agent message
+            let date_user_message_sent = agent_first_message["sent_at"].as_str().unwrap();
+
+            let format = "%Y-%m-%d %H:%M:%S";
+
+            let date_user_message_sent_parsed = match chrono::NaiveDateTime::parse_from_str(date_user_message_sent, format) {
+                Ok(dt) =>  dt,
+                Err(e) => {println!("Error parsing DateTime: {}", e); panic!("Failed parsing date")},
+            };
+
+            let date_transfer_message_sent_parsed = match chrono::NaiveDateTime::parse_from_str(transfer_message["sent_at"].as_str().unwrap(), format) {
+                Ok(dt) =>  dt,
+                Err(e) => {println!("Error parsing DateTime: {}", e); panic!("Failed parsing date")},
+            };
+
+            let response_time = (date_user_message_sent_parsed - date_transfer_message_sent_parsed).as_seconds_f32();
+            let name = agent_first_message["user"]["name"].as_str().unwrap().to_owned();
+            let id = json["tracking_number"].as_str().unwrap_or("").to_owned();
+            let date = date_user_message_sent.to_owned();
+            println!("response_time: {}s", (date_user_message_sent_parsed - date_transfer_message_sent_parsed).as_seconds_f32());
+
+            format!("{};{};{};{}", name, id, response_time, date)
+        }).reduce(|acc, e|{format!("{}\n{}",acc,e)})
+        .unwrap_or("".to_string());
+
+        // Open file and append to it
+        let mut response_time_file = std::fs::OpenOptions::new().write(true).open(format!("./evaluations/{formatted_day_before}/response_time.csv")).expect("Failed to open response time file for write");
+        response_time_file.write_all(response_time.as_bytes()).expect("Failed to write to file!");
+
     talk_ids
         .iter()
         .cloned()