submission logic and fetching submissions

1 files changed, 2 insertions, 1 deletions

diff --git a/src/main.rs b/src/main.rs
index 73dccef..9659af3 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -14,7 +14,7 @@ use routes::user::{create_user, me};
 use routes::auth::{login, logout};
 use tower_http::services::ServeDir;
 
-use crate::{database::Database, routes::submission::create_submission, utils::register_admin};
+use crate::{database::Database, routes::submission::{create_submission, get_display_submissions, get_submissions_by_problem_id}, utils::register_admin};
 
 #[derive(Clone)]
 struct AppState {
@@ -65,6 +65,7 @@ async fn main() {
         .route("/problem", get(get_problems))
         .route("/problem", post(create_problem))
         .route("/submission", post(create_submission))
+        .route("/submission/{problem_id}", get(get_display_submissions))
         .route("/user", post(create_user))
         .route("/me", get(me))
         .nest_service("/static", static_files)