grapes_generator.rb

require 'rails/generators/rails/scaffold/scaffold_generator'

module Rails::Generators
  class GrapesGenerator < ::Rails::Generators::ScaffoldGenerator
    attr_reader :fields

    def initialize(args, *options) #:nodoc:
      @fields = args[1..-1]
      super
    end

    hook_for :orm, as: :model do |instance, orm|
      instance.invoke orm, [ instance.options[:model_name], instance.fields ]
    end
  end
end