path: root/src/components/new_chat.rs

use std::str::FromStr;

use filamento::{chat::Chat, error::{CommandError, DatabaseError}, user::User};
use jid::{BareJID, JID};
use leptos::{html::Input, prelude::*};
use leptos_fetch::QueryClient;
use reactive_stores::{ArcStore, Store};
use thiserror::Error;

use crate::{chat::MacawChat, client::Client, open_chats::OpenChatsPanel, user::MacawUser};

#[derive(Clone, Debug, Error)]
pub enum NewChatError {
    #[error("Missing JID")]
    MissingJID,
    #[error("Invalid JID: {0}")]
    InvalidJID(#[from] jid::ParseError),
    #[error("Database: {0}")]
    Db(#[from] CommandError<DatabaseError>),
}

// TODO: remove
#[derive(Clone)]
pub struct GotChat(Chat);

async fn get_chat(jid: BareJID) -> ArcStore<Chat> {
    // let client: Client = use_context().expect("no client in context");
    // ArcStore::new(client.get_chat(jid).await.unwrap())
    let GotChat(chat) = use_context().expect("no chat in context");
    ArcStore::new(chat)
}

#[derive(Clone)]
pub struct GotUser(User);

async fn get_user(jid: BareJID) -> ArcStore<User> {
    // let client: Client = use_context().expect("no client in context");
    // ArcStore::new(client.get_user(jid).await.unwrap())
    let GotUser(user) = use_context().expect("no user in context");
    ArcStore::new(user)
}

#[component]
pub fn NewChatWidget(set_open_new_chat: WriteSignal<bool>) -> impl IntoView {
    let jid = RwSignal::new("".to_string());

    // TODO: compartmentalise into error component, form component...
    let (error, set_error) = signal(None::<NewChatError>);
    let error_message = move || {
        error.with(|error| {
            if let Some(error) = error {
                view! { <div class="error">{error.to_string()}</div> }.into_any()
            } else {
                view! {}.into_any()
            }
        })
    };
    let (new_chat_pending, set_new_chat_pending) = signal(false);
 
    let open_chats: Store<OpenChatsPanel> =
        use_context().expect("no open chats panel store in context");
    let client = use_context::<Client>().expect("client not in context");

    let query_client: QueryClient = expect_context();

    let open_chat = Action::new_local(move |_| {
        let client = client.clone();
        async move {
            set_new_chat_pending.set(true);

            if jid.read_untracked().is_empty() {
                set_error.set(Some(NewChatError::MissingJID));
                set_new_chat_pending.set(false);
                return;
            }

            let jid = match JID::from_str(&jid.read_untracked()) {
                // TODO: ability to direct address a resource?
                Ok(j) => j.to_bare(),
                Err(e) => {
                    set_error.set(Some(e.into()));
                    set_new_chat_pending.set(false);
                    return;
                }
            };

            let chat_jid = jid;
            let (chat, user) = match client.get_chat_and_user(chat_jid).await {
                Ok(c) => c,
                Err(e) => {
                    set_error.set(Some(e.into()));
                    set_new_chat_pending.set(false);
                    return;
                },
            };

            let chat = {
                // let user = MacawUser::got_user(user);
                let jid = chat.correspondent.clone();
                provide_context(GotChat(chat));
                let chat = query_client.local_resource(get_chat, move || jid.clone());
                let user = {
                    let jid = user.jid.clone();
                    provide_context(GotUser(user));
                    MacawUser {
                        user: query_client.local_resource(get_user, move || jid.clone())
                    }
                    
                };
                MacawChat { chat, user }
            };
            open_chats.update(|open_chats| open_chats.open(chat.clone()));
            set_open_new_chat.set(false);
        }
    });

    let jid_input = NodeRef::<Input>::new();
    let _focus = Effect::new(move |_| {
        if let Some(input) = jid_input.get() {
            let _ = input.focus();
            input.set_text_content(Some(""));
            // input.style("height: 0");
            // let height = input.scroll_height();
            // input.style(format!("height: {}px", height));
        }
    });

    view! {
        <div class="new-chat-widget">
            <form on:submit=move |ev| {
                ev.prevent_default();
                open_chat.dispatch(());
            }>
                {error_message}
                <input
                    disabled=new_chat_pending
                    placeholder="JID"
                    type="text"
                    node_ref=jid_input
                    bind:value=jid
                    name="jid"
                    id="jid"
                    autofocus="true"
                />
                <input disabled=new_chat_pending class="button" type="submit" value="Start Chat" />
            </form>
        </div>
    }
}