Add support for SOURCE_DATE_EPOCH

Setting `SOURCE_DATE_EPOCH` in the environment will set the time stamps
of all files and directories in the produced wheels to the specified
value.

src/auditwheel/tools.py

@@ -2,7 +2,8 @@
 import os
 import subprocess
 import zipfile
-from typing import Any, Iterable, List
+from datetime import datetime, timezone
+from typing import Any, Iterable, List, Optional
 
 
 def unique_by_index(sequence: Iterable[Any]) -> List[Any]:
@@ -49,7 +50,7 @@ def zip2dir(zip_fname: str, out_dir: str) -> None:
                 os.chmod(extracted_path, attr)
 
 
-def dir2zip(in_dir: str, zip_fname: str) -> None:
+def dir2zip(in_dir: str, zip_fname: str, date_time: Optional[datetime] = None) -> None:
     """Make a zip file `zip_fname` with contents of directory `in_dir`
 
     The recorded filenames are relative to `in_dir`, so doing a standard zip
@@ -62,17 +63,28 @@ def dir2zip(in_dir: str, zip_fname: str) -> None:
         Directory path containing files to go in the zip archive
     zip_fname : str
         Filename of zip archive to write
+    date_time : Optional[datetime]
+        Time stamp to set on each file in the archive
     """
+    if date_time is None:
+        st = os.stat(in_dir)
+        date_time = datetime.fromtimestamp(st.st_mtime, tz=timezone.utc)
+    date_time_args = date_time.timetuple()[:6]
     with zipfile.ZipFile(zip_fname, "w", compression=zipfile.ZIP_DEFLATED) as z:
         for root, dirs, files in os.walk(in_dir):
             for dir in dirs:
                 dname = os.path.join(root, dir)
-                out_dname = os.path.relpath(dname, in_dir)
-                z.write(dname, out_dname)
+                out_dname = os.path.relpath(dname, in_dir) + "/"
+                zinfo = zipfile.ZipInfo(out_dname, date_time=date_time_args)
+                zinfo.external_attr = os.stat(dname).st_mode << 16
+                z.writestr(zinfo, "")
             for file in files:
                 fname = os.path.join(root, file)
                 out_fname = os.path.relpath(fname, in_dir)
-                z.write(fname, out_fname)
+                zinfo = zipfile.ZipInfo(out_fname, date_time=date_time_args)
+                zinfo.external_attr = os.stat(fname).st_mode << 16
+                with open(fname, "rb") as fp:
+                    z.writestr(zinfo, fp.read())
 
 
 def tarbz2todir(tarbz2_fname: str, out_dir: str) -> None:

src/auditwheel/wheeltools.py

@@ -9,6 +9,7 @@
 import logging
 import os
 from base64 import urlsafe_b64encode
+from datetime import datetime, timezone
 from itertools import product
 from os.path import abspath, basename, dirname, exists
 from os.path import join as pjoin
@@ -127,7 +128,11 @@ def __exit__(
     ) -> None:
         if self.out_wheel is not None:
             rewrite_record(self.name)
-            dir2zip(self.name, self.out_wheel)
+            date_time = None
+            timestamp = os.environ.get("SOURCE_DATE_EPOCH")
+            if timestamp:
+                date_time = datetime.fromtimestamp(int(timestamp), tz=timezone.utc)
+            dir2zip(self.name, self.out_wheel, date_time)
         return super().__exit__(exc, value, tb)
 
 

tests/integration/test_bundled_wheels.py

@@ -1,8 +1,15 @@
 import platform
+import subprocess
+import sys
+import zipfile
+from argparse import Namespace
+from datetime import datetime, timezone
 from pathlib import Path
+from unittest.mock import Mock
 
 import pytest
 
+from auditwheel import main_repair
 from auditwheel.wheel_abi import analyze_wheel_abi
 
 HERE = Path(__file__).parent.resolve()
@@ -30,3 +37,46 @@ def test_analyze_wheel_abi_pyfpe():
     assert (
         winfo.pyfpe_tag == "linux_x86_64"
     )  # but for having the pyfpe reference, it gets just linux
+
+
+@pytest.mark.skipif(platform.machine() != "x86_64", reason="only checked on x86_64")
+def test_wheel_source_date_epoch(tmp_path, monkeypatch):
+    wheel_build_path = tmp_path / "wheel"
+    subprocess.run(
+        [
+            sys.executable,
+            "-m",
+            "pip",
+            "wheel",
+            "--no-deps",
+            "-w",
+            wheel_build_path,
+            HERE / "sample_extension",
+        ],
+        check=True,
+    )
+
+    wheel_path, *_ = list(wheel_build_path.glob("*.whl"))
+    wheel_output_path = tmp_path / "out"
+    args = Namespace(
+        LIB_SDIR=".libs",
+        ONLY_PLAT=False,
+        PLAT="manylinux_2_5_x86_64",
+        STRIP=False,
+        UPDATE_TAGS=True,
+        WHEEL_DIR=str(wheel_output_path),
+        WHEEL_FILE=str(wheel_path),
+        cmd="repair",
+        func=Mock(),
+        prog="auditwheel",
+        verbose=1,
+    )
+    monkeypatch.setenv("SOURCE_DATE_EPOCH", "650203200")
+    main_repair.execute(args, Mock())
+
+    output_wheel, *_ = list(wheel_output_path.glob("*.whl"))
+    with zipfile.ZipFile(output_wheel) as wheel_file:
+        for file in wheel_file.infolist():
+            assert (
+                datetime(*file.date_time, tzinfo=timezone.utc).timestamp() == 650203200
+            )