1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
#[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<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(req: HttpRequest<AppState>) -> FutureResponse<HttpResponse> {
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<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(req: HttpRequest<AppState>) -> FutureResponse<HttpResponse> {
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::<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())
.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();
}
|
