81 lines
2.7 KiB
Rust
81 lines
2.7 KiB
Rust
use rocket::http::ContentType;
|
|
use rocket::response::Content;
|
|
use webfinger::*;
|
|
|
|
use plume_models::{ap_url, blogs::Blog, db_conn::DbConn, users::User, CONFIG};
|
|
|
|
#[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 = CONFIG.base_url.as_str()))
|
|
},
|
|
{
|
|
"rel": "http://nodeinfo.diaspora.software/ns/schema/2.1",
|
|
"href": ap_url(&format!("{domain}/nodeinfo/2.1", domain = CONFIG.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 = CONFIG.base_url.as_str()
|
|
))
|
|
)
|
|
}
|
|
|
|
struct WebfingerResolver;
|
|
|
|
impl Resolver<DbConn> for WebfingerResolver {
|
|
fn instance_domain<'a>() -> &'a str {
|
|
CONFIG.base_url.as_str()
|
|
}
|
|
|
|
fn find(prefix: Prefix, acct: String, conn: DbConn) -> Result<Webfinger, ResolverError> {
|
|
match prefix {
|
|
Prefix::Acct => User::find_by_fqn(&conn, &acct)
|
|
.and_then(|usr| usr.webfinger(&*conn))
|
|
.or(Err(ResolverError::NotFound)),
|
|
Prefix::Group => Blog::find_by_fqn(&conn, &acct)
|
|
.and_then(|blog| blog.webfinger(&*conn))
|
|
.or(Err(ResolverError::NotFound)),
|
|
Prefix::Custom(_) => 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::WrongDomain => "This is not the instance of the requested resource",
|
|
}),
|
|
),
|
|
}
|
|
}
|