AstralRinth/packages/app-lib/src/api/profile/update.rs

use crate::state::CacheBehaviour;
use crate::{
    event::{
        emit::{emit_profile, init_loading},
        ProfilePayloadType,
    },
    pack::{self, install_from::generate_pack_from_version_id},
    profile::get,
    state::ProfileInstallStage,
    LoadingBarType,
};
use futures::try_join;

/// Updates a managed modrinth pack to the version specified by new_version_id
#[tracing::instrument]

pub async fn update_managed_modrinth_version(
    profile_path: &String,
    new_version_id: &String,
) -> crate::Result<()> {
    let profile = get(profile_path).await?.ok_or_else(|| {
        crate::ErrorKind::UnmanagedProfileError(profile_path.to_string())
            .as_error()
    })?;

    let unmanaged_err = || {
        crate::ErrorKind::InputError(
            format!("Profile at {} is not a managed modrinth pack, or has been disconnected.", profile_path),
        )
    };

    // Extract modrinth pack information, if appropriate
    let linked_data = profile.linked_data.as_ref().ok_or_else(unmanaged_err)?;

    // Replace the pack with the new version
    replace_managed_modrinth(
        profile_path,
        &profile,
        &linked_data.project_id,
        &linked_data.version_id,
        Some(new_version_id),
        true, // switching versions should ignore the lock
    )
    .await?;

    emit_profile(profile_path, ProfilePayloadType::Edited).await?;

    Ok(())
}

/// Repair a managed modrinth pack by 'updating' it to the current version
#[tracing::instrument]

pub async fn repair_managed_modrinth(profile_path: &str) -> crate::Result<()> {
    let profile = get(profile_path).await?.ok_or_else(|| {
        crate::ErrorKind::UnmanagedProfileError(profile_path.to_string())
            .as_error()
    })?;

    let unmanaged_err = || {
        crate::ErrorKind::InputError(
            format!("Profile at {} is not a managed modrinth pack, or has been disconnected.", profile_path),
        )
    };

    // For repairing specifically, first we remove all installed projects (to ensure we do remove ones that aren't in the pack)
    // We do a project removal followed by removing everything in the .mrpack, to ensure we only
    // remove relevant projects and not things like save files
    let state = crate::State::get().await?;
    let projects_map = profile
        .get_projects(
            Some(CacheBehaviour::MustRevalidate),
            &state.pool,
            &state.api_semaphore,
        )
        .await?;

    for (file, _) in projects_map {
        crate::state::Profile::remove_project(&profile.path, &file).await?;
    }

    // Extract modrinth pack information, if appropriate
    let linked_data = profile.linked_data.as_ref().ok_or_else(unmanaged_err)?;

    // Replace the pack with the same version
    replace_managed_modrinth(
        profile_path,
        &profile,
        &linked_data.project_id,
        &linked_data.version_id,
        None,
        false, // do not ignore lock, as repairing can reset the lock
    )
    .await?;

    emit_profile(profile_path, ProfilePayloadType::Edited).await?;

    Ok(())
}

/// Replace a managed modrinth pack with a new version
/// If new_version_id is None, the pack is 'reinstalled' in-place
#[tracing::instrument(skip(profile))]

async fn replace_managed_modrinth(
    profile_path: &str,
    profile: &crate::state::Profile,
    project_id: &String,
    version_id: &String,
    new_version_id: Option<&String>,
    ignore_lock: bool,
) -> crate::Result<()> {
    crate::profile::edit(profile_path, |profile| {
        profile.install_stage = ProfileInstallStage::MinecraftInstalling;
        async { Ok(()) }
    })
    .await?;

    // Fetch .mrpacks for both old and new versions
    // TODO: this will need to be updated if we revert the hacky pack method we needed for compiler speed

    let (old_pack_creator, new_pack_creator) = if let Some(new_version_id) =
        new_version_id
    {
        let shared_loading_bar = init_loading(
            LoadingBarType::PackFileDownload {
                profile_path: crate::api::profile::get_full_path(profile_path)
                    .await?
                    .to_string_lossy()
                    .to_string(),
                pack_name: profile.name.clone(),
                icon: None,
                pack_version: version_id.clone(),
            },
            200.0, // These two downloads will share the same loading bar
            "Downloading pack file",
        )
        .await?;

        // download in parallel, then join.
        try_join!(
            generate_pack_from_version_id(
                project_id.clone(),
                version_id.clone(),
                profile.name.clone(),
                None,
                profile_path.to_string(),
                Some(shared_loading_bar.clone())
            ),
            generate_pack_from_version_id(
                project_id.clone(),
                new_version_id.clone(),
                profile.name.clone(),
                None,
                profile_path.to_string(),
                Some(shared_loading_bar)
            )
        )?
    } else {
        // If new_version_id is None, we don't need to download the new pack, so we clone the old one
        let mut old_pack_creator = generate_pack_from_version_id(
            project_id.clone(),
            version_id.clone(),
            profile.name.clone(),
            None,
            profile_path.to_string(),
            None,
        )
        .await?;
        old_pack_creator.description.existing_loading_bar = None;
        (old_pack_creator.clone(), old_pack_creator)
    };

    // Removal - remove all files that were added by the old pack
    // - remove all installed projects
    // - remove all overrides
    pack::install_mrpack::remove_all_related_files(
        profile_path.to_string(),
        old_pack_creator.file,
    )
    .await?;

    // Reinstallation - install all files that are added by the new pack
    // - install all projects
    // - install all overrides
    // - edits the profile to update the new data
    // - (functionals almost identically to rteinstalling the pack 'in-place')
    pack::install_mrpack::install_zipped_mrpack_files(
        new_pack_creator,
        ignore_lock,
    )
    .await?;

    Ok(())
}