summaryrefslogtreecommitdiff
path: root/lib/gitlab/import_export/project/tree_saver.rb
blob: 58f33a0485110ae7063f9f954c62d73973527be6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
# frozen_string_literal: true

module Gitlab
  module ImportExport
    module Project
      class TreeSaver
        attr_reader :full_path

        def initialize(project:, current_user:, shared:, params: {})
          @params       = params
          @project      = project
          @current_user = current_user
          @shared       = shared
          @full_path    = File.join(@shared.export_path, ImportExport.project_filename)
        end

        def save
          project_tree = tree_saver.serialize(@project, reader.project_tree)
          fix_project_tree(project_tree)
          tree_saver.save(project_tree, @shared.export_path, ImportExport.project_filename)

          true
        rescue => e
          @shared.error(e)
          false
        end

        private

        # Aware that the resulting hash needs to be pure-hash and
        # does not include any AR objects anymore, only objects that run `.to_json`
        def fix_project_tree(project_tree)
          if @params[:description].present?
            project_tree['description'] = @params[:description]
          end

          project_tree['project_members'] += group_members_array

          RelationRenameService.add_new_associations(project_tree)
        end

        def reader
          @reader ||= Gitlab::ImportExport::Reader.new(shared: @shared)
        end

        def group_members_array
          group_members.as_json(reader.group_members_tree).each do |group_member|
            group_member['source_type'] = 'Project' # Make group members project members of the future import
          end
        end

        def group_members
          return [] unless @current_user.can?(:admin_group, @project.group)

          # We need `.where.not(user_id: nil)` here otherwise when a group has an
          # invitee, it would make the following query return 0 rows since a NULL
          # user_id would be present in the subquery
          # See http://stackoverflow.com/questions/129077/not-in-clause-and-null-values
          non_null_user_ids = @project.project_members.where.not(user_id: nil).select(:user_id)

          GroupMembersFinder.new(@project.group).execute.where.not(user_id: non_null_user_ids)
        end

        def tree_saver
          @tree_saver ||= RelationTreeSaver.new
        end
      end
    end
  end
end