correct download

This commit is contained in:
nora 2022-05-15 18:26:05 +02:00
parent f9c5a18720
commit 89168e9a1a
4 changed files with 65 additions and 27 deletions

12
Cargo.lock generated
View file

@ -742,9 +742,9 @@ dependencies = [
[[package]] [[package]]
name = "os_str_bytes" name = "os_str_bytes"
version = "6.0.0" version = "6.0.1"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "8e22443d1643a904602595ba1cd8f7d896afe56d26712531c5ff73a15b2fbf64" checksum = "029d8d0b2f198229de29dca79676f2738ff952edf3fde542eb8bf94d8c21b435"
[[package]] [[package]]
name = "owo-colors" name = "owo-colors"
@ -947,9 +947,9 @@ checksum = "7ef03e0a2b150c7a90d01faf6254c9c48a41e95fb2a8c2ac1c6f0d2b9aefc342"
[[package]] [[package]]
name = "rustls" name = "rustls"
version = "0.20.4" version = "0.20.5"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "4fbfeb8d0ddb84706bc597a5574ab8912817c52a397f819e5b614e2265206921" checksum = "a024a432ae760ab3bff924ad91ce1cfa52cb57ed16e1ef32d0d249cfee1a6c13"
dependencies = [ dependencies = [
"log", "log",
"ring", "ring",
@ -1265,9 +1265,9 @@ dependencies = [
[[package]] [[package]]
name = "tokio-util" name = "tokio-util"
version = "0.7.1" version = "0.7.2"
source = "registry+https://github.com/rust-lang/crates.io-index" source = "registry+https://github.com/rust-lang/crates.io-index"
checksum = "0edfdeb067411dba2044da6d1cb2df793dd35add7888d73c16e3381ded401764" checksum = "f988a1a1adc2fb21f9c12aa96441da33a1728193ae0b95d2be22dbd17fcb4e5c"
dependencies = [ dependencies = [
"bytes", "bytes",
"futures-core", "futures-core",

View file

@ -8,10 +8,11 @@ use indexmap::IndexMap;
use node_semver::Version; use node_semver::Version;
use reqwest::Client; use reqwest::Client;
use serde::Deserialize; use serde::Deserialize;
use tar::Archive; use tokio::task::spawn_blocking;
use tracing::{debug, info}; use tracing::{debug, info};
use crate::{ use crate::{
create_dir_if_not_exists,
manifest::{Bugs, Human, Person, Repository}, manifest::{Bugs, Human, Person, Repository},
PackageJson, WrapErr, PackageJson, WrapErr,
}; };
@ -103,6 +104,10 @@ impl NpmClient {
#[tracing::instrument(skip(self))] #[tracing::instrument(skip(self))]
pub async fn download_package(&self, name: &str, url: &str) -> Result<()> { pub async fn download_package(&self, name: &str, url: &str) -> Result<()> {
let module = Path::new("node_modules").join(name);
create_dir_if_not_exists(&module).await?;
let response = self let response = self
.reqwest .reqwest
.get(url) .get(url)
@ -111,11 +116,29 @@ impl NpmClient {
.wrap_err("getting response")?; .wrap_err("getting response")?;
let tarball = response.bytes().await.wrap_err("fetching body")?; let tarball = response.bytes().await.wrap_err("fetching body")?;
spawn_blocking(move || -> Result<()> {
let tar = flate2::read::GzDecoder::new(tarball.reader()); let tar = flate2::read::GzDecoder::new(tarball.reader());
let mut archive = tar::Archive::new(tar); let mut archive = tar::Archive::new(tar);
archive
.unpack(Path::new("node_modules").join(name)) for entry in archive.entries()? {
.wrap_err("unpack tarball")?; let mut entry = entry?;
let path = entry.path()?;
let path = path
.strip_prefix("package")
.wrap_err("file name must start with package/")?;
let path = module.join(path);
info!(?path, "Unpacking file");
entry
.unpack(&path)
.wrap_err(format!("unpacking file {}", path.display()))?;
}
Ok(())
})
.await??;
info!("successfully downloaded package"); info!("successfully downloaded package");

19
src/helper.rs Normal file
View file

@ -0,0 +1,19 @@
use std::{io, path::Path};
use color_eyre::{eyre::bail, Result};
use tokio::fs;
use crate::WrapErr;
pub async fn create_dir_if_not_exists(path: impl AsRef<Path>) -> Result<()> {
match fs::metadata(&path).await {
Ok(_) => {}
Err(e) if e.kind() == io::ErrorKind::NotFound => {
fs::create_dir(&path)
.await
.wrap_err("creating node_modules directory")?;
}
Err(e) => bail!(e),
}
Ok(())
}

View file

@ -1,15 +1,15 @@
use std::{fs, io}; use color_eyre::{eyre::WrapErr, Result};
use tokio::fs;
use color_eyre::{
eyre::{bail, WrapErr},
Result,
};
use tracing::metadata::LevelFilter; use tracing::metadata::LevelFilter;
use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Registry}; use tracing_subscriber::{layer::SubscriberExt, util::SubscriberInitExt, EnvFilter, Registry};
use crate::{download::NpmClient, manifest::PackageJson, resolve::ResolveContext}; use crate::{
download::NpmClient, helper::create_dir_if_not_exists, manifest::PackageJson,
resolve::ResolveContext,
};
mod download; mod download;
mod helper;
mod manifest; mod manifest;
mod resolve; mod resolve;
@ -19,19 +19,15 @@ async fn main() -> Result<()> {
setup_tracing()?; setup_tracing()?;
let manifest = "package.json"; let manifest = "package.json";
let manifest = fs::read_to_string(manifest).wrap_err("Opening package.json file")?; let manifest = fs::read_to_string(manifest)
.await
.wrap_err("Opening package.json file")?;
let manifest: PackageJson = serde_json::from_str(&manifest)?; let manifest: PackageJson = serde_json::from_str(&manifest)?;
let resolve_context = ResolveContext::new(); let resolve_context = ResolveContext::new();
match fs::metadata("node_modules") { create_dir_if_not_exists("node_modules").await?;
Ok(_) => {}
Err(e) if e.kind() == io::ErrorKind::NotFound => {
fs::create_dir("node_modules").wrap_err("creating node_modules directory")?;
}
Err(e) => bail!(e),
}
for (name, requested_version) in &manifest.dependencies.unwrap() { for (name, requested_version) in &manifest.dependencies.unwrap() {
resolve_context resolve_context