1 files changed, 59 insertions, 12 deletions

diff --git a/filamento/src/logic/local_only.rs b/filamento/src/logic/local_only.rs
index 3f6fe8d..7f3a2e6 100644
--- a/filamento/src/logic/local_only.rs
+++ b/filamento/src/logic/local_only.rs
@@ -1,47 +1,94 @@
-use jid::JID;
+use jid::{BareJID, JID};
 use uuid::Uuid;
 
 use crate::{
     chat::{Chat, Message},
     error::DatabaseError,
+    files::FileStore,
     user::User,
 };
 
 use super::ClientLogic;
 
-pub async fn handle_get_chats(logic: &ClientLogic) -> Result<Vec<Chat>, DatabaseError> {
+pub async fn handle_get_chats<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+) -> Result<Vec<Chat>, DatabaseError> {
     Ok(logic.db().read_chats().await?)
 }
 
-pub async fn handle_get_chats_ordered(logic: &ClientLogic) -> Result<Vec<Chat>, DatabaseError> {
+pub async fn handle_get_chats_ordered<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+) -> Result<Vec<Chat>, DatabaseError> {
     Ok(logic.db().read_chats_ordered().await?)
 }
 
-pub async fn handle_get_chats_ordered_with_latest_messages(
-    logic: &ClientLogic,
+pub async fn handle_get_chats_ordered_with_latest_messages<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
 ) -> Result<Vec<(Chat, Message)>, DatabaseError> {
     Ok(logic.db().read_chats_ordered_with_latest_messages().await?)
 }
 
-pub async fn handle_get_chat(logic: &ClientLogic, jid: JID) -> Result<Chat, DatabaseError> {
+pub async fn handle_get_chats_ordered_with_latest_messages_and_users<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+) -> Result<Vec<((Chat, User), (Message, User))>, DatabaseError> {
+    Ok(logic
+        .db()
+        .read_chats_ordered_with_latest_messages_and_users()
+        .await?)
+}
+
+pub async fn handle_get_chat<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    jid: BareJID,
+) -> Result<Chat, DatabaseError> {
     Ok(logic.db().read_chat(jid).await?)
 }
 
-pub async fn handle_get_messages(
-    logic: &ClientLogic,
-    jid: JID,
+pub async fn handle_get_chat_and_user<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    jid: BareJID,
+) -> Result<(Chat, User), DatabaseError> {
+    Ok(logic.db().read_chat_and_user(jid).await?)
+}
+
+pub async fn handle_get_message<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    id: Uuid,
+) -> Result<Message, DatabaseError> {
+    Ok(logic.db().read_message(id).await?)
+}
+
+pub async fn handle_get_messages<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    jid: BareJID,
 ) -> Result<Vec<Message>, DatabaseError> {
     Ok(logic.db().read_message_history(jid).await?)
 }
 
-pub async fn handle_delete_chat(logic: &ClientLogic, jid: JID) -> Result<(), DatabaseError> {
+pub async fn handle_get_messages_with_users<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    jid: BareJID,
+) -> Result<Vec<(Message, User)>, DatabaseError> {
+    Ok(logic.db().read_message_history_with_users(jid).await?)
+}
+
+pub async fn handle_delete_chat<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    jid: BareJID,
+) -> Result<(), DatabaseError> {
     Ok(logic.db().delete_chat(jid).await?)
 }
 
-pub async fn handle_delete_messaage(logic: &ClientLogic, uuid: Uuid) -> Result<(), DatabaseError> {
+pub async fn handle_delete_messaage<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    uuid: Uuid,
+) -> Result<(), DatabaseError> {
     Ok(logic.db().delete_message(uuid).await?)
 }
 
-pub async fn handle_get_user(logic: &ClientLogic, jid: JID) -> Result<User, DatabaseError> {
+pub async fn handle_get_user<Fs: FileStore + Clone>(
+    logic: &ClientLogic<Fs>,
+    jid: BareJID,
+) -> Result<User, DatabaseError> {
     Ok(logic.db().read_user(jid).await?)
 }