ferric_crypto_lib 0.2.7

use crate::crypto_systems::vigenere::Vigenere;
use crate::error::VigenereError;
use crate::prelude::ALPHABET_LEN;
use crate::utils::{decode_digit, encode_char, BaseString, EncodedString, StringType};
use crate::Traits::Encrypt;
use rayon::prelude::*;

impl Encrypt<VigenereError, BaseString, BaseString> for Vigenere {
    fn encrypt(&self, input: BaseString) -> Result<BaseString, VigenereError> {
        let encoded = input.encode()?;

        let key_cycle: Vec<_> = self
            .key
            .data
            .iter()
            .cycle()
            .take(encoded.data.len())
            .copied()
            .collect();

        let encrypted_data: Vec<_> = encoded
            .par_iter()
            .zip(key_cycle.par_iter()) // Use a parallel iterator of the key cycle
            .map(|(&x, &y)| (x + y) % *ALPHABET_LEN) // perform encryption
            .collect();

        let encrypted_data = EncodedString::new(encrypted_data, StringType::Standard).decode()?;

        Ok(encrypted_data)
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn encrypts_input_correctly() {
        let vigenere = Vigenere::new("hello".to_string());
        let result = vigenere.encrypt("hello".into());
        assert!(result.is_ok());
        assert_eq!(result.unwrap(), "oixxa".into()); // replace with expected encrypted string
    }
}