engine/lib/locomotive/import/job.rb

require 'zip/zipfilesystem'

module Locomotive
  module Import
    class Job

      def initialize(identifier, site = nil, enabled = {})
        raise "Theme identifier not found" if identifier.blank?

        @identifier = identifier
        @site = site
        @enabled = enabled
      end

      def before(worker)
        @worker = worker
      end

      def perform
        self.log "theme identifier #{@identifier} / enabled steps = #{@enabled.inspect}"

        self.unzip!

        raise "No database.yml found in the theme zipfile" if @database.nil?

        context = {
          :database => @database,
          :site => @site,
          :theme_path => @theme_path,
          :error => nil,
          :worker => @worker
        }

        %w(site content_types assets asset_collections snippets pages).each do |step|
          if @enabled[step] != false
            self.log "performing '#{step}' step"
            "Locomotive::Import::#{step.camelize}".constantize.process(context)
            @worker.update_attributes :step => step if @worker
          else
            self.log "skipping #{step}"
          end
        end
      end

      def success(worker)
        self.log 'deleting original zip file'

        uploader = ThemeUploader.new(@site)

        uploader.retrieve_from_store!(@identifier)

        uploader.remove!

        self.log 'deleting working folder'

        FileUtils.rm_rf(themes_folder) rescue nil
      end

      protected

      def log(message)
        puts "\t[import_theme] #{message}"
      end

      def themes_folder
        File.join(Rails.root, 'tmp', 'themes', @site.id.to_s)
      end

      def prepare_folder
        FileUtils.rm_rf self.themes_folder if File.exists?(self.themes_folder)

        FileUtils.mkdir_p(self.themes_folder)
      end

      def retrieve_zip_file
        uploader = ThemeUploader.new(@site)

        uploader.retrieve_from_store!(@identifier)

        if uploader.file.respond_to?(:url)
          self.log 'file from remote storage'

          @theme_file = File.join(self.themes_folder, @identifier)

          File.open(@theme_file, 'w') { |f| f.write(uploader.file.read) }
        else # local filesystem
          self.log 'file from local storage'

          @theme_file = uploader.path
        end
      end

      def unzip!
        self.prepare_folder

        self.retrieve_zip_file

        self.log "unzip #{@theme_file}"

        Zip::ZipFile.open(@theme_file) do |zipfile|
          destination_path = self.themes_folder

          zipfile.each do |entry|
            next if entry.name =~ /__MACOSX/

            if entry.name =~ /database.yml$/

              @database = YAML.load(zipfile.read(entry.name))
              @theme_path = File.join(destination_path, entry.name.gsub('database.yml', ''))

              next
            end

            FileUtils.mkdir_p(File.dirname(File.join(destination_path, entry.name)))

            zipfile.extract(entry, File.join(destination_path, entry.name))
          end
        end

      end

    end
  end
end