enigma_ruby 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2e53ea3ed0ad1827e09e9f3f5a05842af0e4cb87aeb855922736c20526f9867c
-  data.tar.gz: 986f7539cb3633e3aa55c8dcd46c52ae7c54e90e63a3a8cdd17e28e89617e1d6
+  metadata.gz: d2aa5a5b1b3a00b9462ddf323f15adcd89c56b1d788f6366223d398b04cfdefd
+  data.tar.gz: 4110e1e5ff0f19f873c081cebe52da2ee574286be70eea4177f5997ee3de205a
 SHA512:
-  metadata.gz: df1107dbd9fa1154605f3ffc17d282d517111f0cfe3ca2e45e72551fbf0150bd1cad466ad11dff13e20e502682f285c26bcfa97aa3f6a4008215dd9e402ef79e
-  data.tar.gz: 8e11afbdbea5fe6e8cc8d0a5409e9d7f762946b47039b090a9cb29e6b4fddb04481fa80fbe012bf6d37d16ce76ffc3ff0ae3b05adea2ffca022c537d2d9b626a
+  metadata.gz: 65cb27db3efe47b644560594c9d03e7ebefc557103fd5feb1be8235b6c17e3fdbc213d2baab1dd3ca4bb93e6cc10e909342dd4e1d67268859d5f399b420ae5a0
+  data.tar.gz: e40bb17690c5bd0c748a6bca21a0337d3247dfa41364fb3d246d1ce679ed441aaf77921e9ebf17d43f6d130d79fbb020b868fd326af147f25d5de18e5f56d0ca

data/lib/enigma_ruby/commands/advance_rotors.rb

@@ -0,0 +1,19 @@
+module EnigmaRuby
+  module Commands
+    class AdvanceRotors < Base
+      def initialize(rotors)
+        @rotors = rotors
+      end
+
+      def execute(_)
+        @rotors[0].advance
+        @rotors.each_with_index do |rotor, i|
+          next if i == 0
+          break unless @rotors[i - 1].at_notch?
+
+          rotor.advance
+        end
+      end
+    end
+  end
+end

data/lib/enigma_ruby/commands/base.rb

@@ -0,0 +1,9 @@
+module EnigmaRuby
+  module Commands
+    class Base
+      def execute(context)
+        raise NotImplementedError, 'Subclasses must implement execute(context)'
+      end
+    end
+  end
+end

data/lib/enigma_ruby/commands/encode_rotor_backward.rb

@@ -0,0 +1,15 @@
+module EnigmaRuby
+  module Commands
+    class EncodeRotorBackward < Base
+      def initialize(rotors)
+        @rotors = rotors
+      end
+
+      def execute(context)
+        @rotors.reverse_each do |rotor|
+          context[:encoded_char] = rotor.encode_backward(context[:encoded_char])
+        end
+      end
+    end
+  end
+end

data/lib/enigma_ruby/commands/encode_rotor_forward.rb

@@ -0,0 +1,15 @@
+module EnigmaRuby
+  module Commands
+    class EncodeRotorForward < Base
+      def initialize(rotors)
+        @rotors = rotors
+      end
+
+      def execute(context)
+        @rotors.each do |rotor|
+          context[:encoded_char] = rotor.encode_forward(context[:encoded_char])
+        end
+      end
+    end
+  end
+end

data/lib/enigma_ruby/commands/reflect.rb

@@ -0,0 +1,9 @@
+module EnigmaRuby
+  module Commands
+    class Reflect < Base
+      def execute(context)
+        context[:encoded_char] = Reflector.reflect(context[:encoded_char])
+      end
+    end
+  end
+end

data/lib/enigma_ruby/commands/swap_plugboard.rb

@@ -0,0 +1,9 @@
+module EnigmaRuby
+  module Commands
+    class SwapPlugboard < Base
+      def execute(context)
+        context[:encoded_char] = Plugboard.swap(context[:encoded_char])
+      end
+    end
+  end
+end

data/lib/enigma_ruby/enigma.rb

