🚧 WIP Migrate from Rocket to Axum

RemiBardon · Aug 24, 2024 · ff260f9 · ff260f9
1 parent 2660d40
commit ff260f9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/src/helpers/src/config.rs b/src/helpers/src/config.rs
@@ -6,6 +6,7 @@ pub const THEME_NAME: &str = "Orangutan";
 pub const DATA_FILE_EXTENSION: &str = "orangutan";
 pub const DEFAULT_PROFILE: &str = "_default";
 pub const ROOT_KEY_NAME: &'static str = "_biscuit_root";
+pub const COOKIE_KEY_ENV_VAR_NAME: &'static str = "COOKIE_ENCRYPTION_KEY";
 
 pub(super) const WEBSITE_DIR_NAME: &'static str = "website";
 

diff --git a/src/orangutan-server/src/main.rs b/src/orangutan-server/src/main.rs
@@ -15,6 +15,7 @@ use axum::{
 };
 use axum_extra::extract::cookie::Key;
 use orangutan_helpers::{
+    config::COOKIE_KEY_ENV_VAR_NAME,
     generate::{self, *},
     website_id::WebsiteId,
 };
@@ -63,7 +64,13 @@ async fn main() -> ExitCode {
     let mut app_state = AppState {
         website_root,
         // FIXME: Use predefined key.
-        cookie_key: Key::generate(),
+        cookie_key: Key::from(
+            std::env::var(COOKIE_KEY_ENV_VAR_NAME)
+                .expect(&format!(
+                    "Environment variable '{COOKIE_KEY_ENV_VAR_NAME}' not defined."
+                ))
+                .as_bytes(),
+        ),
         #[cfg(feature = "templating")]
         tera: Default::default(),
     };