1 files changed, 9 insertions, 1 deletions
diff --git a/src/main.rs b/src/main.rs
index afa9821..b446402 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -14,11 +14,12 @@ use routes::user::{create_user, me};
 use routes::auth::{login, logout};
 use tower_http::services::ServeDir;
 
-use crate::{database::Database, routes::submission::{create_submission, get_display_submission, get_display_submissions, get_submissions_by_problem_id}, utils::register_admin};
+use crate::{database::Database, routes::submission::{create_submission, get_display_submission, get_display_submissions}, utils::register_admin};
 
 #[derive(Clone)]
 struct AppState {
     secret: String,
+    register_code: String,
     database: Database,
 }
 
@@ -33,6 +34,12 @@ async fn main() {
         eprintln!("missing environment variable JWT_SECRET");
         return;
     };
+
+    let Ok(register_code) = env::var("REGISTER_CODE") else {
+        eprintln!("missing environment variable REGISTER_CODE");
+        return;
+    };
+    
     let Ok(admin_email) = env::var("ADMIN_EMAIL") else {
         eprintln!("missing environment variable ADMIN_EMAIL");
         return;
@@ -53,6 +60,7 @@ async fn main() {
     
     let state = AppState {
         secret: secret,
+        register_code: register_code,
         database: database,
     };