Contents

module DeviseInvitable::Controllers::Registrations
  def self.included(controller)
    controller.send :around_filter, :keep_invitation_info, :only => :create
  end

  protected

  def destroy_if_previously_invited
    hash = params[resource_name]
    if hash && hash[:email]
      resource = resource_class.where(:email => hash[:email], :encrypted_password => '').first
      if resource
        @invitation_info = Hash[resource.invitation_fields.map {|field|
          [field, resource[field]]
        }]
        resource.destroy
      end
    end
  end

  def keep_invitation_info
    resource_invitable = resource_class.devise_modules.include?(:invitable)
    destroy_if_previously_invited if resource_invitable
    yield
    reset_invitation_info if resource_invitable
  end

  def reset_invitation_info
    # Restore info about the last invitation (for later reference)
    # Reset the invitation_info only, if invited_by_id is still nil at this stage:
    resource = resource_class.where(:email => params[resource_name][:email], :invited_by_id => nil).first
    if resource && @invitation_info
      resource.invitation_fields.each do |field|
        resource[field] = @invitation_info[field]
      end
      resource.save!
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
devise_invitable-1.1.0	lib/devise_invitable/controllers/registrations.rb
devise_invitable-1.0.3	lib/devise_invitable/controllers/registrations.rb