lemmy/src/routes/nodeinfo.rs

100 lines
2.5 KiB
Rust
Raw Normal View History

2020-05-16 14:04:08 +00:00
use actix_web::{body::Body, error::ErrorBadRequest, *};
use anyhow::anyhow;
use lemmy_api::version;
use lemmy_db_views::site_view::SiteView;
2020-09-16 13:31:30 +00:00
use lemmy_structs::blocking;
use lemmy_utils::{settings::Settings, LemmyError};
use lemmy_websocket::LemmyContext;
2020-05-16 14:04:08 +00:00
use serde::{Deserialize, Serialize};
use url::Url;
2019-11-15 02:08:25 +00:00
pub fn config(cfg: &mut web::ServiceConfig) {
cfg
.route("/nodeinfo/2.0.json", web::get().to(node_info))
.route("/.well-known/nodeinfo", web::get().to(node_info_well_known));
}
async fn node_info_well_known() -> Result<HttpResponse<Body>, LemmyError> {
2020-01-19 11:32:02 +00:00
let node_info = NodeInfoWellKnown {
links: NodeInfoWellKnownLinks {
2020-04-08 12:37:05 +00:00
rel: Url::parse("http://nodeinfo.diaspora.software/ns/schema/2.0")?,
href: Url::parse(&format!(
"{}/nodeinfo/2.0.json",
Settings::get().get_protocol_and_hostname()
2020-04-08 12:37:05 +00:00
))?,
2020-01-19 11:32:02 +00:00
},
};
2020-04-08 12:37:05 +00:00
Ok(HttpResponse::Ok().json(node_info))
2019-11-15 02:08:25 +00:00
}
async fn node_info(context: web::Data<LemmyContext>) -> Result<HttpResponse, Error> {
let site_view = blocking(context.pool(), SiteView::read)
.await?
.map_err(|_| ErrorBadRequest(LemmyError::from(anyhow!("not_found"))))?;
let protocols = if Settings::get().federation.enabled {
vec!["activitypub".to_string()]
} else {
vec![]
};
let json = NodeInfo {
version: "2.0".to_string(),
software: NodeInfoSoftware {
name: "lemmy".to_string(),
version: version::VERSION.to_string(),
},
protocols,
usage: NodeInfoUsage {
users: NodeInfoUsers {
total: site_view.counts.users,
},
local_posts: site_view.counts.posts,
local_comments: site_view.counts.comments,
},
open_registrations: site_view.site.open_registration,
};
Ok(HttpResponse::Ok().json(json))
2019-11-15 02:08:25 +00:00
}
2020-01-19 11:32:02 +00:00
#[derive(Serialize, Deserialize, Debug)]
struct NodeInfoWellKnown {
pub links: NodeInfoWellKnownLinks,
2020-01-19 11:32:02 +00:00
}
#[derive(Serialize, Deserialize, Debug)]
struct NodeInfoWellKnownLinks {
2020-04-08 12:37:05 +00:00
pub rel: Url,
pub href: Url,
2020-01-19 11:32:02 +00:00
}
#[derive(Serialize, Deserialize, Debug)]
#[serde(rename_all = "camelCase")]
struct NodeInfo {
pub version: String,
pub software: NodeInfoSoftware,
pub protocols: Vec<String>,
pub usage: NodeInfoUsage,
pub open_registrations: bool,
2020-01-19 11:32:02 +00:00
}
#[derive(Serialize, Deserialize, Debug)]
struct NodeInfoSoftware {
pub name: String,
pub version: String,
2020-01-19 11:32:02 +00:00
}
#[derive(Serialize, Deserialize, Debug)]
2020-01-19 11:32:02 +00:00
#[serde(rename_all = "camelCase")]
struct NodeInfoUsage {
pub users: NodeInfoUsers,
pub local_posts: i64,
pub local_comments: i64,
2020-01-19 11:32:02 +00:00
}
#[derive(Serialize, Deserialize, Debug)]
struct NodeInfoUsers {
pub total: i64,
2020-01-19 11:32:02 +00:00
}