This repository has been archived on 2023-07-04. You can view files and clone it, but cannot push or open issues/pull-requests.
blog/src/main.rs

105 lines
2.9 KiB
Rust
Raw Normal View History

2021-11-23 09:32:08 +01:00
#[macro_use]
extern crate rocket;
#[macro_use]
extern crate diesel_migrations;
use figment::{
providers::{Env, Format, Yaml},
Figment,
};
use rb::auth::JwtConf;
2021-11-23 09:32:08 +01:00
use rocket::{
fairing::AdHoc,
http::Status,
serde::json::{json, Value},
Build, Request, Rocket,
};
use rocket_sync_db_pools::database;
use serde::{Deserialize, Serialize};
2021-11-24 17:51:52 +01:00
pub mod v1;
2021-11-23 09:32:08 +01:00
2021-11-24 18:52:02 +01:00
/// This function just returns a pre-built rocket header containing a valid JWT for the secret
/// "secret", valid until 2034 or something. It's easy for testing the API.
#[cfg(test)]
pub fn auth_header() -> rocket::http::Header<'static>
{
return rocket::http::Header::new("Authorization", "Bearer eyJhbGciOiJIUzI1NiJ9.eyJpZCI6IjVjMjM2OTI0NjY4ZDQzZWFiNGNmNDczYjk1YWZiNzgzIiwidXNlcm5hbWUiOiJKb2huIERvZSIsImFkbWluIjp0cnVlLCJleHAiOjE1MTYyMzkwMjIwfQ.if939L9le8LP-dtXnQs-mHPkb-VieRAvAfSu20755jY");
}
2021-11-28 21:45:29 +01:00
/// Used by Rocket to store database connections.
2021-11-23 09:32:08 +01:00
#[database("postgres_rb")]
pub struct RbDbConn(diesel::PgConnection);
2021-11-28 21:45:29 +01:00
/// Handles all error status codes.
2021-11-23 09:32:08 +01:00
#[catch(default)]
fn default_catcher(status: Status, _: &Request) -> Value
{
2021-11-23 20:21:29 +01:00
json!({"status": status.code, "message": ""})
2021-11-23 09:32:08 +01:00
}
2021-11-28 21:45:29 +01:00
/// Rocket fairing that executes the necessary migrations in our database.
2021-11-23 09:32:08 +01:00
async fn run_db_migrations(rocket: Rocket<Build>) -> Result<Rocket<Build>, Rocket<Build>>
{
2021-11-28 21:45:29 +01:00
embed_migrations!();
2021-11-23 09:32:08 +01:00
let conn = RbDbConn::get_one(&rocket)
.await
.expect("database connection");
conn.run(|c| match embedded_migrations::run(c) {
Ok(()) => Ok(rocket),
Err(_) => Err(rocket),
})
.await
}
2021-11-28 21:45:29 +01:00
/// Struct to deserialize from the config file. It contains any custom configuration our
/// application might need besides the default Rocket variables.
2021-11-23 09:32:08 +01:00
#[derive(Debug, Deserialize, Serialize)]
pub struct RbConfig
{
2021-11-23 17:49:31 +01:00
jwt: JwtConf,
2021-11-23 09:32:08 +01:00
}
2021-11-28 21:45:29 +01:00
/// The main entrypoint of our program. It launches the Rocket instance.
2021-11-23 09:32:08 +01:00
#[launch]
fn rocket() -> _
{
let figment = Figment::from(rocket::config::Config::default())
.merge(Yaml::file("Rb.yaml").nested())
.merge(Env::prefixed("RB_").global());
2021-11-23 18:00:10 +01:00
let rocket = rocket::custom(figment)
2021-11-23 09:32:08 +01:00
.attach(RbDbConn::fairing())
.attach(AdHoc::try_on_ignite(
"Run database migrations",
run_db_migrations,
))
.register("/", catchers![default_catcher])
2021-11-23 20:21:29 +01:00
.mount(
2021-11-24 17:51:52 +01:00
"/v1/sections",
2021-11-23 20:21:29 +01:00
routes![
2021-11-24 17:51:52 +01:00
v1::sections::get,
v1::sections::create,
v1::sections::find,
v1::sections::patch,
v1::sections::delete
2021-11-23 20:21:29 +01:00
],
)
.mount(
2021-11-24 17:51:52 +01:00
"/v1/posts",
2021-11-23 20:21:29 +01:00
routes![
2021-11-24 17:51:52 +01:00
v1::posts::get,
v1::posts::create,
v1::posts::find,
v1::posts::patch,
v1::posts::delete
2021-11-23 20:21:29 +01:00
],
);
2021-11-23 18:00:10 +01:00
let new_figment = rocket.figment();
let jwt_conf: JwtConf = new_figment.extract_inner("jwt").expect("jwt config");
rocket.manage(jwt_conf)
2021-11-23 09:32:08 +01:00
}