Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
128 changes: 1 addition & 127 deletions whatsrust/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ pub use callbacks::CallbackTranslator;
pub use events::set_event_handler;
pub use lifecycle::{connect, disconnect, logout, new_client, pair_phone};
pub use media::{download_file, get_community_profile_picture, get_profile_picture};
use message_send::{build_content_for_ffi, quote_to_ffi};
pub use message_send::{TextSendResult, forward_message, send_message, send_text_message};
pub(crate) use models::file_kind_discriminant;
pub use models::{
ChatSettings, CommunitiesError, CommunityInfo, Contact, DownloadFailed, Event, FileContent,
Expand Down Expand Up @@ -311,132 +311,6 @@ impl CallbackTranslator<i64> for i64 {
}
}

pub fn forward_message(source: &Message, destinations: &[JID]) -> ForwardReport {
if destinations.is_empty() {
return ForwardReport::default();
}
let Some(forward_source) = forward_source(&source.info) else {
return ForwardReport::with_reason(
0,
destinations.len(),
ForwardFailure::SourceUnavailable,
);
};
let source_id = match CString::new(source.info.id.as_ref()) {
Ok(value) => value,
Err(_) => {
return ForwardReport::with_reason(
0,
destinations.len(),
ForwardFailure::InvalidSource,
);
}
};
let destination_values: Result<Vec<_>, _> = destinations
.iter()
.map(|jid| CString::new(jid.0.as_ref()))
.collect();
let Ok(destination_values) = destination_values else {
return ForwardReport::with_reason(
0,
destinations.len(),
ForwardFailure::InvalidDestination,
);
};
let destination_pointers: Vec<_> = destination_values.iter().map(|jid| jid.as_ptr()).collect();
let result = unsafe {
C_ForwardMessage(
source_id.as_ptr(),
CJID::from(&source.info.chat),
CJID::from(&source.info.sender),
source.info.is_from_me,
destination_pointers.as_ptr(),
destination_pointers.len(),
forward_source.as_ptr(),
forward_source.len(),
)
};
ForwardReport {
succeeded: result.succeeded as usize,
failed: result.failed as usize,
failure: ForwardFailure::from_repr(result.failure).unwrap_or(ForwardFailure::SendFailed),
}
}

pub fn send_message(
jid: &JID,
content: &MessageContent,
quoted_message: Option<&Message>,
mentions: &[Mention],
) {
let jid_c = CJID::from(jid);
let (msg_type, content_ptr, _holder) = build_content_for_ffi(content, mentions);
let (_quote_id_owner, quote_id, quote_sender, quote_chat) = quote_to_ffi(quoted_message);
let quote_content = quoted_message.map(|message| build_content_for_ffi(&message.message, &[]));
let (quote_message_type, quote_message_content) = quote_content
.as_ref()
.map_or((0, std::ptr::null()), |(message_type, pointer, _)| {
(*message_type, *pointer)
});

unsafe {
C_SendMessage(
jid_c,
msg_type,
content_ptr,
quote_id,
quote_sender,
quote_chat,
quote_message_type,
quote_message_content,
)
}
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum TextSendResult {
Sent,
Failed,
}

pub fn send_text_message(
jid: &JID,
content: &MessageContent,
quoted_message: Option<&Message>,
mentions: &[Mention],
local_send_id: u64,
) -> TextSendResult {
let MessageContent::Text(_) = content else {
return TextSendResult::Failed;
};
let jid_c = CJID::from(jid);
let (_message_type, content_ptr, _holder) = build_content_for_ffi(content, mentions);
let (_quote_id_owner, quote_id, quote_sender, quote_chat) = quote_to_ffi(quoted_message);
let quote_content = quoted_message.map(|message| build_content_for_ffi(&message.message, &[]));
let (quote_message_type, quote_message_content) = quote_content
.as_ref()
.map_or((0, std::ptr::null()), |(message_type, pointer, _)| {
(*message_type, *pointer)
});
let status = unsafe {
C_SendTextMessage(
jid_c,
content_ptr,
quote_id,
quote_sender,
quote_chat,
quote_message_type,
quote_message_content,
local_send_id,
)
};
if status == 0 {
TextSendResult::Sent
} else {
TextSendResult::Failed
}
}

/// Returns all contacts and groups as (JID, display name). Includes LID aliases for contacts.
pub fn get_contacts() -> Vec<(JID, Arc<str>)> {
let result = unsafe { C_GetContacts() };
Expand Down
128 changes: 127 additions & 1 deletion whatsrust/src/message_send.rs
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@
use std::ffi::{CString, c_char, c_void};

use crate::{
ForwardFailure, ForwardReport,
abi::{CFileMessage, CJID, CTextMessage, MessageType},
file_kind_discriminant,
models::{Mention, Message, MessageContent},
models::{JID, Mention, Message, MessageContent},
};

/// Keeps CStrings and C structs alive for the duration of an FFI call.
Expand Down Expand Up @@ -118,6 +119,131 @@ pub(crate) fn quote_to_ffi(quoted: Option<&Message>) -> (CString, *const c_char,
),
}
}

