summaryrefslogtreecommitdiff
path: root/app/services/projects/fork_service.rb
blob: 03be7039b2a5c8267e29effd98de6b1aca5a9b4c (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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
module Projects
  class ForkService < BaseService
    def execute(fork_to_project = nil)
      if fork_to_project
        link_existing_project(fork_to_project)
      else
        fork_new_project
      end
    end

    private

    def link_existing_project(fork_to_project)
      return if fork_to_project.forked?

      link_fork_network(fork_to_project)

      fork_to_project
    end

    def fork_new_project
      new_params = {
        forked_from_project_id: @project.id,
        visibility_level:       allowed_visibility_level,
        description:            @project.description,
        name:                   @project.name,
        path:                   @project.path,
        shared_runners_enabled: @project.shared_runners_enabled,
        namespace_id:           @params[:namespace].try(:id) || current_user.namespace.id
      }

      if @project.avatar.present? && @project.avatar.image?
        new_params[:avatar] = @project.avatar
      end

      new_project = CreateService.new(current_user, new_params).execute
      return new_project unless new_project.persisted?

      builds_access_level = @project.project_feature.builds_access_level
      new_project.project_feature.update_attributes(builds_access_level: builds_access_level)

      link_fork_network(new_project)

      new_project
    end

    def fork_network
      if @project.fork_network
        @project.fork_network
      elsif forked_from_project = @project.forked_from_project
        # TODO: remove this case when all background migrations have completed
        # this only happens when a project had a `forked_project_link` that was
        # not migrated to the `fork_network` relation
        forked_from_project.fork_network || forked_from_project.create_root_of_fork_network
      else
        @project.create_root_of_fork_network
      end
    end

    def link_fork_network(fork_to_project)
      fork_network.fork_network_members.create(project: fork_to_project,
                                               forked_from_project: @project)

      # TODO: remove this when ForkedProjectLink model is removed
      unless fork_to_project.forked_project_link
        fork_to_project.create_forked_project_link(forked_to_project: fork_to_project,
                                                   forked_from_project: @project)
      end

      refresh_forks_count
    end

    def refresh_forks_count
      Projects::ForksCountService.new(@project).refresh_cache
    end

    def allowed_visibility_level
      project_level = @project.visibility_level

      if Gitlab::VisibilityLevel.non_restricted_level?(project_level)
        project_level
      else
        Gitlab::VisibilityLevel.highest_allowed_level
      end
    end
  end
end