class Mongoid::Generators::AASMGenerator

def generate_model
  invoke "mongoid:model", [name] unless model_exists?
end

def inject_aasm_content
  inject_into_file model_path, model_contents, after: "include Mongoid::Document\n" if model_exists?
end

def inject_field_types
  inject_into_file model_path, migration_data, after: "include Mongoid::Document\n" if model_exists?
end

def migration_data
  "  field :#{column_name}"
end