oiupoyt/OneLauncher-Cracked
1
1
Fork 0
Code Issues Pull requests Projects Releases 2 Packages Wiki Activity

LAUNCHER-9 feat: Send notification when downloading a package

Browse source
This commit is contained in:
LynithDev 2024-12-27 12:09:19 +01:00
parent 3770d7b896
commit d4a33ad348
1 changed files with 39 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
packages/core/src/api/cluster/content/package.rs
View file

@ -3,11 +3,12 @@
use crate::data::{Loader, ManagedPackage, ManagedVersion, PackageType}; use crate::data::{Loader, ManagedPackage, ManagedVersion, PackageType};
use crate::prelude::PackagePath; use crate::prelude::PackagePath;
use crate::processor::Cluster; use crate::processor::Cluster;
use crate::proxy::send::send_internet; use crate::proxy::send::{init_ingress, send_ingress, send_internet};
use crate::store::{ClusterPath, ManagedVersionFile, Package, PackageMetadata}; use crate::store::{ClusterPath, ManagedVersionFile, Package, PackageMetadata};
use crate::utils::http; use crate::utils::http;
use crate::{Result, State}; use crate::{Result, State};
use onelauncher_utils::io; use onelauncher_utils::io;
use reqwest::Method;
// TODO: Implement proper error handling // TODO: Implement proper error handling
/// Find a managed version using filters /// Find a managed version using filters
@ -85,14 +86,9 @@ pub async fn download_package(
let file = managed_version let file = managed_version
.get_primary_file() .get_primary_file()
.ok_or_else(|| anyhow::anyhow!("no primary file found"))?; .ok_or_else(|| anyhow::anyhow!("no primary file found"))?;
tracing::info!(
"downloading file '{}' version '{}'",
file.file_name,
managed_version.name
);
let package_path = download_file(file, &package.package_type, cluster).await?; let package_path = download_file(package, &managed_version, file, &package.package_type, cluster).await?;
let sha1 = file.hashes.get("sha1").unwrap_or(&"".to_string()).clone(); let sha1 = file.hashes.get("sha1").unwrap_or(&String::new()).clone();
let package = Package { let package = Package {
file_name: file.file_name.clone(), file_name: file.file_name.clone(),
@ -105,28 +101,54 @@ pub async fn download_package(
} }
/// Download a file to a cluster from a managed version file. /// Download a file to a cluster from a managed version file.
#[tracing::instrument(skip(file, cluster))] #[tracing::instrument(skip_all)]
async fn download_file( async fn download_file(
package: &ManagedPackage,
version: &ManagedVersion,
file: &ManagedVersionFile, file: &ManagedVersionFile,
package_type: &PackageType, package_type: &PackageType,
cluster: &Cluster, cluster: &Cluster,
) -> Result<PackagePath> { ) -> Result<PackagePath> {
// TODO: Implement hash checking // TODO: Implement hash checking
let cluster_path = &cluster.get_full_path().await?;
let path = PackagePath::new( let path = PackagePath::new(
&cluster &cluster_path
.get_full_path()
.await?
.join(package_type.get_folder()) .join(package_type.get_folder())
.join(&file.file_name), .join(&file.file_name),
); );
let ingress_id = init_ingress(
crate::IngressType::DownloadPackage {
cluster_path: cluster_path.to_owned(),
package_name: package.title.clone(),
icon: None,
package_id: Some(package.id.clone()),
package_version: Some(version.id.clone()),
}, 100.0, "downloading package").await?;
tracing::info!(
"downloading file '{}' version '{}'",
file.file_name,
version.name
);
let state = State::get().await?; let state = State::get().await?;
let bytes = http::fetch( let bytes = http::fetch_advanced(
Method::GET,
&file.url, &file.url,
file.hashes.get("sha1").map(String::as_str), file.hashes.get("sha1").map(String::as_str),
None,
None,
Some((&ingress_id, 90.0)),
&state.fetch_semaphore, &state.fetch_semaphore,
) )
.await?; .await?;
if let Err(err) = send_ingress(&ingress_id, 5.0, Some("saving file")).await {
tracing::error!("{err}");
}
if let Err(err) = http::write(&path.0, &bytes, &state.io_semaphore).await { if let Err(err) = http::write(&path.0, &bytes, &state.io_semaphore).await {
tracing::error!("failed to write file to cluster: {err}"); tracing::error!("failed to write file to cluster: {err}");
if path.0.exists() { if path.0.exists() {
@ -136,6 +158,10 @@ async fn download_file(
return Err(err); return Err(err);
}; };
if let Err(err) = send_ingress(&ingress_id, 100.0, Some(format!("downloaded package {}", package.title).as_str())).await {
tracing::error!("{err}");
}
drop(state); drop(state);
Ok(path) Ok(path)