Plume/src/main.rs

183 lines
5.7 KiB
Rust
Raw Normal View History

#![feature(custom_derive, plugin, decl_macro)]
#![plugin(rocket_codegen)]
2018-09-02 22:55:42 +02:00
2018-06-10 13:13:07 +02:00
extern crate activitypub;
2018-09-01 22:08:26 +02:00
extern crate atom_syndication;
2018-09-25 21:45:32 +02:00
extern crate canapi;
extern crate chrono;
extern crate colored;
extern crate diesel;
extern crate dotenv;
2018-05-16 20:20:44 +02:00
extern crate failure;
2018-06-15 15:08:38 +02:00
extern crate gettextrs;
2018-09-02 22:55:42 +02:00
extern crate guid_create;
2018-04-24 11:21:39 +02:00
extern crate heck;
2018-09-02 22:55:42 +02:00
extern crate multipart;
2018-09-25 21:45:32 +02:00
extern crate plume_api;
extern crate plume_common;
extern crate plume_models;
#[macro_use]
extern crate rocket;
extern crate rocket_contrib;
2018-06-24 18:58:57 +02:00
extern crate rocket_csrf;
2018-06-17 16:28:44 +02:00
extern crate rocket_i18n;
2018-06-19 17:14:52 +02:00
extern crate rpassword;
extern crate serde;
#[macro_use]
extern crate serde_derive;
2018-05-01 13:48:19 +02:00
#[macro_use]
2018-04-23 17:09:05 +02:00
extern crate serde_json;
2018-09-29 16:45:27 +02:00
extern crate serde_qs;
2018-06-29 14:22:43 +02:00
extern crate validator;
#[macro_use]
extern crate validator_derive;
2018-06-18 23:50:40 +02:00
extern crate webfinger;
2018-07-26 17:32:52 +02:00
extern crate workerpool;
2018-10-06 13:06:02 +02:00
use diesel::r2d2::ConnectionManager;
use rocket::State;
use rocket_contrib::Template;
2018-06-24 18:58:57 +02:00
use rocket_csrf::CsrfFairingBuilder;
use plume_models::{DATABASE_URL, Connection, db_conn::DbPool};
2018-07-26 17:32:52 +02:00
use workerpool::{Pool, thunk::ThunkWorker};
2018-09-19 16:49:34 +02:00
mod api;
mod inbox;
mod routes;
type Worker<'a> = State<'a, Pool<ThunkWorker<()>>>;
2018-10-06 13:06:02 +02:00
/// Initializes a database pool.
fn init_pool() -> Option<DbPool> {
2018-10-06 13:06:02 +02:00
dotenv::dotenv().ok();
let manager = ConnectionManager::<Connection>::new(DATABASE_URL.as_str());
DbPool::new(manager).ok()
2018-10-06 13:06:02 +02:00
}
fn main() {
2018-10-06 13:06:02 +02:00
let pool = init_pool();
rocket::ignite()
.mount("/", routes![
routes::blogs::paginated_details,
routes::blogs::details,
2018-06-17 17:26:15 +02:00
routes::blogs::activity_details,
routes::blogs::outbox,
routes::blogs::new,
2018-06-04 21:57:03 +02:00
routes::blogs::new_auth,
routes::blogs::create,
2018-09-01 22:08:26 +02:00
routes::blogs::atom_feed,
2018-05-10 20:01:16 +02:00
routes::comments::create,
routes::comments::activity_pub,
2018-04-24 10:35:45 +02:00
2018-04-29 19:50:46 +02:00
routes::instance::index,
2018-09-04 21:56:27 +02:00
routes::instance::paginated_local,
routes::instance::local,
2018-09-05 16:21:50 +02:00
routes::instance::paginated_feed,
routes::instance::feed,
routes::instance::paginated_federated,
routes::instance::federated,
routes::instance::admin,
routes::instance::admin_instances,
routes::instance::admin_instances_paginated,
2018-09-09 12:25:55 +02:00
routes::instance::admin_users,
routes::instance::admin_users_paginated,
routes::instance::ban,
routes::instance::toggle_block,
routes::instance::update_settings,
2018-05-13 19:39:18 +02:00
routes::instance::shared_inbox,
2018-06-10 21:33:42 +02:00
routes::instance::nodeinfo,
2018-09-01 18:39:40 +02:00
routes::instance::about,
2018-09-10 16:08:22 +02:00
routes::instance::web_manifest,
2018-04-22 20:13:12 +02:00
routes::likes::create,
2018-06-04 21:57:03 +02:00
routes::likes::create_auth,
2018-09-02 22:55:42 +02:00
routes::medias::list,
routes::medias::new,
routes::medias::upload,
routes::medias::details,
2018-09-02 23:10:15 +02:00
routes::medias::delete,
2018-09-03 14:04:17 +02:00
routes::medias::set_avatar,
2018-09-02 22:55:42 +02:00
routes::medias::static_files,
routes::notifications::paginated_notifications,
2018-05-13 15:35:55 +02:00
routes::notifications::notifications,
2018-06-04 21:57:03 +02:00
routes::notifications::notifications_auth,
2018-05-13 15:35:55 +02:00
routes::posts::details,
routes::posts::details_response,
routes::posts::activity_details,
2018-09-07 19:51:53 +02:00
routes::posts::edit,
routes::posts::update,
routes::posts::new,
routes::posts::new_auth,
routes::posts::create,
routes::posts::delete,
2018-05-19 11:51:10 +02:00
routes::reshares::create,
2018-06-04 21:57:03 +02:00
routes::reshares::create_auth,
2018-05-19 11:51:10 +02:00
routes::session::new,
routes::session::new_message,
routes::session::create,
routes::session::delete,
routes::static_files,
2018-09-06 14:06:04 +02:00
routes::tags::tag,
routes::tags::paginated_tag,
2018-04-23 11:52:44 +02:00
routes::user::me,
2018-04-22 20:13:12 +02:00
routes::user::details,
2018-06-10 19:55:08 +02:00
routes::user::dashboard,
routes::user::dashboard_auth,
2018-07-25 15:50:29 +02:00
routes::user::followers_paginated,
2018-06-17 17:26:15 +02:00
routes::user::followers,
2018-05-12 17:30:14 +02:00
routes::user::edit,
routes::user::edit_auth,
2018-05-12 17:30:14 +02:00
routes::user::update,
2018-09-09 21:49:24 +02:00
routes::user::delete,
2018-05-01 21:57:30 +02:00
routes::user::follow,
routes::user::follow_auth,
2018-04-24 14:31:02 +02:00
routes::user::activity_details,
2018-04-29 20:01:42 +02:00
routes::user::outbox,
2018-05-01 16:00:29 +02:00
routes::user::inbox,
routes::user::ap_followers,
2018-04-22 20:13:12 +02:00
routes::user::new,
routes::user::create,
2018-09-01 22:08:26 +02:00
routes::user::atom_feed,
2018-04-23 11:52:44 +02:00
routes::well_known::host_meta,
2018-06-10 21:33:42 +02:00
routes::well_known::nodeinfo,
2018-06-24 18:58:57 +02:00
routes::well_known::webfinger,
routes::errors::csrf_violation
])
2018-09-19 16:49:34 +02:00
.mount("/api/v1", routes![
2018-09-25 21:45:32 +02:00
api::posts::get,
2018-09-29 16:45:27 +02:00
api::posts::list,
2018-09-19 16:49:34 +02:00
])
2018-06-18 17:59:49 +02:00
.catch(catchers![
routes::errors::not_found,
routes::errors::server_error
])
.manage(pool)
2018-07-26 17:32:52 +02:00
.manage(Pool::<ThunkWorker<()>>::new(4))
.attach(Template::custom(|engines| {
2018-06-17 16:28:44 +02:00
rocket_i18n::tera(&mut engines.tera);
}))
2018-06-17 16:28:44 +02:00
.attach(rocket_i18n::I18n::new("plume"))
.attach(CsrfFairingBuilder::new()
.set_default_target("/csrf-violation?target=<uri>".to_owned(), rocket::http::Method::Post)
.add_exceptions(vec![
("/inbox".to_owned(), "/inbox".to_owned(), rocket::http::Method::Post),
("/@/<name>/inbox".to_owned(), "/@/<name>/inbox".to_owned(), rocket::http::Method::Post),
("/login".to_owned(), "/login".to_owned(), rocket::http::Method::Post),
("/users/new".to_owned(), "/users/new".to_owned(), rocket::http::Method::Post),
])
.finalize().unwrap())
.launch();
}