forked from Plume/Plume
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
61 lines
2.3 KiB
Rust
61 lines
2.3 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/2.0", domain = BASE_URL.as_str()))
|
|
},
|
|
{
|
|
"rel": "http://nodeinfo.diaspora.software/ns/schema/2.1",
|
|
"href": ap_url(&format!("{domain}/nodeinfo/2.1", 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_by_fqn(&*conn, &acct)
|
|
.and_then(|usr| usr.webfinger(&*conn))
|
|
.or_else(|_| Blog::find_by_fqn(&*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"
|
|
}))
|
|
}
|
|
}
|