Anjing Webshells

Website : rimsha.abasa.com
: / var / canvas / app / models / importers /
Filename : attachment_importer.rb
back
# frozen_string_literal: true

#
# Copyright (C) 2014 - present Instructure, Inc.
#
# This file is part of Canvas.
#
# Canvas is free software: you can redistribute it and/or modify it under
# the terms of the GNU Affero General Public License as published by the Free
# Software Foundation, version 3 of the License.
#
# Canvas is distributed in the hope that it will be useful, but WITHOUT ANY
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
# A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
# details.
#
# You should have received a copy of the GNU Affero General Public License along
# with this program. If not, see <http://www.gnu.org/licenses/>.
#

module Importers
  class AttachmentImporter < Importer
    self.item_class = Attachment

    class << self
      def process_migration(data, migration)
        created_usage_rights_map = {}
        attachments = data["file_map"] || {}
        attachments = attachments.with_indifferent_access
        attachments.each_value do |att|
          next unless !att["is_folder"] && (migration.import_object?("attachments", att["migration_id"]) || migration.import_object?("files", att["migration_id"]))

          begin
            import_from_migration(att, migration.context, migration, nil, created_usage_rights_map)
          rescue
            migration.add_import_warning(I18n.t("#migration.file_type", "File"), (att[:display_name] || att[:path_name]), $!)
          end
        end

        data[:locked_folders]&.each do |path|
          next if migration.migration_settings[:importer_skips]&.include?("folders")

          # TODO: i18n
          if (f = migration.context.active_folders.where(full_name: "course files/#{path}").first)
            f.locked = true
            f.save
          end
        end

        data[:hidden_folders]&.each do |path|
          next if migration.migration_settings[:importer_skips]&.include?("folders")

          # TODO: i18n
          if (f = migration.context.active_folders.where(full_name: "course files/#{path}").first)
            f.workflow_state = "hidden"
            f.save
          end
        end
      end

      private

      def import_from_migration(hash, context, migration, item = nil, created_usage_rights_map = {})
        return nil if hash[:files_to_import] && !hash[:files_to_import][hash[:migration_id]]

        item ||= Attachment.where(context_type: context.class.to_s, context_id: context, id: hash[:id]).first
        item ||= Attachment.where(context_type: context.class.to_s, context_id: context, migration_id: hash[:migration_id]).first if hash[:migration_id]
        item ||= Attachment.find_from_path(hash[:path_name], context)
        if item
          item.mark_as_importing!(migration)
          item.category = hash[:category] if hash[:category].present?
          item.context = context
          item.migration_id = hash[:migration_id]
          item.locked = true if hash[:locked]
          item.lock_at = Canvas::Migration::MigratorHelper.get_utc_time_from_timestamp(hash[:lock_at]) if hash[:lock_at]
          item.unlock_at = Canvas::Migration::MigratorHelper.get_utc_time_from_timestamp(hash[:unlock_at]) if hash[:unlock_at]
          item.file_state = hash[:hidden] ? "hidden" : "available"
          item.display_name = hash[:display_name] if hash[:display_name]
          item.usage_rights_id = find_or_create_usage_rights(context, hash[:usage_rights], created_usage_rights_map) if hash[:usage_rights]
          item.set_publish_state_for_usage_rights unless hash[:locked]
          item.save_without_broadcasting!
          item.handle_duplicates(:rename)
          migration.add_imported_item(item)
        end
        item
      end

      def find_or_create_usage_rights(context, usage_rights_hash, created_usage_rights_map)
        attrs = usage_rights_hash.slice("use_justification", "license", "legal_copyright")
        key = attrs.values_at("use_justification", "license", "legal_copyright").join("/")
        id = created_usage_rights_map[key]
        return id if id

        usage_rights = context.usage_rights.create!(attrs)
        created_usage_rights_map[key] = usage_rights.id
      end
    end
  end
end