aboutsummaryrefslogtreecommitdiff
path: root/server/src/ui/error.rs
blob: 05249afdf2c87e5652c9df471d0755d1814ec683 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
/*
    This file is part of jellything (https://codeberg.org/metamuffin/jellything)
    which is licensed under the GNU Affero General Public License (version 3); see /COPYING.
    Copyright (C) 2025 metamuffin <metamuffin.org>
*/
use crate::CONF;
use log::info;
use rocket::{
    catch,
    http::{ContentType, Status},
    response::{self, content::RawHtml, Responder},
    Request,
};
use serde_json::{json, Value};
use std::{fmt::Display, fs::File, io::Read, sync::LazyLock};

static ERROR_IMAGE: LazyLock<Vec<u8>> = LazyLock::new(|| {
    info!("loading error image");
    let mut f = File::open(CONF.asset_path.join("error.avif"))
        .expect("please create error.avif in the asset dir");
    let mut o = Vec::new();
    f.read_to_end(&mut o).unwrap();
    o
});

#[catch(default)]
pub fn r_catch<'a>(status: Status, _request: &Request) -> RawHtml<String> {
    // LayoutPage {
    //     title: "Not found".to_string(),
    //     content: markup::new! {
    //         h2 { "Error" }
    //         p { @format!("{status}") }
    //         @if status == Status::NotFound {
    //             p { "You might need to " a[href=uri!(r_account_login())] { "log in" } ", to see this page" }
    //         }
    //     },
    //     ..Default::default()
    // }
    RawHtml("as".to_string())
}

#[catch(default)]
pub fn r_api_catch(status: Status, _request: &Request) -> Value {
    json!({ "error": format!("{status}") })
}

pub type MyResult<T> = Result<T, MyError>;

// TODO an actual error enum would be useful for status codes

pub struct MyError(pub anyhow::Error);

impl<'r> Responder<'r, 'static> for MyError {
    fn respond_to(self, req: &'r Request<'_>) -> response::Result<'static> {
        match req.accept().map(|a| a.preferred()) {
            Some(x) if x.is_json() => json!({ "error": format!("{}", self.0) }).respond_to(req),
            Some(x) if x.is_avif() || x.is_png() || x.is_jpeg() => {
                (ContentType::AVIF, ERROR_IMAGE.as_slice()).respond_to(req)
            }
            _ => r_catch(Status::InternalServerError, req).respond_to(req),
        }
    }
}

impl std::fmt::Debug for MyError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_fmt(format_args!("{:?}", self.0))
    }
}

impl Display for MyError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        self.0.fmt(f)
    }
}
impl From<anyhow::Error> for MyError {
    fn from(err: anyhow::Error) -> MyError {
        MyError(err)
    }
}
impl From<std::fmt::Error> for MyError {
    fn from(err: std::fmt::Error) -> MyError {
        MyError(anyhow::anyhow!("{err}"))
    }
}
impl From<std::io::Error> for MyError {
    fn from(err: std::io::Error) -> Self {
        MyError(anyhow::anyhow!("{err}"))
    }
}
impl From<serde_json::Error> for MyError {
    fn from(err: serde_json::Error) -> Self {
        MyError(anyhow::anyhow!("{err}"))
    }
}