started development
This commit is contained in:
@@ -0,0 +1,99 @@
|
||||
use serde::{Deserialize, Serialize};
|
||||
use yew::prelude::*;
|
||||
use web_sys::HtmlInputElement;
|
||||
use chrono::prelude::*;
|
||||
use crate::hooks::websocket::use_websocket;
|
||||
|
||||
#[derive(Clone, Debug, Serialize, Deserialize)]
|
||||
pub struct RealTimeMessage {
|
||||
pub message_id: i32,
|
||||
pub user_id: i32,
|
||||
pub display_name: String,
|
||||
pub created_at: i64,
|
||||
pub content: String,
|
||||
}
|
||||
|
||||
#[function_component(Chat)]
|
||||
pub fn chat() -> Html {
|
||||
let ws = use_websocket("ws://localhost:8000/messenger/connect/1");
|
||||
let input_ref = use_node_ref();
|
||||
let dark_theme = use_state(|| true);
|
||||
|
||||
// let theme_toggle = {
|
||||
// let dark_theme = dark_theme.clone();
|
||||
// Callback::from(move |_| {
|
||||
// dark_theme.set(!*dark_theme);
|
||||
// })
|
||||
// };
|
||||
|
||||
let onsubmit = {
|
||||
let ws = ws.ws.clone();
|
||||
let input_ref = input_ref.clone();
|
||||
|
||||
Callback::from(move |e: SubmitEvent| {
|
||||
e.prevent_default();
|
||||
if let Some(input) = input_ref.cast::<HtmlInputElement>() {
|
||||
let message = input.value();
|
||||
if !message.is_empty() {
|
||||
ws.send_with_str(&message).unwrap();
|
||||
input.set_value("");
|
||||
}
|
||||
}
|
||||
})
|
||||
};
|
||||
|
||||
html! {
|
||||
<div class={classes!("app-container", if *dark_theme { "dark-theme" } else { "light-theme" })}>
|
||||
<nav class="navbar">
|
||||
<div class="nav-brand">{"Chat App"}</div>
|
||||
// <div class="theme-toggle">
|
||||
// <button onclick={theme_toggle} class="theme-button">
|
||||
// if *dark_theme {
|
||||
// {"🌞"}
|
||||
// } else {
|
||||
// {"🌙"}
|
||||
// }
|
||||
// </button>
|
||||
// </div>
|
||||
</nav>
|
||||
<div class="chat-container">
|
||||
<div class="messages-container">
|
||||
{ws.messages.messages().iter().map(|msg| {
|
||||
let timestamp = Local.timestamp_millis_opt(msg.created_at).unwrap();
|
||||
let formatted_time = timestamp.format("%d/%m/%y %H:%M").to_string();
|
||||
let userid = msg.user_id;
|
||||
|
||||
html! {
|
||||
<div class="message">
|
||||
<div class="profile-picture" style={ format!(
|
||||
"background-image: url('http://localhost:8000/static/pfp/{userid}.png')"
|
||||
)}></div>
|
||||
<div class="message-bubble">
|
||||
<div class="message-header">
|
||||
<span class="username">{&msg.display_name}</span>
|
||||
<span class="timestamp">{formatted_time}</span>
|
||||
</div>
|
||||
<div class="message-content">{&msg.content}</div>
|
||||
</div>
|
||||
</div>
|
||||
}
|
||||
}).collect::<Html>()}
|
||||
</div>
|
||||
if let Some(error) = (*ws.error).as_ref() {
|
||||
<div class="error-message">
|
||||
{format!("Error: {}", error)}
|
||||
</div>
|
||||
}
|
||||
<form {onsubmit} class="message-form">
|
||||
<input
|
||||
type="text"
|
||||
ref={input_ref}
|
||||
class="message-input"
|
||||
placeholder="Type a message..."
|
||||
/>
|
||||
<button type="submit" class="send-button">{"Send"}</button>
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,115 @@
|
||||
use gloo_net::http::Request;
|
||||
use serde::{Deserialize, Serialize};
|
||||
use yew::prelude::*;
|
||||
use yew_router::{navigator, prelude::*};
|
||||
use gloo::storage::{LocalStorage, Storage};
|
||||
use web_sys::HtmlInputElement;
|
||||
use wasm_bindgen::JsCast;
|
||||
use wasm_bindgen_futures::spawn_local;
|
||||
|
||||
use crate::Route;
|
||||
|
||||
#[function_component(Login)]
|
||||
pub fn login_page() -> Html {
|
||||
let navigator = use_navigator().unwrap();
|
||||
let username_ref = use_node_ref();
|
||||
let password_ref = use_node_ref();
|
||||
let login_success = use_state(|| true);
|
||||
|
||||
let navigator_clone = navigator.clone();
|
||||
let username_ref_clone = username_ref.clone();
|
||||
let password_ref_clone = password_ref.clone();
|
||||
let login_success_clone = login_success.clone();
|
||||
let onsubmit = Callback::from(move |e: SubmitEvent| {
|
||||
e.prevent_default();
|
||||
let username = username_ref_clone.cast::<HtmlInputElement>().unwrap().value();
|
||||
let password = password_ref_clone.cast::<HtmlInputElement>().unwrap().value();
|
||||
|
||||
// TODO: Replace this with actual authentication
|
||||
// For now, we'll just set a dummy token
|
||||
if !( username.is_empty() || password.is_empty() ) {
|
||||
let navigator = navigator_clone.clone();
|
||||
let login_success = login_success_clone.clone();
|
||||
spawn_local(async move {
|
||||
match login(username, password).await {
|
||||
Ok(_) => navigator.push(&Route::Chat),
|
||||
Err(_) => login_success.set(false),
|
||||
}
|
||||
});
|
||||
}
|
||||
});
|
||||
|
||||
let go_to_signup = {
|
||||
let navigator_clone = navigator.clone();
|
||||
Callback::from(move |_| {
|
||||
navigator_clone.push(&Route::Signup);
|
||||
})
|
||||
};
|
||||
|
||||
html! {
|
||||
<div class="login-container">
|
||||
<form {onsubmit} class="login-form">
|
||||
<h2 class="login-title">{"Login"}</h2>
|
||||
<input
|
||||
ref={username_ref}
|
||||
class="login-input"
|
||||
type="text"
|
||||
id="username"
|
||||
name="username"
|
||||
placeholder="Username"
|
||||
/>
|
||||
<input
|
||||
ref={password_ref}
|
||||
class="login-input"
|
||||
type="password"
|
||||
id="password"
|
||||
name="password"
|
||||
placeholder="Password"
|
||||
/>
|
||||
<button class="login-button" type="submit">{"Login"}</button>
|
||||
{
|
||||
if !(*login_success) {
|
||||
html! {
|
||||
<p class="login-error">{"Incorrect username or password"}</p>
|
||||
}
|
||||
} else {
|
||||
html! {}
|
||||
}
|
||||
}
|
||||
|
||||
<p class="login-text">{"Don't have an account?"}</p>
|
||||
<a onclick={go_to_signup}
|
||||
href=""
|
||||
class="login-button"
|
||||
>
|
||||
{"Create Account"}
|
||||
</a>
|
||||
</form>
|
||||
</div>
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
struct LoginRequest {
|
||||
username: String,
|
||||
password: String,
|
||||
}
|
||||
|
||||
async fn login(username: String, password: String) -> Result<(), String> {
|
||||
let login_request = LoginRequest {
|
||||
username,
|
||||
password,
|
||||
};
|
||||
|
||||
match Request::post("http://127.0.0.1:8000/login")
|
||||
.json(&login_request)
|
||||
.map_err(|e| e.to_string())?
|
||||
.send()
|
||||
.await
|
||||
.map_err(|e| e.to_string())?
|
||||
.status()
|
||||
{
|
||||
200 => Ok(()),
|
||||
_ => Err("Login failed".to_string()),
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,134 @@
|
||||
use gloo_net::http::Request;
|
||||
use serde::{Deserialize, Serialize};
|
||||
use yew::{function_component, html, use_node_ref, use_state, Callback, Html, SubmitEvent};
|
||||
use yew_router::{navigator, prelude::*};
|
||||
use gloo::storage::{LocalStorage, Storage};
|
||||
use web_sys::HtmlInputElement;
|
||||
use wasm_bindgen::JsCast;
|
||||
use wasm_bindgen_futures::spawn_local;
|
||||
|
||||
use crate::Route;
|
||||
|
||||
#[function_component(Signup)]
|
||||
pub fn signup_page() -> Html {
|
||||
let navigator = use_navigator().unwrap();
|
||||
let username_ref = use_node_ref();
|
||||
let password_ref = use_node_ref();
|
||||
let confirm_password_ref = use_node_ref();
|
||||
let signup_error = use_state(|| None::<String>);
|
||||
|
||||
let navigator_clone = navigator.clone();
|
||||
let username_ref_clone = username_ref.clone();
|
||||
let password_ref_clone = password_ref.clone();
|
||||
let confirm_password_ref_clone = confirm_password_ref.clone();
|
||||
let signup_error_clone = signup_error.clone();
|
||||
|
||||
let onsubmit = Callback::from(move |e: SubmitEvent| {
|
||||
e.prevent_default();
|
||||
let username = username_ref_clone.cast::<HtmlInputElement>().unwrap().value();
|
||||
let password = password_ref_clone.cast::<HtmlInputElement>().unwrap().value();
|
||||
let confirm_password = confirm_password_ref_clone.cast::<HtmlInputElement>().unwrap().value();
|
||||
|
||||
if username.is_empty() || password.is_empty() {
|
||||
signup_error_clone.set(Some("Please fill in all fields".to_string()));
|
||||
return;
|
||||
}
|
||||
|
||||
if password != confirm_password {
|
||||
signup_error_clone.set(Some("Passwords do not match".to_string()));
|
||||
return;
|
||||
}
|
||||
|
||||
let navigator = navigator_clone.clone();
|
||||
let signup_error = signup_error_clone.clone();
|
||||
spawn_local(async move {
|
||||
match signup(username, password).await {
|
||||
Ok(_) => navigator.push(&Route::Chat),
|
||||
Err(e) => signup_error.set(Some(e)),
|
||||
}
|
||||
});
|
||||
});
|
||||
|
||||
let go_to_login = {
|
||||
let navigator = navigator.clone();
|
||||
Callback::from(move |_| {
|
||||
navigator.push(&Route::Login);
|
||||
})
|
||||
};
|
||||
|
||||
html! {
|
||||
<div class="login-container">
|
||||
<form {onsubmit} class="login-form">
|
||||
<h2 class="login-title">{"Sign Up"}</h2>
|
||||
<input
|
||||
ref={username_ref}
|
||||
class="login-input"
|
||||
type="text"
|
||||
id="username"
|
||||
name="username"
|
||||
placeholder="Username"
|
||||
/>
|
||||
<input
|
||||
ref={password_ref}
|
||||
class="login-input"
|
||||
type="password"
|
||||
id="password"
|
||||
name="password"
|
||||
placeholder="Password"
|
||||
/>
|
||||
<input
|
||||
ref={confirm_password_ref}
|
||||
class="login-input"
|
||||
type="password"
|
||||
id="confirm_password"
|
||||
name="confirm_password"
|
||||
placeholder="Confirm Password"
|
||||
/>
|
||||
<button class="login-button" type="submit">{"Sign Up"}</button>
|
||||
{
|
||||
if let Some(error) = (*signup_error).clone() {
|
||||
html! {
|
||||
<p class="login-error">{error}</p>
|
||||
}
|
||||
} else {
|
||||
html! {}
|
||||
}
|
||||
}
|
||||
|
||||
<p class="login-text">{"Already have an account?"}</p>
|
||||
<a onclick={go_to_login}
|
||||
href=""
|
||||
class="login-button"
|
||||
>
|
||||
{"Login"}
|
||||
</a>
|
||||
</form>
|
||||
</div>
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
struct SignupRequest {
|
||||
username: String,
|
||||
password: String,
|
||||
}
|
||||
|
||||
async fn signup(username: String, password: String) -> Result<(), String> {
|
||||
let signup_request = SignupRequest {
|
||||
username,
|
||||
password,
|
||||
};
|
||||
|
||||
match Request::post("http://127.0.0.1:8000/signup")
|
||||
.json(&signup_request)
|
||||
.map_err(|e| e.to_string())?
|
||||
.send()
|
||||
.await
|
||||
.map_err(|e| e.to_string())?
|
||||
.status()
|
||||
{
|
||||
200 => Ok(()),
|
||||
409 => Err("Username already exists".to_string()),
|
||||
x => Err(format!("Signup failed with status code {}", x)),
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,102 @@
|
||||
use yew::prelude::*;
|
||||
use web_sys::{WebSocket, MessageEvent, ErrorEvent};
|
||||
use wasm_bindgen::{prelude::*, JsCast};
|
||||
use std::rc::Rc;
|
||||
use crate::components::chat::RealTimeMessage;
|
||||
|
||||
pub struct UseWebSocketHandle {
|
||||
pub ws: Rc<WebSocket>,
|
||||
pub messages: UseReducerHandle<MessagesState>,
|
||||
pub error: UseStateHandle<Option<String>>,
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
pub struct MessagesState {
|
||||
messages: Vec<RealTimeMessage>,
|
||||
}
|
||||
|
||||
pub enum MessagesAction {
|
||||
AddMessage(RealTimeMessage),
|
||||
}
|
||||
|
||||
impl Reducible for MessagesState {
|
||||
type Action = MessagesAction;
|
||||
|
||||
fn reduce(self: Rc<Self>, action: Self::Action) -> Rc<Self> {
|
||||
match action {
|
||||
MessagesAction::AddMessage(msg) => {
|
||||
let mut new_messages = self.messages.clone();
|
||||
new_messages.push(msg);
|
||||
Rc::new(MessagesState {
|
||||
messages: new_messages,
|
||||
})
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
impl MessagesState {
|
||||
pub fn messages(&self) -> Vec<RealTimeMessage> {
|
||||
self.messages.clone()
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
#[hook]
|
||||
pub fn use_websocket(url: &str) -> UseWebSocketHandle {
|
||||
use web_sys::js_sys;
|
||||
|
||||
let messages = use_reducer(|| MessagesState { messages: Vec::new() });
|
||||
let error = use_state(|| None::<String>);
|
||||
|
||||
let ws = use_state_eq(|| {
|
||||
Rc::new(WebSocket::new(url).unwrap_or_else(|e| panic!("Failed to open WebSocket: {:?}", e)))
|
||||
});
|
||||
|
||||
{
|
||||
let messages = messages.clone();
|
||||
let error = error.clone();
|
||||
let ws = (*ws).clone();
|
||||
|
||||
use_effect_with((), move |_| {
|
||||
let ws_clone = ws.clone();
|
||||
let error_clone = error.clone();
|
||||
|
||||
// Set up message handler
|
||||
let onmessage_callback = {
|
||||
let messages = messages.clone();
|
||||
Closure::wrap(Box::new(move |e: MessageEvent| {
|
||||
if let Ok(txt) = e.data().dyn_into::<js_sys::JsString>() {
|
||||
if let Ok(msg) = serde_json::from_str::<RealTimeMessage>(&txt.as_string().unwrap()) {
|
||||
messages.dispatch(MessagesAction::AddMessage(msg));
|
||||
} else {
|
||||
error_clone.set(Some("Failed to parse message".to_string()));
|
||||
}
|
||||
}
|
||||
}) as Box<dyn FnMut(MessageEvent)>)
|
||||
};
|
||||
|
||||
ws.set_onmessage(Some(onmessage_callback.as_ref().unchecked_ref()));
|
||||
onmessage_callback.forget();
|
||||
|
||||
// Set up error handler
|
||||
let error_clone = error.clone();
|
||||
let onerror_callback = Closure::wrap(Box::new(move |e: ErrorEvent| {
|
||||
error_clone.set(Some(e.message()));
|
||||
}) as Box<dyn FnMut(ErrorEvent)>);
|
||||
|
||||
ws.set_onerror(Some(onerror_callback.as_ref().unchecked_ref()));
|
||||
onerror_callback.forget();
|
||||
|
||||
move || {
|
||||
ws_clone.close().unwrap_or_else(|_| {});
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
UseWebSocketHandle {
|
||||
ws: (*ws).clone(),
|
||||
messages,
|
||||
error,
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,63 @@
|
||||
use yew::prelude::*;
|
||||
use yew_router::prelude::*;
|
||||
use gloo::storage::{LocalStorage, Storage};
|
||||
|
||||
mod hooks {
|
||||
pub mod websocket;
|
||||
}
|
||||
|
||||
mod components {
|
||||
pub mod chat;
|
||||
pub mod signup;
|
||||
pub mod login;
|
||||
}
|
||||
|
||||
use components::{
|
||||
chat::Chat,
|
||||
login::Login,
|
||||
signup::Signup,
|
||||
};
|
||||
|
||||
#[derive(Clone, Routable, PartialEq)]
|
||||
enum Route {
|
||||
#[at("/")]
|
||||
Root,
|
||||
#[at("/login")]
|
||||
Login,
|
||||
#[at("/signup")]
|
||||
Signup,
|
||||
#[at("/chat")]
|
||||
Chat,
|
||||
#[not_found]
|
||||
#[at("/404")]
|
||||
NotFound,
|
||||
}
|
||||
|
||||
fn switch(route: Route) -> Html {
|
||||
match route {
|
||||
Route::Root => html! { <Redirect<Route> to={Route::Login}/> },
|
||||
Route::Login => html! { <Login /> },
|
||||
Route::Signup => html! { <Signup /> },
|
||||
Route::Chat => {
|
||||
if let Ok(token) = LocalStorage::get::<String>("auth_token") {
|
||||
html! { <Chat /> }
|
||||
} else {
|
||||
html! { <Redirect<Route> to={Route::Login}/> }
|
||||
}
|
||||
}
|
||||
Route::NotFound => html! { <h1>{"404 Not Found"}</h1> },
|
||||
}
|
||||
}
|
||||
|
||||
#[function_component(App)]
|
||||
fn app() -> Html {
|
||||
html! {
|
||||
<BrowserRouter>
|
||||
<Switch<Route> render={switch} />
|
||||
</BrowserRouter>
|
||||
}
|
||||
}
|
||||
|
||||
fn main() {
|
||||
yew::Renderer::<App>::new().render();
|
||||
}
|
||||
Reference in New Issue
Block a user