You've already forked AstralRinth
6ff7fa74e2
* refactor: improve error handling * fix: specify bind address instead of port * fix: remove temporary testing file * fix(errors): change error names to snake_case * refactor(errors): split indexing error types, remove unused errors * feat: add env variable checking at program start This just checks whether the enviroment variables exist and can parse to the given type and gives a warning if they can't. This should prevent cases where the program fails at runtime due to checking an environment variable that doesn't exist.
66 lines
2.0 KiB
Rust
66 lines
2.0 KiB
Rust
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);
|
|
}
|
|
|
|
docs_to_add.push(SearchMod {
|
|
mod_id: result.id,
|
|
author: result.author,
|
|
title: result.title,
|
|
description: result.description,
|
|
keywords: result.categories,
|
|
versions: mod_game_versions,
|
|
downloads: result.downloads,
|
|
page_url: "".to_string(),
|
|
icon_url: result.icon_path,
|
|
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)
|
|
}
|