Contents

# frozen_string_literal: true

class String
  # "Encrypt / Decrypt" the string with Caesar cipher. This will shift the
  # alphabet letters by `n`, where `n` is the integer key. The same function
  # is used for encryption / decryption.
  # @param opts [Hash] optional parameters
  # @option opts [Integer] :shift The shift key. Default value: 13.
  # @see https://en.wikipedia.org/wiki/Caesar_cipher
  # @return [String] the (de)ciphered string
  # @example
  #   'Hello world!'.rot # => "Uryyb jbeyq!"
  #   'Hello world!'.rot(shift: 11) # => "Spwwz hzcwo!"
  #   'Uryyb jbeyq!'.rot # => "Hello world!"
  #   'Spwwz hzcwo!'.rot(shift: 26-11) # => "Hello world!"
  def rot(opts = {})
    opts[:shift] ||= 13
    alphabet = Array('a'..'z')
    lowercase = alphabet.zip(alphabet.rotate(opts[:shift])).to_h
    alphabet = Array('A'..'Z')
    uppercasecase = alphabet.zip(alphabet.rotate(opts[:shift])).to_h
    encrypter = lowercase.merge(uppercasecase)
    chars.map { |c| encrypter.fetch(c, c) }.join
  end

  # "Encrypt / Decrypt" the string with Caesar cipher in place as described for
  # {String#rot}.
  # @return [String] the (de)ciphered string as well as changing changing the
  #   object in place.
  # @example
  #   a = 'Bonjour le monde !' # => "Bonjour le monde !"
  #   a.rot! # => "Obawbhe yr zbaqr !"
  #   a # => "Obawbhe yr zbaqr !"
  def rot!(opts = {})
    replace(rot(opts))
  end

  # Alias for {String#rot} with default value ( `rot(shift: 13)` ).
  def rot13
    rot
  end

  # Alias for {String#rot!} with default value ( `rot!(shift: 13)` ).
  def rot13!
    rot!
  end

  # Compute all possibilities with {String#rot}
  # @return [Hash] All possibilities with the shift index as key and the
  #   (de)ciphered text as value
  # @example
  #   'noraj'.rot_all # => {1=>"opsbk", 2=>"pqtcl", 3=>"qrudm", ... }
  def rot_all
    (1..26).each_with_object({}) { |i, h| h[i] = rot(shift: i) }
  end
end

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
ctf-party-3.0.0	lib/ctf_party/rot.rb
ctf-party-2.3.0	lib/ctf_party/rot.rb
ctf-party-2.2.0	lib/ctf_party/rot.rb
ctf-party-2.1.0	lib/ctf_party/rot.rb
ctf-party-2.0.0	lib/ctf_party/rot.rb