Function zip_binary

crates/cargo-lambda-build/src/archive.rs:257–322

Create a zip file from a function binary. The binary inside the zip file is called `bootstrap` for function binaries. The binary inside the zip file is called by its name, and put inside the `extensions` directory, for extension binaries.

(
    binary_path: BP,
    destination_directory: DD,
    data: &BinaryData,
    include: Option<Vec<String>>,
)

/// The binary inside the zip file is called by its name, and put inside the `extensions`
/// directory, for extension binaries.
pub fn zip_binary<BP: AsRef<Path>, DD: AsRef<Path>>(
    binary_path: BP,
    destination_directory: DD,
    data: &BinaryData,
    include: Option<Vec<String>>,
) -> Result<BinaryArchive> {
    let path = binary_path.as_ref();
    let dir = destination_directory.as_ref();

    let zipped = dir.join(data.zip_name());
    debug!(?data, ?path, ?dir, ?zipped, "zipping binary");

    let zipped_binary = File::create(&zipped)
        .into_diagnostic()
        .wrap_err_with(|| format!("failed to create zip file `{zipped:?}`"))?;

    let mut file = File::open(path)
        .into_diagnostic()
        .wrap_err_with(|| format!("failed to open binary file `{path:?}`"))?;

    let file_metadata = file
        .metadata()
        .into_diagnostic()
        .wrap_err_with(|| format!("failed to get metadata from file `{path:?}`"))?;

    let binary_modified_at = file_metadata
        .modified()
        .ok()
        .or_else(|| file_metadata.created().ok());

    let (binary_data, arch) = read_binary(&mut file, path)?;

    let mut zip = ZipWriter::new(zipped_binary);
    if let Some(files) = include {
        include_files_in_zip(&mut zip, &files)?;
    }

    if let Some(parent) = data.parent_dir() {
        let options = SimpleFileOptions::default();
        zip.add_directory(parent, options)
            .into_diagnostic()
            .wrap_err_with(|| {
                format!("failed to add directory `{parent}` to zip file `{zipped:?}`")
            })?;
    }

    let binary_path_in_zip = data.binary_path_in_zip()?;

    let options = zip_file_options(&file, path)?;

    zip.start_file(binary_path_in_zip.to_string(), options)
        .into_diagnostic()
        .wrap_err_with(|| format!("failed to start zip file `{binary_path_in_zip:?}`"))?;
    zip.write_all(&binary_data)
        .into_diagnostic()
        .wrap_err_with(|| format!("failed to write data into zip file `{binary_path_in_zip:?}`"))?;
    zip.finish()
        .into_diagnostic()