//! # Conversations controller //! //! @author Pierre Hubert use crate::api_data::conversation_api::ConversationAPI; use crate::api_data::conversation_message_api::ConversationMessageAPI; use crate::api_data::list_unread_conversations_api::UnreadConversationAPI; use crate::api_data::res_count_unread_conversations::ResultCountUnreadConversations; use crate::api_data::res_create_conversation::ResCreateConversation; use crate::api_data::res_find_private_conversations::ResFindPrivateConversations; use crate::constants::MIN_CONVERSATION_MESSAGE_LENGTH; use crate::controllers::user_ws_controller; use crate::data::base_request_handler::BaseRequestHandler; use crate::data::conversation::{ConversationMemberSetting, NewConversationSettings}; use crate::data::conversation_message::ConversationMessageFile; use crate::data::error::Res; use crate::data::http_request_handler::HttpRequestHandler; use crate::data::new_conversation::NewConversation; use crate::data::new_conversation_message::NewConversationMessage; use crate::data::user::UserID; use crate::data::user_ws_connection::UserWsConnection; use crate::data::user_ws_message::UserWsMessage; use crate::helpers::{conversations_helper, events_helper, user_helper}; use crate::helpers::events_helper::Event; use crate::routes::RequestResult; use crate::utils::string_utils::remove_html_nodes; /// Create a new conversation pub fn create(r: &mut HttpRequestHandler) -> RequestResult { let name = r.post_string("name")?; let mut members = r.post_numbers_list("users", 1)? .iter() .map(|f| UserID::new(f.clone())) .collect::>(); // Adapt name let name = match name.as_str() { "false" => None, s => Some(s.to_string()) }; // Check if members exists for user in &members { if !user_helper::exists(user)? { r.not_found(format!("User {} not found!", user.id()))?; } } // Add current user ID if required let curr_user_id = r.user_id()?; if !members.contains(&curr_user_id) { members.push(curr_user_id); } let conv = NewConversation { owner_id: r.user_id()?, name, owner_following: r.post_bool("follow")?, members, can_everyone_add_members: r.post_bool_opt("canEveryoneAddMembers", true), color: r.post_color_opt("color")?, group_id: None, background: None,// TODO : add support for background }; // Create the conversation let conv_id = conversations_helper::create(&conv)?; r.set_response(ResCreateConversation::new(conv_id)) } /// Get the list of conversations of a user pub fn get_list(r: &mut HttpRequestHandler) -> RequestResult { let list = conversations_helper::get_list_user(&r.user_id()?)?; r.set_response(list.iter().map(|c| ConversationAPI::new(c)).collect::>()) } /// Get information about a single conversation pub fn get_single(r: &mut HttpRequestHandler) -> RequestResult { let conv = r.post_conv("conversationID")?; let conv = conversations_helper::get_single(conv.conv_id)?; r.set_response(ConversationAPI::new(&conv)) } /// Update the settings of a conversation pub fn update_settings(r: &mut HttpRequestHandler) -> RequestResult { let conv_membership = r.post_conv("conversationID")?; let conv = conversations_helper::get_single(conv_membership.conv_id)?; // Update following state, if required if r.has_post_parameter("following") { conversations_helper::set_following( &r.user_id()?, conv_membership.conv_id, r.post_bool("following")?, )?; } // Update members list if r.has_post_parameter("members") && !conv.is_managed() { let mut members = r.post_users_id("members")? .into_iter() .map(|user_id| ConversationMemberSetting { user_id, set_admin: false }) .collect::>(); let admins = r.post_users_id("admins")?; let can_everyone_add_members = conversations_helper::can_everyone_add_members(conv_membership.conv_id)?; if !conv_membership.is_admin && !can_everyone_add_members { r.forbidden("You can not update the list of members of this conversation!".to_string())?; } // Set same admin status as earlier if !conv_membership.is_admin { members = members .into_iter() .map(|mut n| { n.set_admin = conv.is_admin(&n.user_id); n }) .collect() } // If the user is an admin, use the values he gave else { members = members .into_iter() .map(|mut n| { n.set_admin = admins.contains(&n.user_id); n }) .collect() } // Current user can not touch his own membership, so revert it back forcefully members = members .into_iter() .filter(|m| !m.user_id.eq(&conv_membership.user_id)) .collect::>(); members.push(ConversationMemberSetting { user_id: r.user_id()?, set_admin: conv_membership.is_admin, }); conversations_helper::set_members(conv_membership.conv_id, &members, conv_membership.is_admin)?; } // Change moderator settings if r.has_post_parameter("name") && r.has_post_parameter("canEveryoneAddMembers") && r.has_post_parameter("color") { if !conv_membership.is_admin { r.forbidden("You are not allowed to perform changes on this conversation!".to_string())?; } let new_settings = NewConversationSettings { conv_id: conv_membership.conv_id, name: r.post_string_optional("name").map(|s| remove_html_nodes(&s)), color: r.post_color_opt("color")?, can_everyone_add_members: r.post_bool("canEveryoneAddMembers")?, }; conversations_helper::set_settings(new_settings)?; } r.success("Conversation information successfully updated!") } /// Find a private conversation pub fn find_private(r: &mut HttpRequestHandler) -> RequestResult { let other_user = r.post_user_id("otherUser")?; let allow_create = r.post_bool_opt("allowCreate", false); // Query the database let mut list = conversations_helper::find_private(&r.user_id()?, &other_user)?; if list.is_empty() { if !allow_create { return r.not_found(format!("Not any private conversation was found. The server was not allowed to create a new one...")); } let new_conv = NewConversation { owner_id: r.user_id()?, name: None, owner_following: true, members: vec![r.user_id()?, other_user], can_everyone_add_members: true, color: r.post_color_opt("color")?, background: None, group_id: None, }; let conv_id = conversations_helper::create(&new_conv)?; list.push(conv_id); } r.set_response(ResFindPrivateConversations::new(list)) } /// Refresh a single conversation pub fn refresh_single(r: &mut HttpRequestHandler) -> RequestResult { let conv = r.post_conv("conversationID")?; let last_message_id = r.post_u64("last_message_id")?; let messages = match last_message_id { // Get latest messages of the conversation 0 => conversations_helper::get_last_messages(conv.conv_id, 10)?, // Get new messages _ => conversations_helper::get_new_messages(conv.conv_id, last_message_id)?, }; if messages.len() > 0 { conversations_helper::mark_user_seen( conv.conv_id, r.user_id_ref()?, messages.iter().map(|m| m.id).max().unwrap(), )?; } r.set_response(ConversationMessageAPI::for_list(&messages)) } /// Get older messages of a conversation pub fn get_older_messages(r: &mut HttpRequestHandler) -> RequestResult { let conv = r.post_conv("conversationID")?; let max_id = r.post_u64("oldest_message_id")? - 1; // Determine limit let limit = r.post_u64("limit")?; let limit = match limit { 0 => 1, 1..=60 => limit, _ => 60 }; let messages = conversations_helper::get_older_messages(conv.conv_id, max_id, limit)?; r.set_response(ConversationMessageAPI::for_list(&messages)) } /// Send a new message pub fn send_message(r: &mut HttpRequestHandler) -> RequestResult { let conv = r.post_conv("conversationID")?; // TODO : add support for other files type // Get image let file = match r.has_file("image") { false => None, true => { let path = r.save_post_image("image", "conversations", 1200, 1200)?; Some(ConversationMessageFile { path: path.clone(), size: std::fs::metadata(&path)?.len(), name: "picture.png".to_string(), thumbnail: Some(r.save_post_image("image", "conversations", 50, 50)?), r#type: "image/png".to_string(), }) } }; let message = if let None = file { Some(r.post_string_without_html("message", MIN_CONVERSATION_MESSAGE_LENGTH, true)?) } else { None }; conversations_helper::send_message(&NewConversationMessage { user_id: Some(r.user_id()?), conv_id: conv.conv_id, message, file, server_message: None, })?; r.success("Conversation message was successfully sent!") } /// Count the number of unread conversation of the user pub fn count_unread(r: &mut HttpRequestHandler) -> RequestResult { let num = conversations_helper::count_unread_for_user(&r.user_id()?)?; r.set_response(ResultCountUnreadConversations::new(num)) } /// Get the list of unread conversations of a user pub fn list_unread(r: &mut HttpRequestHandler) -> RequestResult { let list = conversations_helper::get_list_unread(&r.user_id()?)?; r.set_response(UnreadConversationAPI::for_list(&list)?) } /// Delete a conversation pub fn delete_conversation(r: &mut HttpRequestHandler) -> RequestResult { let conv_membership = r.post_conv("conversationID")?; let conv = conversations_helper::get_single(conv_membership.conv_id)?; if conv.is_managed() { r.bad_request("This conversation is managed, it can not be deleted by this way!".to_string())?; } conversations_helper::remove_user_from_conversation(&r.user_id()?, &conv, r.user_id_ref()?)?; r.success("The conversation has been deleted") } /// Update a single conversation message pub fn update_message(r: &mut HttpRequestHandler) -> RequestResult { let msg_id = r.post_u64("messageID")?; let new_content = r.post_string_opt("content", 3, true)?; if !conversations_helper::is_message_owner(&r.user_id()?, msg_id)? { r.forbidden("You are not the owner of this message!".to_string())?; } conversations_helper::update_message_content(msg_id, &new_content)?; r.success("Conversation message content successfully updated") } /// Delete a conversation message pub fn delete_message(r: &mut HttpRequestHandler) -> RequestResult { let msg_id = r.post_u64("messageID")?; if !conversations_helper::is_message_owner(&r.user_id()?, msg_id)? { r.forbidden("You are not the owner of this message!".to_string())?; } conversations_helper::delete_message_by_id(msg_id)?; r.success("The message has been successfully deleted!") } /// Events handler pub fn handle_event(e: &events_helper::Event) -> Res { match e { Event::UpdatedNumberUnreadConversations(users) => { for user in users.iter() { if user_ws_controller::is_user_connected(user) { user_ws_controller::send_message_to_user( &UserWsMessage::no_id_message( "number_unread_conversations", conversations_helper::count_unread_for_user(user)?, )?, user, )?; } } } Event::NewConversationMessage(msg) => { user_ws_controller::send_message_to_specific_connections( |f| f.conversations.contains(&msg.conv_id), |_| UserWsMessage::no_id_message("new_conv_message", ConversationMessageAPI::new(msg)), Some(|conn: &UserWsConnection| conversations_helper::mark_user_seen(msg.conv_id, conn.user_id(), msg.id)), )?; } Event::UpdatedConversationMessage(msg) => { user_ws_controller::send_message_to_specific_connections( |f| f.conversations.contains(&msg.conv_id), |_| UserWsMessage::no_id_message("updated_conv_message", ConversationMessageAPI::new(msg)), None:: _>, )?; } Event::DeleteConversationMessage(msg) => { user_ws_controller::send_message_to_specific_connections( |f| f.conversations.contains(&msg.conv_id), |_| UserWsMessage::no_id_message("deleted_conv_message", ConversationMessageAPI::new(msg)), None:: _>, )?; } _ => {} } Ok(()) }