Plume/src/main.rs

131 lines
3.5 KiB
Rust
Raw Normal View History

#![feature(plugin, custom_derive, iterator_find_map)]
#![plugin(rocket_codegen)]
2018-06-10 13:13:07 +02:00
extern crate activitypub;
2018-05-16 20:20:44 +02:00
extern crate activitystreams_types;
extern crate array_tool;
2018-04-29 17:40:10 +02:00
extern crate base64;
2018-04-24 11:21:39 +02:00
extern crate bcrypt;
2018-04-29 17:40:10 +02:00
extern crate chrono;
2018-05-24 12:42:45 +02:00
extern crate comrak;
2018-05-16 20:20:44 +02:00
extern crate failure;
#[macro_use]
extern crate failure_derive;
2018-04-24 11:21:39 +02:00
extern crate heck;
2018-04-29 17:40:10 +02:00
extern crate hex;
#[macro_use]
2018-05-04 17:18:00 +02:00
extern crate hyper;
#[macro_use]
extern crate diesel;
extern crate dotenv;
#[macro_use]
extern crate lazy_static;
2018-04-29 17:40:10 +02:00
extern crate openssl;
extern crate reqwest;
extern crate rocket;
extern crate rocket_contrib;
extern crate serde;
#[macro_use]
2018-05-01 13:48:19 +02:00
extern crate serde_derive;
#[macro_use]
2018-04-23 17:09:05 +02:00
extern crate serde_json;
2018-05-01 20:02:29 +02:00
extern crate url;
2018-05-19 09:39:59 +02:00
use diesel::{pg::PgConnection, r2d2::{ConnectionManager, Pool}};
use dotenv::dotenv;
use rocket_contrib::Template;
2018-04-24 11:21:39 +02:00
use std::env;
2018-04-23 13:57:14 +02:00
mod activity_pub;
mod db_conn;
mod models;
mod schema;
mod routes;
2018-04-23 12:54:37 +02:00
mod utils;
lazy_static! {
pub static ref BASE_URL: String = env::var("BASE_URL")
.unwrap_or(format!("127.0.0.1:{}", env::var("ROCKET_PORT").unwrap_or(String::from("8000"))));
pub static ref DB_URL: String = env::var("DB_URL")
.unwrap_or(format!("postgres://plume:plume@localhost/{}", env::var("DB_NAME").unwrap_or(String::from("plume"))));
}
type PgPool = Pool<ConnectionManager<PgConnection>>;
/// Initializes a database pool.
fn init_pool() -> PgPool {
dotenv().ok();
let manager = ConnectionManager::<PgConnection>::new(DB_URL.as_str());
Pool::new(manager).expect("DB pool error")
}
fn main() {
rocket::ignite()
.mount("/", routes![
routes::blogs::details,
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-05-10 20:01:16 +02:00
routes::comments::new,
2018-06-04 21:57:03 +02:00
routes::comments::new_auth,
routes::comments::create,
2018-04-24 10:35:45 +02:00
2018-04-29 19:50:46 +02:00
routes::instance::index,
routes::instance::configure,
2018-04-22 20:13:12 +02:00
routes::instance::post_config,
2018-05-13 19:39:18 +02:00
routes::instance::shared_inbox,
2018-06-10 21:33:42 +02:00
routes::instance::nodeinfo,
2018-04-22 20:13:12 +02:00
routes::likes::create,
2018-06-04 21:57:03 +02:00
routes::likes::create_auth,
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::activity_details,
routes::posts::new,
routes::posts::new_auth,
routes::posts::create,
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-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,
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-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-04-23 11:52:44 +02:00
routes::well_known::host_meta,
2018-06-10 21:33:42 +02:00
routes::well_known::nodeinfo,
routes::well_known::webfinger
])
.manage(init_pool())
.attach(Template::fairing())
.launch();
}