enigma_ruby 0.1.4 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 30c3260357339774bee49cb4a35783ce07cadd38a6d74679f2e6bbd3a825ec4e
-  data.tar.gz: 1d1b66f1fc5b5822e1010ab97df5edfc5af854fa26ab6cb988d27bd925466c66
+  metadata.gz: d2aa5a5b1b3a00b9462ddf323f15adcd89c56b1d788f6366223d398b04cfdefd
+  data.tar.gz: 4110e1e5ff0f19f873c081cebe52da2ee574286be70eea4177f5997ee3de205a
 SHA512:
-  metadata.gz: ffc012733c489f766cfc2fb640f23c138ba80d11c7efa44941e1b5797a64f9f0f627c77008d5c1e43f908bb25d2d22987172a3707ed301d88233a9e6efb69e09
-  data.tar.gz: 0532310c2969e3abc6fd00d3a48d96a0319cc5756770738ca8ba0910a543eeb076e85987b4bc48bae915ec6c41c06a0c096203f63be36b5d290abecfb5b0eb49
+  metadata.gz: 65cb27db3efe47b644560594c9d03e7ebefc557103fd5feb1be8235b6c17e3fdbc213d2baab1dd3ca4bb93e6cc10e909342dd4e1d67268859d5f399b420ae5a0
+  data.tar.gz: e40bb17690c5bd0c748a6bca21a0337d3247dfa41364fb3d246d1ce679ed441aaf77921e9ebf17d43f6d130d79fbb020b868fd326af147f25d5de18e5f56d0ca

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.4"
+  VERSION = "0.1.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: enigma_ruby
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  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