Modrinth/src/routes/mod.rs

mod v1;
pub use v1::v1_config;

mod admin;
mod auth;
mod health;
mod index;
mod maven;
mod midas;
mod moderation;
mod not_found;
mod notifications;
pub(crate) mod project_creation;
mod projects;
mod reports;
mod statistics;
mod tags;
mod teams;
mod updates;
mod users;
mod version_creation;
mod version_file;
mod versions;

pub use auth::config as auth_config;
pub use tags::config as tags_config;

pub use self::health::health_get;
pub use self::index::index_get;
pub use self::not_found::not_found;
use crate::file_hosting::FileHostingError;
use actix_web::web;

pub fn v2_config(cfg: &mut web::ServiceConfig) {
    cfg.service(
        web::scope("v2")
            .configure(auth_config)
            .configure(tags_config)
            .configure(projects_config)
            .configure(versions_config)
            .configure(teams_config)
            .configure(users_config)
            .configure(moderation_config)
            .configure(reports_config)
            .configure(notifications_config)
            .configure(statistics_config)
            .configure(admin_config)
            .configure(midas_config),
    );
}

pub fn projects_config(cfg: &mut web::ServiceConfig) {
    cfg.service(projects::project_search);
    cfg.service(projects::projects_get);
    cfg.service(project_creation::project_create);

    cfg.service(
        web::scope("project")
            .service(projects::project_get)
            .service(projects::project_get_check)
            .service(projects::project_delete)
            .service(projects::project_edit)
            .service(projects::project_icon_edit)
            .service(projects::delete_project_icon)
            .service(projects::add_gallery_item)
            .service(projects::edit_gallery_item)
            .service(projects::delete_gallery_item)
            .service(projects::project_follow)
            .service(projects::project_unfollow)
            .service(projects::project_schedule)
            .service(teams::team_members_get_project)
            .service(
                web::scope("{project_id}")
                    .service(versions::version_list)
                    .service(projects::dependency_list),
            ),
    );
}

pub fn maven_config(cfg: &mut web::ServiceConfig) {
    cfg.service(maven::maven_metadata);
    cfg.service(maven::version_file_sha512);
    cfg.service(maven::version_file_sha1);
    cfg.service(maven::version_file);
}

pub fn updates(cfg: &mut web::ServiceConfig) {
    cfg.service(updates::forge_updates);
}

pub fn versions_config(cfg: &mut web::ServiceConfig) {
    cfg.service(versions::versions_get);
    cfg.service(version_creation::version_create);
    cfg.service(
        web::scope("version")
            .service(versions::version_get)
            .service(versions::version_delete)
            .service(version_creation::upload_file_to_version)
            .service(versions::version_edit)
            .service(versions::version_schedule),
    );
    cfg.service(
        web::scope("version_file")
            .service(version_file::delete_file)
            .service(version_file::get_version_from_hash)
            .service(version_file::download_version)
            .service(version_file::get_update_from_hash),
    );

    cfg.service(
        web::scope("version_files")
            .service(version_file::get_versions_from_hashes)
            .service(version_file::download_files)
            .service(version_file::update_files),
    );
}

pub fn users_config(cfg: &mut web::ServiceConfig) {
    cfg.service(users::user_auth_get);

    cfg.service(users::users_get);
    cfg.service(
        web::scope("user")
            .service(users::user_get)
            .service(users::projects_list)
            .service(users::user_delete)
            .service(users::user_edit)
            .service(users::user_icon_edit)
            .service(users::user_notifications)
            .service(users::user_follows)
            .service(users::user_payouts)
            .service(users::user_payouts_request),
    );
}

pub fn teams_config(cfg: &mut web::ServiceConfig) {
    cfg.service(teams::teams_get);

    cfg.service(
        web::scope("team")
            .service(teams::team_members_get)
            .service(teams::edit_team_member)
            .service(teams::transfer_ownership)
            .service(teams::add_team_member)
            .service(teams::join_team)
            .service(teams::remove_team_member),
    );
}

pub fn notifications_config(cfg: &mut web::ServiceConfig) {
    cfg.service(notifications::notifications_get);
    cfg.service(notifications::notifications_delete);

    cfg.service(
        web::scope("notification")
            .service(notifications::notification_get)
            .service(notifications::notification_delete),
    );
}

pub fn moderation_config(cfg: &mut web::ServiceConfig) {
    cfg.service(
        web::scope("moderation")
            .service(moderation::get_projects)
            .service(moderation::ban_user)
            .service(moderation::unban_user),
    );
}

pub fn reports_config(cfg: &mut web::ServiceConfig) {
    cfg.service(reports::reports);
    cfg.service(reports::report_create);
    cfg.service(reports::delete_report);
}

pub fn statistics_config(cfg: &mut web::ServiceConfig) {
    cfg.service(statistics::get_stats);
}

