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); }