SecGen/modules/generators/domain/secgen_local/local.rb

#!/usr/bin/ruby
require_relative '../../../../lib/objects/local_string_encoder.rb'
require 'faker'

class DomainEncoder < StringEncoder
  attr_accessor :name

  def initialize
    super
    self.module_name = 'Domain Encoder'
    self.name = ''
  end

  def encode_all
    domain = craft_domain
    tld = %w(org com net co.uk).sample

    self.outputs << "#{domain}.#{tld}"
  end

  # Creates a domain from the business_name
  def craft_domain
    domain = self.name
    # replace spaces
    domain = domain.downcase.tr(' ', %w(_ -).sample)
    # strip punctuation and return
    domain.gsub(/[^0-9a-z\s_-]/i, '')
  end

  def read_arguments
    # Get command line arguments
    opts = GetoptLong.new(
        ['--help', '-h', GetoptLong::NO_ARGUMENT],
        ['--name', GetoptLong::REQUIRED_ARGUMENT],
    )

    # process option arguments
    opts.each do |opt, arg|
      case opt
        when '--help'
          usage
        when '--name'
          self.name << arg;
        else
          Print.err "Argument not valid: #{arg}"
          usage
          exit
      end
    end
  end

  def encoding_print_string
    'name: ' + self.name.to_s
  end
end

DomainEncoder.new.run