engine/app/models/locomotive/account.rb

require 'digest'

module Locomotive
  class Account

    include Locomotive::Mongoid::Document

    devise *Locomotive.config.devise_modules

    ## attributes ##
    field :name
    field :locale, :default => Locomotive.config.default_locale.to_s or 'en'
    field :switch_site_token

    ## validations ##
    validates_presence_of :name

    ## associations ##

    ## callbacks ##
    before_destroy :remove_memberships!

    ## methods ##

    def sites
      @sites ||= Site.where({ 'memberships.account_id' => self._id })
    end

    def reset_switch_site_token!
      self.switch_site_token = SecureRandom.base64(8).gsub("/", "_").gsub(/=+$/, "")
      self.save
    end

    def self.find_using_switch_site_token(token, age = 1.minute)
      return if token.blank?
      self.where(:switch_site_token => token, :updated_at.gt => age.ago.utc).first
    end

    def self.find_using_switch_site_token!(token, age = 1.minute)
      self.find_using_switch_site_token(token, age) || raise(::Mongoid::Errors::DocumentNotFound.new(self, token))
    end

    def devise_mailer
      Locomotive::DeviseMailer
    end

    def as_json(options = {})
      super((options || {}).merge(:only => [:name, :email, :memberships]))
    end

    protected

    def password_required?
      !persisted? || !password.blank? || !password_confirmation.blank?
    end

    def remove_memberships!
      self.sites.each do |site|
        membership = site.memberships.where(:account_id => self._id).first

        if site.admin_memberships.size == 1 && membership.admin?
          raise I18n.t('errors.messages.needs_admin_account')
        else
          membership.destroy
        end
      end
    end

  end
end