module AwesomePrint::NoBrainer

def self.included(base)
  base.send :alias_method, :cast_without_nobrainer, :cast
  base.send :alias_method, :cast, :cast_with_nobrainer
end

def awesome_nobrainer_class(object)
  name = "#{awesome_simple(object, :class)} < #{awesome_simple(object.superclass, :class)}"
  data = Hash[object.fields.map do |field, options|
    [field, (options[:type] || Object).to_s.underscore.to_sym]
  end]
  name = "class #{awesome_simple(object.to_s, :class)}"
  base = "< #{awesome_simple(object.superclass.to_s, :class)}"
  [name, base, awesome_hash(data)].join(' ')
end

def awesome_nobrainer_document(object)
  data = object.inspectable_attributes.symbolize_keys
  data = { errors: object.errors, attributes: data } if object.errors.present?
  "#{object} #{awesome_hash(data)}"
end

def cast_with_nobrainer(object, type)
  cast = cast_without_nobrainer(object, type)
  if defined?(::NoBrainer::Document)
    if object.is_a?(Class) && object < ::NoBrainer::Document
      cast = :nobrainer_class
    elsif object.is_a?(::NoBrainer::Document)
      cast = :nobrainer_document
    end
  end
  cast
end