src/routes/nodeinfo.rs

   1 use actix_web::{body::Body, error::ErrorBadRequest, *};
   2 use anyhow::anyhow;
   3 use lemmy_api::version;
   4 use lemmy_db::site_view::SiteView;
   5 use lemmy_structs::blocking;
   6 use lemmy_utils::{apub::get_apub_protocol_string, settings::Settings, LemmyError};
   7 use lemmy_websocket::LemmyContext;
   8 use serde::{Deserialize, Serialize};
   9 use url::Url;
  10
  11 pub fn config(cfg: &mut web::ServiceConfig) {
  12   cfg
  13     .route("/nodeinfo/2.0.json", web::get().to(node_info))
  14     .route("/.well-known/nodeinfo", web::get().to(node_info_well_known));
  15 }
  16
  17 async fn node_info_well_known() -> Result<HttpResponse<Body>, LemmyError> {
  18   let node_info = NodeInfoWellKnown {
  19     links: NodeInfoWellKnownLinks {
  20       rel: Url::parse("http://nodeinfo.diaspora.software/ns/schema/2.0")?,
  21       href: Url::parse(&format!(
  22         "{}://{}/nodeinfo/2.0.json",
  23         get_apub_protocol_string(),
  24         Settings::get().hostname
  25       ))?,
  26     },
  27   };
  28   Ok(HttpResponse::Ok().json(node_info))
  29 }
  30
  31 async fn node_info(context: web::Data<LemmyContext>) -> Result<HttpResponse, Error> {
  32   let site_view = blocking(context.pool(), SiteView::read)
  33     .await?
  34     .map_err(|_| ErrorBadRequest(LemmyError::from(anyhow!("not_found"))))?;
  35
  36   let protocols = if Settings::get().federation.enabled {
  37     vec!["activitypub".to_string()]
  38   } else {
  39     vec![]
  40   };
  41
  42   let json = NodeInfo {
  43     version: "2.0".to_string(),
  44     software: NodeInfoSoftware {
  45       name: "lemmy".to_string(),
  46       version: version::VERSION.to_string(),
  47     },
  48     protocols,
  49     usage: NodeInfoUsage {
  50       users: NodeInfoUsers {
  51         total: site_view.number_of_users,
  52       },
  53       local_posts: site_view.number_of_posts,
  54       local_comments: site_view.number_of_comments,
  55       open_registrations: site_view.open_registration,
  56     },
  57   };
  58
  59   Ok(HttpResponse::Ok().json(json))
  60 }
  61
  62 #[derive(Serialize, Deserialize, Debug)]
  63 pub struct NodeInfoWellKnown {
  64   pub links: NodeInfoWellKnownLinks,
  65 }
  66
  67 #[derive(Serialize, Deserialize, Debug)]
  68 pub struct NodeInfoWellKnownLinks {
  69   pub rel: Url,
  70   pub href: Url,
  71 }
  72
  73 #[derive(Serialize, Deserialize, Debug)]
  74 pub struct NodeInfo {
  75   pub version: String,
  76   pub software: NodeInfoSoftware,
  77   pub protocols: Vec<String>,
  78   pub usage: NodeInfoUsage,
  79 }
  80
  81 #[derive(Serialize, Deserialize, Debug)]
  82 pub struct NodeInfoSoftware {
  83   pub name: String,
  84   pub version: String,
  85 }
  86
  87 #[derive(Serialize, Deserialize, Debug)]
  88 #[serde(rename_all = "camelCase")]
  89 pub struct NodeInfoUsage {
  90   pub users: NodeInfoUsers,
  91   pub local_posts: i64,
  92   pub local_comments: i64,
  93   pub open_registrations: bool,
  94 }
  95
  96 #[derive(Serialize, Deserialize, Debug)]
  97 pub struct NodeInfoUsers {
  98   pub total: i64,
  99 }