pub fn forward_message(source: &Message, destinations: &[JID]) -> ForwardReport {
if destinations.is_empty() {
return ForwardReport::default();
}
let Some(forward_source) = crate::forward_source(&source.info) else {
return ForwardReport::with_reason(
0,
destinations.len(),
ForwardFailure::SourceUnavailable,
);
};
let source_id = match CString::new(source.info.id.as_ref()) {
Ok(value) => value,
Err(_) => {
return ForwardReport::with_reason(
0,
destinations.len(),
ForwardFailure::InvalidSource,
);
}
};
let destination_values: Result<Vec<_>, _> = destinations
.iter()
.map(|jid| CString::new(jid.0.as_ref()))
.collect();
let Ok(destination_values) = destination_values else {
return ForwardReport::with_reason(
0,
destinations.len(),
ForwardFailure::InvalidDestination,
);
};
let destination_pointers: Vec<_> = destination_values.iter().map(|jid| jid.as_ptr()).collect();
let result = unsafe {
crate::abi::C_ForwardMessage(
source_id.as_ptr(),
CJID::from(&source.info.chat),
CJID::from(&source.info.sender),
source.info.is_from_me,
destination_pointers.as_ptr(),
destination_pointers.len(),
forward_source.as_ptr(),
forward_source.len(),
)
};
ForwardReport {
succeeded: result.succeeded as usize,
failed: result.failed as usize,
failure: ForwardFailure::from_repr(result.failure).unwrap_or(ForwardFailure::SendFailed),
}
}

pub fn send_message(
jid: &JID,
content: &MessageContent,
quoted_message: Option<&Message>,
mentions: &[Mention],
) {
let jid_c = CJID::from(jid);
let (msg_type, content_ptr, _holder) = build_content_for_ffi(content, mentions);
let (_quote_id_owner, quote_id, quote_sender, quote_chat) = quote_to_ffi(quoted_message);
let quote_content = quoted_message.map(|message| build_content_for_ffi(&message.message, &[]));
let (quote_message_type, quote_message_content) = quote_content
.as_ref()
.map_or((0, std::ptr::null()), |(message_type, pointer, _)| {
(*message_type, *pointer)
});
unsafe {
crate::abi::C_SendMessage(
jid_c,
msg_type,
content_ptr,
quote_id,
quote_sender,
quote_chat,
quote_message_type,
quote_message_content,
)
}
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
pub enum TextSendResult {
Sent,
Failed,
}

pub fn send_text_message(
jid: &JID,
content: &MessageContent,
quoted_message: Option<&Message>,
mentions: &[Mention],
local_send_id: u64,
) -> TextSendResult {
let MessageContent::Text(_) = content else {
return TextSendResult::Failed;
};
let jid_c = CJID::from(jid);
let (_message_type, content_ptr, _holder) = build_content_for_ffi(content, mentions);
let (_quote_id_owner, quote_id, quote_sender, quote_chat) = quote_to_ffi(quoted_message);
let quote_content = quoted_message.map(|message| build_content_for_ffi(&message.message, &[]));
let (quote_message_type, quote_message_content) = quote_content
.as_ref()
.map_or((0, std::ptr::null()), |(message_type, pointer, _)| {
(*message_type, *pointer)
});
let status = unsafe {
crate::abi::C_SendTextMessage(
jid_c,
content_ptr,
quote_id,
quote_sender,
quote_chat,
quote_message_type,
quote_message_content,
local_send_id,
)
};
if status == 0 {
TextSendResult::Sent
} else {
TextSendResult::Failed
}
}
#[cfg(test)]
mod tests {
use std::ffi::CStr;
Expand Down
Loading