lemmy/crates/routes/src/webfinger.rs

96 lines
2.7 KiB
Rust
Raw Normal View History

2023-04-05 20:29:08 +00:00
use activitypub_federation::{
config::Data,
fetch::webfinger::{extract_webfinger_name, Webfinger, WebfingerLink},
};
use actix_web::{web, web::Query, HttpResponse};
use lemmy_api_common::context::LemmyContext;
use lemmy_db_schema::{
source::{community::Community, person::Person},
traits::ApubActor,
};
2023-04-05 20:29:08 +00:00
use lemmy_utils::error::LemmyError;
use serde::Deserialize;
use std::collections::HashMap;
2021-11-16 17:03:09 +00:00
use url::Url;
2019-12-18 00:59:47 +00:00
#[derive(Deserialize)]
struct Params {
2019-12-18 00:59:47 +00:00
resource: String,
}
pub fn config(cfg: &mut web::ServiceConfig) {
cfg.route(
".well-known/webfinger",
web::get().to(get_webfinger_response),
);
}
2019-12-18 00:59:47 +00:00
/// Responds to webfinger requests of the following format. There isn't any real documentation for
/// this, but it described in this blog post:
/// https://mastodon.social/.well-known/webfinger?resource=acct:gargron@mastodon.social
///
/// You can also view the webfinger response that Mastodon sends:
/// https://radical.town/.well-known/webfinger?resource=acct:felix@radical.town
async fn get_webfinger_response(
info: Query<Params>,
2023-04-05 20:29:08 +00:00
context: Data<LemmyContext>,
) -> Result<HttpResponse, LemmyError> {
2023-04-05 20:29:08 +00:00
let name = extract_webfinger_name(&info.resource, &context)?;
2019-12-18 00:59:47 +00:00
2021-11-16 17:03:09 +00:00
let name_ = name.clone();
2022-11-09 10:05:00 +00:00
let user_id: Option<Url> = Person::read_from_name(context.pool(), &name_, false)
.await
.ok()
.map(|c| c.actor_id.into());
let community_id: Option<Url> = Community::read_from_name(context.pool(), &name, false)
.await
.ok()
.map(|c| c.actor_id.into());
// Mastodon seems to prioritize the last webfinger item in case of duplicates. Put
// community last so that it gets prioritized. For Lemmy the order doesnt matter.
2021-11-16 17:03:09 +00:00
let links = vec![
webfinger_link_for_actor(user_id, "Person"),
webfinger_link_for_actor(community_id, "Group"),
2021-11-16 17:03:09 +00:00
]
.into_iter()
.flatten()
.collect();
2019-12-18 00:59:47 +00:00
let json = Webfinger {
subject: info.resource.clone(),
2021-11-16 17:03:09 +00:00
links,
..Default::default()
2021-11-16 17:03:09 +00:00
};
Ok(HttpResponse::Ok().json(json))
}
fn webfinger_link_for_actor(url: Option<Url>, kind: &str) -> Vec<WebfingerLink> {
2021-11-16 17:03:09 +00:00
if let Some(url) = url {
let mut properties = HashMap::new();
properties.insert(
"https://www.w3.org/ns/activitystreams#type"
.parse()
.expect("parse url"),
kind.to_string(),
);
2021-11-16 17:03:09 +00:00
vec![
WebfingerLink {
rel: Some("http://webfinger.net/rel/profile-page".to_string()),
kind: Some("text/html".to_string()),
href: Some(url.clone()),
..Default::default()
},
WebfingerLink {
rel: Some("self".to_string()),
kind: Some("application/activity+json".to_string()),
2021-11-16 17:03:09 +00:00
href: Some(url),
properties,
2021-11-16 17:03:09 +00:00
},
]
} else {
vec![]
}
2019-12-18 00:59:47 +00:00
}