pub fn admin_config(cfg: &mut web::ServiceConfig) {
    cfg.service(
        web::scope("admin")
            .service(admin::count_download)
            .service(admin::process_payout),
    );
}

pub fn midas_config(cfg: &mut web::ServiceConfig) {
    cfg.service(
        web::scope("midas")
            .service(midas::init_checkout)
            .service(midas::init_customer_portal)
            .service(midas::handle_stripe_webhook),
    );
}

#[derive(thiserror::Error, Debug)]
pub enum ApiError {
    #[error("Environment Error")]
    Env(#[from] dotenvy::Error),
    #[error("Error while uploading file")]
    FileHosting(#[from] FileHostingError),
    #[error("Database Error: {0}")]
    Database(#[from] crate::database::models::DatabaseError),
    #[error("Database Error: {0}")]
    SqlxDatabase(#[from] sqlx::Error),
    #[error("Internal server error: {0}")]
    Xml(String),
    #[error("Deserialization error: {0}")]
    Json(#[from] serde_json::Error),
    #[error("Authentication Error: {0}")]
    Authentication(#[from] crate::util::auth::AuthenticationError),
    #[error("Authentication Error: {0}")]
    CustomAuthentication(String),
    #[error("Invalid Input: {0}")]
    InvalidInput(String),
    #[error("Error while validating input: {0}")]
    Validation(String),
    #[error("Search Error: {0}")]
    Search(#[from] meilisearch_sdk::errors::Error),
    #[error("Indexing Error: {0}")]
    Indexing(#[from] crate::search::indexing::IndexingError),
    #[error("Ariadne Error: {0}")]
    Analytics(String),
    #[error("Crypto Error: {0}")]
    Crypto(String),
    #[error("Payments Error: {0}")]
    Payments(String),
}

impl actix_web::ResponseError for ApiError {
    fn status_code(&self) -> actix_web::http::StatusCode {
        match self {
            ApiError::Env(..) => {
                actix_web::http::StatusCode::INTERNAL_SERVER_ERROR
            }
            ApiError::Database(..) => {
                actix_web::http::StatusCode::INTERNAL_SERVER_ERROR
            }
            ApiError::SqlxDatabase(..) => {
                actix_web::http::StatusCode::INTERNAL_SERVER_ERROR
            }
            ApiError::Authentication(..) => {
                actix_web::http::StatusCode::UNAUTHORIZED
            }
            ApiError::CustomAuthentication(..) => {
                actix_web::http::StatusCode::UNAUTHORIZED
            }
            ApiError::Xml(..) => {
                actix_web::http::StatusCode::INTERNAL_SERVER_ERROR
            }
            ApiError::Json(..) => actix_web::http::StatusCode::BAD_REQUEST,
            ApiError::Search(..) => {
                actix_web::http::StatusCode::INTERNAL_SERVER_ERROR
            }
            ApiError::Indexing(..) => {
                actix_web::http::StatusCode::INTERNAL_SERVER_ERROR
            }
            ApiError::FileHosting(..) => {
                actix_web::http::StatusCode::INTERNAL_SERVER_ERROR
            }
            ApiError::InvalidInput(..) => {
                actix_web::http::StatusCode::BAD_REQUEST
            }
            ApiError::Validation(..) => {
                actix_web::http::StatusCode::BAD_REQUEST
            }
            ApiError::Analytics(..) => {
                actix_web::http::StatusCode::FAILED_DEPENDENCY
            }
            ApiError::Crypto(..) => actix_web::http::StatusCode::FORBIDDEN,
            ApiError::Payments(..) => {
                actix_web::http::StatusCode::FAILED_DEPENDENCY
            }
        }
    }

    fn error_response(&self) -> actix_web::HttpResponse {
        actix_web::HttpResponse::build(self.status_code()).json(
            crate::models::error::ApiError {
                error: match self {
                    ApiError::Env(..) => "environment_error",
                    ApiError::SqlxDatabase(..) => "database_error",
                    ApiError::Database(..) => "database_error",
                    ApiError::Authentication(..) => "unauthorized",
                    ApiError::CustomAuthentication(..) => "unauthorized",
                    ApiError::Xml(..) => "xml_error",
                    ApiError::Json(..) => "json_error",
                    ApiError::Search(..) => "search_error",
                    ApiError::Indexing(..) => "indexing_error",
                    ApiError::FileHosting(..) => "file_hosting_error",
                    ApiError::InvalidInput(..) => "invalid_input",
                    ApiError::Validation(..) => "invalid_input",
                    ApiError::Analytics(..) => "analytics_error",
                    ApiError::Crypto(..) => "crypto_error",
                    ApiError::Payments(..) => "payments_error",
                },
                description: &self.to_string(),
            },
        )
    }
}