Contents

require 'rails/generators/migration'

module Roles
  module Generators
    class RoleGenerator < Rails::Generators::Base
      include Rails::Generators::Migration

      source_root File.expand_path('../templates', __FILE__)
      argument :role_cname, :type => :string, :default => "Role"
      argument :user_cname, :type => :string, :default => "User"

      desc "Generates a model with the given NAME and a migration file."

      def generate_role
        template "role-active_record.rb", "app/models/#{role_cname.underscore}.rb"
        inject_into_file(model_path, :after => inject_roles_method) do
          "  rolify" + (role_cname == "Role" ? "" : " :role_cname => '#{role_cname.camelize}'") + "\n"
        end
      end

      def copy_role_file
        migration_template "migration.rb", "db/migrate/roles_create_#{role_cname.tableize}"
      end

      def model_path
        File.join("app", "models", "#{user_cname.underscore}.rb")
      end
      
      def self.next_migration_number(path)
        Time.now.utc.strftime("%Y%m%d%H%M%S")
      end
      
      def show_readme
        readme "README-active_record" if behavior == :invoke
      end
      
      def inject_roles_method
        /class #{user_cname.camelize}\n|class #{user_cname.camelize} .*\n/
      end
    end
  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
roles-0.2.1	lib/generators/roles/role/role_generator.rb
roles-0.2.0	lib/generators/roles/role/role_generator.rb
roles-0.1.0	lib/generators/roles/role/role_generator.rb
roles-0.0.5	lib/generators/roles/role/role_generator.rb
roles-0.0.4	lib/generators/roles/role/role_generator.rb
roles-0.0.3	lib/generators/roles/role/role_generator.rb