engine/lib/locomotive/routing/site_dispatcher.rb

module Locomotive
  
  module Routing
  
    module SiteDispatcher
    
      def self.included(base)
        base.class_eval do
          include Locomotive::Routing::SiteDispatcher::InstanceMethods
        
          before_filter :fetch_site
          
          helper_method :current_site
        end
      end

      module InstanceMethods
        
        protected
        
        def fetch_site
          @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_account)          
          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