iqps_backend/routing/

handlers.rs

//! All endpoint handlers and their response types.
//!
//! All endpoints accept JSON or URL query parameters as the request. The response of each handler is a [`BackendResponse`] serialized as JSON and the return type of the handler function determines the schema of the data sent in the response (if successful)
//!
//! The request format is described

use axum::{
    body::Bytes,
    extract::{Json, Multipart},
    http::StatusCode,
    Extension,
};
use color_eyre::eyre::{ContextCompat, Result};
use http::HeaderMap;
use serde::Serialize;
use tokio::fs;

use std::collections::HashMap;

use axum::extract::{Query, State};
use serde::Deserialize;

use crate::{
    auth::{self, Auth},
    pathutils::PaperCategory,
    qp::{self, AdminDashboardQP, Exam, WithUrl},
    slack::send_slack_message,
};

use super::{AppError, BackendResponse, RouterState, Status};

/// The return type of a handler function. T is the data type returned if the operation was a success
type HandlerReturn<T> = Result<(StatusCode, BackendResponse<T>), AppError>;

/// Healthcheck route. Returns a `Hello World.` message if healthy.
pub async fn healthcheck() -> HandlerReturn<()> {
    Ok(BackendResponse::ok("Hello, World.".into(), ()))
}

/// Fetches all the unapproved papers.
pub async fn get_unapproved(
    State(state): State<RouterState>,
) -> HandlerReturn<Vec<AdminDashboardQP>> {
    let papers: Vec<AdminDashboardQP> = state.db.get_unapproved_papers().await?;

    let papers = papers
        .iter()
        .map(|paper| paper.clone().with_url(&state.env_vars))
        .collect::<Result<Vec<qp::AdminDashboardQP>, color_eyre::eyre::Error>>()?;

    Ok(BackendResponse::ok(
        format!("Successfully fetched {} papers.", papers.len()),
        papers,
    ))
}

/// Gets all papers which have been soft-deleted.
pub async fn get_trash(State(state): State<RouterState>) -> HandlerReturn<Vec<AdminDashboardQP>> {
    let papers: Vec<AdminDashboardQP> = state.db.get_soft_deleted_papers().await?;

    let papers = papers
        .iter()
        .map(|paper| paper.clone().with_url(&state.env_vars))
        .collect::<Result<Vec<qp::AdminDashboardQP>, color_eyre::eyre::Error>>()?;

    Ok(BackendResponse::ok(
        format!("Successfully fetched {} papers.", papers.len()),
        papers,
    ))
}

/// Fetches a paper by id.
pub async fn get_paper_details(
    State(state): State<RouterState>,
    Query(params): Query<HashMap<String, String>>,
) -> HandlerReturn<AdminDashboardQP> {
    if let Some(id) = params.get("id") {
        if let Ok(id) = id.parse::<i32>() {
            let paper = state.db.get_paper_by_id(id).await?;
            let paper_with_url = paper.with_url(&state.env_vars)?;
            Ok(BackendResponse::ok(
                "Successfully fetched the paper.".into(),
                paper_with_url,
            ))
        } else {
            Ok(BackendResponse::error(
                "Invalid `id` URL parameter.".into(),
                StatusCode::BAD_REQUEST,
            ))
        }
    } else {
        Ok(BackendResponse::error(
            "`id` URL parameter is required.".into(),
            StatusCode::BAD_REQUEST,
        ))
    }
}

/// Searches for question papers given a query and an optional `exam` parameter.
///
/// # Request Query Parameters
/// * `query`: The query string to search in the question papers (searches course name or code)
/// * `exam` (optional): A comma-separated string of exam types to filter. Leave empty to match any exam.
pub async fn search(
    State(state): State<RouterState>,
    Query(params): Query<HashMap<String, String>>,
) -> HandlerReturn<Vec<qp::BaseQP>> {
    let response = if let Some(query) = params.get("query") {
        let exam_query_str = params
            .get("exam")
            .map(|value| value.to_owned())
            .unwrap_or("".into());

        if let Ok(exam_filter) = exam_query_str
            .split(',')
            .filter(|val| !val.trim().is_empty())
            .map(|val| Exam::try_from(&val.to_owned()))
            .collect::<Result<Vec<Exam>, _>>()
        {
            let papers = state.db.search_papers(query, exam_filter).await?;

            let papers = papers
                .iter()
                .map(|paper| paper.clone().with_url(&state.env_vars))
                .collect::<Result<Vec<qp::BaseQP>, color_eyre::eyre::Error>>()?;

            Ok(BackendResponse::ok(
                format!("Successfully fetched {} papers.", papers.len()),
                papers,
            ))
        } else {
            Ok(BackendResponse::error(
                "Invalid `exam` URL parameter.".into(),
                StatusCode::BAD_REQUEST,
            ))
        }
    } else {
        Ok(BackendResponse::error(
            "`query` URL parameter is required.".into(),
            StatusCode::BAD_REQUEST,
        ))
    };

    response
}

