Function use_zip_in_place

crates/cargo-lambda-build/src/archive.rs:207–251

(
    zip_path: PathBuf,
    data: &BinaryData<'_>,
    include: Option<Vec<String>>,
)

}

fn use_zip_in_place(
    zip_path: PathBuf,
    data: &BinaryData<'_>,
    include: Option<Vec<String>>,
) -> Result<BinaryArchive> {
    let binary_path_in_zip = data.binary_path_in_zip()?;
    let (arch, binary_modified_at) =
        extract_data_from_zipped_binary(&zip_path, &binary_path_in_zip)?;

    if let Some(files) = include {
        let file = File::open(&zip_path)
            .into_diagnostic()
            .wrap_err_with(|| format!("failed to open zip file `{zip_path:?}`"))?;

        // Zip2 doesn't support updating zip files in place, so we need to create a new zip file
        // if we want to include files in the zip archive.
        // Before we do that, we need to read the existing zip file and copy its contents to a new zip file.
        let mut archive = ZipArchive::new(file).into_diagnostic()?;

        let tmp_dir = tempfile::tempdir().into_diagnostic()?;
        let tmp_path = tmp_dir.path().join(data.zip_name());
        let tmp = File::create(&tmp_path).into_diagnostic()?;
        let mut zip = ZipWriter::new(tmp);

        for i in 0..archive.len() {
            let file = archive.by_index_raw(i).into_diagnostic()?;
            zip.raw_copy_file(file).into_diagnostic()?;
        }

        include_files_in_zip(&mut zip, &files)?;

        zip.finish()
            .into_diagnostic()
            .wrap_err_with(|| format!("failed to finish zip file `{}`", tmp_path.display()))?;

        drop(archive);
        copy_and_replace(&tmp_path, &zip_path).into_diagnostic()?;
    }

    Ok(BinaryArchive::new(
        zip_path.clone(),
        arch.to_string(),
        binary_modified_at,
    ))
}

/// Create a zip file from a function binary.
/// The binary inside the zip file is called `bootstrap` for function binaries.