Skip to content

Commit

Permalink
💚 Fix build with no default features
Browse files Browse the repository at this point in the history
  • Loading branch information
RemiBardon committed Aug 25, 2024
1 parent 1e49ad6 commit 3a019c5
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 8 deletions.
16 changes: 9 additions & 7 deletions src/orangutan-server/src/main.rs
Original file line number Diff line number Diff line change
Expand Up @@ -17,17 +17,15 @@ use orangutan_helpers::{
website_id::WebsiteId,
};
use request_guards::{handle_refresh_token, REVOKED_TOKENS};
use tera::Tera;
use tokio::runtime::Handle;
use tower::Service;
use tower_http::{
services::{fs::ServeFileSystemResponseBody, ServeFile},
trace::TraceLayer,
};
#[cfg(feature = "templating")]
use tracing::debug;
use tracing::{info, warn};
use tracing::{debug, info, warn};
use tracing_subscriber::EnvFilter;
#[cfg(feature = "token-generator")]
use util::WebsiteRoot;

#[cfg(feature = "templating")]
Expand All @@ -36,13 +34,15 @@ use crate::{config::NOT_FOUND_FILE, routes::update_content_routes, util::error};

#[derive(Clone)]
struct AppState {
#[cfg(feature = "token-generator")]
website_root: WebsiteRoot,
#[cfg(feature = "templating")]
tera: Tera,
tera: tera::Tera,
}

#[tokio::main]
async fn main() -> ExitCode {
#[cfg(feature = "token-generator")]
let website_root = match WebsiteRoot::try_from_env() {
Ok(r) => r,
Err(err) => {
Expand All @@ -51,7 +51,8 @@ async fn main() -> ExitCode {
},
};

let mut app_state = AppState {
let app_state = AppState {
#[cfg(feature = "token-generator")]
website_root,
#[cfg(feature = "templating")]
tera: Default::default(),
Expand All @@ -64,6 +65,8 @@ async fn main() -> ExitCode {

// Add support for templating if needed
#[cfg(feature = "templating")]
let mut app_state = app_state;
#[cfg(feature = "templating")]
{
info!("Initializing templating engine…");
if let Err(err) = app_state.tera.add_raw_templates(routes::templates()) {
Expand Down Expand Up @@ -163,7 +166,6 @@ enum Error {
#[cfg(feature = "templating")]
#[error("Templating error: {0}")]
TemplatingError(#[from] templating::Error),
#[cfg(feature = "templating")]
#[error("Internal server error: {0}")]
InternalServerError(String),
#[error("Client error: {0}")]
Expand Down
6 changes: 5 additions & 1 deletion src/orangutan-server/src/routes/debug_routes.rs
Original file line number Diff line number Diff line change
Expand Up @@ -23,19 +23,23 @@ lazy_static! {
}

pub(super) fn router() -> Router<AppState> {
let mut router = Router::<AppState>::new()
let router = Router::<AppState>::new()
.route("/clear-cookies", get(clear_cookies).put(clear_cookies))
.route("/_info", get(get_user_info))
.route("/_errors", get(errors))
.route("/_access-logs", get(access_logs))
.route("/_revoked-tokens", get(revoked_tokens));

#[cfg(feature = "token-generator")]
let mut router = router;
#[cfg(feature = "token-generator")]
{
router = router.route(
"/_generate-token",
get(token_generator::token_generation_form).post(token_generator::generate_token),
);
}

router
}

Expand Down

0 comments on commit 3a019c5

Please sign in to comment.