app/services/create_branch_service.rb


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

require_relative 'base_service'

class CreateBranchService < BaseService
  def execute(branch_name, ref, source_project: @project, with_hooks: true)
    valid_branch = Gitlab::GitRefValidator.validate(branch_name)

    unless valid_branch
      return error('Branch name is invalid')
    end

    repository = project.repository
    existing_branch = repository.find_branch(branch_name)

    if existing_branch
      return error('Branch already exists')
    end

    new_branch = if source_project != @project
                   repository.fetch_ref(
                     source_project.repository.path_to_repo,
                     "refs/heads/#{ref}",
                     "refs/heads/#{branch_name}"
                   )

                   repository.after_create_branch

                   repository.find_branch(branch_name)
                 else
                   repository.add_branch(current_user, branch_name, ref, with_hooks: with_hooks)
                 end

    if new_branch
      success(new_branch)
    else
      error('Invalid reference name')
    end
  rescue GitHooksService::PreReceiveError => ex
    error(ex.message)
  end

  def success(branch)
    super().merge(branch: branch)
  end
end