AstralRinth/src/search/indexing/local_import.rs

use bson::doc;
use futures::StreamExt;
use log::info;

use crate::database::models::Item;
use crate::database::{DatabaseError, Mod, Version};

use super::IndexingError;
use crate::search::SearchMod;

pub async fn index_local(client: mongodb::Client) -> Result<Vec<SearchMod>, IndexingError> {
    info!("Indexing local mods!");

    let mut docs_to_add: Vec<SearchMod> = vec![];

    let db = client.database("modrinth");

    let mods = db.collection("mods");
    let versions = db.collection("versions");

    let mut results = mods
        .find(None, None)
        .await
        .map_err(DatabaseError::LocalDatabaseError)?;

    while let Some(unparsed_result) = results.next().await {
        let result: Mod =
            *Mod::from_doc(unparsed_result.map_err(DatabaseError::LocalDatabaseError)?)?;

        let mut mod_versions = versions
            .find(doc! { "mod_id": result.id }, None)
            .await
            .map_err(DatabaseError::LocalDatabaseError)?;

        let mut mod_game_versions = vec![];

        while let Some(unparsed_version) = mod_versions.next().await {
            let mut version = unparsed_version
                .map_err(DatabaseError::LocalDatabaseError)
                .and_then(Version::from_doc)?;
            mod_game_versions.append(&mut version.game_versions);
        }

        let mut icon_url = "".to_string();

        if let Some(url) = result.icon_url {
            icon_url = url;
        }

        docs_to_add.push(SearchMod {
            mod_id: result.id,
            author: "".to_string(),
            title: result.title,
            description: result.description,
            keywords: result.categories,
            versions: mod_game_versions,
            downloads: result.downloads,
            page_url: "".to_string(),
            icon_url,
            author_url: "".to_string(),
            date_created: "".to_string(),
            created: 0,
            date_modified: "".to_string(),
            updated: 0,
            latest_version: "".to_string(),
            empty: String::from("{}{}{}"),
        });
    }

    Ok(docs_to_add)
}