mirror of https://github.com/astral-sh/uv
Remove unused `seek` methods (#3526)
This commit is contained in:
parent
3b728c16cc
commit
2c3a6796aa
|
|
@ -3,7 +3,6 @@ pub use sync::*;
|
|||
|
||||
mod error;
|
||||
pub mod hash;
|
||||
pub mod seek;
|
||||
pub mod stream;
|
||||
mod sync;
|
||||
mod tar;
|
||||
|
|
|
|||
|
|
@ -1,145 +0,0 @@
|
|||
use std::path::Path;
|
||||
|
||||
use rustc_hash::FxHashSet;
|
||||
use tokio_util::compat::FuturesAsyncReadCompatExt;
|
||||
use tokio_util::compat::TokioAsyncReadCompatExt;
|
||||
|
||||
use crate::Error;
|
||||
|
||||
/// Unzip a `.zip` archive into the target directory, requiring `Seek`.
|
||||
///
|
||||
/// This is useful for unzipping files asynchronously that already exist on disk.
|
||||
pub async fn unzip<R: tokio::io::AsyncRead + tokio::io::AsyncSeek + Unpin>(
|
||||
reader: R,
|
||||
target: impl AsRef<Path>,
|
||||
) -> Result<(), Error> {
|
||||
let target = target.as_ref();
|
||||
let mut reader = futures::io::BufReader::new(reader.compat());
|
||||
let mut zip = async_zip::base::read::seek::ZipFileReader::new(&mut reader).await?;
|
||||
|
||||
let mut directories = FxHashSet::default();
|
||||
|
||||
for index in 0..zip.file().entries().len() {
|
||||
let reader = zip.reader_with_entry(index).await?;
|
||||
|
||||
// Construct the (expected) path to the file on-disk.
|
||||
let path = reader.entry().filename().as_str()?;
|
||||
let path = target.join(path);
|
||||
let is_dir = reader.entry().dir()?;
|
||||
|
||||
// Either create the directory or write the file to disk.
|
||||
if is_dir {
|
||||
if directories.insert(path.clone()) {
|
||||
fs_err::tokio::create_dir_all(path).await?;
|
||||
}
|
||||
} else {
|
||||
if let Some(parent) = path.parent() {
|
||||
if directories.insert(parent.to_path_buf()) {
|
||||
fs_err::tokio::create_dir_all(parent).await?;
|
||||
}
|
||||
}
|
||||
|
||||
// Copy the mode.
|
||||
#[cfg(unix)]
|
||||
let mode = reader.entry().unix_permissions();
|
||||
|
||||
// Copy the file contents.
|
||||
let file = fs_err::tokio::File::create(&path).await?;
|
||||
let mut writer = if let Ok(size) = usize::try_from(reader.entry().uncompressed_size()) {
|
||||
tokio::io::BufWriter::with_capacity(size, file)
|
||||
} else {
|
||||
tokio::io::BufWriter::new(file)
|
||||
};
|
||||
tokio::io::copy(&mut reader.compat(), &mut writer).await?;
|
||||
|
||||
// See `uv_extract::stream::unzip`.
|
||||
#[cfg(unix)]
|
||||
{
|
||||
use std::fs::Permissions;
|
||||
use std::os::unix::fs::PermissionsExt;
|
||||
|
||||
let Some(mode) = mode else {
|
||||
continue;
|
||||
};
|
||||
|
||||
// The executable bit is the only permission we preserve, otherwise we use the OS defaults.
|
||||
// https://github.com/pypa/pip/blob/3898741e29b7279e7bffe044ecfbe20f6a438b1e/src/pip/_internal/utils/unpacking.py#L88-L100
|
||||
let has_any_executable_bit = mode & 0o111;
|
||||
if has_any_executable_bit != 0 {
|
||||
let permissions = fs_err::tokio::metadata(&path).await?.permissions();
|
||||
fs_err::tokio::set_permissions(
|
||||
&path,
|
||||
Permissions::from_mode(permissions.mode() | 0o111),
|
||||
)
|
||||
.await?;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
/// Unzip a `.zip`, `.tar.gz`, or `.tar.bz2` archive into the target directory, requiring `Seek`.
|
||||
pub async fn archive<R: tokio::io::AsyncRead + tokio::io::AsyncSeek + Unpin>(
|
||||
reader: R,
|
||||
source: impl AsRef<Path>,
|
||||
target: impl AsRef<Path>,
|
||||
) -> Result<(), Error> {
|
||||
// `.zip`
|
||||
if source
|
||||
.as_ref()
|
||||
.extension()
|
||||
.is_some_and(|ext| ext.eq_ignore_ascii_case("zip"))
|
||||
{
|
||||
unzip(reader, target).await?;
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// `.tar.gz`
|
||||
if source
|
||||
.as_ref()
|
||||
.extension()
|
||||
.is_some_and(|ext| ext.eq_ignore_ascii_case("gz"))
|
||||
&& source.as_ref().file_stem().is_some_and(|stem| {
|
||||
Path::new(stem)
|
||||
.extension()
|
||||
.is_some_and(|ext| ext.eq_ignore_ascii_case("tar"))
|
||||
})
|
||||
{
|
||||
crate::stream::untar_gz(reader, target).await?;
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// `.tar.bz2`
|
||||
if source
|
||||
.as_ref()
|
||||
.extension()
|
||||
.is_some_and(|ext| ext.eq_ignore_ascii_case("bz2"))
|
||||
&& source.as_ref().file_stem().is_some_and(|stem| {
|
||||
Path::new(stem)
|
||||
.extension()
|
||||
.is_some_and(|ext| ext.eq_ignore_ascii_case("tar"))
|
||||
})
|
||||
{
|
||||
crate::stream::untar_bz2(reader, target).await?;
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
// `.tar.zst`
|
||||
if source
|
||||
.as_ref()
|
||||
.extension()
|
||||
.is_some_and(|ext| ext.eq_ignore_ascii_case("zst"))
|
||||
&& source.as_ref().file_stem().is_some_and(|stem| {
|
||||
Path::new(stem)
|
||||
.extension()
|
||||
.is_some_and(|ext| ext.eq_ignore_ascii_case("tar"))
|
||||
})
|
||||
{
|
||||
crate::stream::untar_zst(reader, target).await?;
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
Err(Error::UnsupportedArchive(source.as_ref().to_path_buf()))
|
||||
}
|
||||
Loading…
Reference in New Issue