HomePhorge
Diffusion Process Management Controller (master) Code

pmc/src/daemon/api/fairing.rs
e855a1e2c0ecmaster

Last Change
Raw File

fairing.rs
View Options

use rocket::{
async_trait,
fairing::{Fairing, Info, Kind},
http::{ContentType, Header},
Data, Orbit, Request, Response, Rocket,
};
#[async_trait]
impl Fairing for super::Logger {
fn info(&self) -> Info {
Info {
name: "Logger Fairing",
kind: Kind::Liftoff | Kind::Request | Kind::Response,
}
}
async fn on_liftoff(&self, rocket: &Rocket<Orbit>) {
let config = rocket.config();
log!("[rocket] launched",
"tls" => config.tls_enabled(),
"keep_alive" => config.keep_alive,
"workers" => config.workers,
"profile" => config.profile.to_string(),
);
log!("[rocket] limits", "limits" => config.limits);
log!("[api] server started", "port" => config.port, "host" => config.address);
}
async fn on_request(&self, request: &mut Request<'_>, _: &mut Data<'_>) {
log!("[api] req",
"method" => request.method(),
"uri" => request.uri(),
"content_type" => request.content_type().unwrap_or(&ContentType::Plain),
);
}
async fn on_response<'r>(&self, _request: &'r Request<'_>, response: &mut Response<'r>) {
log!("[api] res",
"status" => response.status(),
"size" => response.body_mut().size().await.unwrap_or(0),
"content_type" => response.content_type().unwrap_or(ContentType::Plain),
);
}
}
#[async_trait]
impl Fairing for super::AddCORS {
fn info(&self) -> Info {
Info {
name: "Add CORS headers to responses",
kind: Kind::Response,
}
}
async fn on_response<'r>(&self, _request: &'r Request<'_>, response: &mut Response<'r>) {
response.set_header(Header::new("Access-Control-Allow-Origin", "*"));
response.set_header(Header::new("Access-Control-Allow-Credentials", "true"));
response.set_header(Header::new("Access-Control-Allow-Methods", "POST, GET, OPTIONS"));
response.set_header(Header::new("Access-Control-Allow-Headers", "token, Content-Type, Accept"));
response.set_header(Header::new("Access-Control-Expose-Headers", "Content-Encoding, Content-Type"));
}
}

Owner Packages

  • No Owners