module Locomotive class Site include Locomotive::Mongoid::Document ## Extensions ## extend Extensions::Site::SubdomainDomains extend Extensions::Site::FirstInstallation include Extensions::Shared::Seo ## 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 ## methods ## def all_pages_in_once Page.quick_tree(self) end def accounts Account.criteria.in(:_id => self.memberships.collect(&: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 to_json Locomotive::SitePresenter.new(self).as_json end protected def create_default_pages! %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 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