module Locomotive class Site include Locomotive::Mongoid::Document ## Extensions ## extend Extensions::Site::SubdomainDomains extend Extensions::Site::FirstInstallation include Extensions::Shared::Seo include Extensions::Site::Locales ## fields ## field :name field :robots_txt ## associations ## references_many :pages, :class_name => 'Locomotive::Page', :validate => false references_many :snippets, :class_name => 'Locomotive::Snippet', :dependent => :destroy, :validate => false references_many :theme_assets, :class_name => 'Locomotive::ThemeAsset', :dependent => :destroy, :validate => false references_many :content_assets, :class_name => 'Locomotive::ContentAsset', :dependent => :destroy, :validate => false references_many :content_types, :class_name => 'Locomotive::ContentType', :dependent => :destroy, :validate => false embeds_many :memberships, :class_name => 'Locomotive::Membership' ## validations ## validates_presence_of :name ## callbacks ## after_create :create_default_pages! after_destroy :destroy_pages ## behaviours ## enable_subdomain_n_domains_if_multi_sites accepts_nested_attributes_for :memberships, :allow_destroy => true ## methods ## def all_pages_in_once Page.quick_tree(self) end def fetch_page(path, logged_in) Locomotive::Page.fetch_page_from_path self, path, logged_in end def accounts Account.criteria.in(:_id => self.memberships.map(&:account_id)) end def admin_memberships self.memberships.find_all { |m| m.admin? } end def to_liquid Locomotive::Liquid::Drops::Site.new(self) end def as_json(options = {}) Locomotive::SitePresenter.new(self, options).as_json end protected # FIXME: Currently there is no t/translate method on the I18n module # Extensions::Site::I18n which is breaking the testing. The # namespaced ::I18n should be changed to just I18n when the t() # method is available def create_default_pages! ::Mongoid::Fields::I18n.with_locale(self.default_locale) do %w{index 404}.each do |slug| self.pages.create({ :slug => slug, :title => ::I18n.t("attributes.defaults.pages.#{slug}.title"), :raw_template => ::I18n.t("attributes.defaults.pages.#{slug}.body"), :published => true }) end end end def destroy_pages # pages is a tree so we just need to delete the root (as well as the page not found page) self.pages.root.first.try(:destroy) && self.pages.not_found.first.try(:destroy) end end end