locomotivecms/engine

View on GitHub
app/services/locomotive/site_service.rb

Summary

Maintainability
A
1 hr
Test Coverage
module Locomotive
  class SiteService < Struct.new(:account)

    include Locomotive::Concerns::ActivityService
    include Morphine

    register :page_service do
      Locomotive::PageService.new(nil, account)
    end

    register :content_entry_service do
      Locomotive::ContentEntryService.new(nil, account)
    end

    def list
      sorter = lambda do |site_a, site_b|
        site_a.name.downcase <=> site_b.name.downcase
      end

      account.sites.order_by(:name.asc).to_a.sort(&sorter)
    end

    def build_new
      Site.new(handle: unique_handle)
    end

    def create(attributes, raise_if_not_valid = false)
      if attributes[:handle].blank?
        attributes[:handle] = unique_handle
      end

      Site.new(attributes).tap do |site|
        site.created_by = account
        site.memberships.build account: account, role: 'admin'

        success = raise_if_not_valid ? site.save! : site.save

        ActiveSupport::Notifications.instrument 'locomotive.site.created', account: account, site: site

        track_activity 'site.created', site: site, parameters: { name: site.name } if success
      end
    end

    def create!(attributes)
      create(attributes, true)
    end

    def update(site, attributes, raise_if_not_valid = false)
      site.attributes = attributes

      new_locales = site.locales_changed? ? site.locales - site.locales_was : nil
      previous_default_locale = site.default_locale_was

      (raise_if_not_valid ? site.save! : site.save).tap do |success|
        if success
          localize_pages_and_content_entries(site, new_locales, previous_default_locale)
        end
      end
    end

    def update!(site, attributes)
      update(site, attributes, true)
    end

    private

    def localize_pages_and_content_entries(site, new_locales, previous_default_locale)
      return unless new_locales.present?

      # localize all the existing pages
      page_service.site = site
      page_service.localize(new_locales, previous_default_locale)

      # localize all the content entries of localized content types
      site.localized_content_types.each do |content_type|
        content_entry_service.content_type = content_type
        content_entry_service.localize(new_locales, previous_default_locale)
      end
    end

    def unique_handle
      Bazaar.heroku do |value|
        Site.where(handle: value).exists?
      end
    end

  end
end