Set the location of the zip via environment variable

.cargo/config.toml

@@ -8,3 +8,9 @@ benchmark = "bench -p ruff_benchmark --bench linter --bench formatter --"
 # See: https://github.com/astral-sh/ruff/issues/11503
 [target.'cfg(all(target_env="msvc", target_os = "windows"))']
 rustflags = ["-C", "target-feature=+crt-static"]
+
+[env]
+# Location for where to put the zip containing our vendored typeshed stubs
+# that the red-knot crate creates at build time.
+# The location is relative to the `OUT_DIR` environment variable that cargo sets.
+RUFF_VENDORED_TYPESHED_ZIP = "/vendored_typeshed.zip"

crates/red_knot/build.rs

@@ -13,7 +13,6 @@ use zip::write::{FileOptions, ZipWriter};
 use zip::CompressionMethod;
 
 const TYPESHED_SOURCE_DIR: &str = "vendor/typeshed";
-const TYPESHED_ZIP_LOCATION: &str = "zipped_typeshed.zip";
 
 /// Recursively zip the contents of an entire directory.
 ///
@@ -60,13 +59,16 @@ fn main() {
     );
     let out_dir = std::env::var("OUT_DIR").unwrap();
 
+    // This environment variable is set in `.cargo/config.toml`
+    let typeshed_destination = std::env::var("RUFF_VENDORED_TYPESHED_ZIP").unwrap();
+
     // N.B. Deliberately using `format!()` instead of `Path::join()` here,
     // so that we use `/` as a path separator on all platforms.
     // That enables us to load the typeshed zip at compile time in `module.rs`
     // (otherwise we'd have to dynamically determine the exact path to the typeshed zip
     // based on the default path separator for the specific platform we're on,
     // which can't be done at compile time.)
-    let zipped_typeshed_location = format!("{out_dir}/{TYPESHED_ZIP_LOCATION}");
+    let zipped_typeshed_location = format!("{out_dir}{typeshed_destination}");
 
     let zipped_typeshed = File::create(zipped_typeshed_location).unwrap();
     zip_dir(TYPESHED_SOURCE_DIR, zipped_typeshed).unwrap();

crates/red_knot/src/module.rs

@@ -928,10 +928,9 @@ mod tests {
 
     #[test]
     fn typeshed_zip_created_at_build_time() -> anyhow::Result<()> {
-        // The file path here is hardcoded in this crate's `build.rs` script.
-        // Luckily this crate will fail to build if this file isn't available at build time.
+        // The RUFF_VENDORED_TYPESHED_ZIP environment variable is set in `.cargo/config.toml`
         const TYPESHED_ZIP_BYTES: &[u8] =
-            include_bytes!(concat!(env!("OUT_DIR"), "/zipped_typeshed.zip"));
+            include_bytes!(concat!(env!("OUT_DIR"), env!("RUFF_VENDORED_TYPESHED_ZIP")));
         assert!(!TYPESHED_ZIP_BYTES.is_empty());
         let mut typeshed_zip_archive = ZipArchive::new(Cursor::new(TYPESHED_ZIP_BYTES))?;