#[derive(Deserialize)]
/// The request format for the OAuth endpoint
pub struct OAuthReq {
    code: String,
}

#[derive(Serialize)]
/// The response format for the OAuth endpoint
pub struct OAuthRes {
    token: String,
}

/// Takes a Github OAuth code and returns a JWT auth token to log in a user if authorized
///
/// Request format - [`OAuthReq`]
pub async fn oauth(
    State(state): State<RouterState>,
    Json(body): Json<OAuthReq>,
) -> HandlerReturn<OAuthRes> {
    if let Some(token) = auth::authenticate_user(&body.code, &state.env_vars).await? {
        Ok(BackendResponse::ok(
            "Successfully authorized the user.".into(),
            OAuthRes { token },
        ))
    } else {
        Ok(BackendResponse::error(
            "Error: User unauthorized.".into(),
            StatusCode::UNAUTHORIZED,
        ))
    }
}

#[derive(Serialize)]
/// The response format for the user profile endpoint
pub struct ProfileRes {
    token: String,
    username: String,
}

/// Returns a user's profile (the JWT and username) if authorized and the token is valid. Can be used to check if the user is logged in.
pub async fn profile(Extension(auth): Extension<Auth>) -> HandlerReturn<ProfileRes> {
    Ok(BackendResponse::ok(
        "Successfully authorized the user.".into(),
        ProfileRes {
            token: auth.jwt,
            username: auth.username,
        },
    ))
}

#[derive(Deserialize)]
/// The request format for the paper edit endpoint
pub struct EditReq {
    pub id: i32,
    pub course_code: Option<String>,
    pub course_name: Option<String>,
    pub year: Option<i32>,
    pub semester: Option<String>,
    pub exam: Option<String>,
    pub note: Option<String>,
    pub approve_status: Option<bool>,
    pub replace: Vec<i32>,
}

/// Paper edit endpoint (for admin dashboard)
/// Takes a JSON request body. The `id` field is required.
/// Other optional fields can be set to change that particular value in the paper.
///
/// Request format - [`EditReq`]
pub async fn edit(
    Extension(auth): Extension<Auth>,
    State(state): State<RouterState>,
    Json(body): Json<EditReq>,
) -> HandlerReturn<AdminDashboardQP> {
    // Edit the database entry
    let (tx, old_filelink, new_qp) = state
        .db
        .edit_paper(body, &auth.username, &state.env_vars)
        .await?;

    // Copy the actual file
    let old_filepath = state.env_vars.paths.get_path_from_slug(&old_filelink);
    let new_filepath = state.env_vars.paths.get_path_from_slug(&new_qp.qp.filelink);

    if old_filepath != new_filepath {
        if let Err(e) = fs::copy(old_filepath, new_filepath).await {
            tracing::error!("Error copying file: {}", e);

            tx.rollback().await?;
            Ok(BackendResponse::error(
                "Error copying question paper file.".into(),
                StatusCode::INTERNAL_SERVER_ERROR,
            ))
        } else {
            // Commit the transaction
            tx.commit().await?;

            Ok(BackendResponse::ok(
                "Successfully updated paper details.".into(),
                new_qp.with_url(&state.env_vars)?,
            ))
        }
    } else {
        // Commit the transaction
        tx.commit().await?;
        Ok(BackendResponse::ok(
            "Successfully updated paper details.".into(),
            new_qp.with_url(&state.env_vars)?,
        ))
    }
}

#[derive(Deserialize)]
/// The details for an uploaded question paper file
pub struct FileDetails {
    pub course_code: String,
    pub course_name: String,
    pub year: i32,
    pub exam: String,
    pub semester: String,
    pub filename: String,
    pub note: String,
}

/// 10 MiB file size limit
const FILE_SIZE_LIMIT: usize = 10 << 20;
#[derive(Serialize)]
/// The status of an uploaded question paper file
pub struct UploadStatus {
    /// The filename
    filename: String,
    /// Whether the file was successfully uploaded
    status: Status,
    /// A message describing the status
    message: String,
}

