path: root/src/main.rs

#[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 actix_web::http::Method;

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<Syn, DbExecutor>,

    /// Compiled templates
    tera: tera::Tera,
}

/// Really inefficient renderer example!
fn render_threads(tpl: &tera::Tera, threads: Vec<Thread>) -> String {
    let mut ctx = tera::Context::new();
    ctx.add("threads", &threads);
    tpl.render("index.html", &ctx).expect("Oh no")
}

fn forum_index(state: State<AppState>) -> FutureResponse<HttpResponse> {
    state.db.send(ListThreads)
        .from_err()
        .and_then(move |res| match res {
            Ok(threads) => Ok(HttpResponse::Ok()
                              .content_type("text/html")
                              .body(render_threads(&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<Post>) -> 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(state: State<AppState>, thread_id: Path<i32>)
                -> FutureResponse<HttpResponse> {
    let id = thread_id.into_inner();
    state.db.send(GetThread(id))
        .from_err()
        .and_then(move |res| match res {
            Ok((thread, posts)) => Ok(render_thread(&state.tera, thread, posts)),
            Err(err) => {
                error!("Error loading 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::<PgConnection>::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())
            .resource("/", |r| r.method(Method::GET).with(forum_index))
            .resource("/thread/{id}", |r| r.method(Method::GET).with2(forum_thread))
    }).bind("127.0.0.1:4567").unwrap().start();

    let _ = sys.run();
}