datafusion-common 49.0.0

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

// Support optional features for encryption in Parquet files.
//! This module provides types and functions related to encryption in Parquet files.

#[cfg(feature = "parquet_encryption")]
pub use parquet::encryption::decrypt::FileDecryptionProperties;
#[cfg(feature = "parquet_encryption")]
pub use parquet::encryption::encrypt::FileEncryptionProperties;

#[cfg(not(feature = "parquet_encryption"))]
pub struct FileDecryptionProperties;
#[cfg(not(feature = "parquet_encryption"))]
pub struct FileEncryptionProperties;

#[cfg(feature = "parquet")]
use crate::config::ParquetEncryptionOptions;
pub use crate::config::{ConfigFileDecryptionProperties, ConfigFileEncryptionProperties};
#[cfg(feature = "parquet")]
use parquet::file::properties::WriterPropertiesBuilder;

#[cfg(feature = "parquet")]
pub fn add_crypto_to_writer_properties(
    #[allow(unused)] crypto: &ParquetEncryptionOptions,
    #[allow(unused_mut)] mut builder: WriterPropertiesBuilder,
) -> WriterPropertiesBuilder {
    #[cfg(feature = "parquet_encryption")]
    if let Some(file_encryption_properties) = &crypto.file_encryption {
        builder = builder
            .with_file_encryption_properties(file_encryption_properties.clone().into());
    }
    builder
}

#[cfg(feature = "parquet_encryption")]
pub fn map_encryption_to_config_encryption(
    encryption: Option<&FileEncryptionProperties>,
) -> Option<ConfigFileEncryptionProperties> {
    encryption.map(|fe| fe.into())
}

#[cfg(not(feature = "parquet_encryption"))]
pub fn map_encryption_to_config_encryption(
    _encryption: Option<&FileEncryptionProperties>,
) -> Option<ConfigFileEncryptionProperties> {
    None
}

#[cfg(feature = "parquet_encryption")]
pub fn map_config_decryption_to_decryption(
    decryption: Option<&ConfigFileDecryptionProperties>,
) -> Option<FileDecryptionProperties> {
    decryption.map(|fd| fd.clone().into())
}

#[cfg(not(feature = "parquet_encryption"))]
pub fn map_config_decryption_to_decryption(
    _decryption: Option<&ConfigFileDecryptionProperties>,
) -> Option<FileDecryptionProperties> {
    None
}