/// Uploads question papers to the server
///
/// Request format - Multipart form with a `file_details` field of the format [`FileDetails`]
pub async fn upload(
    State(state): State<RouterState>,
    mut multipart: Multipart,
) -> HandlerReturn<Vec<UploadStatus>> {
    let mut files = Vec::<(HeaderMap, Bytes)>::new();
    let mut file_details: String = "".into();

    while let Some(field) = multipart.next_field().await.unwrap() {
        let name = field.name().unwrap().to_string();

        if name == "files" {
            files.push((field.headers().clone(), field.bytes().await?));
        } else if name == "file_details" {
            if file_details.is_empty() {
                file_details = field.text().await?;
            } else {
                return Ok(BackendResponse::error(
                    "Error: Multiple `file_details` fields found.".into(),
                    StatusCode::BAD_REQUEST,
                ));
            }
        }
    }

    let files = files;
    let file_details: Vec<FileDetails> = serde_json::from_str(&file_details)?;

    if files.len() > state.env_vars.max_upload_limit {
        return Ok(BackendResponse::error(
            format!(
                "Only upto {} files can be uploaded. Found {}.",
                state.env_vars.max_upload_limit,
                files.len()
            ),
            StatusCode::BAD_REQUEST,
        ));
    }

    if files.len() != file_details.len() {
        return Ok(BackendResponse::error(
            "Error: Number of files and file details array length do not match.".into(),
            StatusCode::BAD_REQUEST,
        ));
    }

    let files_iter = files.iter().zip(file_details.iter());
    let mut upload_statuses = Vec::<UploadStatus>::new();

    for ((file_headers, file_data), details) in files_iter {
        let filename = details.filename.to_owned();

        if file_data.len() > FILE_SIZE_LIMIT {
            upload_statuses.push(UploadStatus {
                filename,
                status: Status::Error,
                message: format!(
                    "File size too big. Only files upto {} MiB are allowed.",
                    FILE_SIZE_LIMIT >> 20
                ),
            });
            continue;
        }

        if let Some(content_type) = file_headers.get("content-type") {
            if content_type != "application/pdf" {
                upload_statuses.push(UploadStatus {
                    filename: filename.to_owned(),
                    status: Status::Error,
                    message: "Only PDFs are supported.".into(),
                });
                continue;
            }
        } else {
            upload_statuses.push(UploadStatus {
                filename,
                status: Status::Error,
                message: "`content-type` header not found. File type could not be determined."
                    .into(),
            });
            continue;
        }

        // Insert the db entry
        let (mut tx, id) = state.db.insert_new_uploaded_qp(details).await?;

        // Create the new filelink (slug)
        let filelink_slug = state
            .env_vars
            .paths
            .get_slug(&format!("{}.pdf", id), PaperCategory::Unapproved);

        // Update the filelink in the db
        if state
            .db
            .update_filelink(&mut tx, id, &filelink_slug)
            .await
            .is_ok()
        {
            let filepath = state.env_vars.paths.get_path_from_slug(&filelink_slug);

            // Write the file data
            if fs::write(&filepath, file_data).await.is_ok() {
                if tx.commit().await.is_ok() {
                    upload_statuses.push(UploadStatus {
                        filename,
                        status: Status::Success,
                        message: "Succesfully uploaded file.".into(),
                    });
                    continue;
                } else {
                    // Transaction commit failed, delete the file
                    fs::remove_file(filepath).await?;
                    upload_statuses.push(UploadStatus {
                        filename,
                        status: Status::Success,
                        message: "Succesfully uploaded file.".into(),
                    });
                    continue;
                }
            } else {
                tx.rollback().await?;
            }

            // If the write fails, rollback the transaction, else commit it.
        } else {
            tx.rollback().await?;

            upload_statuses.push(UploadStatus {
                filename,
                status: Status::Error,
                message: "Error updating the filelink".into(),
            });
            continue;
        }

        upload_statuses.push(UploadStatus {
            filename,
            status: Status::Error,
            message: "THIS SHOULD NEVER HAPPEN. REPORT IMMEDIATELY. ALSO THIS WOULDN'T HAPPEN IF RUST HAD STABLE ASYNC CLOSURES.".into(),
        });
    }

    let total_count = state.db.get_unapproved_papers_count().await?;
    let count = upload_statuses.len();
    let message = format!(
        "🔔 {} uploaded to IQPS!\n\n<https://qp.metakgp.org/admin|Review> | Total Unapproved papers: *{}*",
        if count == 1 {
            "A new paper was".into()
        } else {
            format!("{} new papers were", count)
        },
        total_count
    );

    let _ = send_slack_message(&state.env_vars.slack_webhook_url, &message).await;

    Ok(BackendResponse::ok(
        format!("Successfully processed {} files", upload_statuses.len()),
        upload_statuses,
    ))
}

