require "dejavu/version"

module Dejavu
  module ViewHelpers
    def has_dejavu?(obj)
      obj_name = ActiveRecord::Base === obj ? obj.class.model_name.underscore : obj.to_s
      !!flash[:"saved_#{obj_name}_for_redisplay"]
    end

    def get_dejavu_for(obj)
      is_instance = ActiveRecord::Base === obj
      model_name = is_instance ? obj.class.model_name.underscore : obj.to_s

      if has_dejavu?(obj)
        foo = if is_instance
                if Rails::VERSION::MINOR >= 1
                  obj.assign_attributes(flash[:"saved_#{model_name}_for_redisplay"], :without_protection => true)
                else
                  obj.attributes = flash[:"saved_#{model_name}_for_redisplay"]
                end
                obj
              else
                obj.to_s.classify.constantize.new flash[:"saved_#{model_name}_for_redisplay"]
              end
        foo.valid?
        foo
      else
        is_instance ? obj : obj.to_s.classify.constantize.new
      end
    end
  end

  module ControllerMethods
    def save_for_dejavu(obj, opts = {})
      attrs = obj.attributes
      if opts[:nested]
        attrs["#{opts[:nested]}_attributes"] = obj.send(opts[:nested]).attributes
      end
      flash[:"saved_#{obj.class.model_name.underscore}_for_redisplay"] = attrs
    end
  end
end

ActionController::Base.send(:include, Dejavu::ControllerMethods)
ActionView::Base.send(:include, Dejavu::ViewHelpers)