class NoBrainer::Generators::AASMGenerator

def generate_model
  invoke 'nobrainer:model', [name] unless model_exists?
end

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

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

def migration_data
  "  field :#{column_name}"
end