#[derive(Deserialize)]
/// The request format for the delete endpoint
pub struct DeleteReq {
    id: i32,
}

/// (Soft) Deletes a given paper.
///
/// Request format - [`DeleteReq`]
pub async fn delete(
    State(state): State<RouterState>,
    Json(body): Json<DeleteReq>,
) -> HandlerReturn<()> {
    let paper_deleted = state.db.soft_delete(body.id).await?;

    if paper_deleted {
        Ok(BackendResponse::ok(
            "Succesfully deleted the paper.".into(),
            (),
        ))
    } else {
        Ok(BackendResponse::error(
            "No paper was changed. Either the paper does not exist, or is already deleted.".into(),
            StatusCode::BAD_REQUEST,
        ))
    }
}

#[derive(Deserialize)]
/// The request format for the hard delete endpoint
pub struct HardDeleteReq {
    ids: Vec<i32>,
}

#[derive(Serialize)]
/// The status of a paper to be deleted
pub struct DeleteStatus {
    id: i32,
    status: Status,
    message: String,
}

/// Hard deletes papers from a list of ids.
///
/// Request format - [`HardDeleteReq`]
pub async fn hard_delete(
    State(state): State<RouterState>,
    Json(body): Json<HardDeleteReq>,
) -> HandlerReturn<Vec<DeleteStatus>> {
    let mut delete_statuses = Vec::<DeleteStatus>::new();
    let mut deleted_count = 0;
    for id in body.ids {
        if let Ok(paper) = state.db.get_paper_by_id(id).await {
            let tx = state.db.hard_delete(id).await?;
            let filepath = state.env_vars.paths.get_path_from_slug(&paper.qp.filelink);
            if fs::remove_file(&filepath).await.is_ok() {
                if tx.commit().await.is_ok() {
                    delete_statuses.push(DeleteStatus {
                        id,
                        status: Status::Success,
                        message: "Successfully hard deleted the paper.".into(),
                    });
                    deleted_count += 1;
                } else {
                    delete_statuses.push(DeleteStatus {
                        id,
                        status: Status::Error,
                        message: "Error committing the transaction.".into(),
                    });
                }
            } else {
                tx.rollback().await?;
                delete_statuses.push(DeleteStatus {
                    id,
                    status: Status::Error,
                    message: "Failed to delete file.".into(),
                });
            }
        }
    }

    let message = if deleted_count > 0 {
        format!("Successfully hard deleted {} papers.", deleted_count)
    } else {
        "No papers were deleted.".into()
    };

    Ok(BackendResponse::ok(message, delete_statuses))
}

/// Fetches all question papers that match one or more properties specified. `course_name` is compulsory.
///
/// # Request Query Parameters
/// * `course_code`: The course code of the question paper. (required)
/// * `year` (optional): The year of the question paper.
/// * `course_name` (optional): The course name (exact).
/// * `semester` (optional): The semester (autumn/spring)
/// * `exam` (optional): The exam field (midsem/endsem/ct)
pub async fn similar(
    State(state): State<RouterState>,
    Query(body): Query<HashMap<String, String>>,
) -> HandlerReturn<Vec<AdminDashboardQP>> {
    if !body.contains_key("course_code") {
        return Ok(BackendResponse::error(
            "Error: `course_code` is required.".into(),
            StatusCode::BAD_REQUEST,
        ));
    }

    let papers = state
        .db
        .get_similar_papers(
            body.get("course_code")
                .context("Expected course code to be here.")?,
            body.get("year")
                .map(|year| year.parse::<i32>())
                .transpose()?,
            body.get("semester"),
            body.get("exam"),
        )
        .await?;

    Ok(BackendResponse::ok(
        format!("Found {} similar papers.", papers.len()),
        papers
            .iter()
            .map(|paper| paper.to_owned().with_url(&state.env_vars))
            .collect::<Result<Vec<AdminDashboardQP>>>()?,
    ))
}