move to monorepo dir

apps/labrinth/src/routes/v2/version_creation.rs

@@ -0,0 +1,298 @@
+use crate::database::models::loader_fields::VersionField;
+use crate::database::models::{project_item, version_item};
+use crate::database::redis::RedisPool;
+use crate::file_hosting::FileHost;
+use crate::models::ids::ImageId;
+use crate::models::projects::{
+    Dependency, FileType, Loader, ProjectId, Version, VersionId, VersionStatus, VersionType,
+};
+use crate::models::v2::projects::LegacyVersion;
+use crate::queue::moderation::AutomatedModerationQueue;
+use crate::queue::session::AuthQueue;
+use crate::routes::v3::project_creation::CreateError;
+use crate::routes::v3::version_creation;
+use crate::routes::{v2_reroute, v3};
+use actix_multipart::Multipart;
+use actix_web::http::header::ContentDisposition;
+use actix_web::web::Data;
+use actix_web::{post, web, HttpRequest, HttpResponse};
+use serde::{Deserialize, Serialize};
+use serde_json::json;
+use sqlx::postgres::PgPool;
+use std::collections::HashMap;
+use std::sync::Arc;
+use validator::Validate;
+
+pub fn default_requested_status() -> VersionStatus {
+    VersionStatus::Listed
+}
+
+#[derive(Serialize, Deserialize, Validate, Clone)]
+pub struct InitialVersionData {
+    #[serde(alias = "mod_id")]
+    pub project_id: Option<ProjectId>,
+    #[validate(length(min = 1, max = 256))]
+    pub file_parts: Vec<String>,
+    #[validate(
+        length(min = 1, max = 32),
+        regex = "crate::util::validate::RE_URL_SAFE"
+    )]
+    pub version_number: String,
+    #[validate(
+        length(min = 1, max = 64),
+        custom(function = "crate::util::validate::validate_name")
+    )]
+    #[serde(alias = "name")]
+    pub version_title: String,
+    #[validate(length(max = 65536))]
+    #[serde(alias = "changelog")]
+    pub version_body: Option<String>,
+    #[validate(
+        length(min = 0, max = 4096),
+        custom(function = "crate::util::validate::validate_deps")
+    )]
+    pub dependencies: Vec<Dependency>,
+    #[validate(length(min = 1))]
+    pub game_versions: Vec<String>,
+    #[serde(alias = "version_type")]
+    pub release_channel: VersionType,
+    #[validate(length(min = 1))]
+    pub loaders: Vec<Loader>,
+    pub featured: bool,
+    pub primary_file: Option<String>,
+    #[serde(default = "default_requested_status")]
+    pub status: VersionStatus,
+    #[serde(default = "HashMap::new")]
+    pub file_types: HashMap<String, Option<FileType>>,
+    // Associations to uploaded images in changelog
+    #[validate(length(max = 10))]
+    #[serde(default)]
+    pub uploaded_images: Vec<ImageId>,
+
+    // The ordering relative to other versions
+    pub ordering: Option<i32>,
+}
+
+#[derive(Serialize, Deserialize, Clone)]
+struct InitialFileData {
+    #[serde(default = "HashMap::new")]
+    pub file_types: HashMap<String, Option<FileType>>,
+}
+
+// under `/api/v1/version`
+#[post("version")]
+pub async fn version_create(
+    req: HttpRequest,
+    payload: Multipart,
+    client: Data<PgPool>,
+    redis: Data<RedisPool>,
+    file_host: Data<Arc<dyn FileHost + Send + Sync>>,
+    session_queue: Data<AuthQueue>,
+    moderation_queue: Data<AutomatedModerationQueue>,
+) -> Result<HttpResponse, CreateError> {
+    let payload = v2_reroute::alter_actix_multipart(
+        payload,
+        req.headers().clone(),
+        |legacy_create: InitialVersionData, content_dispositions: Vec<ContentDisposition>| {
+            let client = client.clone();
+            let redis = redis.clone();
+            async move {
+                // Convert input data to V3 format
+                let mut fields = HashMap::new();
+                fields.insert(
+                    "game_versions".to_string(),
+                    json!(legacy_create.game_versions),
+                );
+
+                // Get all possible side-types for loaders given- we will use these to check if we need to convert/apply singleplayer, etc.
+                let loaders = match v3::tags::loader_list(client.clone(), redis.clone()).await {
+                    Ok(loader_response) => {
+                        (v2_reroute::extract_ok_json::<Vec<v3::tags::LoaderData>>(loader_response)
+                            .await)
+                            .unwrap_or_default()
+                    }
+                    Err(_) => vec![],
+                };
+
+                let loader_fields_aggregate = loaders
+                    .into_iter()
+                    .filter_map(|loader| {
+                        if legacy_create.loaders.contains(&Loader(loader.name.clone())) {
+                            Some(loader.supported_fields)
+                        } else {
+                            None
+                        }
+                    })
+                    .flatten()
+                    .collect::<Vec<_>>();
+
+                // Copies side types of another version of the project.
+                // If no version exists, defaults to all false.
+                // This is inherently lossy, but not much can be done about it, as side types are no longer associated with projects,
+                // so the 'missing' ones can't be easily accessed, and versions do need to have these fields explicitly set.
+                let side_type_loader_field_names = [
+                    "singleplayer",
+                    "client_and_server",
+                    "client_only",
+                    "server_only",
+                ];
+
+                // Check if loader_fields_aggregate contains any of these side types
+                // We assume these four fields are linked together.
+                if loader_fields_aggregate
+                    .iter()
+                    .any(|f| side_type_loader_field_names.contains(&f.as_str()))
+                {
+                    // If so, we get the fields of the example version of the project, and set the side types to match.
+                    fields.extend(
+                        side_type_loader_field_names
+                            .iter()
+                            .map(|f| (f.to_string(), json!(false))),
+                    );
+                    if let Some(example_version_fields) =
+                        get_example_version_fields(legacy_create.project_id, client, &redis).await?
+                    {
+                        fields.extend(example_version_fields.into_iter().filter_map(|f| {
+                            if side_type_loader_field_names.contains(&f.field_name.as_str()) {
+                                Some((f.field_name, f.value.serialize_internal()))
+                            } else {
+                                None
+                            }
+                        }));
+                    }
+                }
+                // Handle project type via file extension prediction
+                let mut project_type = None;
+                for file_part in &legacy_create.file_parts {
+                    if let Some(ext) = file_part.split('.').last() {
+                        match ext {
+                            "mrpack" | "mrpack-primary" => {
+                                project_type = Some("modpack");
+                                break;
+                            }
+                            // No other type matters
+                            _ => {}
+                        }
+                        break;
+                    }
+                }
+
+                // Similarly, check actual content disposition for mrpacks, in case file_parts is wrong
+                for content_disposition in content_dispositions {
+                    // Uses version_create functions to get the file name and extension
+                    let (_, file_extension) = version_creation::get_name_ext(&content_disposition)?;
+                    crate::util::ext::project_file_type(file_extension)
+                        .ok_or_else(|| CreateError::InvalidFileType(file_extension.to_string()))?;
+
+                    if file_extension == "mrpack" {
+                        project_type = Some("modpack");
+                        break;
+                    }
+                }
+
+                // Modpacks now use the "mrpack" loader, and loaders are converted to loader fields.
+                // Setting of 'project_type' directly is removed, it's loader-based now.
+                if project_type == Some("modpack") {
+                    fields.insert("mrpack_loaders".to_string(), json!(legacy_create.loaders));
+                }
+
+                let loaders = if project_type == Some("modpack") {
+                    vec![Loader("mrpack".to_string())]
+                } else {
+                    legacy_create.loaders
+                };
+
+                Ok(v3::version_creation::InitialVersionData {
+                    project_id: legacy_create.project_id,
+                    file_parts: legacy_create.file_parts,
+                    version_number: legacy_create.version_number,
+                    version_title: legacy_create.version_title,
+                    version_body: legacy_create.version_body,
+                    dependencies: legacy_create.dependencies,
+                    release_channel: legacy_create.release_channel,
+                    loaders,
+                    featured: legacy_create.featured,
+                    primary_file: legacy_create.primary_file,
+                    status: legacy_create.status,
+                    file_types: legacy_create.file_types,
+                    uploaded_images: legacy_create.uploaded_images,
+                    ordering: legacy_create.ordering,
+                    fields,
+                })
+            }
+        },
+    )
+    .await?;
+
+    // Call V3 project creation
+    let response = v3::version_creation::version_create(
+        req,
+        payload,
+        client.clone(),
+        redis.clone(),
+        file_host,
+        session_queue,
+        moderation_queue,
+    )
+    .await?;
+
+    // Convert response to V2 format
+    match v2_reroute::extract_ok_json::<Version>(response).await {
+        Ok(version) => {
+            let v2_version = LegacyVersion::from(version);
+            Ok(HttpResponse::Ok().json(v2_version))
+        }
+        Err(response) => Ok(response),
+    }
+}
+
+// Gets version fields of an example version of a project, if one exists.
+async fn get_example_version_fields(
+    project_id: Option<ProjectId>,
+    pool: Data<PgPool>,
+    redis: &RedisPool,
+) -> Result<Option<Vec<VersionField>>, CreateError> {
+    let project_id = match project_id {
+        Some(project_id) => project_id,
+        None => return Ok(None),
+    };
+
+    let vid = match project_item::Project::get_id(project_id.into(), &**pool, redis)
+        .await?
+        .and_then(|p| p.versions.first().cloned())
+    {
+        Some(vid) => vid,
+        None => return Ok(None),
+    };
+
+    let example_version = match version_item::Version::get(vid, &**pool, redis).await? {
+        Some(version) => version,
+        None => return Ok(None),
+    };
+    Ok(Some(example_version.version_fields))
+}
+
+// under /api/v1/version/{version_id}
+#[post("{version_id}/file")]
+pub async fn upload_file_to_version(
+    req: HttpRequest,
+    url_data: web::Path<(VersionId,)>,
+    payload: Multipart,
+    client: Data<PgPool>,
+    redis: Data<RedisPool>,
+    file_host: Data<Arc<dyn FileHost + Send + Sync>>,
+    session_queue: web::Data<AuthQueue>,
+) -> Result<HttpResponse, CreateError> {
+    // Returns NoContent, so no need to convert to V2
+    let response = v3::version_creation::upload_file_to_version(
+        req,
+        url_data,
+        payload,
+        client.clone(),
+        redis.clone(),
+        file_host,
+        session_queue,
+    )
+    .await?;
+    Ok(response)
+}