#[macro_use] extern crate diesel; #[macro_use] extern crate log; #[macro_use] extern crate tera; #[macro_use] extern crate serde_derive; extern crate chrono; extern crate actix; extern crate actix_web; extern crate env_logger; extern crate r2d2; extern crate futures; extern crate serde; pub mod schema; pub mod models; pub mod db; use actix::prelude::*; use actix_web::*; use diesel::pg::PgConnection; use diesel::r2d2::{ConnectionManager, Pool}; use std::env; use db::*; use futures::Future; use models::*; /// Represents the state carried by the web server actors. struct AppState { /// Address of the database actor db: Addr, /// Compiled templates tera: tera::Tera, } /// Really inefficient renderer example! fn render_threads(tpl: &tera::Tera, threads: Vec) -> String { let mut ctx = tera::Context::new(); ctx.add("threads", &threads); tpl.render("index.html", &ctx).expect("Oh no") } fn forum_index(req: HttpRequest) -> FutureResponse { req.state().db.send(ListThreads) .from_err() .and_then(move |res| match res { Ok(threads) => Ok(HttpResponse::Ok() .content_type("text/html") .body(render_threads(&req.state().tera, threads))), Err(err) => { error!("Error loading threads: {}", err); Ok(HttpResponse::InternalServerError().into()) } }) .responder() } fn render_thread(tpl: &tera::Tera, thread: Thread, posts: Vec) -> HttpResponse { let mut ctx = tera::Context::new(); ctx.add("thread", &thread); ctx.add("posts", &posts); let body = tpl.render("thread.html", &ctx).expect("Oh no"); HttpResponse::Ok() .content_type("text/html") .body(body) } fn forum_thread(req: HttpRequest) -> FutureResponse { let thread_id = req.match_info().query("id").unwrap(); req.state().db.send(GetThread(thread_id)) .from_err() .and_then(move |res| match res { Ok((thread, posts)) => Ok(render_thread(&req.state().tera, thread, posts)), Err(err) => { error!("Error loading thread {}: {}", thread_id, err); Ok(HttpResponse::InternalServerError().into()) } }) .responder() } fn main() { env_logger::init(); info!("Welcome to Converse! Hold on tight while we're getting ready."); let sys = actix::System::new("converse"); info!("Initialising database connection pool ..."); let db_url = env::var("DATABASE_URL") .expect("DATABASE_URL must be set"); let manager = ConnectionManager::::new(db_url); let pool = Pool::builder().build(manager).expect("Failed to initialise DB pool"); let db_addr = SyncArbiter::start(2, move || DbExecutor(pool.clone())); info!("Initialising HTTP server ..."); server::new(move || { let template_path = concat!(env!("CARGO_MANIFEST_DIR"), "/templates/**/*"); let tera = compile_templates!(template_path); App::with_state(AppState { db: db_addr.clone(), tera }) .middleware(middleware::Logger::default()) .route("/", http::Method::GET, &forum_index) .route("/thread/{id}", http::Method::GET, &forum_thread) }).bind("127.0.0.1:4567").unwrap().start(); let _ = sys.run(); }