@@ -6,12 +6,12 @@ module EnigmaRuby
     def initialize(rotor_settings = [])
       initialize_rotors(rotor_settings)
       @commands = [
-        SwapPlugboardCommand.new,
-        EncodeRotorForwardCommand.new(@rotors),
-        ReflectCommand.new,
-        EncodeRotorBackwardCommand.new(@rotors),
-        SwapPlugboardCommand.new,
-        AdvanceRotorsCommand.new(@rotors)
+        Commands::SwapPlugboard.new,
+        Commands::EncodeRotorForward.new(@rotors),
+        Commands::Reflect.new,
+        Commands::EncodeRotorBackward.new(@rotors),
+        Commands::SwapPlugboard.new,
+        Commands::AdvanceRotors.new(@rotors)
       ]
     end
 

data/lib/enigma_ruby/plugboard.rb

@@ -1,14 +1,22 @@
 
 module EnigmaRuby
   class Plugboard
-    WIRING = {
-      'A' => 'F', 'F' => 'A',
-      'B' => 'E', 'E' => 'B',
-      'C' => 'D', 'D' => 'C'
-    }.freeze
+    PAIRS = %w[AF BE CD].freeze
 
     def self.swap(char)
-      WIRING[char] || char
+      check_duplicates
+
+      PAIRS.each do |pair|
+        return pair[1] if pair[0] == char
+        return pair[0] if pair[1] == char
+      end
+      char
+    end
+
+    def self.check_duplicates
+      characters = PAIRS.join.chars
+      duplicates = characters.select { |char| characters.count(char) > 1 }.uniq
+      raise "Duplicate characters found: #{duplicates.join(', ')}" if duplicates.any?
     end
   end
 end

data/lib/enigma_ruby/reflector.rb

@@ -1,10 +1,15 @@
 module EnigmaRuby
   class Reflector
-    WIRING = 'YRUHQSLDPXNGOKMIEBFZCWVJAT'.freeze
+    WIRING = {
+      'ORIGIN' => 'YRUHQSLDPXNGOKMIEBFZCWVJAT',
+      'UKW-A' => 'EJMZALYXVBWFCRQUONTSPIKHGD',
+      'UKW-B' => 'YRUHQSLDPXNGOKMIEBFZCWVJAT',
+      'UKW-C' => 'FVPJIAOYEDRZXWGCTKUQSBNMHL'
+    }.freeze
 
-    def self.reflect(char)
+    def self.reflect(char, wiring_key = 'ORIGIN')
       index = char.ord - 'A'.ord
-      WIRING[index]
+      WIRING[wiring_key][index]
     end
   end
 end

data/lib/enigma_ruby/rotor.rb

@@ -6,21 +6,27 @@ module EnigmaRuby
     end
 
     def encode_forward(char)
-      index = (char.ord - 'A'.ord + @position) % 26
+      index = (char.ord - 'A'.ord + @position) % wiring_size
       @wiring[index]
     end
 
     def encode_backward(char)
       index = @wiring.index(char)
-      ((index - @position) % 26 + 'A'.ord).chr
+      ((index - @position) % wiring_size + 'A'.ord).chr
     end
 
     def advance
-      @position = (@position + 1) % 26
+      @position = (@position + 1) % wiring_size
     end
 
     def at_notch?
       @position == @notch
     end
+
+    private
+
+    def wiring_size
+      @wiring.size
+    end
   end
 end

data/lib/enigma_ruby/version.rb

@@ -1,3 +1,3 @@
 module EnigmaRuby
-  VERSION = "0.1.3"
+  VERSION = "0.1.5"
 end

data/lib/enigma_ruby.rb

@@ -3,12 +3,12 @@ require "enigma_ruby/rotor"
 require "enigma_ruby/reflector"
 require "enigma_ruby/plugboard"
 require "enigma_ruby/enigma"
