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
/*
* This file is part of laurelin/api
* Copyright (C) 2023 Jonni Liljamo <jonni@liljamo.com>
*
* Licensed under GPL-3.0-only.
* See LICENSE for licensing information.
*/
use actix_session::{
config::{PersistentSession, SessionLifecycle, TtlExtensionPolicy},
storage::RedisActorSessionStore,
Session, SessionMiddleware,
};
use actix_web::{
cookie::{time::Duration, Key},
get,
middleware::Logger,
web, App, HttpResponse, HttpServer, Responder,
};
use diesel::{
self,
r2d2::{self, ConnectionManager},
PgConnection,
};
use diesel_migrations::{embed_migrations, EmbeddedMigrations, MigrationHarness};
type PgPool = r2d2::Pool<ConnectionManager<PgConnection>>;
const MIGRATIONS: EmbeddedMigrations = embed_migrations!();
#[get("/api/ping")]
async fn ping() -> impl Responder {
HttpResponse::Ok().body("pong")
}
#[get("/api/ping_sec")]
async fn ping_sec(session: Session) -> impl Responder {
let session_validation = session::validate_session(&session);
match session_validation {
Err(err) => err,
Ok(user_id) => {
// NOTE: this is where one would spawn an action to do... something.
// the user id can be used to check if the user can, e.g. start a game.
return HttpResponse::Ok().body(format!("pong_sec for user_id: '{}'", user_id));
}
}
}
fn run_migrations(conn: &mut PgConnection) {
conn.run_pending_migrations(MIGRATIONS).unwrap();
}
mod schema;
mod actions;
mod handlers;
mod models;
mod session;
#[actix_web::main]
async fn main() -> std::io::Result<()> {
pretty_env_logger::init();
// timestamp status_code latency remote_ip request_details
let log_format = "%t | %s | %Dms | %a | \"%r\"";
// setup db pool
let db_url = std::env::var("LAURELIN_DB_URL").expect("LAURELIN_DB_URL");
let manager = ConnectionManager::<PgConnection>::new(db_url);
let pg_pool = r2d2::Pool::builder()
.build(manager)
.expect("failed to create pool");
// run migrations
log::info!("running migrations");
run_migrations(&mut pg_pool.get().unwrap());
let redis_url = std::env::var("LAURELIN_REDIS_URL").expect("LAURELIN_REDIS_URL");
let cookie_key = std::env::var("LAURELIN_COOKIE_KEY").expect("LAURELIN_COOKIE_KEY");
HttpServer::new(move || {
App::new()
.app_data(web::Data::new(pg_pool.clone()))
.wrap(Logger::new(log_format))
.wrap(
SessionMiddleware::builder(
RedisActorSessionStore::new(&redis_url),
Key::from(cookie_key.as_bytes()),
)
.session_lifecycle(SessionLifecycle::PersistentSession(
PersistentSession::default()
.session_ttl(Duration::days(30))
.session_ttl_extension_policy(TtlExtensionPolicy::OnStateChanges),
))
.build(),
)
.service(ping)
.service(ping_sec)
.service(handlers::info)
.service(handlers::user::create)
.service(handlers::user::login)
.service(handlers::user::logout)
})
.bind(("0.0.0.0", 8080))?
.run()
.await
}