forked from jmug/stoatchat
70 lines
1.6 KiB
Rust
70 lines
1.6 KiB
Rust
#![feature(proc_macro_hygiene, decl_macro)]
|
|
#![feature(async_closure)]
|
|
|
|
#[macro_use]
|
|
extern crate rocket;
|
|
#[macro_use]
|
|
extern crate rocket_contrib;
|
|
#[macro_use]
|
|
extern crate lazy_static;
|
|
#[macro_use]
|
|
extern crate impl_ops;
|
|
#[macro_use]
|
|
extern crate bitfield;
|
|
extern crate ctrlc;
|
|
|
|
pub mod database;
|
|
pub mod notifications;
|
|
pub mod routes;
|
|
pub mod util;
|
|
|
|
use futures::join;
|
|
use log::info;
|
|
use rauth::{self, options::Options};
|
|
use rocket_cors::AllowedOrigins;
|
|
|
|
#[async_std::main]
|
|
async fn main() {
|
|
dotenv::dotenv().ok();
|
|
env_logger::init_from_env(env_logger::Env::default().filter_or("RUST_LOG", "info"));
|
|
|
|
info!("Starting REVOLT server.");
|
|
|
|
util::variables::preflight_checks();
|
|
database::connect().await;
|
|
notifications::hive::init_hive().await;
|
|
|
|
ctrlc::set_handler(move || {
|
|
// Force ungraceful exit to avoid hang.
|
|
std::process::exit(0);
|
|
})
|
|
.expect("Error setting Ctrl-C handler");
|
|
|
|
join!(
|
|
launch_web(),
|
|
notifications::websocket::launch_server(),
|
|
notifications::hive::listen(),
|
|
);
|
|
}
|
|
|
|
async fn launch_web() {
|
|
let cors = rocket_cors::CorsOptions {
|
|
allowed_origins: AllowedOrigins::All,
|
|
..Default::default()
|
|
}
|
|
.to_cors()
|
|
.expect("Failed to create CORS.");
|
|
|
|
let auth = rauth::auth::Auth::new(database::get_collection("accounts"), Options::new());
|
|
|
|
routes::mount(rocket::ignite())
|
|
.mount("/", rocket_cors::catch_all_options_routes())
|
|
.mount("/auth", rauth::routes::routes())
|
|
.manage(auth)
|
|
.manage(cors.clone())
|
|
.attach(cors)
|
|
.launch()
|
|
.await
|
|
.unwrap();
|
|
}
|