started development

This commit is contained in:
FantasyPvP
2024-12-02 03:41:18 +00:00
commit 0ae26d46bf
35 changed files with 9749 additions and 0 deletions
+99
View File
@@ -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>
}
}
+115
View File
@@ -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()),
}
}
+134
View File
@@ -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)),
}
}
+102
View File
@@ -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,
}
}
+63
View File
@@ -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();
}