From cd6eba8b2b70404e93d9d4b0832faa7c9f00d217 Mon Sep 17 00:00:00 2001 From: Harald Eilertsen Date: Mon, 4 Mar 2019 17:33:20 +0100 Subject: Refactoring, move db connection and room controller to separate modules. --- src/controllers.rs | 19 +++++++++++++++++++ src/controllers/room.rs | 29 +++++++++++++++++++++++++++++ src/db.rs | 26 ++++++++++++++++++++++++++ src/lib.rs | 27 +++++++-------------------- 4 files changed, 81 insertions(+), 20 deletions(-) create mode 100644 src/controllers.rs create mode 100644 src/controllers/room.rs create mode 100644 src/db.rs diff --git a/src/controllers.rs b/src/controllers.rs new file mode 100644 index 0000000..d8f9d5d --- /dev/null +++ b/src/controllers.rs @@ -0,0 +1,19 @@ +/* + Social program for Ramaskrik. + Copyright (C) 2019 Harald Eilertsen + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see . +*/ + +pub mod room; diff --git a/src/controllers/room.rs b/src/controllers/room.rs new file mode 100644 index 0000000..27f6a41 --- /dev/null +++ b/src/controllers/room.rs @@ -0,0 +1,29 @@ +/* + Social program for Ramaskrik. + Copyright (C) 2019 Harald Eilertsen + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see . +*/ + +use crate::db; +use rocket::get; +use rocket_contrib::{ + json, + json::JsonValue, +}; + +#[get("/")] +pub fn get_rooms(_db: db::Connection) -> JsonValue { + json!([]) +} diff --git a/src/db.rs b/src/db.rs new file mode 100644 index 0000000..f427e14 --- /dev/null +++ b/src/db.rs @@ -0,0 +1,26 @@ +/* + Social program for Ramaskrik. + Copyright (C) 2019 Harald Eilertsen + + This program is free software: you can redistribute it and/or modify + it under the terms of the GNU Affero General Public License as published by + the Free Software Foundation, either version 3 of the License, or + (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU Affero General Public License for more details. + + You should have received a copy of the GNU Affero General Public License + along with this program. If not, see . +*/ + +use rocket_contrib::{ + database, + databases::diesel, +}; + +#[database("main")] +pub struct Connection(diesel::PgConnection); + diff --git a/src/lib.rs b/src/lib.rs index e6bb0f6..7929ab2 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -17,28 +17,15 @@ */ #![feature(proc_macro_hygiene, decl_macro)] -use rocket::{ - get, - routes, -}; -use rocket_contrib::{ - database, - databases::diesel, - json, - json::JsonValue, -}; +mod controllers; +mod db; +use crate::controllers::room; + +use rocket::routes; use std::collections::HashMap; use std::error::Error; use std::result::Result; -#[database("main")] -struct DbConn(diesel::PgConnection); - -#[get("/")] -fn get_rooms(_db: DbConn) -> JsonValue { - json!([]) -} - pub fn build_rocket() -> Result> { let db_url = dotenv::var("DATABASE_URL") .map_err(|_| "No database! Set DATABASE_URL env var and try again.")?; @@ -53,7 +40,7 @@ pub fn build_rocket() -> Result> { config.extras.insert("databases".into(), rocket::config::Value::from(databases)); Ok(rocket::custom(config) - .attach(DbConn::fairing()) + .attach(db::Connection::fairing()) .mount("/", rocket_contrib::serve::StaticFiles::from("./public")) - .mount("/rooms", routes![get_rooms])) + .mount("/rooms", routes![room::get_rooms])) } -- cgit v1.2.3