feat: list submission

.changes/submission.md

@@ -0,0 +1,5 @@
+---
+"algohub-server": patch:feat
+---
+
+Support list submissions by user or contest or user and contest

src/models/submission.rs

@@ -10,16 +10,17 @@ pub enum Status {
     Ready,
 }
 
-#[derive(Serialize, Deserialize)]
+#[derive(Serialize, Deserialize, Debug)]
 pub struct Submission {
     pub id: Option<Thing>,
 
     pub lang: Language,
-    pub problem_id: String,
+    pub problem: Thing,
     pub code: String,
     pub status: Status,
     pub results: Vec<JudgeResult>,
     pub creator: Thing,
+    pub contest: Option<Thing>,
 
     pub created_at: chrono::NaiveDateTime,
     pub updated_at: chrono::NaiveDateTime,

src/routes/submission.rs

@@ -34,12 +34,17 @@ pub async fn submit(
     }
 
     let data = data.into_inner();
-    let submission = submission::create(db, &data.auth.id, id, data.code, data.lang)
-        .await
-        .map_err(|e| Error::ServerError(Json(e.to_string().into())))?
-        .ok_or(Error::ServerError(Json(
-            "Failed to submit, please try again later.".into(),
-        )))?;
+    let submission = submission::create(
+        db,
+        &data.auth.id,
+        id,
+        data.code,
+        data.lang,
+    )
+    .await?
+    .ok_or(Error::ServerError(Json(
+        "Failed to submit, please try again later.".into(),
+    )))?;
 
     Ok(Json(Response {
         success: true,
@@ -57,8 +62,7 @@ pub async fn get(
     _auth: Json<Credentials<'_>>,
 ) -> Result<Submission> {
     let submission = submission::get_by_id(db, id)
-        .await
-        .map_err(|e| Error::ServerError(Json(e.to_string().into())))?
+        .await?
         .ok_or(Error::NotFound(Json("Submission not found".into())))?;
 
     Ok(Json(Response {
@@ -68,7 +72,80 @@ pub async fn get(
     }))
 }
 
+#[post("/list/user/<id>", data = "<_auth>")]
+pub async fn list_by_user(
+    db: &State<Surreal<Client>>,
+    id: &str,
+    _auth: Json<Credentials<'_>>,
+) -> Result<Vec<Submission>> {
+    let submissions = submission::list_by_user(db, ("account", id).into()).await?;
+
+    Ok(Json(Response {
+        success: true,
+        message: "submission fetched successfully".to_string(),
+        data: Some(submissions),
+    }))
+}
+
+#[post("/list/contest/<id>", data = "<_auth>")]
+pub async fn list_by_contest(
+    db: &State<Surreal<Client>>,
+    id: &str,
+    _auth: Json<Credentials<'_>>,
+) -> Result<Vec<Submission>> {
+    let submissions = submission::list_by_contest(db, ("contest", id).into()).await?;
+
+    Ok(Json(Response {
+        success: true,
+        message: "submission fetched successfully".to_string(),
+        data: Some(submissions),
+    }))
+}
+
+#[post("/list/problem/<id>", data = "<_auth>")]
+pub async fn list_by_problem(
+    db: &State<Surreal<Client>>,
+    id: &str,
+    _auth: Json<Credentials<'_>>,
+) -> Result<Vec<Submission>> {
+    let submissions = submission::list_by_problem(db, ("problem", id).into()).await?;
+
+    Ok(Json(Response {
+        success: true,
+        message: "submission fetched successfully".to_string(),
+        data: Some(submissions),
+    }))
+}
+
+#[post("/list/contest/<contest_id>/<user_id>", data = "<_auth>")]
+pub async fn list_within_contest(
+    db: &State<Surreal<Client>>,
+    contest_id: &str,
+    user_id: &str,
+    _auth: Json<Credentials<'_>>,
+) -> Result<Vec<Submission>> {
+    let submissions = submission::list_within_contest(
+        db,
+        ("contest", contest_id).into(),
+        ("contest", user_id).into(),
+    )
+    .await?;
+
+    Ok(Json(Response {
+        success: true,
+        message: "submission fetched successfully".to_string(),
+        data: Some(submissions),
+    }))
+}
+
 pub fn routes() -> Vec<rocket::Route> {
     use rocket::routes;
-    routes![submit, get]
+    routes![
+        submit,
+        get,
+        list_by_user,
+        list_by_contest,
+        list_within_contest,
+        list_by_problem,
+    ]
 }

src/utils/submission.rs

@@ -1,12 +1,12 @@
 use crate::models::submission::Submission;
 use anyhow::Result;
 use eval_stack::compile::Language;
-use surrealdb::{engine::remote::ws::Client, Surreal};
+use surrealdb::{engine::remote::ws::Client, sql::Thing, Surreal};
 
 pub async fn create(
     db: &Surreal<Client>,
     account_id: &str,
-    problem_id: &str,
+    problem: &str,
     code: String,
     lang: Language,
 ) -> Result<Option<Submission>> {
@@ -16,12 +16,14 @@ pub async fn create(
             id: None,
             lang,
             code,
-            problem_id: problem_id.to_string(),
+            problem: ("problem", problem).into(),
             status: crate::models::submission::Status::InQueue,
 
             creator: ("account", account_id).into(),
             results: vec![],
 
+            contest: None,
+
             created_at: chrono::Local::now().naive_local(),
             updated_at: chrono::Local::now().naive_local(),
         })
@@ -31,3 +33,40 @@ pub async fn create(
 pub async fn get_by_id(db: &Surreal<Client>, id: &str) -> Result<Option<Submission>> {
     Ok(db.select(("submission", id)).await?)
 }
+
+pub async fn list_by_user(db: &Surreal<Client>, creator: Thing) -> Result<Vec<Submission>> {
+    Ok(db
+        .query("SELECT * FROM submission WHERE creator = $creator")
+        .bind(("creator", creator))
+        .await?
+        .take(0)?)
+}
+
+pub async fn list_by_contest(db: &Surreal<Client>, contest_id: Thing) -> Result<Vec<Submission>> {
+    Ok(db
+        .query("SELECT * FROM submission WHERE contest_id = $contest_id")
+        .bind(("contest_id", contest_id))
+        .await?
+        .take(0)?)
+}
+
+pub async fn list_within_contest(
+    db: &Surreal<Client>,
+    contest_id: Thing,
+    user_id: Thing,
+) -> Result<Vec<Submission>> {
+    let submissions = list_by_contest(db, contest_id).await?;
+
+    Ok(submissions
+        .into_iter()
+        .filter(|s| s.creator == user_id)
+        .collect())
+}
+
+pub async fn list_by_problem(db: &Surreal<Client>, problem: Thing) -> Result<Vec<Submission>> {
+    Ok(db
+        .query("SELECT * FROM submission WHERE problem = $problem")
+        .bind(("problem", problem))
+        .await?
+        .take(0)?)
+}