Install generator

lib/generators/enumerate_it/enum/enum_generator.rb

@@ -0,0 +1,47 @@
+module EnumerateIt
+  module Generators
+    class EnumGenerator < Rails::Generators::NamedBase
+      source_root File.expand_path('../templates', __FILE__)
+
+      argument :attributes, :type => 'array', :default => []
+
+      class_option :singular, :type => 'string', :desc => 'Singular name for i18n'
+
+      class_option :lang, :type => 'string', :desc => 'Lang to use in i18n', :default => 'en'
+
+      desc "Creates an initializer file on config/initializers that extends ActiveRecord::Base with Enumerate_it"
+      def create_locale
+        template "locale.yml", File.join('config/locales', "#{singular_name}.yml")
+      end
+
+      desc "Creates an initializer file on config/initializers that extends ActiveRecord::Base with Enumerate_it"
+      def create_enumerate_it
+        template "enumerate_it.rb", File.join('app/enumerations', "#{singular_name}.rb")
+      end
+
+      protected
+
+      def default_lang
+        options[:lang]
+      end
+
+      def singular
+        singular_name
+      end
+
+      def locale_fields
+        attributes.map(&:name)
+      end
+
+      def fields
+        if attributes.first.type == :string
+          attributes.map(&:name)
+        else
+          attributes.map do |attribute|
+            [attribute.name, attribute.type]
+          end
+        end
+      end
+    end
+  end
+end

lib/generators/enumerate_it/install/USAGE

@@ -0,0 +1,8 @@
+Description:
+    Creates an initializer file for EnumerateIt
+
+Example:
+    rails generate enumerate_it:install
+
+    This will create:
+        config/initializers/enumerate_it.rb

lib/generators/enumerate_it/install/install_generator.rb

@@ -0,0 +1,11 @@
+module EnumerateIt
+  module Generators
+    class InstallGenerator < Rails::Generators::Base
+      source_root File.expand_path('../templates', __FILE__)
+
+      def copy_initializer_file
+        template "enumerate_it_initializer.rb", File.join('config/initializers/', "enumerate_it.rb")
+      end
+    end
+  end
+end

lib/generators/enumerate_it/install/templates/enumerate_it_initializer.rb

@@ -0,0 +1 @@
+ActiveRecord::Base.extend EnumerateIt