147 lines
3.8 KiB
Rust
147 lines
3.8 KiB
Rust
extern crate rand;
|
|
extern crate uuid;
|
|
extern crate bcrypt;
|
|
extern crate chrono;
|
|
|
|
extern crate dotenv;
|
|
extern crate postgres;
|
|
extern crate r2d2;
|
|
extern crate r2d2_postgres;
|
|
extern crate fallible_iterator;
|
|
|
|
extern crate serde;
|
|
extern crate serde_cbor;
|
|
#[macro_use] extern crate serde_derive;
|
|
#[macro_use] extern crate failure;
|
|
|
|
extern crate fern;
|
|
extern crate log_panics;
|
|
#[macro_use] extern crate log;
|
|
|
|
extern crate stripe;
|
|
|
|
extern crate iron;
|
|
extern crate bodyparser;
|
|
extern crate urlencoded;
|
|
extern crate persistent;
|
|
extern crate router;
|
|
extern crate mount;
|
|
extern crate cookie;
|
|
|
|
extern crate lettre;
|
|
extern crate lettre_email;
|
|
|
|
extern crate ws;
|
|
extern crate crossbeam_channel;
|
|
|
|
extern crate mnml_core;
|
|
|
|
mod account;
|
|
// mod acp;
|
|
mod img;
|
|
mod mail;
|
|
mod mtx;
|
|
mod names;
|
|
mod http;
|
|
mod payments;
|
|
mod pg;
|
|
mod events;
|
|
pub mod rpc;
|
|
mod warden;
|
|
mod user_authenticated;
|
|
mod user_anonymous;
|
|
|
|
use std::thread::{spawn};
|
|
use std::path::{Path};
|
|
use fern::colors::{Color, ColoredLevelConfig};
|
|
use crossbeam_channel::{unbounded};
|
|
|
|
#[cfg(unix)]
|
|
pub fn setup_logger() -> Result<(), fern::InitError> {
|
|
log_panics::init();
|
|
|
|
let formatter = syslog::Formatter3164 {
|
|
facility: syslog::Facility::LOG_USER,
|
|
hostname: None,
|
|
process: "mnml".to_owned(),
|
|
pid: 0,
|
|
};
|
|
|
|
let syslog = fern::Dispatch::new()
|
|
.level(log::LevelFilter::Info)
|
|
.chain(syslog::unix(formatter).unwrap());
|
|
|
|
let colors_line = ColoredLevelConfig::new()
|
|
.error(Color::Red)
|
|
.warn(Color::Yellow)
|
|
.info(Color::BrightWhite)
|
|
.debug(Color::BrightWhite)
|
|
.trace(Color::BrightBlack);
|
|
|
|
let colors_level = colors_line.clone().info(Color::Green);
|
|
let term = fern::Dispatch::new()
|
|
.format(move |out, message, record| {
|
|
out.finish(format_args!(
|
|
"{color_line}{date} {target} {level}{color_line} {message}\x1B[0m",
|
|
color_line = format_args!("\x1B[{}m", colors_line.get_color(&record.level()).to_fg_str()),
|
|
date = chrono::Local::now().format("%Y-%m-%d %H:%M:%S"),
|
|
target = record.target(),
|
|
level = colors_level.color(record.level()),
|
|
message = message,
|
|
));
|
|
})
|
|
.chain(fern::log_file("/var/log/mnml/mnml.log")?)
|
|
.chain(std::io::stdout());
|
|
|
|
fern::Dispatch::new()
|
|
.level_for("postgres", log::LevelFilter::Info)
|
|
.level_for("ws", log::LevelFilter::Warn)
|
|
.level_for("iron", log::LevelFilter::Info)
|
|
.level(log::LevelFilter::Info)
|
|
.chain(term)
|
|
.chain(syslog)
|
|
.apply()?;
|
|
|
|
Ok(())
|
|
}
|
|
|
|
pub fn start() {
|
|
#[cfg(unix)]
|
|
setup_logger().unwrap();
|
|
dotenv::from_path(Path::new("/etc/mnml/gs.conf")).ok();
|
|
|
|
let pool = pg::create_pool();
|
|
let http_pool = pool.clone();
|
|
|
|
let (events_tx, events_rx) = unbounded();
|
|
let pg_events_tx = events_tx.clone();
|
|
let rpc_events_tx = events_tx.clone();
|
|
|
|
let (warden_tx, warden_rx) = unbounded();
|
|
let events_warden_tx = warden_tx.clone();
|
|
let warden_tick_tx = warden_tx.clone();
|
|
|
|
let (mail_tx, mail_rx) = unbounded();
|
|
let _http_mail_tx = mail_tx.clone();
|
|
|
|
// create a clone of the tx so ws handler can tell events
|
|
// about connection status
|
|
let events = events::Events::new(events_tx, events_rx, events_warden_tx);
|
|
let warden = warden::Warden::new(warden_tx, warden_rx, events.tx.clone(), pool.clone());
|
|
|
|
let pg_pool = pool.clone();
|
|
let mailer = mail::listen(mail_rx);
|
|
|
|
let stripe = payments::stripe_client();
|
|
|
|
spawn(move || http::start(http_pool, mailer));
|
|
spawn(move || warden.listen());
|
|
spawn(move || warden::upkeep_tick(warden_tick_tx));
|
|
spawn(move || pg::listen(pg_pool, pg_events_tx));
|
|
spawn(move || events.listen());
|
|
|
|
// the main thread becomes this ws listener
|
|
let rpc_pool = pool.clone();
|
|
rpc::start(rpc_pool, rpc_events_tx, stripe);
|
|
}
|