lemmy/server/src/routes/nodeinfo.rs

103 lines
2.4 KiB
Rust
Raw Normal View History

2020-01-19 11:32:02 +00:00
extern crate lazy_static;
use crate::db::site_view::SiteView;
2019-11-15 02:08:25 +00:00
use crate::version;
2019-11-15 17:10:56 +00:00
use crate::Settings;
use actix_web::body::Body;
use actix_web::web;
2019-11-21 19:27:52 +00:00
use actix_web::HttpResponse;
use diesel::r2d2::{ConnectionManager, Pool};
use diesel::PgConnection;
2020-01-19 11:32:02 +00:00
use serde::Serialize;
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));
}
2020-01-11 12:30:45 +00:00
async fn node_info_well_known() -> HttpResponse<Body> {
2020-01-19 11:32:02 +00:00
let node_info = NodeInfoWellKnown {
links: NodeInfoWellKnownLinks {
rel: "http://nodeinfo.diaspora.software/ns/schema/2.0".to_string(),
href: format!("https://{}/nodeinfo/2.0.json", Settings::get().hostname),
},
};
HttpResponse::Ok().json(node_info)
2019-11-15 02:08:25 +00:00
}
async fn node_info(
db: web::Data<Pool<ConnectionManager<PgConnection>>>,
) -> Result<HttpResponse, actix_web::Error> {
let res = web::block(move || {
let conn = db.get()?;
let site_view = match SiteView::read(&conn) {
Ok(site_view) => site_view,
Err(_) => return Err(format_err!("not_found")),
};
let protocols = if Settings::get().federation_enabled {
2020-01-19 11:32:02 +00:00
vec!["activitypub".to_string()]
} else {
vec![]
};
2020-01-19 11:32:02 +00:00
Ok(NodeInfo {
version: "2.0".to_string(),
software: NodeInfoSoftware {
name: "lemmy".to_string(),
version: version::VERSION.to_string(),
2019-11-15 17:10:56 +00:00
},
2020-01-19 11:32:02 +00:00
protocols,
usage: NodeInfoUsage {
users: NodeInfoUsers {
total: site_view.number_of_users,
},
2020-01-19 11:32:02 +00:00
local_posts: site_view.number_of_posts,
local_comments: site_view.number_of_comments,
open_registrations: site_view.open_registration,
},
})
})
.await
.map(|json| HttpResponse::Ok().json(json))
.map_err(|_| HttpResponse::InternalServerError())?;
Ok(res)
2019-11-15 02:08:25 +00:00
}
2020-01-19 11:32:02 +00:00
#[derive(Serialize)]
struct NodeInfoWellKnown {
links: NodeInfoWellKnownLinks,
}
#[derive(Serialize)]
struct NodeInfoWellKnownLinks {
rel: String,
href: String,
}
#[derive(Serialize)]
struct NodeInfo {
version: String,
software: NodeInfoSoftware,
protocols: Vec<String>,
usage: NodeInfoUsage,
}
#[derive(Serialize)]
struct NodeInfoSoftware {
name: String,
version: String,
}
#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
struct NodeInfoUsage {
users: NodeInfoUsers,
local_posts: i64,
local_comments: i64,
open_registrations: bool,
}
#[derive(Serialize)]
struct NodeInfoUsers {
total: i64,
}