enigma-cli 1.0.0

A command-line interface for an Enigma machine simulator. Configurable with any reflector, number of rotors, and plugboard pairs.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51

use crate::{
    enigma_int::{FromEnigmaInt, ToEnigmaInt},
    Enigma,
};

impl Enigma {
    pub fn encode(&mut self, message: &String) -> String {
        let mut output = String::new();

        for c in message.chars() {
            if !c.is_ascii_alphabetic() {
                output.push(c);
                continue;
            }

            let mut internal_int = c.to_internal_int();

            if self.plugboard.is_some() {
                internal_int = self.plugboard.as_ref().unwrap().reflect(internal_int);
            }

            let mut turn_next = true;
            for rotor in self.rotors.iter_mut().enumerate() {
                if turn_next {
                    rotor.1.turn();
                    turn_next = false;
                } else {
                    let position = rotor.1.position;
                    let turnover = rotor.1.turnover;

                    if position == turnover {
                        turn_next = true;
                    }
                }

                internal_int = rotor.1.encode(internal_int);
            }

            internal_int = self.reflector.reflect(internal_int);

            self.rotors.reverse();
            for rotor in &self.rotors {
                internal_int = rotor.decode(internal_int);
            }

            output.push(internal_int.from_internal_int());
        }

        output
    }
}