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 generate_id_method
  RUBY_VERSION =~ /1\.8/
end

def parent_class_name
  if options[:parent]
    options[:parent]
  # Only works on 3.2
  # elsif (n = Rails::Generators.namespace) && n.const_defined?(:ApplicationSerializer)
  #   "ApplicationSerializer"
  elsif defined?(::ApplicationSerializer)
    "ApplicationSerializer"
  else
    "ActiveModel::Serializer"
  end
end