Implement logout

Also introduce a library to handle cookies
This commit is contained in:
Valentin Tolmer 2021-05-18 19:04:06 +02:00
parent d57cd1230c
commit 4d9f554fe6
5 changed files with 150 additions and 70 deletions

View File

@ -1,6 +1,6 @@
use crate::cookies::set_cookie;
use anyhow::{anyhow, Result};
use lldap_model::*;
use wasm_bindgen::JsCast;
use yew::callback::Callback;
use yew::format::Json;
@ -30,19 +30,17 @@ impl HostService {
let url = "/api/users";
let handler = move |response: Response<Result<String>>| {
let (meta, maybe_data) = response.into_parts();
match maybe_data {
Ok(data) => {
let message = maybe_data
.map_err(|e| anyhow!("Could not fetch: {}", e))
.and_then(|data| {
if meta.status.is_success() {
callback.emit(
serde_json::from_str(&data)
.map_err(|e| anyhow!("Could not parse response: {}", e)),
)
serde_json::from_str(&data)
.map_err(|e| anyhow!("Could not parse response: {}", e))
} else {
callback.emit(Err(anyhow!("[{}]: {}", meta.status, data)))
Err(anyhow!("[{}]: {}", meta.status, data))
}
}
Err(e) => callback.emit(Err(anyhow!("Could not fetch: {}", e))),
}
});
callback.emit(message)
};
let request = Request::post(url)
.header("Content-Type", "application/json")
@ -57,43 +55,28 @@ impl HostService {
let url = "/api/authorize";
let handler = move |response: Response<Result<String>>| {
let (meta, maybe_data) = response.into_parts();
match maybe_data {
Ok(data) => {
let message = maybe_data
.map_err(|e| anyhow!("Could not reach authentication server: {}", e))
.and_then(|data| {
if meta.status.is_success() {
match get_claims_from_jwt(&data) {
Ok(jwt_claims) => {
let document = web_sys::window()
.unwrap()
.document()
.unwrap()
.dyn_into::<web_sys::HtmlDocument>()
.unwrap();
document
.set_cookie(&format!(
"user_id={}; expires={}",
&jwt_claims.user, &jwt_claims.exp
))
.unwrap();
callback.emit(Ok(jwt_claims.user.clone()))
}
Err(e) => {
callback.emit(Err(anyhow!("Could not parse response: {}", e)))
}
}
get_claims_from_jwt(&data)
.map_err(|e| anyhow!("Could not parse response: {}", e))
.and_then(|jwt_claims| {
set_cookie("user_id", &jwt_claims.user, &jwt_claims.exp)
.map(|_| jwt_claims.user.clone())
.map_err(|e| anyhow!("Error clearing cookie: {}", e))
})
} else if meta.status == 401 {
callback.emit(Err(anyhow!("Invalid username or password")))
Err(anyhow!("Invalid username or password"))
} else {
callback.emit(Err(anyhow!(
Err(anyhow!(
"Could not authenticate: [{}]: {}",
meta.status,
data
)))
))
}
}
Err(e) => {
callback.emit(Err(anyhow!("Could not reach authentication server: {}", e)))
}
}
});
callback.emit(message)
};
let request = Request::post(url)
.header("Content-Type", "application/json")

View File

@ -1,8 +1,9 @@
use crate::cookies::get_cookie;
use crate::login::LoginForm;
use crate::logout::LogoutButton;
use crate::user_table::UserTable;
use anyhow::{anyhow, Result};
use wasm_bindgen::JsCast;
use yew::prelude::*;
use yew::services::ConsoleService;
pub struct App {
link: ComponentLink<Self>,
@ -11,30 +12,7 @@ pub struct App {
pub enum Msg {
Login(String),
}
fn extract_user_id_cookie() -> Result<String> {
let document = web_sys::window()
.unwrap()
.document()
.unwrap()
.dyn_into::<web_sys::HtmlDocument>()
.unwrap();
let cookies = document.cookie().unwrap();
yew::services::ConsoleService::info(&cookies);
cookies
.split(";")
.filter_map(|c| c.split_once('='))
.map(|(name, value)| {
if name == "user_id" {
Ok(value.into())
} else {
Err(anyhow!("Wrong cookie"))
}
})
.filter(Result::is_ok)
.next()
.unwrap_or(Err(anyhow!("User ID cookie not found in response")))
Logout,
}
impl Component for App {
@ -44,7 +22,10 @@ impl Component for App {
fn create(_: Self::Properties, link: ComponentLink<Self>) -> Self {
App {
link: link.clone(),
user_name: extract_user_id_cookie().ok(),
user_name: get_cookie("user_id").unwrap_or_else(|e| {
ConsoleService::error(&e.to_string());
None
}),
}
}
@ -53,6 +34,9 @@ impl Component for App {
Msg::Login(user_name) => {
self.user_name = Some(user_name);
}
Msg::Logout => {
self.user_name = None;
}
}
true
}
@ -66,9 +50,14 @@ impl Component for App {
<div>
<h1>{ "LLDAP" }</h1>
{if self.user_name.is_some() {
html! {<UserTable />}
html! {
<div>
<LogoutButton on_logged_out=self.link.callback(|_| Msg::Logout) />
<UserTable />
</div>
}
} else {
html! {<LoginForm on_logged_in=self.link.callback(|u| { Msg::Login(u) })/>}
html! {<LoginForm on_logged_in=self.link.callback(|u| Msg::Login(u))/>}
}}
</div>
}

52
app/src/cookies.rs Normal file
View File

@ -0,0 +1,52 @@
use anyhow::{anyhow, Result};
use chrono::prelude::*;
use wasm_bindgen::JsCast;
use web_sys::HtmlDocument;
fn get_document() -> Result<HtmlDocument> {
web_sys::window()
.map(|w| w.document())
.flatten()
.ok_or(anyhow!("Could not get window document"))
.and_then(|d| {
d.dyn_into::<web_sys::HtmlDocument>()
.map_err(|_| anyhow!("Document is not an HTMLDocument"))
})
}
pub fn set_cookie(cookie_name: &str, value: &str, expiration: &DateTime<Utc>) -> Result<()> {
let doc = web_sys::window()
.map(|w| w.document())
.flatten()
.ok_or(anyhow!("Could not get window document"))
.and_then(|d| {
d.dyn_into::<web_sys::HtmlDocument>()
.map_err(|_| anyhow!("Document is not an HTMLDocument"))
})?;
doc.set_cookie(&format!("{}={};expires={}", cookie_name, value, expiration))
.map_err(|_| anyhow!("Could not set cookie"))
}
pub fn get_cookie(cookie_name: &str) -> Result<Option<String>> {
let cookies = get_document()?
.cookie()
.map_err(|_| anyhow!("Could not access cookies"))?;
Ok(cookies
.split(";")
.filter_map(|c| c.split_once('='))
.find_map(|(name, value)| {
if name == cookie_name {
Some(value.to_string())
} else {
None
}
}))
}
pub fn delete_cookie(cookie_name: &str) -> Result<()> {
if let Some(_) = get_cookie(cookie_name)? {
set_cookie(cookie_name, "", &Utc.ymd(1970, 1, 1).and_hms(0, 0, 0))
} else {
Ok(())
}
}

View File

@ -1,7 +1,9 @@
#![recursion_limit = "256"]
mod api;
mod app;
mod cookies;
mod login;
mod logout;
mod user_table;
use wasm_bindgen::prelude::{wasm_bindgen, JsValue};

54
app/src/logout.rs Normal file
View File

@ -0,0 +1,54 @@
use crate::cookies::delete_cookie;
use yew::prelude::*;
use yew::services::ConsoleService;
pub struct LogoutButton {
link: ComponentLink<Self>,
on_logged_out: Callback<()>,
}
#[derive(Clone, PartialEq, Properties)]
pub struct Props {
pub on_logged_out: Callback<()>,
}
pub enum Msg {
Logout,
}
impl Component for LogoutButton {
type Message = Msg;
type Properties = Props;
fn create(props: Self::Properties, link: ComponentLink<Self>) -> Self {
LogoutButton {
link: link.clone(),
on_logged_out: props.on_logged_out,
}
}
fn update(&mut self, msg: Self::Message) -> ShouldRender {
match msg {
Msg::Logout => match delete_cookie("user_id") {
Err(e) => {
ConsoleService::error(&e.to_string());
false
}
Ok(()) => {
self.on_logged_out.emit(());
true
}
},
}
}
fn change(&mut self, _: Self::Properties) -> ShouldRender {
false
}
fn view(&self) -> Html {
html! {
<button onclick=self.link.callback(|_| { Msg::Logout })>{"Logout"}</button>
}
}
}