zyl/zyllian.github.io
1
0
Fork 0
mirror of https://github.com/zyllian/zyllian.github.io.git synced 2025-01-18 03:32:30 -08:00
Code Issues Projects Releases Packages Wiki Activity

Improve 404 page

Browse source
This commit is contained in:
Zoey 2021-08-19 12:48:38 -07:00
parent de4c97c74b
commit 821c94c609
1 changed files with 8 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/serving.rs
View file

@ -10,6 +10,7 @@ use std::{
use futures::SinkExt;
use hotwatch::{Event, Hotwatch};
use warp::{
hyper::StatusCode,
path::FullPath,
reply::Response,
ws::{Message, WebSocket},
@ -242,9 +243,13 @@ impl Site {
async move {
// Handle missing files
println!("404 - {}", path.as_str());
Ok::<_, std::convert::Infallible>(warp::reply::html(
std::fs::read_to_string(build_path.join("404.html")).unwrap(),
))
let body = match std::fs::read_to_string(build_path.join("404.html")) {
Ok(body) => body,
_ => "404 Not Found".to_string(),
};
let mut res = Response::new(body.into());
*res.status_mut() = StatusCode::NOT_FOUND;
Ok::<_, std::convert::Infallible>(res)
}
}));