use crate::common::alphabet::Alphabet;
use crate::common::{alphabet, substitute};
pub fn encrypt(message: &str) -> String {
substitute::shift_substitution(message, |i| alphabet::STANDARD.modulo((i + 13) as isize))
}
pub fn decrypt(message: &str) -> String {
substitute::shift_substitution(message, |i| alphabet::STANDARD.modulo((i + 13) as isize))
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn with_utf8() {
let message = "Peace, Freedom and Liberty! 🗡️";
let encrypted = encrypt(message);
let decrypted = decrypt(&encrypted);
assert_eq!(decrypted, message);
}
#[test]
fn alphabet_encrypt() {
let message = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";
let encrypted = encrypt(message);
let decrypted = decrypt(&encrypted);
assert_eq!(decrypted, message);
}
}