Contents

module Imap::Backup
  class Uploader
    attr_reader :folder
    attr_reader :serializer

    def initialize(folder, serializer)
      @folder = folder
      @serializer = serializer
    end

    def run
      if folder.uids.any?
        rename_serialized_folder
      else
        folder.create
        serializer.force_uid_validity(folder.uid_validity)
      end

      return if count.zero?

      Logger.logger.debug "[#{folder.name}] #{count} to restore"
      serializer.each_message(missing_uids).with_index do |(uid, message, flags), i|
        upload_message uid, message, flags, i + 1
      end
    end

    private

    def upload_message(uid, message, flags, index)
      return if message.nil?

      log_prefix = "[#{folder.name}] uid: #{uid} (#{index}/#{count}) -"
      Logger.logger.debug(
        "#{log_prefix} #{message.supplied_body.size} bytes"
      )

      begin
        new_uid = folder.append(message, flags: flags)
        serializer.update_uid(uid, new_uid)
      rescue StandardError => e
        Logger.logger.warn "#{log_prefix} append error: #{e}"
      end
    end

    def count
      @count ||= missing_uids.count
    end

    def missing_uids
      serializer.uids - folder.uids
    end

    def rename_serialized_folder
      Logger.logger.debug(
        "There's already a '#{folder.name}' folder with emails"
      )

      # Rename the local folder to a unique name
      new_name = serializer.apply_uid_validity(folder.uid_validity)

      return if !new_name

      # Restore the renamed folder
      Logger.logger.debug(
        "Backup '#{serializer.folder}' renamed and restored to '#{new_name}'"
      )
      @folder = Account::Folder.new(folder.connection, new_name)
      folder.create
      @serializer = Serializer.new(serializer.path, new_name)
      serializer.force_uid_validity(@folder.uid_validity)
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
imap-backup-7.0.0.rc1	lib/imap/backup/uploader.rb