FarmBot/Farmbot-Web-App

# Creates asymmetric key pairs for cryptographically secure operations.
# Mostly used for creation of jwt.pem- which is used to verify authenticity of
# JSON Web Tokens
class KeyGen
  SAVE_PATH = "jwt.#{Rails.env}.pem"

  def self.try_file(path = SAVE_PATH)
    OpenSSL::PKey::RSA.new(File.read(path)) if File.file?(path)
  end

  def self.generate_new_key(path = SAVE_PATH)
    rsa = OpenSSL::PKey::RSA.generate(2048)
    File.open(path, "w") { |f| f.write(rsa.to_pem) }
    return rsa
  end

  # Heroku / 12Factor users can't store stuff on the file system.
  # Store your pem file in ENV['RSA_KEY'] if that applies to you.
  def self.try_env
    OpenSSL::PKey::RSA.new(ENV["RSA_KEY"]) if ENV["RSA_KEY"]
  end

  # Lazy loads a crypto key.  Will look in ENV['RSA_KEY'], then
  # jwt.environment_name.pem. Generates new key if neither is available.
  def self.current
    @current ||= (try_env || try_file || generate_new_key)
  end
end