-require "enigma_ruby/commands/command"
-require "enigma_ruby/commands/advance_rotors_command"
-require "enigma_ruby/commands/encode_rotor_backward_command"
-require "enigma_ruby/commands/encode_rotor_forward_command"
-require "enigma_ruby/commands/reflect_command"
-require "enigma_ruby/commands/swap_plugboard_command"
+require "enigma_ruby/commands/base"
+require "enigma_ruby/commands/advance_rotors"
+require "enigma_ruby/commands/encode_rotor_backward"
+require "enigma_ruby/commands/encode_rotor_forward"
+require "enigma_ruby/commands/reflect"
+require "enigma_ruby/commands/swap_plugboard"
 
 module EnigmaRuby
   DEFAULT_ROTOR_SETTINGS = [

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enigma_ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - chibicco
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-07-13 00:00:00.000000000 Z
+date: 2024-07-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -62,12 +62,12 @@ extra_rdoc_files: []
 files:
 - README.md
 - lib/enigma_ruby.rb
-- lib/enigma_ruby/commands/advance_rotors_command.rb
-- lib/enigma_ruby/commands/command.rb
-- lib/enigma_ruby/commands/encode_rotor_backward_command.rb
-- lib/enigma_ruby/commands/encode_rotor_forward_command.rb
-- lib/enigma_ruby/commands/reflect_command.rb
-- lib/enigma_ruby/commands/swap_plugboard_command.rb
+- lib/enigma_ruby/commands/advance_rotors.rb
+- lib/enigma_ruby/commands/base.rb
+- lib/enigma_ruby/commands/encode_rotor_backward.rb
+- lib/enigma_ruby/commands/encode_rotor_forward.rb
+- lib/enigma_ruby/commands/reflect.rb
+- lib/enigma_ruby/commands/swap_plugboard.rb
 - lib/enigma_ruby/enigma.rb
 - lib/enigma_ruby/plugboard.rb
 - lib/enigma_ruby/reflector.rb

data/lib/enigma_ruby/commands/advance_rotors_command.rb

@@ -1,17 +0,0 @@
-module EnigmaRuby
-  class AdvanceRotorsCommand < Command
-    def initialize(rotors)
-      @rotors = rotors
-    end
-
-    def execute(_)
-      @rotors[0].advance
-      @rotors.each_with_index do |rotor, i|
-        next if i == 0
-        break unless @rotors[i - 1].at_notch?
-
-        rotor.advance
-      end
-    end
-  end
-end

data/lib/enigma_ruby/commands/command.rb

@@ -1,7 +0,0 @@
-module EnigmaRuby
-  class Command
-    def execute(context)
-      raise NotImplementedError, 'Subclasses must implement execute(context)'
-    end
-  end
-end

data/lib/enigma_ruby/commands/encode_rotor_backward_command.rb

@@ -1,13 +0,0 @@
-module EnigmaRuby
-  class EncodeRotorBackwardCommand < Command
-    def initialize(rotors)
-      @rotors = rotors
-    end
-
-    def execute(context)
-      @rotors.reverse_each do |rotor|
-        context[:encoded_char] = rotor.encode_backward(context[:encoded_char])
-      end
-    end
-  end
-end

data/lib/enigma_ruby/commands/encode_rotor_forward_command.rb

@@ -1,13 +0,0 @@
-module EnigmaRuby
-  class EncodeRotorForwardCommand < Command
-    def initialize(rotors)
-      @rotors = rotors
-    end
-
-    def execute(context)
-      @rotors.each do |rotor|
-        context[:encoded_char] = rotor.encode_forward(context[:encoded_char])
-      end
-    end
-  end
-end

data/lib/enigma_ruby/commands/reflect_command.rb

@@ -1,7 +0,0 @@
-module EnigmaRuby
-  class ReflectCommand < Command
-    def execute(context)
-      context[:encoded_char] = Reflector.reflect(context[:encoded_char])
-    end
-  end
-end

data/lib/enigma_ruby/commands/swap_plugboard_command.rb

@@ -1,7 +0,0 @@
-module EnigmaRuby
-  class SwapPlugboardCommand < Command
-    def execute(context)
-      context[:encoded_char] = Plugboard.swap(context[:encoded_char])
-    end
-  end
-end