class RegistrationsController < Devise::RegistrationsController
  helper :billing
  before_filter :set_billing_object, :only => [:edit, :update]
  before_filter :set_layout
  before_filter :set_page

  def update
    if self.resource.update_with_password(params[resource_name])
      set_flash_message :notice, :updated
      #redirect_to edit_registration_path(self.resource, :page => params[:page] || 'profile')
      redirect_to "/dashboard?page=#{params[:page]}"
    else
      render_with_scope :edit
    end
  end

  private

  def set_layout
    if request.get? && params[:action] != "new"
      render :layout => false
    end
  end

  def set_page
    @page = case params[:page]
    when 'balance'
      'balance'
    when 'shipping'
      'shipping'
    when 'billing'
      'billing'
    else
      'profile'
    end
  end
end