algohub_server/utils/
problem.rs

use anyhow::Result;
use serde::Deserialize;
use surrealdb::{engine::remote::ws::Client, sql::Thing, Surreal};

use crate::models::problem::{CreateProblem, Problem, ServerTestCase};

pub async fn create(db: &Surreal<Client>, problem: CreateProblem<'_>) -> Result<Option<Problem>> {
    Ok(db
        .create("problem")
        .content(Into::<Problem>::into(problem))
        .await?)
}

pub async fn update(db: &Surreal<Client>, problem: Problem) -> Result<Option<Problem>> {
    Ok(db
        .update((
            "problem",
            problem.id.clone().expect("empty id").id.to_string(),
        ))
        .content(problem)
        .await?)
}

const DELETE_ALL_ASSETS_QUERY: &str = r#"
IF $problem.exists() {
    FOR $asset IN (SELECT VALUE test_cases FROM $problem) {
        DELETE $asset.input, $asset.output;
    };
    (DELETE ONLY $problem RETURN BEFORE)
}
"#;
pub async fn delete(db: &Surreal<Client>, id: &str) -> Result<Option<Problem>> {
    Ok(db
        .query(DELETE_ALL_ASSETS_QUERY)
        .bind(("problem", Thing::from(("problem", id))))
        .await?
        .take(0)?)
}

pub async fn get<M>(db: &Surreal<Client>, id: &str) -> Result<Option<M>>
where
    for<'de> M: Deserialize<'de>,
{
    Ok(db.select(("problem", id)).await?)
}

const LIST_QUERY: &str = r#"
IF $authed THEN
    IF $limit THEN
        SELECT * FROM problem WHERE owner = type::thing("account", $id) LIMIT $limit
    ELSE
        SELECT * FROM problem WHERE owner = type::thing("account", $id)
    END;
ELSE
    IF $limit THEN
        SELECT * FROM problem WHERE owner = type::thing("account", $id) AND private = false LIMIT $limit
    ELSE
        SELECT * FROM problem WHERE owner = type::thing("account", $id) AND private = false
    END;
END;"#;

pub async fn list_for_account<M>(
    db: &Surreal<Client>,
    account_id: Option<String>,
    authed_id: Option<String>,
    limit: Option<u32>,
) -> Result<Vec<M>>
where
    for<'de> M: Deserialize<'de>,
{
    let mut response = db
        .query(LIST_QUERY)
        .bind((
            "authed",
            authed_id.is_some() && account_id.is_some() && authed_id == account_id,
        ))
        .bind(("id", account_id))
        .bind(("limit", limit))
        .await?;

    Ok(response.take(0)?)
}

const SELECT_TEST_CASES_QUERY: &str = r#"
IF $problem.exists() THEN
    SELECT input.path AS input, output.path AS output FROM $problem.test_cases
ELSE
    []
END;
"#;
pub async fn get_test_cases_by_id(db: &Surreal<Client>, id: &str) -> Result<Vec<ServerTestCase>> {
    Ok(db
        .query(SELECT_TEST_CASES_QUERY)
        .bind(("problem", Thing::from(("problem", id))))
        .await?
        .take(0)?)
}
algohub_server/utils/problem.rs

algohub_server/utils/
problem.rs
