engine/app/models/locomotive/site.rb

71 lines
2.1 KiB
Ruby

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 :assets, :class_name => 'Locomotive::Asset', :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
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