class Rails::Generators::SerializerGenerator

def association_names
  attributes.select { |attr| attr.reference? }.map { |a| a.name.to_sym }
end

def attributes_names
  [:id] + attributes.select { |attr| !attr.reference? }.map { |a| a.name.to_sym }
end

def create_serializer_file
  template 'serializer.rb', File.join('app/serializers', class_path, "#{file_name}_serializer.rb")
end

def parent_class_name
  if options[:parent]
    options[:parent]
  elsif (ns = Rails::Generators.namespace) && ns.const_defined?(:ApplicationSerializer) ||
         defined?(::ApplicationSerializer)
    'ApplicationSerializer'
  else
    'ActiveModel::Serializer'
  end
end