58 lines
1.4 KiB
Ruby
58 lines
1.4 KiB
Ruby
module Locomotive
|
|
module Routing
|
|
module SiteDispatcher
|
|
|
|
extend ActiveSupport::Concern
|
|
|
|
included do
|
|
if self.respond_to?(:before_filter)
|
|
before_filter :fetch_site
|
|
|
|
helper_method :current_site
|
|
end
|
|
end
|
|
|
|
module InstanceMethods
|
|
|
|
protected
|
|
|
|
def fetch_site
|
|
Locomotive.log "[fetch site] host = #{request.host} / #{request.env['HTTP_HOST']}"
|
|
|
|
if Locomotive.config.multi_sites?
|
|
@current_site ||= Site.match_domain(request.host).first
|
|
else
|
|
@current_site ||= Site.first
|
|
end
|
|
end
|
|
|
|
def current_site
|
|
@current_site || fetch_site
|
|
end
|
|
|
|
def require_site
|
|
return true if current_site
|
|
|
|
redirect_to locomotive_installation_url and return false if Account.count == 0 || Site.count == 0
|
|
|
|
render_no_site_error and return false
|
|
end
|
|
|
|
def render_no_site_error
|
|
render :template => '/admin/errors/no_site', :layout => false
|
|
end
|
|
|
|
def validate_site_membership
|
|
return true if current_site.present? && current_site.accounts.include?(current_account)
|
|
|
|
sign_out(current_account)
|
|
flash[:alert] = I18n.t(:no_membership, :scope => [:devise, :failure, :locomotive])
|
|
redirect_to new_locomotive_session_url and return false
|
|
end
|
|
|
|
end
|
|
|
|
end
|
|
end
|
|
end
|