engine/lib/tasks/locomotive.rake

# encoding: utf-8

require 'locomotive'

namespace :locomotive do

  desc 'Fetch the Locomotive default site template for the installation'
  task :fetch_default_site_template => :environment do
    puts "Downloading default site template from '#{Locomotive::Import.DEFAULT_SITE_TEMPLATE}'"
    `curl -L -s -o #{Rails.root}/tmp/default_site_template.zip #{Locomotive::Import.DEFAULT_SITE_TEMPLATE}`
    puts '...done'
  end

  desc 'Rebuild the serialized template of all the site pages'
  task :rebuild_serialized_page_templates do
    Page.all.each do |page|
      next unless page.template.nil?
      page.send :_parse_and_serialize_template
      page.save
    end
  end

  desc 'Import a remote template described by its URL -- 2 options: SITE=name or id, RESET=by default false'
  task :import => :environment do
    url, site_name_or_id, reset = ENV['URL'], ENV['SITE'], Boolean.set(ENV['RESET']) || false

    if url.blank? || (url =~ /https?:\/\//).nil?
      raise "URL is missing or it is not a valid http url."
    end

    site = Site.find(site_name_or_id) || Site.where(:name => site_name_or_id).first || Site.first

    if site.nil?
      raise "No site found. Please give a correct value (name or id) for the SITE env variable."
    end

    ::Locomotive::Import::Job.run!(url, site, { :samples => true, :reset => reset })
  end

  namespace :upgrade do

    desc 'Set roles to the existing users'
    task :set_roles => :environment do
      Site.all.each do |site|
        site.memberships.each do |membership|
          if membership.attributes['admin'] == true
            puts "...[#{site.name}] #{membership.account.name} has now the admin role"
            membership.role = 'admin'
          else
            puts "...[#{site.name}] #{membership.account.name} has now the author role"
            membership.role = 'author'
          end
        end
        site.save
      end
    end

    desc "Index page is sometimes after the 404 error page. Fix this"
    task :place_index_before_404 => :environment do
      Site.all.each do |site|
        site.pages.root.first.update_attribute :position, 0
        site.pages.not_found.first.update_attribute :position, 1
      end
    end

    desc 'Remove asset collections and convert them into content types'
    task :remove_asset_collections => :environment do
      puts "Processing #{AssetCollection.count} asset collection(s)..."

      Asset.destroy_all

      AssetCollection.all.each do |collection|
        site = Site.find(collection.attributes['site_id'])

        if collection.internal?
          # internal collection => create simple assets without associated to a collection

          collection.assets.each do |tmp_asset|
            # puts "tmp asset = #{tmp_asset.inspect} / #{tmp_asset.source.url.inspect}" TODO

            sanitized_attributes = tmp_asset.attributes.dup
            sanitized_attributes[:_id] = tmp_asset._id

            asset = site.assets.build(sanitized_attributes)

            asset.save(:validate => false)

            # puts "asset = #{asset.inspect} / #{asset.source.url.inspect}" TODO
          end
        else
          collection.fetch_asset_klass.class_eval { def self.model_name; 'Asset'; end }

          # create content_types reflection of an asset collection
          ContentType.where(:slug => collection.slug).all.collect(&:destroy) # TODO

          content_type = site.content_types.build({
            :name => collection.name,
            :slug => collection.slug,
            :order_by => '_position_in_list'
          })

          content_type._id = collection._id

          # extra custom fields
          collection.asset_custom_fields.each_with_index do |field, i|
            content_type.content_custom_fields.build(field.attributes.merge(:position => i + 3))
          end

          # add default custom fields
          content_type.content_custom_fields.build(:label => 'Name', :_alias => 'name', :kind => 'string', :required => true, :position => 1)
          content_type.content_custom_fields.build(:label => 'Source', :_alias => 'source', :kind => 'file', :required => true, :position => 2)

          content_type.save!

          content_type = ContentType.find(content_type._id) # hard reload

          # set the highlighted field name
          field = content_type.content_custom_fields.detect { |f| f._alias == 'name' }
          content_type.highlighted_field_name = field._name
          content_type.save

          # insert data
          collection.ordered_assets.each do |asset|
            attributes = (if asset.custom_fields.blank?
              { :created_at => asset.created_at, :updated_at => asset.updated_at }
            else
              {}.tap do |h|
                asset.custom_fields.each do |field|
                  case field.kind
                  when 'file' then h["#{field._name}"] = asset.send("#{field._name}".to_sym)
                  else
                    h[field._alias] = asset.send(field._name.to_sym)
                  end
                end
              end
            end)

            attributes.merge!(:name => asset.name, :_position_in_list => asset.position)

            content = content_type.contents.build(attributes)

            content._id = asset._id

            content.source = asset.source.file

            content.save(:validate => false)
          end
        end

        puts "...the collection named '#{collection.slug}' for the '#{site.name}' site has been migrated with success !"
      end
    end

  end

end


class TmpAssetUploader < CarrierWave::Uploader::Base
  include CarrierWave::RMagick
  include Locomotive::CarrierWave::Uploader::Asset
  version :thumb, :if => :image? do
    process :resize_to_fill => [50, 50]
    process :convert => 'png'
  end
  version :medium, :if => :image? do
    process :resize_to_fill => [80, 80]
    process :convert => 'png'
  end
  version :preview, :if => :image? do
    process :resize_to_fit => [880, 1100]
    process :convert => 'png'
  end
  def store_dir
    self.build_store_dir('sites', model.collection.site_id, 'assets', model.id)
  end
end

# Classes only used during the upgrade mechanism. Will be removed in a few weeks
class AssetCollection
  include Locomotive::Mongoid::Document
  field :name
  field :slug
  field :internal, :type => Boolean, :default => false
  referenced_in :site
  embeds_many :assets, :class_name => 'TmpAsset', :validate => false
  custom_fields_for :assets
  after_destroy :remove_uploaded_files
  scope :internal, :where => { :internal => true }
  scope :not_internal, :where => { :internal => false }
  def ordered_assets
    self.assets.sort { |a, b| (a.position || 0) <=> (b.position || 0) }
  end
  def self.find_or_create_internal(site)
    site.asset_collections.internal.first || site.asset_collections.create(:name => 'system', :slug => 'system', :internal => true)
  end
end

class TmpAsset
  include Mongoid::Document
  include Mongoid::Timestamps
  include CustomFields::ProxyClassEnabler
  field :name, :type => String
  field :content_type, :type => String
  field :width, :type => Integer
  field :height, :type => Integer
  field :size, :type => Integer
  field :position, :type => Integer, :default => 0
  mount_uploader :source, TmpAssetUploader
  embedded_in :collection, :class_name => 'AssetCollection', :inverse_of => :assets
  def site_id; self.collection.site_id; end
end