Sha256: 90ba8e65acfa649b2864518512ee9401ee36434d9056bc2ae714a40b5e91e898

Contents?: true

Size: 1.42 KB

Versions: 23

Compression:

Stored size: 1.42 KB

Contents

module Riveter
  module Generators
    class EnumGenerator < ::Rails::Generators::NamedBase
      source_root File.expand_path('../templates', __FILE__)

      check_class_collision :suffix => 'Enum'

      argument :enum_members,
               :type => :array,
               :default => [],
               :banner => "[name[:value] name[:value]]"

      def initialize(args, *options)
        super
        parse_enum_members!
      end

      def create_enum_file
        template 'enum.rb', File.join('app/enums', class_path, "#{file_name}_enum.rb")
      end

      def create_module_file
        return if regular_class_path.empty?
        template 'module.rb', File.join('app/enums', "#{class_path.join('/')}.rb") if behavior == :invoke
      end

      def create_locale_file
        template 'enums.en.yml', File.join('config/locales', 'enums.en.yml') unless locale_file_exists?
      end

      hook_for :test_framework, :as => :enum

    protected

      def parse_enum_members!
        self.enum_members = (enum_members || []).map do |member|
          # expected in the form "name:value" or "name"
          parts = member.split(':')
          OpenStruct.new(
            :name => parts.first.titleize,
            :value => (parts.length > 1 ? parts.last : nil)
          )
        end
      end

      def locale_file_exists?
        File.exists?(File.join(destination_root, 'config', 'locales', 'enums.en.yml'))
      end
    end
  end
end

Version data entries

23 entries across 23 versions & 1 rubygems

Version Path
riveter-0.8.3 lib/generators/riveter/enum/enum_generator.rb
riveter-0.8.2 lib/generators/riveter/enum/enum_generator.rb
riveter-0.8.1 lib/generators/riveter/enum/enum_generator.rb
riveter-0.8.0 lib/generators/riveter/enum/enum_generator.rb
riveter-0.7.0 lib/generators/riveter/enum/enum_generator.rb
riveter-0.6.0 lib/generators/riveter/enum/enum_generator.rb
riveter-0.3.0 lib/generators/riveter/enum/enum_generator.rb
riveter-0.2.1 lib/generators/riveter/enum/enum_generator.rb
riveter-0.2.0 lib/generators/riveter/enum/enum_generator.rb
riveter-0.1.1 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.15 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.14 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.13 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.12 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.11 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.10 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.9 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.8 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.7 lib/generators/riveter/enum/enum_generator.rb
riveter-0.0.6 lib/generators/riveter/enum/enum_generator.rb