mirror of
https://git.joinplu.me/Plume/Plume.git
synced 2025-04-20 10:44:09 +00:00
80a4dae8bd
- Use `Result` as much as possible - Display errors instead of panicking TODO (maybe in another PR? this one is already quite big): - Find a way to merge Ructe/ErrorPage types, so that we can have routes returning `Result<X, ErrorPage>` instead of panicking when we have an `Error` - Display more details about the error, to make it easier to debug (sorry, this isn't going to be fun to review, the diff is huge, but it is always the same changes)
56 lines
2.1 KiB
Rust
56 lines
2.1 KiB
Rust
use rocket::http::ContentType;
|
|
use rocket::response::Content;
|
|
use serde_json;
|
|
use webfinger::*;
|
|
|
|
use plume_models::{BASE_URL, ap_url, db_conn::DbConn, blogs::Blog, users::User};
|
|
|
|
#[get("/.well-known/nodeinfo")]
|
|
pub fn nodeinfo() -> Content<String> {
|
|
Content(ContentType::new("application", "jrd+json"), json!({
|
|
"links": [
|
|
{
|
|
"rel": "http://nodeinfo.diaspora.software/ns/schema/2.0",
|
|
"href": ap_url(&format!("{domain}/nodeinfo", domain = BASE_URL.as_str()))
|
|
}
|
|
]
|
|
}).to_string())
|
|
}
|
|
|
|
#[get("/.well-known/host-meta")]
|
|
pub fn host_meta() -> String {
|
|
format!(r#"
|
|
<?xml version="1.0"?>
|
|
<XRD xmlns="http://docs.oasis-open.org/ns/xri/xrd-1.0">
|
|
<Link rel="lrdd" type="application/xrd+xml" template="{url}"/>
|
|
</XRD>
|
|
"#, url = ap_url(&format!("{domain}/.well-known/webfinger?resource={{uri}}", domain = BASE_URL.as_str())))
|
|
}
|
|
|
|
struct WebfingerResolver;
|
|
|
|
impl Resolver<DbConn> for WebfingerResolver {
|
|
fn instance_domain<'a>() -> &'a str {
|
|
BASE_URL.as_str()
|
|
}
|
|
|
|
fn find(acct: String, conn: DbConn) -> Result<Webfinger, ResolverError> {
|
|
User::find_local(&*conn, &acct)
|
|
.and_then(|usr| usr.webfinger(&*conn))
|
|
.or_else(|_| Blog::find_local(&*conn, &acct)
|
|
.and_then(|blog| blog.webfinger(&*conn))
|
|
.or(Err(ResolverError::NotFound)))
|
|
}
|
|
}
|
|
|
|
#[get("/.well-known/webfinger?<resource>")]
|
|
pub fn webfinger(resource: String, conn: DbConn) -> Content<String> {
|
|
match WebfingerResolver::endpoint(resource, conn).and_then(|wf| serde_json::to_string(&wf).map_err(|_| ResolverError::NotFound)) {
|
|
Ok(wf) => Content(ContentType::new("application", "jrd+json"), wf),
|
|
Err(err) => Content(ContentType::new("text", "plain"), String::from(match err {
|
|
ResolverError::InvalidResource => "Invalid resource. Make sure to request an acct: URI",
|
|
ResolverError::NotFound => "Requested resource was not found",
|
|
ResolverError::WrongInstance => "This is not the instance of the requested resource"
|
|
}))
|
|
}
|
|
}
|