forked from didirus/AstralRinth
Fix issues
This commit is contained in:
@@ -1,8 +1,14 @@
|
||||
use crate::{format_url, upload_file_to_bucket, Error};
|
||||
use crate::{
|
||||
download_file, download_file_mirrors, format_url, upload_file_to_bucket,
|
||||
Error,
|
||||
};
|
||||
use chrono::{DateTime, Utc};
|
||||
use daedalus::download_file;
|
||||
use daedalus::minecraft::{Argument, ArgumentType, Library, VersionManifest, VersionType};
|
||||
use daedalus::modded::{LoaderVersion, Manifest, PartialVersionInfo, Processor, SidedDataEntry};
|
||||
use daedalus::minecraft::{
|
||||
Argument, ArgumentType, Library, VersionManifest, VersionType,
|
||||
};
|
||||
use daedalus::modded::{
|
||||
LoaderVersion, Manifest, PartialVersionInfo, Processor, SidedDataEntry,
|
||||
};
|
||||
use lazy_static::lazy_static;
|
||||
use log::info;
|
||||
use semver::{Version, VersionReq};
|
||||
@@ -10,8 +16,8 @@ use serde::{Deserialize, Serialize};
|
||||
use std::collections::HashMap;
|
||||
use std::io::Read;
|
||||
use std::sync::Arc;
|
||||
use std::time::{Duration, Instant};
|
||||
use tokio::sync::Mutex;
|
||||
use std::time::Instant;
|
||||
use tokio::sync::{Mutex, Semaphore};
|
||||
|
||||
lazy_static! {
|
||||
static ref FORGE_MANIFEST_V1_QUERY: VersionReq =
|
||||
@@ -20,14 +26,16 @@ lazy_static! {
|
||||
VersionReq::parse(">=23.5.2851, <31.2.52").unwrap();
|
||||
static ref FORGE_MANIFEST_V2_QUERY_P2: VersionReq =
|
||||
VersionReq::parse(">=32.0.1, <37.0.0").unwrap();
|
||||
static ref FORGE_MANIFEST_V3_QUERY: VersionReq = VersionReq::parse(">=37.0.0").unwrap();
|
||||
static ref FORGE_MANIFEST_V3_QUERY: VersionReq =
|
||||
VersionReq::parse(">=37.0.0").unwrap();
|
||||
}
|
||||
|
||||
pub async fn retrieve_data(
|
||||
minecraft_versions: &VersionManifest,
|
||||
uploaded_files: &mut Vec<String>,
|
||||
semaphore: Arc<Semaphore>,
|
||||
) -> Result<(), Error> {
|
||||
let maven_metadata = fetch_maven_metadata(None).await?;
|
||||
let maven_metadata = fetch_maven_metadata(None, semaphore.clone()).await?;
|
||||
let old_manifest = daedalus::modded::fetch_manifest(&format_url(&format!(
|
||||
"forge/v{}/manifest.json",
|
||||
daedalus::modded::CURRENT_FORGE_FORMAT_VERSION,
|
||||
@@ -35,11 +43,12 @@ pub async fn retrieve_data(
|
||||
.await
|
||||
.ok();
|
||||
|
||||
let old_versions = Arc::new(Mutex::new(if let Some(old_manifest) = old_manifest {
|
||||
old_manifest.game_versions
|
||||
} else {
|
||||
Vec::new()
|
||||
}));
|
||||
let old_versions =
|
||||
Arc::new(Mutex::new(if let Some(old_manifest) = old_manifest {
|
||||
old_manifest.game_versions
|
||||
} else {
|
||||
Vec::new()
|
||||
}));
|
||||
|
||||
let versions = Arc::new(Mutex::new(Vec::new()));
|
||||
|
||||
@@ -52,13 +61,14 @@ pub async fn retrieve_data(
|
||||
let mut loaders = Vec::new();
|
||||
|
||||
for loader_version_full in loader_versions {
|
||||
let loader_version = loader_version_full.split('-').into_iter().nth(1);
|
||||
let loader_version = loader_version_full.split('-').nth(1);
|
||||
|
||||
if let Some(loader_version_raw) = loader_version {
|
||||
// This is a dirty hack to get around Forge not complying with SemVer, but whatever
|
||||
// Most of this is a hack anyways :(
|
||||
// Works for all forge versions!
|
||||
let split = loader_version_raw.split('.').collect::<Vec<&str>>();
|
||||
let split =
|
||||
loader_version_raw.split('.').collect::<Vec<&str>>();
|
||||
let loader_version = if split.len() >= 4 {
|
||||
if split[0].parse::<i32>().unwrap_or(0) < 6 {
|
||||
format!("{}.{}.{}", split[0], split[1], split[3])
|
||||
@@ -80,203 +90,209 @@ pub async fn retrieve_data(
|
||||
}
|
||||
}
|
||||
}
|
||||
version_futures.push(async {
|
||||
let mut loaders_versions = Vec::new();
|
||||
|
||||
{
|
||||
let mut loaders_futures = loaders.into_iter().map(|(loader_version_full, version)| async {
|
||||
let versions_mutex = Arc::clone(&old_versions);
|
||||
let visited_assets = Arc::clone(&visited_assets_mutex);
|
||||
let uploaded_files_mutex = Arc::clone(&uploaded_files_mutex);
|
||||
let minecraft_version = minecraft_version.clone();
|
||||
if !loaders.is_empty() {
|
||||
version_futures.push(async {
|
||||
let loaders_versions = Vec::new();
|
||||
|
||||
async move {
|
||||
/// These forge versions are not worth supporting!
|
||||
const WHITELIST : [&str; 1] = [
|
||||
// Not supported due to `data` field being `[]` even though the type is a map
|
||||
"1.12.2-14.23.5.2851"
|
||||
];
|
||||
{
|
||||
let loaders_futures = loaders.into_iter().map(|(loader_version_full, version)| async {
|
||||
let versions_mutex = Arc::clone(&old_versions);
|
||||
let visited_assets = Arc::clone(&visited_assets_mutex);
|
||||
let uploaded_files_mutex = Arc::clone(&uploaded_files_mutex);
|
||||
let semaphore = Arc::clone(&semaphore);
|
||||
let minecraft_version = minecraft_version.clone();
|
||||
|
||||
if WHITELIST.contains(&&*loader_version_full) {
|
||||
return Ok(None);
|
||||
}
|
||||
async move {
|
||||
/// These forge versions are not worth supporting!
|
||||
const WHITELIST : [&str; 1] = [
|
||||
// Not supported due to `data` field being `[]` even though the type is a map
|
||||
"1.12.2-14.23.5.2851",
|
||||
];
|
||||
|
||||
{
|
||||
let versions = versions_mutex.lock().await;
|
||||
let version = versions.iter().find(|x|
|
||||
x.id == minecraft_version).and_then(|x| x.loaders.iter().find(|x| x.id == loader_version_full));
|
||||
|
||||
if let Some(version) = version {
|
||||
return Ok::<Option<LoaderVersion>, Error>(Some(version.clone()));
|
||||
if WHITELIST.contains(&&*loader_version_full) {
|
||||
return Ok(None);
|
||||
}
|
||||
}
|
||||
|
||||
info!("Forge - Installer Start {}", loader_version_full.clone());
|
||||
let bytes = download_file(&format!("https://maven.minecraftforge.net/net/minecraftforge/forge/{0}/forge-{0}-installer.jar", loader_version_full), None).await?;
|
||||
{
|
||||
let versions = versions_mutex.lock().await;
|
||||
let version = versions.iter().find(|x|
|
||||
x.id == minecraft_version).and_then(|x| x.loaders.iter().find(|x| x.id == loader_version_full));
|
||||
|
||||
let reader = std::io::Cursor::new(bytes);
|
||||
if let Some(version) = version {
|
||||
return Ok::<Option<LoaderVersion>, Error>(Some(version.clone()));
|
||||
}
|
||||
}
|
||||
|
||||
if let Ok(archive) = zip::ZipArchive::new(reader) {
|
||||
if FORGE_MANIFEST_V1_QUERY.matches(&version) {
|
||||
let mut archive_clone = archive.clone();
|
||||
let profile = tokio::task::spawn_blocking(move || {
|
||||
let mut install_profile = archive_clone.by_name("install_profile.json")?;
|
||||
info!("Forge - Installer Start {}", loader_version_full.clone());
|
||||
let bytes = download_file(&format!("https://maven.minecraftforge.net/net/minecraftforge/forge/{0}/forge-{0}-installer.jar", loader_version_full), None, semaphore.clone()).await?;
|
||||
|
||||
let mut contents = String::new();
|
||||
install_profile.read_to_string(&mut contents)?;
|
||||
let reader = std::io::Cursor::new(bytes);
|
||||
|
||||
Ok::<ForgeInstallerProfileV1, Error>(serde_json::from_str::<ForgeInstallerProfileV1>(&contents)?)
|
||||
}).await??;
|
||||
if let Ok(archive) = zip::ZipArchive::new(reader) {
|
||||
if FORGE_MANIFEST_V1_QUERY.matches(&version) {
|
||||
let mut archive_clone = archive.clone();
|
||||
let profile = tokio::task::spawn_blocking(move || {
|
||||
let mut install_profile = archive_clone.by_name("install_profile.json")?;
|
||||
|
||||
let mut archive_clone = archive.clone();
|
||||
let file_path = profile.install.file_path.clone();
|
||||
let forge_universal_bytes = tokio::task::spawn_blocking(move || {
|
||||
let mut forge_universal_file = archive_clone.by_name(&file_path)?;
|
||||
let mut forge_universal = Vec::new();
|
||||
forge_universal_file.read_to_end(&mut forge_universal)?;
|
||||
let mut contents = String::new();
|
||||
install_profile.read_to_string(&mut contents)?;
|
||||
|
||||
Ok::<ForgeInstallerProfileV1, Error>(serde_json::from_str::<ForgeInstallerProfileV1>(&contents)?)
|
||||
}).await??;
|
||||
|
||||
let mut archive_clone = archive.clone();
|
||||
let file_path = profile.install.file_path.clone();
|
||||
let forge_universal_bytes = tokio::task::spawn_blocking(move || {
|
||||
let mut forge_universal_file = archive_clone.by_name(&file_path)?;
|
||||
let mut forge_universal = Vec::new();
|
||||
forge_universal_file.read_to_end(&mut forge_universal)?;
|
||||
|
||||
|
||||
Ok::<bytes::Bytes, Error>(bytes::Bytes::from(forge_universal))
|
||||
}).await??;
|
||||
let forge_universal_path = profile.install.path.clone();
|
||||
Ok::<bytes::Bytes, Error>(bytes::Bytes::from(forge_universal))
|
||||
}).await??;
|
||||
let forge_universal_path = profile.install.path.clone();
|
||||
|
||||
let now = Instant::now();
|
||||
let libs = futures::future::try_join_all(profile.version_info.libraries.into_iter().map(|mut lib| async {
|
||||
if let Some(url) = lib.url {
|
||||
{
|
||||
let mut visited_assets = visited_assets.lock().await;
|
||||
let now = Instant::now();
|
||||
let libs = futures::future::try_join_all(profile.version_info.libraries.into_iter().map(|mut lib| async {
|
||||
if let Some(url) = lib.url {
|
||||
{
|
||||
let mut visited_assets = visited_assets.lock().await;
|
||||
|
||||
if visited_assets.contains(&lib.name) {
|
||||
lib.url = Some(format_url("maven/"));
|
||||
if visited_assets.contains(&lib.name) {
|
||||
lib.url = Some(format_url("maven/"));
|
||||
|
||||
return Ok::<Library, Error>(lib);
|
||||
} else {
|
||||
visited_assets.push(lib.name.clone())
|
||||
return Ok::<Library, Error>(lib);
|
||||
} else {
|
||||
visited_assets.push(lib.name.clone())
|
||||
}
|
||||
}
|
||||
|
||||
let artifact_path =
|
||||
daedalus::get_path_from_artifact(&lib.name)?;
|
||||
|
||||
let artifact = if lib.name == forge_universal_path {
|
||||
forge_universal_bytes.clone()
|
||||
} else {
|
||||
let mirrors = vec![&*url, "https://maven.creeperhost.net/", "https://libraries.minecraft.net/"];
|
||||
|
||||
download_file_mirrors(
|
||||
&artifact_path,
|
||||
&mirrors,
|
||||
None,
|
||||
semaphore.clone(),
|
||||
)
|
||||
.await?
|
||||
};
|
||||
|
||||
lib.url = Some(format_url("maven/"));
|
||||
|
||||
upload_file_to_bucket(
|
||||
format!("{}/{}", "maven", artifact_path),
|
||||
artifact.to_vec(),
|
||||
Some("application/java-archive".to_string()),
|
||||
uploaded_files_mutex.as_ref(),
|
||||
semaphore.clone(),
|
||||
).await?;
|
||||
}
|
||||
|
||||
let artifact_path =
|
||||
daedalus::get_path_from_artifact(&lib.name)?;
|
||||
Ok::<Library, Error>(lib)
|
||||
})).await?;
|
||||
|
||||
let artifact = if lib.name == forge_universal_path {
|
||||
forge_universal_bytes.clone()
|
||||
} else {
|
||||
let mirrors = vec![&*url, "https://maven.creeperhost.net/", "https://libraries.minecraft.net/"];
|
||||
let elapsed = now.elapsed();
|
||||
info!("Elapsed lib DL: {:.2?}", elapsed);
|
||||
|
||||
daedalus::download_file_mirrors(
|
||||
&artifact_path,
|
||||
&mirrors,
|
||||
None,
|
||||
)
|
||||
.await?
|
||||
};
|
||||
let new_profile = PartialVersionInfo {
|
||||
id: profile.version_info.id,
|
||||
inherits_from: profile.install.minecraft,
|
||||
release_time: profile.version_info.release_time,
|
||||
time: profile.version_info.time,
|
||||
main_class: profile.version_info.main_class,
|
||||
minecraft_arguments: profile.version_info.minecraft_arguments.clone(),
|
||||
arguments: profile.version_info.minecraft_arguments.map(|x| [(ArgumentType::Game, x.split(' ').map(|x| Argument::Normal(x.to_string())).collect())].iter().cloned().collect()),
|
||||
libraries: libs,
|
||||
type_: profile.version_info.type_,
|
||||
data: None,
|
||||
processors: None
|
||||
};
|
||||
|
||||
lib.url = Some(format_url("maven/"));
|
||||
let version_path = format!(
|
||||
"forge/v{}/versions/{}.json",
|
||||
daedalus::modded::CURRENT_FORGE_FORMAT_VERSION,
|
||||
new_profile.id
|
||||
);
|
||||
|
||||
upload_file_to_bucket(
|
||||
format!("{}/{}", "maven", artifact_path),
|
||||
artifact.to_vec(),
|
||||
Some("application/java-archive".to_string()),
|
||||
uploaded_files_mutex.as_ref(),
|
||||
).await?;
|
||||
upload_file_to_bucket(
|
||||
version_path.clone(),
|
||||
serde_json::to_vec(&new_profile)?,
|
||||
Some("application/json".to_string()),
|
||||
uploaded_files_mutex.as_ref(),
|
||||
semaphore.clone(),
|
||||
).await?;
|
||||
|
||||
return Ok(Some(LoaderVersion {
|
||||
id: loader_version_full,
|
||||
url: format_url(&version_path),
|
||||
stable: false
|
||||
}));
|
||||
} else if FORGE_MANIFEST_V2_QUERY_P1.matches(&version) || FORGE_MANIFEST_V2_QUERY_P2.matches(&version) || FORGE_MANIFEST_V3_QUERY.matches(&version) {
|
||||
let mut archive_clone = archive.clone();
|
||||
let mut profile = tokio::task::spawn_blocking(move || {
|
||||
let mut install_profile = archive_clone.by_name("install_profile.json")?;
|
||||
|
||||
let mut contents = String::new();
|
||||
install_profile.read_to_string(&mut contents)?;
|
||||
|
||||
Ok::<ForgeInstallerProfileV2, Error>(serde_json::from_str::<ForgeInstallerProfileV2>(&contents)?)
|
||||
}).await??;
|
||||
|
||||
let mut archive_clone = archive.clone();
|
||||
let version_info = tokio::task::spawn_blocking(move || {
|
||||
let mut install_profile = archive_clone.by_name("version.json")?;
|
||||
|
||||
let mut contents = String::new();
|
||||
install_profile.read_to_string(&mut contents)?;
|
||||
|
||||
Ok::<PartialVersionInfo, Error>(serde_json::from_str::<PartialVersionInfo>(&contents)?)
|
||||
}).await??;
|
||||
|
||||
|
||||
let mut libs : Vec<Library> = version_info.libraries.into_iter().chain(profile.libraries.into_iter().map(|x| Library {
|
||||
downloads: x.downloads,
|
||||
extract: x.extract,
|
||||
name: x.name,
|
||||
url: x.url,
|
||||
natives: x.natives,
|
||||
rules: x.rules,
|
||||
checksums: x.checksums,
|
||||
include_in_classpath: false
|
||||
})).collect();
|
||||
|
||||
let mut local_libs : HashMap<String, bytes::Bytes> = HashMap::new();
|
||||
|
||||
for lib in &libs {
|
||||
if lib.downloads.as_ref().and_then(|x| x.artifact.as_ref().map(|x| x.url.is_empty())).unwrap_or(false) {
|
||||
let mut archive_clone = archive.clone();
|
||||
let lib_name_clone = lib.name.clone();
|
||||
|
||||
let lib_bytes = tokio::task::spawn_blocking(move || {
|
||||
let mut lib_file = archive_clone.by_name(&format!("maven/{}", daedalus::get_path_from_artifact(&lib_name_clone)?))?;
|
||||
let mut lib_bytes = Vec::new();
|
||||
lib_file.read_to_end(&mut lib_bytes)?;
|
||||
|
||||
Ok::<bytes::Bytes, Error>(bytes::Bytes::from(lib_bytes))
|
||||
}).await??;
|
||||
|
||||
local_libs.insert(lib.name.clone(), lib_bytes);
|
||||
}
|
||||
}
|
||||
|
||||
Ok::<Library, Error>(lib)
|
||||
})).await?;
|
||||
let path = profile.path.clone();
|
||||
let version = profile.version.clone();
|
||||
|
||||
let elapsed = now.elapsed();
|
||||
info!("Elapsed lib DL: {:.2?}", elapsed);
|
||||
|
||||
let new_profile = PartialVersionInfo {
|
||||
id: profile.version_info.id,
|
||||
inherits_from: profile.install.minecraft,
|
||||
release_time: profile.version_info.release_time,
|
||||
time: profile.version_info.time,
|
||||
main_class: profile.version_info.main_class,
|
||||
minecraft_arguments: profile.version_info.minecraft_arguments.clone(),
|
||||
arguments: profile.version_info.minecraft_arguments.map(|x| [(ArgumentType::Game, x.split(' ').map(|x| Argument::Normal(x.to_string())).collect())].iter().cloned().collect()),
|
||||
libraries: libs,
|
||||
type_: profile.version_info.type_,
|
||||
data: None,
|
||||
processors: None
|
||||
};
|
||||
|
||||
let version_path = format!(
|
||||
"forge/v{}/versions/{}.json",
|
||||
daedalus::modded::CURRENT_FORGE_FORMAT_VERSION,
|
||||
new_profile.id
|
||||
);
|
||||
|
||||
upload_file_to_bucket(
|
||||
version_path.clone(),
|
||||
serde_json::to_vec(&new_profile)?,
|
||||
Some("application/json".to_string()),
|
||||
uploaded_files_mutex.as_ref()
|
||||
).await?;
|
||||
|
||||
return Ok(Some(LoaderVersion {
|
||||
id: loader_version_full,
|
||||
url: format_url(&version_path),
|
||||
stable: false
|
||||
}));
|
||||
} else if FORGE_MANIFEST_V2_QUERY_P1.matches(&version) || FORGE_MANIFEST_V2_QUERY_P2.matches(&version) || FORGE_MANIFEST_V3_QUERY.matches(&version) {
|
||||
let mut archive_clone = archive.clone();
|
||||
let mut profile = tokio::task::spawn_blocking(move || {
|
||||
let mut install_profile = archive_clone.by_name("install_profile.json")?;
|
||||
|
||||
let mut contents = String::new();
|
||||
install_profile.read_to_string(&mut contents)?;
|
||||
|
||||
Ok::<ForgeInstallerProfileV2, Error>(serde_json::from_str::<ForgeInstallerProfileV2>(&contents)?)
|
||||
}).await??;
|
||||
|
||||
let mut archive_clone = archive.clone();
|
||||
let version_info = tokio::task::spawn_blocking(move || {
|
||||
let mut install_profile = archive_clone.by_name("version.json")?;
|
||||
|
||||
let mut contents = String::new();
|
||||
install_profile.read_to_string(&mut contents)?;
|
||||
|
||||
Ok::<PartialVersionInfo, Error>(serde_json::from_str::<PartialVersionInfo>(&contents)?)
|
||||
}).await??;
|
||||
|
||||
|
||||
let mut libs : Vec<daedalus::minecraft::Library> = version_info.libraries.into_iter().chain(profile.libraries.into_iter().map(|x| Library {
|
||||
downloads: x.downloads,
|
||||
extract: x.extract,
|
||||
name: x.name,
|
||||
url: x.url,
|
||||
natives: x.natives,
|
||||
rules: x.rules,
|
||||
checksums: x.checksums,
|
||||
include_in_classpath: false
|
||||
})).collect();
|
||||
|
||||
let mut local_libs : HashMap<String, bytes::Bytes> = HashMap::new();
|
||||
|
||||
for lib in &libs {
|
||||
if lib.downloads.as_ref().and_then(|x| x.artifact.as_ref().map(|x| x.url.is_empty())).unwrap_or(false) {
|
||||
let mut archive_clone = archive.clone();
|
||||
let lib_name_clone = lib.name.clone();
|
||||
|
||||
let lib_bytes = tokio::task::spawn_blocking(move || {
|
||||
let mut lib_file = archive_clone.by_name(&format!("maven/{}", daedalus::get_path_from_artifact(&lib_name_clone)?))?;
|
||||
let mut lib_bytes = Vec::new();
|
||||
lib_file.read_to_end(&mut lib_bytes)?;
|
||||
|
||||
Ok::<bytes::Bytes, Error>(bytes::Bytes::from(lib_bytes))
|
||||
}).await??;
|
||||
|
||||
local_libs.insert(lib.name.clone(), lib_bytes);
|
||||
}
|
||||
}
|
||||
|
||||
let path = profile.path.clone();
|
||||
let version = profile.version.clone();
|
||||
|
||||
for entry in profile.data.values_mut() {
|
||||
if entry.client.starts_with('/') || entry.server.starts_with('/') {
|
||||
macro_rules! read_data {
|
||||
for entry in profile.data.values_mut() {
|
||||
if entry.client.starts_with('/') || entry.server.starts_with('/') {
|
||||
macro_rules! read_data {
|
||||
($value:expr) => {
|
||||
let mut archive_clone = archive.clone();
|
||||
let value_clone = $value.clone();
|
||||
@@ -315,179 +331,149 @@ pub async fn retrieve_data(
|
||||
}
|
||||
}
|
||||
|
||||
if entry.client.starts_with('/') {
|
||||
read_data!(entry.client);
|
||||
}
|
||||
|
||||
// Do we really need to support server installs? Keeping this here
|
||||
// just in case
|
||||
//
|
||||
// if entry.server.starts_with('/') {
|
||||
// read_data!(entry.server);
|
||||
// }
|
||||
}
|
||||
}
|
||||
|
||||
let now = Instant::now();
|
||||
let libs = futures::future::try_join_all(libs.into_iter().map(|mut lib| async {
|
||||
let artifact_path =
|
||||
daedalus::get_path_from_artifact(&lib.name)?;
|
||||
|
||||
{
|
||||
let mut visited_assets = visited_assets.lock().await;
|
||||
|
||||
if visited_assets.contains(&lib.name) {
|
||||
if let Some(ref mut downloads) = lib.downloads {
|
||||
if let Some(ref mut artifact) = downloads.artifact {
|
||||
artifact.url = format_url(&format!("maven/{}", artifact_path));
|
||||
}
|
||||
} else if lib.url.is_some() {
|
||||
lib.url = Some(format_url("maven/"));
|
||||
if entry.client.starts_with('/') {
|
||||
read_data!(entry.client);
|
||||
}
|
||||
|
||||
return Ok::<Library, Error>(lib);
|
||||
} else {
|
||||
visited_assets.push(lib.name.clone())
|
||||
if entry.server.starts_with('/') {
|
||||
read_data!(entry.server);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
let artifact_bytes = if let Some(ref mut downloads) = lib.downloads {
|
||||
if let Some(ref mut artifact) = downloads.artifact {
|
||||
let res = if artifact.url.is_empty() {
|
||||
let now = Instant::now();
|
||||
let libs = futures::future::try_join_all(libs.into_iter().map(|mut lib| async {
|
||||
let artifact_path =
|
||||
daedalus::get_path_from_artifact(&lib.name)?;
|
||||
|
||||
{
|
||||
let mut visited_assets = visited_assets.lock().await;
|
||||
|
||||
if visited_assets.contains(&lib.name) {
|
||||
if let Some(ref mut downloads) = lib.downloads {
|
||||
if let Some(ref mut artifact) = downloads.artifact {
|
||||
artifact.url = format_url(&format!("maven/{}", artifact_path));
|
||||
}
|
||||
} else if lib.url.is_some() {
|
||||
lib.url = Some(format_url("maven/"));
|
||||
}
|
||||
|
||||
return Ok::<Library, Error>(lib);
|
||||
} else {
|
||||
visited_assets.push(lib.name.clone())
|
||||
}
|
||||
}
|
||||
|
||||
let artifact_bytes = if let Some(ref mut downloads) = lib.downloads {
|
||||
if let Some(ref mut artifact) = downloads.artifact {
|
||||
let res = if artifact.url.is_empty() {
|
||||
local_libs.get(&lib.name).cloned()
|
||||
} else {
|
||||
Some(download_file(
|
||||
&artifact.url,
|
||||
Some(&*artifact.sha1),
|
||||
semaphore.clone(),
|
||||
)
|
||||
.await?)
|
||||
};
|
||||
|
||||
if res.is_some() {
|
||||
artifact.url = format_url(&format!("maven/{}", artifact_path));
|
||||
}
|
||||
|
||||
res
|
||||
} else { None }
|
||||
} else if let Some(ref mut url) = lib.url {
|
||||
let res = if url.is_empty() {
|
||||
local_libs.get(&lib.name).cloned()
|
||||
} else {
|
||||
Some(daedalus::download_file(
|
||||
&artifact.url,
|
||||
Some(&*artifact.sha1),
|
||||
Some(download_file(
|
||||
url,
|
||||
None,
|
||||
semaphore.clone(),
|
||||
)
|
||||
.await?)
|
||||
};
|
||||
|
||||
if res.is_some() {
|
||||
artifact.url = format_url(&format!("maven/{}", artifact_path));
|
||||
lib.url = Some(format_url("maven/"));
|
||||
}
|
||||
|
||||
res
|
||||
} else { None }
|
||||
} else if let Some(ref mut url) = lib.url {
|
||||
let res = if url.is_empty() {
|
||||
local_libs.get(&lib.name).cloned()
|
||||
} else {
|
||||
Some(daedalus::download_file(
|
||||
url,
|
||||
None,
|
||||
)
|
||||
.await?)
|
||||
};
|
||||
} else { None };
|
||||
|
||||
if res.is_some() {
|
||||
lib.url = Some(format_url("maven/"));
|
||||
if let Some(bytes) = artifact_bytes {
|
||||
upload_file_to_bucket(
|
||||
format!("{}/{}", "maven", artifact_path),
|
||||
bytes.to_vec(),
|
||||
Some("application/java-archive".to_string()),
|
||||
uploaded_files_mutex.as_ref(),
|
||||
semaphore.clone(),
|
||||
).await?;
|
||||
}
|
||||
|
||||
res
|
||||
} else { None };
|
||||
Ok::<Library, Error>(lib)
|
||||
})).await?;
|
||||
|
||||
if let Some(bytes) = artifact_bytes {
|
||||
upload_file_to_bucket(
|
||||
format!("{}/{}", "maven", artifact_path),
|
||||
bytes.to_vec(),
|
||||
Some("application/java-archive".to_string()),
|
||||
uploaded_files_mutex.as_ref()
|
||||
).await?;
|
||||
}
|
||||
let elapsed = now.elapsed();
|
||||
info!("Elapsed lib DL: {:.2?}", elapsed);
|
||||
|
||||
Ok::<Library, Error>(lib)
|
||||
})).await?;
|
||||
let new_profile = PartialVersionInfo {
|
||||
id: version_info.id,
|
||||
inherits_from: version_info.inherits_from,
|
||||
release_time: version_info.release_time,
|
||||
time: version_info.time,
|
||||
main_class: version_info.main_class,
|
||||
minecraft_arguments: version_info.minecraft_arguments,
|
||||
arguments: version_info.arguments,
|
||||
libraries: libs,
|
||||
type_: version_info.type_,
|
||||
data: Some(profile.data),
|
||||
processors: Some(profile.processors),
|
||||
};
|
||||
|
||||
let elapsed = now.elapsed();
|
||||
info!("Elapsed lib DL: {:.2?}", elapsed);
|
||||
let version_path = format!(
|
||||
"forge/v{}/versions/{}.json",
|
||||
daedalus::modded::CURRENT_FORGE_FORMAT_VERSION,
|
||||
new_profile.id
|
||||
);
|
||||
|
||||
let new_profile = PartialVersionInfo {
|
||||
id: version_info.id,
|
||||
inherits_from: version_info.inherits_from,
|
||||
release_time: version_info.release_time,
|
||||
time: version_info.time,
|
||||
main_class: version_info.main_class,
|
||||
minecraft_arguments: version_info.minecraft_arguments,
|
||||
arguments: version_info.arguments,
|
||||
libraries: libs,
|
||||
type_: version_info.type_,
|
||||
data: Some(profile.data),
|
||||
processors: Some(profile.processors),
|
||||
};
|
||||
upload_file_to_bucket(
|
||||
version_path.clone(),
|
||||
serde_json::to_vec(&new_profile)?,
|
||||
Some("application/json".to_string()),
|
||||
uploaded_files_mutex.as_ref(),
|
||||
semaphore.clone(),
|
||||
).await?;
|
||||
|
||||
let version_path = format!(
|
||||
"forge/v{}/versions/{}.json",
|
||||
daedalus::modded::CURRENT_FORGE_FORMAT_VERSION,
|
||||
new_profile.id
|
||||
);
|
||||
|
||||
upload_file_to_bucket(
|
||||
version_path.clone(),
|
||||
serde_json::to_vec(&new_profile)?,
|
||||
Some("application/json".to_string()),
|
||||
uploaded_files_mutex.as_ref()
|
||||
).await?;
|
||||
|
||||
return Ok(Some(LoaderVersion {
|
||||
id: loader_version_full,
|
||||
url: format_url(&version_path),
|
||||
stable: false
|
||||
}));
|
||||
return Ok(Some(LoaderVersion {
|
||||
id: loader_version_full,
|
||||
url: format_url(&version_path),
|
||||
stable: false
|
||||
}));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(None)
|
||||
}.await
|
||||
}).into_iter().peekable()/*.into_iter().flatten().collect()*/;
|
||||
Ok(None)
|
||||
}.await
|
||||
});
|
||||
|
||||
let mut chunk_index = 0;
|
||||
while loaders_futures.peek().is_some() {
|
||||
info!("Loader Chunk {} Start", chunk_index);
|
||||
let now = Instant::now();
|
||||
|
||||
let chunk: Vec<_> = loaders_futures.by_ref().take(10).collect();
|
||||
|
||||
let res = futures::future::try_join_all(chunk).await?;
|
||||
loaders_versions.extend(res.into_iter().flatten());
|
||||
|
||||
tokio::time::sleep(Duration::from_secs(1)).await;
|
||||
|
||||
chunk_index += 1;
|
||||
|
||||
let elapsed = now.elapsed();
|
||||
info!("Loader Chunk {} Elapsed: {:.2?}", chunk_index, elapsed);
|
||||
futures::future::try_join_all(loaders_futures).await?;
|
||||
}
|
||||
}
|
||||
|
||||
versions.lock().await.push(daedalus::modded::Version {
|
||||
id: minecraft_version,
|
||||
loaders: loaders_versions
|
||||
versions.lock().await.push(daedalus::modded::Version {
|
||||
id: minecraft_version,
|
||||
stable: true,
|
||||
loaders: loaders_versions
|
||||
});
|
||||
|
||||
Ok::<(), Error>(())
|
||||
});
|
||||
|
||||
Ok::<(), Error>(())
|
||||
});
|
||||
}
|
||||
|
||||
{
|
||||
let mut versions_peek = version_futures.into_iter().peekable();
|
||||
let mut chunk_index = 0;
|
||||
while versions_peek.peek().is_some() {
|
||||
info!("Chunk {} Start", chunk_index);
|
||||
let now = Instant::now();
|
||||
|
||||
let chunk: Vec<_> = versions_peek.by_ref().take(1).collect();
|
||||
futures::future::try_join_all(chunk).await?;
|
||||
|
||||
tokio::time::sleep(Duration::from_secs(1)).await;
|
||||
|
||||
chunk_index += 1;
|
||||
|
||||
let elapsed = now.elapsed();
|
||||
info!("Chunk {} Elapsed: {:.2?}", chunk_index, elapsed);
|
||||
}
|
||||
}
|
||||
|
||||
futures::future::try_join_all(version_futures).await?;
|
||||
|
||||
if let Ok(versions) = Arc::try_unwrap(versions) {
|
||||
let mut versions = versions.into_inner();
|
||||
|
||||
@@ -495,13 +481,17 @@ pub async fn retrieve_data(
|
||||
minecraft_versions
|
||||
.versions
|
||||
.iter()
|
||||
.position(|z| x.id == z.id)
|
||||
.position(|z| {
|
||||
x.id.replace("1.7.10_pre4", "1.7.10-pre4") == z.id
|
||||
})
|
||||
.unwrap_or_default()
|
||||
.cmp(
|
||||
&minecraft_versions
|
||||
.versions
|
||||
.iter()
|
||||
.position(|z| y.id == z.id)
|
||||
.position(|z| {
|
||||
y.id.replace("1.7.10_pre4", "1.7.10-pre4") == z.id
|
||||
})
|
||||
.unwrap_or_default(),
|
||||
)
|
||||
});
|
||||
@@ -524,6 +514,8 @@ pub async fn retrieve_data(
|
||||
}
|
||||
}
|
||||
|
||||
println!("{:?}", versions);
|
||||
|
||||
upload_file_to_bucket(
|
||||
format!(
|
||||
"forge/v{}/manifest.json",
|
||||
@@ -534,6 +526,7 @@ pub async fn retrieve_data(
|
||||
})?,
|
||||
Some("application/json".to_string()),
|
||||
uploaded_files_mutex.as_ref(),
|
||||
semaphore,
|
||||
)
|
||||
.await?;
|
||||
}
|
||||
@@ -554,9 +547,15 @@ const DEFAULT_MAVEN_METADATA_URL: &str =
|
||||
/// the specified Minecraft version
|
||||
pub async fn fetch_maven_metadata(
|
||||
url: Option<&str>,
|
||||
semaphore: Arc<Semaphore>,
|
||||
) -> Result<HashMap<String, Vec<String>>, Error> {
|
||||
Ok(serde_json::from_slice(
|
||||
&download_file(url.unwrap_or(DEFAULT_MAVEN_METADATA_URL), None).await?,
|
||||
&download_file(
|
||||
url.unwrap_or(DEFAULT_MAVEN_METADATA_URL),
|
||||
None,
|
||||
semaphore,
|
||||
)
|
||||
.await?,
|
||||
)?)
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user