importer.rb 2.23 KB
Newer Older
1 2 3 4
module Gitlab
  module ImportExport
    class Importer

5 6 7
      def initialize(project)
        @archive_file = project.import_source
        @current_user = project.creator
8 9
        @project = project
        @shared = Gitlab::ImportExport::Shared.new(relative_path: path_with_namespace)
10 11
      end

12 13 14 15
      def execute
        Gitlab::ImportExport::FileImporter.import(archive_file: @archive_file,
                                                  shared: @shared)
        if check_version! && [project_tree, repo_restorer, wiki_restorer, uploads_restorer].all?(&:restore)
16
          project_tree.restored_project
17
        else
18 19 20
          project_tree.restored_project.destroy if project_tree.restored_project

          raise Projects::ImportService::Error.new, @shared.errors.join(', ')
21
        end
22 23 24 25
      end

      private

26 27 28 29 30 31 32 33 34 35 36 37 38
      def check_version!
        Gitlab::ImportExport::VersionChecker.check!(shared: @shared)
      end

      def project_tree
        @project_tree ||= Gitlab::ImportExport::ProjectTreeRestorer.new(user: @current_user,
                                                                        shared: @shared,
                                                                        project: @project)
      end

      def repo_restorer
        Gitlab::ImportExport::RepoRestorer.new(path_to_bundle: repo_path,
                                               shared: @shared,
39
                                               project: project_tree.restored_project)
40 41 42 43 44
      end

      def wiki_restorer
        Gitlab::ImportExport::RepoRestorer.new(path_to_bundle: wiki_repo_path,
                                               shared: @shared,
45
                                               project: ProjectWiki.new(project_tree.restored_project),
46 47 48 49
                                               wiki: true)
      end

      def uploads_restorer
50
        Gitlab::ImportExport::UploadsRestorer.new(project: project_tree.restored_project, shared: @shared)
51 52
      end

53 54
      def path_with_namespace
        File.join(@project.namespace.path, @project.path)
55 56 57 58 59 60 61 62 63
      end

      def repo_path
        File.join(@shared.export_path, 'project.bundle')
      end

      def wiki_repo_path
        File.join(@shared.export_path, 'project.wiki.bundle')
      end
64 65 66
    end
  end
end