summaryrefslogtreecommitdiff
path: root/src/commands/setup.rs
blob: be06040ac9eed89bfc20abe018a79530961ac1a0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
use std::{fs::File, path::Path, io::{Error, ErrorKind}};

use crate::{config::Cfg, db::{db_setup, s_config_get_version, s_config_create_version, s_insert_column, lists_get_all_ids, lists_get, userlist_get_all_current_version_ids, s_userlist_update_download, s_config_update_version}, modrinth::get_raw_versions};

pub async fn setup(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
    let db_file = format!("{}/data.db", String::from(&config.data));
    
    if !Path::new(&db_file).exists() {
        return create(config, db_file);
    }
    
    match s_config_get_version(config.clone()) {
        Ok(ver) => {
            match ver.as_str() {
                "0.2" => to_03(config)?,
                _ => return Err(Box::new(Error::new(ErrorKind::Other, "UNKNOWN_VERSION")))
            }
        },
        Err(..) => to_02(config).await?
    };
    
    Ok(())
}

fn create(config: Cfg, db_file: String) -> Result<(), Box<dyn std::error::Error>> {
    File::create(db_file)?;
    db_setup(config)?;
    Ok(())
}

async fn to_02(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
    let lists = lists_get_all_ids(config.clone())?;

    for list in lists {
        println!("Updating {}", list);
        s_insert_column(config.clone(), String::from(&list), String::from("current_download"), String::from("TEXT"))?;

        let full_list = lists_get(config.clone(), String::from(&list))?;

        let versions = userlist_get_all_current_version_ids(config.clone(), full_list.clone().id)?;
        
        let raw_versions = get_raw_versions(String::from(&config.apis.modrinth), versions).await;

        for ver in raw_versions {
            println!("Adding link for {}", ver.project_id);
            let file = ver.files.into_iter().find(|f| f.primary).unwrap();
            s_userlist_update_download(config.clone(), String::from(&full_list.id), ver.project_id, file.url)?;
        }
    };
    s_config_create_version(config)?;

    Ok(())
}

fn to_03(config: Cfg) -> Result<(), Box<dyn std::error::Error>> {
    s_insert_column(config.clone(), String::from("lists"), String::from("download_folder"), String::from("TEXT"))?;
    s_config_update_version(config, String::from("0.3"))
}