71 lines
2.1 KiB
Ruby
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 |