path: root/src/commands/modification.rs

use std::io::{Error, ErrorKind};

use crate::{modrinth::{project, versions, extract_current_version, Version, projects}, config::Cfg, db::{mods_insert, userlist_remove, mods_get_id, userlist_insert, mods_get_all_ids, userlist_get_all_ids, userlist_get_current_version, lists_get_all_ids, mods_remove}, input::{Input, Subcmd}, get_current_list, files::{delete_version, download_versions}, List};

pub async fn modification(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {
    match input.subcommand.as_ref().ok_or("")? {
        Subcmd::Add => {
            add(config, input).await
        },
        Subcmd::Remove => {
            remove(config, input.args.ok_or("")?)
        },
        _ => Err(Box::new(Error::new(ErrorKind::InvalidInput, "SUBCOMMAND_NOT_AVAILABLE")))
    }
}

async fn add(config: Cfg, input: Input) -> Result<(), Box<dyn std::error::Error>> {

    let args = input.args.ok_or("")?;

    if args.is_empty() { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_FEW_ARGUMENTS"))); };
    
    let current_list = get_current_list(config.clone())?;
    
    mod_add(config, vec![String::from(&args[0])], current_list, input.disable_download).await?;

    Ok(())
}

pub async fn mod_add(config: Cfg, mod_id: Vec<String>, list: List, disable_download: bool) -> Result<(), Box<dyn std::error::Error>> {
    
    println!("Adding mod(s) {:?}", mod_id);
    let projects = if mod_id.len() == 1 {
        vec![project(String::from(&config.apis.modrinth), &mod_id[0]).await]
    } else {
        projects(String::from(&config.apis.modrinth), mod_id).await
    };
    
    for project in projects {
        let available_versions = versions(String::from(&config.apis.modrinth), String::from(&project.id), list.clone()).await;

        let mut available_versions_vec: Vec<String> = Vec::new();
        let current_version: Option<Version>;
        let current_version_id: String;
        let file: String;
        if !available_versions.is_empty() {
            let current_id = extract_current_version(available_versions.clone())?;

            current_version = Some(available_versions.clone().into_iter().find(|v| v.id == current_id).unwrap());

            current_version_id = current_version.clone().unwrap().id;

            file = current_version.clone().ok_or("VERSION_CORRUPTED")?.files.into_iter().find(|f| f.primary).unwrap().url;

            for ver in available_versions {
                available_versions_vec.push(ver.id);
            };
        } else {
            println!("There's currently no mod version for your specified target");
            current_version = None;
            current_version_id = String::from("NONE");
            file = String::from("NONE");
            available_versions_vec.push(String::from("NONE"));
        }

        //add to current list and mod table
        match userlist_get_all_ids(config.clone(), list.clone().id) {
            Ok(mods) => {
                if mods.contains(&project.id) {
                    return Err(Box::new(Error::new(ErrorKind::Other, "MOD_ALREADY_ON_LIST"))); } 
                else {
                    userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?;
                } 
            },
            Err(..) => userlist_insert(config.clone(), String::from(&list.id), String::from(&project.id), String::from(&current_version_id), available_versions_vec, file)?,
        };
        
        match mods_get_all_ids(config.clone()) {
            Ok(mods) => {
                if mods.contains(&project.id) {
                    //return Err(Box::new(Error::new(ErrorKind::Other, "MOD_ALREADY_IN_DATABASE")))
                } else {
                    mods_insert(config.clone(), String::from(&project.id), String::from(&project.title), project.versions)?;
                } 
            },
            Err(..) => {
                mods_insert(config.clone(), String::from(&project.id), String::from(&project.title), project.versions)?;
            },
        };

        if !disable_download && current_version.is_some() { download_versions(list.clone(), vec![current_version.unwrap()]).await?; };
            
    }

    Ok(())
}

fn remove(config: Cfg, args: Vec<String>) -> Result<(), Box<dyn std::error::Error>> {
    if args.is_empty() { return Err(Box::new(Error::new(ErrorKind::InvalidInput, "TOO_FEW_ARGUMENTS"))); };

    let current_list = get_current_list(config.clone())?;
    let mod_id = mods_get_id(config.clone(), String::from(&args[0]))?;
    
    let version = userlist_get_current_version(config.clone(), String::from(&current_list.id), String::from(&mod_id))?;

    //TODO implement remove from modlist if not in any other lists && config clean is true
    userlist_remove(config.clone(), String::from(&current_list.id), String::from(&mod_id))?;
    delete_version(current_list, version)?;
    
    let list_ids = lists_get_all_ids(config.clone())?;
    
    let mut mod_used = false;
    for id in list_ids {
        let mods = userlist_get_all_ids(config.clone(), id)?;
        if mods.contains(&mod_id) { mod_used = true; break; };
    };

    if !mod_used { mods_remove(config, mod_id)?; };

    Ok(())
}