Moving settings to Database. (#2492)

[lemmy.git] / crates / routes / src / webfinger.rs

diff --git a/crates/routes/src/webfinger.rs b/crates/routes/src/webfinger.rs
index 4059e0f893d0491e774c22fa4d76e9f0c87df70b..a13aad3ebeeb20077b30a3182d3382ad1a8f8669 100644 (file)
--- a/crates/routes/src/webfinger.rs
+++ b/crates/routes/src/webfinger.rs
@@ -1,16 +1,15 @@
-use actix_web::{error::ErrorBadRequest, web::Query, *};
-use anyhow::anyhow;
-use lemmy_db_queries::source::{community::Community_, user::User};
-use lemmy_db_schema::source::{community::Community, user::User_};
-use lemmy_structs::{blocking, WebFingerLink, WebFingerResponse};
-use lemmy_utils::{
-  settings::Settings,
-  LemmyError,
-  WEBFINGER_COMMUNITY_REGEX,
-  WEBFINGER_USER_REGEX,
+use actix_web::{web, web::Query, HttpResponse};
+use anyhow::Context;
+use lemmy_api_common::utils::blocking;
+use lemmy_apub::fetcher::webfinger::{WebfingerLink, WebfingerResponse};
+use lemmy_db_schema::{
+  source::{community::Community, person::Person},
+  traits::ApubActor,
 };
+use lemmy_utils::{error::LemmyError, location_info};
 use lemmy_websocket::LemmyContext;
 use serde::Deserialize;
+use url::Url;
 
 #[derive(Deserialize)]
 struct Params {
@@ -18,12 +17,10 @@ struct Params {
 }
 
 pub fn config(cfg: &mut web::ServiceConfig) {
-  if Settings::get().federation.enabled {
-    cfg.route(
-      ".well-known/webfinger",
-      web::get().to(get_webfinger_response),
-    );
-  }
+  cfg.route(
+    ".well-known/webfinger",
+    web::get().to(get_webfinger_response),
+  );
 }
 
 /// Responds to webfinger requests of the following format. There isn't any real documentation for
@@ -35,61 +32,63 @@ pub fn config(cfg: &mut web::ServiceConfig) {
 async fn get_webfinger_response(
   info: Query<Params>,
   context: web::Data<LemmyContext>,
-) -> Result<HttpResponse, Error> {
-  let community_regex_parsed = WEBFINGER_COMMUNITY_REGEX
+) -> Result<HttpResponse, LemmyError> {
+  let name = context
+    .settings()
+    .webfinger_regex()
     .captures(&info.resource)
-    .map(|c| c.get(1))
-    .flatten();
+    .and_then(|c| c.get(1))
+    .context(location_info!())?
+    .as_str()
+    .to_string();
 
-  let user_regex_parsed = WEBFINGER_USER_REGEX
-    .captures(&info.resource)
-    .map(|c| c.get(1))
-    .flatten();
+  let name_ = name.clone();
+  let user_id: Option<Url> = blocking(context.pool(), move |conn| {
+    Person::read_from_name(conn, &name_, false)
+  })
+  .await?
+  .ok()
+  .map(|c| c.actor_id.into());
+  let community_id: Option<Url> = blocking(context.pool(), move |conn| {
+    Community::read_from_name(conn, &name, false)
+  })
+  .await?
+  .ok()
+  .map(|c| c.actor_id.into());
 
-  let url = if let Some(community_name) = community_regex_parsed {
-    let community_name = community_name.as_str().to_owned();
-    // Make sure the requested community exists.
-    blocking(context.pool(), move |conn| {
-      Community::read_from_name(conn, &community_name)
-    })
-    .await?
-    .map_err(|_| ErrorBadRequest(LemmyError::from(anyhow!("not_found"))))?
-    .actor_id
-  } else if let Some(user_name) = user_regex_parsed {
-    let user_name = user_name.as_str().to_owned();
-    // Make sure the requested user exists.
-    blocking(context.pool(), move |conn| {
-      User_::read_from_name(conn, &user_name)
-    })
-    .await?
-    .map_err(|_| ErrorBadRequest(LemmyError::from(anyhow!("not_found"))))?
-    .actor_id
-  } else {
-    return Err(ErrorBadRequest(LemmyError::from(anyhow!("not_found"))));
-  };
+  // 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.
+  let links = vec![
+    webfinger_link_for_actor(user_id),
+    webfinger_link_for_actor(community_id),
+  ]
+  .into_iter()
+  .flatten()
+  .collect();
 
-  let json = WebFingerResponse {
+  let json = WebfingerResponse {
     subject: info.resource.to_owned(),
-    aliases: vec![url.to_owned().into()],
-    links: vec![
-      WebFingerLink {
-        rel: Some("http://webfinger.net/rel/profile-page".to_string()),
-        type_: Some("text/html".to_string()),
-        href: Some(url.to_owned().into()),
-        template: None,
-      },
-      WebFingerLink {
-        rel: Some("self".to_string()),
-        type_: Some("application/activity+json".to_string()),
-        href: Some(url.into()),
-        template: None,
-      }, // TODO: this also needs to return the subscribe link once that's implemented
-         //{
-         //  "rel": "http://ostatus.org/schema/1.0/subscribe",
-         //  "template": "https://my_instance.com/authorize_interaction?uri={uri}"
-         //}
-    ],
+    links,
   };
 
   Ok(HttpResponse::Ok().json(json))
 }
+
+fn webfinger_link_for_actor(url: Option<Url>) -> Vec<WebfingerLink> {
+  if let Some(url) = url {
+    vec![
+      WebfingerLink {
+        rel: Some("http://webfinger.net/rel/profile-page".to_string()),
+        kind: Some("text/html".to_string()),
+        href: Some(url.to_owned()),
+      },
+      WebfingerLink {
+        rel: Some("self".to_string()),
+        kind: Some("application/activity+json".to_string()),
+        href: Some(url),
+      },
+    ]
+  } else {
+    vec![]
+  }
+}