Refactor Caesar Algorithm Implementation

src/ciphers/caesar.rs

@@ -1,43 +1,118 @@
-//! Caesar Cipher
-//! Based on cipher_crypt::caesar
-//!
-//! # Algorithm
-//!
-//! Rotate each ascii character by shift. The most basic example is ROT 13, which rotates 'a' to
-//! 'n'. This implementation does not rotate unicode characters.
-
-/// Caesar cipher to rotate cipher text by shift and return an owned String.
-pub fn caesar(cipher: &str, shift: u8) -> String {
-    cipher
+const ERROR_MESSAGE: &str = "Rotation must be in the range [0, 25]";
+const ALPHABET_LENGTH: u8 = b'z' - b'a' + 1;
+
+/// Encrypts a given text using the Caesar cipher technique.
+///
+/// In cryptography, a Caesar cipher, also known as Caesar's cipher, the shift cipher, Caesar's code,
+/// or Caesar shift, is one of the simplest and most widely known encryption techniques.
+/// It is a type of substitution cipher in which each letter in the plaintext is replaced by a letter
+/// some fixed number of positions down the alphabet.
+///
+/// # Arguments
+///
+/// * `text` - The text to be encrypted.
+/// * `rotation` - The number of rotations (shift) to be applied. It should be within the range [0, 25].
+///
+/// # Returns
+///
+/// Returns a `Result` containing the encrypted string if successful, or an error message if the rotation
+/// is out of the valid range.
+///
+/// # Errors
+///
+/// Returns an error if the rotation value is out of the valid range [0, 25]
+pub fn caesar(text: &str, rotation: isize) -> Result<String, &'static str> {
+    if !(0..ALPHABET_LENGTH as isize).contains(&rotation) {
+        return Err(ERROR_MESSAGE);
+    }
+
+    let result = text
         .chars()
         .map(|c| {
             if c.is_ascii_alphabetic() {
-                let first = if c.is_ascii_lowercase() { b'a' } else { b'A' };
-                // modulo the distance to keep character range
-                (first + (c as u8 + shift - first) % 26) as char
+                shift_char(c, rotation)
             } else {
                 c
             }
         })
-        .collect()
+        .collect();
+
+    Ok(result)
+}
+
+/// Shifts a single ASCII alphabetic character by a specified number of positions in the alphabet.
+///
+/// # Arguments
+///
+/// * `c` - The ASCII alphabetic character to be shifted.
+/// * `rotation` - The number of positions to shift the character. Should be within the range [0, 25].
+///
+/// # Returns
+///
+/// Returns the shifted ASCII alphabetic character.
+fn shift_char(c: char, rotation: isize) -> char {
+    let first = if c.is_ascii_lowercase() { b'a' } else { b'A' };
+    let rotation = rotation as u8; // Safe cast as rotation is within [0, 25]
+
+    (((c as u8 - first) + rotation) % ALPHABET_LENGTH + first) as char
 }
 
 #[cfg(test)]
 mod tests {
     use super::*;
 
-    #[test]
-    fn empty() {
-        assert_eq!(caesar("", 13), "");
+    macro_rules! test_caesar_happy_path {
+        ($($name:ident: $test_case:expr,)*) => {
+            $(
+                #[test]
+                fn $name() {
+                    let (text, rotation, expected) = $test_case;
+                    assert_eq!(caesar(&text, rotation).unwrap(), expected);
+
+                    let backward_rotation = if rotation == 0 { 0 } else { ALPHABET_LENGTH as isize - rotation };
+                    assert_eq!(caesar(&expected, backward_rotation).unwrap(), text);
+                }
+            )*
+        };
     }
 
-    #[test]
-    fn caesar_rot_13() {
-        assert_eq!(caesar("rust", 13), "ehfg");
+    macro_rules! test_caesar_error_cases {
+        ($($name:ident: $test_case:expr,)*) => {
+            $(
+                #[test]
+                fn $name() {
+                    let (text, rotation) = $test_case;
+                    assert_eq!(caesar(&text, rotation), Err(ERROR_MESSAGE));
+                }
+            )*
+        };
     }
 
     #[test]
-    fn caesar_unicode() {
-        assert_eq!(caesar("attack at dawn 攻", 5), "fyyfhp fy ifbs 攻");
+    fn alphabet_length_should_be_26() {
+        assert_eq!(ALPHABET_LENGTH, 26);
+    }
+
+    test_caesar_happy_path! {
+        empty_text: ("", 13, ""),
+        rot_13: ("rust", 13, "ehfg"),
+        unicode: ("attack at dawn 攻", 5, "fyyfhp fy ifbs 攻"),
+        rotation_within_alphabet_range: ("Hello, World!", 3, "Khoor, Zruog!"),
+        no_rotation: ("Hello, World!", 0, "Hello, World!"),
+        rotation_at_alphabet_end: ("Hello, World!", 25, "Gdkkn, Vnqkc!"),
+        longer: ("The quick brown fox jumps over the lazy dog.", 5, "Ymj vznhp gwtbs ktc ozrux tajw ymj qfed itl."),
+        non_alphabetic_characters: ("12345!@#$%", 3, "12345!@#$%"),
+        uppercase_letters: ("ABCDEFGHIJKLMNOPQRSTUVWXYZ", 1, "BCDEFGHIJKLMNOPQRSTUVWXYZA"),
+        mixed_case: ("HeLlO WoRlD", 7, "OlSsV DvYsK"),
+        with_whitespace: ("Hello, World!", 13, "Uryyb, Jbeyq!"),
+        with_special_characters: ("Hello!@#$%^&*()_+World", 4, "Lipps!@#$%^&*()_+Asvph"),
+        with_numbers: ("Abcd1234XYZ", 10, "Klmn1234HIJ"),
+    }
+
+    test_caesar_error_cases! {
+        negative_rotation: ("Hello, World!", -5),
+        empty_input_negative_rotation: ("", -1),
+        empty_input_large_rotation: ("", 27),
+        large_rotation: ("Large rotation", 139),
     }
 }