Implement zip/tar.gz sdist metadata parsing

Author: messense

Cargo.toml

@@ -12,6 +12,8 @@ repository = "https://github.com/messense/python-pkginfo-rs"
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
 
 [dependencies]
+flate2 = "1.0.20"
 fs-err = "2.6.0"
 mailparse = "0.13.4"
+tar = "0.4.35"
 zip = "0.5.12"

src/distribution.rs

@@ -1,6 +1,7 @@
 use std::io::{BufReader, Read};
 use std::path::Path;
 
+use flate2::read::GzDecoder;
 use zip::ZipArchive;
 
 use crate::{Error, Metadata};
@@ -66,7 +67,32 @@ impl Distribution {
     }
 
     fn parse_sdist(path: &Path, sdist_type: SDistType) -> Result<Metadata, Error> {
-        todo!()
+        match sdist_type {
+            SDistType::Zip => Self::parse_zip(path, "PKG-INFO"),
+            SDistType::TarGz => {
+                let mut reader =
+                    tar::Archive::new(GzDecoder::new(BufReader::new(fs_err::File::open(path)?)));
+                let metadata_file = reader
+                    .entries()?
+                    .map(|entry| -> Result<_, Error> {
+                        let entry = entry?;
+                        if entry.path()?.ends_with("PKG-INFO") {
+                            Ok(Some(entry))
+                        } else {
+                            Ok(None)
+                        }
+                    })
+                    .find_map(|x| x.transpose());
+                if let Some(metadata_file) = metadata_file {
+                    let mut entry = metadata_file?;
+                    let mut buf = Vec::new();
+                    entry.read_to_end(&mut buf)?;
+                    Metadata::parse(&buf)
+                } else {
+                    Err(Error::MetadataNotFound)
+                }
+            }
+        }
     }
 
     fn parse_egg(path: &Path) -> Result<Metadata, Error> {
@@ -92,6 +118,14 @@ impl Distribution {
                 archive.by_name(metadata_file)?.read_to_end(&mut buf)?;
                 Metadata::parse(&buf)
             }
+            [file1, file2]
+                if file1.ends_with(".egg-info/PKG-INFO")
+                    || file2.ends_with(".egg-info/PKG-INFO") =>
+            {
+                let mut buf = Vec::new();
+                archive.by_name(file1)?.read_to_end(&mut buf)?;
+                Metadata::parse(&buf)
+            }
             _ => Err(Error::MultipleMetadataFiles(metadata_files)),
         }
     }

tests/fixtures/build-0.4.0.tar.gz

@@ -21,3 +21,25 @@ fn test_parse_egg() {
     assert!(metadata.home_page.is_none());
     assert!(metadata.download_url.is_none());
 }
+
+#[test]
+fn test_parse_sdist_zip() {
+    let dist = Distribution::new("tests/fixtures/build-0.4.0.zip").unwrap();
+    assert_eq!(dist.r#type(), DistributionType::SDist);
+    let metadata = dist.metadata();
+    assert_eq!(metadata.metadata_version, "2.1");
+    assert_eq!(metadata.name, "build");
+    assert!(metadata.home_page.is_none());
+    assert!(metadata.download_url.is_none());
+}
+
+#[test]
+fn test_parse_sdist_tar_gz() {
+    let dist = Distribution::new("tests/fixtures/build-0.4.0.tar.gz").unwrap();
+    assert_eq!(dist.r#type(), DistributionType::SDist);
+    let metadata = dist.metadata();
+    assert_eq!(metadata.metadata_version, "2.1");
+    assert_eq!(metadata.name, "build");
+    assert!(metadata.home_page.is_none());
+    assert!(metadata.download_url.is_none());
+}