2020-01-19 11:32:02 +00:00
|
|
|
extern crate lazy_static;
|
2020-03-18 15:08:08 +00:00
|
|
|
use crate::apub::get_apub_protocol_string;
|
2019-12-11 20:21:47 +00:00
|
|
|
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;
|
2019-12-31 12:55:33 +00:00
|
|
|
use actix_web::web;
|
2019-11-21 19:27:52 +00:00
|
|
|
use actix_web::HttpResponse;
|
2020-01-12 15:31:51 +00:00
|
|
|
use diesel::r2d2::{ConnectionManager, Pool};
|
|
|
|
use diesel::PgConnection;
|
2020-03-18 15:08:08 +00:00
|
|
|
use serde::{Deserialize, Serialize};
|
|
|
|
use std::fmt::Debug;
|
2019-11-15 02:08:25 +00:00
|
|
|
|
2019-12-31 12:55:33 +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(),
|
2020-03-18 15:08:08 +00:00
|
|
|
href: format!(
|
|
|
|
"{}://{}/nodeinfo/2.0.json",
|
|
|
|
get_apub_protocol_string(),
|
|
|
|
Settings::get().hostname
|
|
|
|
),
|
2020-01-19 11:32:02 +00:00
|
|
|
},
|
|
|
|
};
|
|
|
|
HttpResponse::Ok().json(node_info)
|
2019-11-15 02:08:25 +00:00
|
|
|
}
|
|
|
|
|
2020-01-12 15:31:51 +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")),
|
|
|
|
};
|
2020-03-18 21:09:00 +00:00
|
|
|
let protocols = if Settings::get().federation.enabled {
|
2020-01-19 11:32:02 +00:00
|
|
|
vec!["activitypub".to_string()]
|
2020-01-12 15:31:51 +00:00
|
|
|
} 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-12 15:31:51 +00:00
|
|
|
},
|
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,
|
|
|
|
},
|
|
|
|
})
|
2020-01-12 15:31:51 +00:00
|
|
|
})
|
|
|
|
.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
|
|
|
|
2020-03-18 15:08:08 +00:00
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct NodeInfoWellKnown {
|
|
|
|
pub links: NodeInfoWellKnownLinks,
|
2020-01-19 11:32:02 +00:00
|
|
|
}
|
|
|
|
|
2020-03-18 15:08:08 +00:00
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct NodeInfoWellKnownLinks {
|
|
|
|
pub rel: String,
|
|
|
|
pub href: String,
|
2020-01-19 11:32:02 +00:00
|
|
|
}
|
|
|
|
|
2020-03-18 15:08:08 +00:00
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct NodeInfo {
|
|
|
|
pub version: String,
|
|
|
|
pub software: NodeInfoSoftware,
|
|
|
|
pub protocols: Vec<String>,
|
|
|
|
pub usage: NodeInfoUsage,
|
2020-01-19 11:32:02 +00:00
|
|
|
}
|
|
|
|
|
2020-03-18 15:08:08 +00:00
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct NodeInfoSoftware {
|
|
|
|
pub name: String,
|
|
|
|
pub version: String,
|
2020-01-19 11:32:02 +00:00
|
|
|
}
|
|
|
|
|
2020-03-18 15:08:08 +00:00
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
2020-01-19 11:32:02 +00:00
|
|
|
#[serde(rename_all = "camelCase")]
|
2020-03-18 15:08:08 +00:00
|
|
|
pub struct NodeInfoUsage {
|
|
|
|
pub users: NodeInfoUsers,
|
|
|
|
pub local_posts: i64,
|
|
|
|
pub local_comments: i64,
|
|
|
|
pub open_registrations: bool,
|
2020-01-19 11:32:02 +00:00
|
|
|
}
|
|
|
|
|
2020-03-18 15:08:08 +00:00
|
|
|
#[derive(Serialize, Deserialize, Debug)]
|
|
|
|
pub struct NodeInfoUsers {
|
|
|
|
pub total: i64,
|
2020-01-19 11:32:02 +00:00
|
|
|
}
|