engine/lib/locomotive/routing/site_dispatcher.rb

module Locomotive
  module Routing  
    module SiteDispatcher
    
      extend ActiveSupport::Concern
    
      included do
        before_filter :fetch_site
        
        helper_method :current_site
      end

      module InstanceMethods
        
        protected
        
        def fetch_site
          Locomotive.logger "[fetch site]  host = #{request.host} / #{request.env['HTTP_HOST']}"
          @current_site ||= Site.match_domain(request.host).first
        end
      
        def current_site
          @current_site || fetch_site
        end
      
        def require_site
          redirect_to application_root_url and return false if current_site.nil?
        end
        
        def validate_site_membership
          return if current_site && current_site.accounts.include?(current_admin)
          redirect_to application_root_url
        end
        
        def application_root_url
          root_url(:host => Locomotive.config.default_domain, :port => request.port)
        end
        
      end
    
    end  
  end  
end