engine/app/models/locomotive/site.rb

83 lines
2.6 KiB
Ruby

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 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,
:depth => 0
})
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