repository.rb 30.4 KB
Newer Older
1 2
require 'securerandom'

3
class Repository
L
Lin Jen-Shin 已提交
4 5 6
  REF_MERGE_REQUEST = 'merge-requests'.freeze
  REF_KEEP_AROUND = 'keep-around'.freeze
  REF_ENVIRONMENTS = 'environments'.freeze
7 8 9 10

  RESERVED_REFS_NAMES = %W[
    heads
    tags
11
    replace
12 13 14 15 16
    #{REF_ENVIRONMENTS}
    #{REF_KEEP_AROUND}
    #{REF_ENVIRONMENTS}
  ].freeze

17 18
  include Gitlab::ShellAdapter

19
  attr_accessor :full_path, :disk_path, :project, :is_wiki
20

21 22
  delegate :ref_name_for_sha, to: :raw_repository

23
  CreateTreeError = Class.new(StandardError)
24

25 26 27 28 29 30
  # Methods that cache data from the Git repository.
  #
  # Each entry in this Array should have a corresponding method with the exact
  # same name. The cache key used by those methods must also match method's
  # name.
  #
31 32 33
  # For example, for entry `:commit_count` there's a method called `commit_count` which
  # stores its data in the `commit_count` cache key.
  CACHED_METHODS = %i(size commit_count rendered_readme contribution_guide
34 35
                      changelog license_blob license_key gitignore koding_yml
                      gitlab_ci_yml branch_names tag_names branch_count
S
Sean McGivern 已提交
36 37
                      tag_count avatar exists? empty? root_ref has_visible_content?
                      issue_template_names merge_request_template_names).freeze
38 39 40

  # Methods that use cache_method but only memoize the value
  MEMOIZED_CACHED_METHODS = %i(license empty_repo?).freeze
41 42 43 44 45

  # Certain method caches should be refreshed when certain types of files are
  # changed. This Hash maps file types (as returned by Gitlab::FileDetector) to
  # the corresponding methods to call for refreshing caches.
  METHOD_CACHES_FOR_FILE_TYPES = {
46
    readme: :rendered_readme,
47
    changelog: :changelog,
48
    license: %i(license_blob license_key license),
49 50 51 52
    contributing: :contribution_guide,
    gitignore: :gitignore,
    koding: :koding_yml,
    gitlab_ci: :gitlab_ci_yml,
S
Sean McGivern 已提交
53 54 55
    avatar: :avatar,
    issue_template: :issue_template_names,
    merge_request_template: :merge_request_template_names
D
Douwe Maan 已提交
56
  }.freeze
57 58 59 60 61

  # Wraps around the given method and caches its output in Redis and an instance
  # variable.
  #
  # This only works for methods that do not take any arguments.
62
  def self.cache_method(name, fallback: nil, memoize_only: false)
63
    original = :"_uncached_#{name}"
64

65
    alias_method(original, name)
66

67
    define_method(name) do
68 69 70
      cache_method_output(name, fallback: fallback, memoize_only: memoize_only) do
        __send__(original) # rubocop:disable GitlabSecurity/PublicSend
      end
71
    end
72
  end
73

74
  def initialize(full_path, project, disk_path: nil, is_wiki: false)
75
    @full_path = full_path
76
    @disk_path = disk_path || full_path
77
    @project = project
78
    @commit_cache = {}
79
    @is_wiki = is_wiki
80
  end
81

82
  def ==(other)
H
http://jneen.net/ 已提交
83 84 85
    @disk_path == other.disk_path
  end

86
  def raw_repository
87
    return nil unless full_path
88

89
    @raw_repository ||= initialize_raw_repository
90 91
  end

92 93
  alias_method :raw, :raw_repository

94
  # Return absolute path to repository
95
  def path_to_repo
96
    @path_to_repo ||= File.expand_path(
97
      File.join(repository_storage_path, disk_path + '.git')
98
    )
99 100
  end

101 102 103 104
  def inspect
    "#<#{self.class.name}:#{@disk_path}>"
  end

L
Lin Jen-Shin 已提交
105
  def commit(ref = 'HEAD')
106
    return nil unless exists?
107
    return ref if ref.is_a?(::Commit)
108

109 110
    find_commit(ref)
  end
111

112 113 114 115 116 117
  # Finding a commit by the passed SHA
  # Also takes care of caching, based on the SHA
  def commit_by(oid:)
    return @commit_cache[oid] if @commit_cache.key?(oid)

    @commit_cache[oid] = find_commit(oid)
118 119
  end

120
  def commits(ref, path: nil, limit: nil, offset: nil, skip_merges: false, after: nil, before: nil)
121
    options = {
122 123 124 125 126
      repo: raw_repository,
      ref: ref,
      path: path,
      limit: limit,
      offset: offset,
127 128
      after: after,
      before: before,
129
      follow: Array(path).length == 1,
130
      skip_merges: skip_merges
131 132 133
    }

    commits = Gitlab::Git::Commit.where(options)
134
    commits = Commit.decorate(commits, @project) if commits.present?
135 136 137
    commits
  end

138 139
  def commits_between(from, to)
    commits = Gitlab::Git::Commit.between(raw_repository, from, to)
140
    commits = Commit.decorate(commits, @project) if commits.present?
141 142 143
    commits
  end

J
Jacob Vosmaer 已提交
144
  # Gitaly migration: https://gitlab.com/gitlab-org/gitaly/issues/384
145
  def find_commits_by_message(query, ref = nil, path = nil, limit = 1000, offset = 0)
146 147 148 149
    unless exists? && has_visible_content? && query.present?
      return []
    end

150 151 152 153 154 155 156
    raw_repository.gitaly_migrate(:commits_by_message) do |is_enabled|
      if is_enabled
        find_commits_by_message_by_gitaly(query, ref, path, limit, offset)
      else
        find_commits_by_message_by_shelling_out(query, ref, path, limit, offset)
      end
    end
157 158
  end

159 160 161 162 163 164 165
  def find_branch(name, fresh_repo: true)
    # Since the Repository object may have in-memory index changes, invalidating the memoized Repository object may
    # cause unintended side effects. Because finding a branch is a read-only operation, we can safely instantiate
    # a new repo here to ensure a consistent state to avoid a libgit2 bug where concurrent access (e.g. via git gc)
    # may cause the branch to "disappear" erroneously or have the wrong SHA.
    #
    # See: https://github.com/libgit2/libgit2/issues/1534 and https://gitlab.com/gitlab-org/gitlab-ce/issues/15392
166
    raw_repo = fresh_repo ? initialize_raw_repository : raw_repository
167 168

    raw_repo.find_branch(name)
169 170 171
  end

  def find_tag(name)
172
    tags.find { |tag| tag.name == name }
173 174
  end

175
  def add_branch(user, branch_name, ref)
176
    branch = raw_repository.add_branch(branch_name, user: user, target: ref)
177

178
    after_create_branch
179 180 181 182

    branch
  rescue Gitlab::Git::Repository::InvalidRef
    false
183 184
  end

185
  def add_tag(user, tag_name, target, message = nil)
186
    raw_repository.add_tag(tag_name, user: user, target: target, message: message)
187 188
  rescue Gitlab::Git::Repository::InvalidRef
    false
189 190
  end

191
  def rm_branch(user, branch_name)
192
    before_remove_branch
193

194
    raw_repository.rm_branch(branch_name, user: user)
195

196
    after_remove_branch
197
    true
198 199
  end

L
Lin Jen-Shin 已提交
200
  def rm_tag(user, tag_name)
Y
Yorick Peterse 已提交
201
    before_remove_tag
202

203
    raw_repository.rm_tag(tag_name, user: user)
L
Lin Jen-Shin 已提交
204 205 206

    after_remove_tag
    true
207 208
  end

209 210 211 212
  def ref_names
    branch_names + tag_names
  end

213
  def branch_exists?(branch_name)
214 215 216 217 218 219 220
    return false unless raw_repository

    @branch_exists_memo ||= Hash.new do |hash, key|
      hash[key] = raw_repository.branch_exists?(key)
    end

    @branch_exists_memo[branch_name]
221 222
  end

223
  def ref_exists?(ref)
224 225
    !!raw_repository&.ref_exists?(ref)
  rescue ArgumentError
226
    false
227 228
  end

D
Douwe Maan 已提交
229 230 231 232
  # Makes sure a commit is kept around when Git garbage collection runs.
  # Git GC will delete commits from the repository that are no longer in any
  # branches or tags, but we want to keep some of these commits around, for
  # example if they have comments or CI builds.
233
  def keep_around(sha)
234
    return unless sha && commit_by(oid: sha)
235 236 237

    return if kept_around?(sha)

238 239
    # This will still fail if the file is corrupted (e.g. 0 bytes)
    begin
240
      write_ref(keep_around_ref_name(sha), sha)
241
    rescue Rugged::ReferenceError => ex
242
      Rails.logger.error "Unable to create #{REF_KEEP_AROUND} reference for repository #{path}: #{ex}"
243 244
    rescue Rugged::OSError => ex
      raise unless ex.message =~ /Failed to create locked file/ && ex.message =~ /File exists/
245
      Rails.logger.error "Unable to create #{REF_KEEP_AROUND} reference for repository #{path}: #{ex}"
246
    end
247 248 249
  end

  def kept_around?(sha)
250
    ref_exists?(keep_around_ref_name(sha))
251
  end
252

253 254 255 256
  def write_ref(ref_path, sha)
    rugged.references.create(ref_path, sha, force: true)
  end

257
  def diverging_commit_counts(branch)
258
    root_ref_hash = raw_repository.rev_parse_target(root_ref).oid
J
Jeff Stubler 已提交
259
    cache.fetch(:"diverging_commit_counts_#{branch.name}") do
260 261
      # Rugged seems to throw a `ReferenceError` when given branch_names rather
      # than SHA-1 hashes
262 263
      number_commits_behind = raw_repository
        .count_commits_between(branch.dereferenced_target.sha, root_ref_hash)
264

265 266
      number_commits_ahead = raw_repository
        .count_commits_between(root_ref_hash, branch.dereferenced_target.sha)
267

268 269 270
      { behind: number_commits_behind, ahead: number_commits_ahead }
    end
  end
271

272 273 274
  def expire_tags_cache
    expire_method_caches(%i(tag_names tag_count))
    @tags = nil
275
  end
276

277
  def expire_branches_cache
278
    expire_method_caches(%i(branch_names branch_count has_visible_content?))
279
    @local_branches = nil
280
    @branch_exists_memo = nil
281 282
  end

283 284
  def expire_statistics_caches
    expire_method_caches(%i(size commit_count))
285 286
  end

287 288
  def expire_all_method_caches
    expire_method_caches(CACHED_METHODS)
D
Douwe Maan 已提交
289 290
  end

291 292 293 294 295 296 297 298 299
  # Expires the caches of a specific set of methods
  def expire_method_caches(methods)
    methods.each do |key|
      cache.expire(key)

      ivar = cache_instance_variable_name(key)

      remove_instance_variable(ivar) if instance_variable_defined?(ivar)
    end
D
Douwe Maan 已提交
300 301
  end

302 303 304 305 306 307 308 309 310 311 312 313 314 315 316
  def expire_avatar_cache
    expire_method_caches(%i(avatar))
  end

  # Refreshes the method caches of this repository.
  #
  # types - An Array of file types (e.g. `:readme`) used to refresh extra
  #         caches.
  def refresh_method_caches(types)
    to_refresh = []

    types.each do |type|
      methods = METHOD_CACHES_FOR_FILE_TYPES[type.to_sym]

      to_refresh.concat(Array(methods)) if methods
317
    end
318

319
    expire_method_caches(to_refresh)
320

321
    to_refresh.each { |method| send(method) } # rubocop:disable GitlabSecurity/PublicSend
322
  end
323

324 325 326 327 328 329 330
  def expire_branch_cache(branch_name = nil)
    # When we push to the root branch we have to flush the cache for all other
    # branches as their statistics are based on the commits relative to the
    # root branch.
    if !branch_name || branch_name == root_ref
      branches.each do |branch|
        cache.expire(:"diverging_commit_counts_#{branch.name}")
331
        cache.expire(:"commit_count_#{branch.name}")
332 333 334 335 336
      end
    # In case a commit is pushed to a non-root branch we only have to flush the
    # cache for said branch.
    else
      cache.expire(:"diverging_commit_counts_#{branch_name}")
337
      cache.expire(:"commit_count_#{branch_name}")
338
    end
D
Dmitriy Zaporozhets 已提交
339 340
  end

341
  def expire_root_ref_cache
342
    expire_method_caches(%i(root_ref))
343 344
  end

345 346
  # Expires the cache(s) used to determine if a repository is empty or not.
  def expire_emptiness_caches
347
    return unless empty?
348

349
    expire_method_caches(%i(empty? has_visible_content?))
Y
Yorick Peterse 已提交
350 351
  end

352 353 354 355
  def lookup_cache
    @lookup_cache ||= {}
  end

356
  def expire_exists_cache
357
    expire_method_caches(%i(exists?))
358 359
  end

360 361 362 363 364 365 366
  # expire cache that doesn't depend on repository data (when expiring)
  def expire_content_cache
    expire_tags_cache
    expire_branches_cache
    expire_root_ref_cache
    expire_emptiness_caches
    expire_exists_cache
367
    expire_statistics_caches
368 369 370 371 372
  end

  # Runs code after a repository has been created.
  def after_create
    expire_exists_cache
373 374
    expire_root_ref_cache
    expire_emptiness_caches
Y
Yorick Peterse 已提交
375 376

    repository_event(:create_repository)
377 378
  end

379 380
  # Runs code just before a repository is deleted.
  def before_delete
381
    expire_exists_cache
382 383
    expire_all_method_caches
    expire_branch_cache if exists?
384
    expire_content_cache
Y
Yorick Peterse 已提交
385 386

    repository_event(:remove_repository)
387 388 389 390 391 392 393
  end

  # Runs code just before the HEAD of a repository is changed.
  def before_change_head
    # Cached divergent commit counts are based on repository head
    expire_branch_cache
    expire_root_ref_cache
Y
Yorick Peterse 已提交
394 395

    repository_event(:change_default_branch)
396 397
  end

Y
Yorick Peterse 已提交
398 399
  # Runs code before pushing (= creating or removing) a tag.
  def before_push_tag
400 401
    expire_statistics_caches
    expire_emptiness_caches
402
    expire_tags_cache
Y
Yorick Peterse 已提交
403 404

    repository_event(:push_tag)
Y
Yorick Peterse 已提交
405 406 407 408 409
  end

  # Runs code before removing a tag.
  def before_remove_tag
    expire_tags_cache
410
    expire_statistics_caches
Y
Yorick Peterse 已提交
411 412

    repository_event(:remove_tag)
413 414
  end

L
Lin Jen-Shin 已提交
415 416 417 418 419
  # Runs code after removing a tag.
  def after_remove_tag
    expire_tags_cache
  end

420 421 422
  # Runs code after the HEAD of a repository is changed.
  def after_change_head
    expire_method_caches(METHOD_CACHES_FOR_FILE_TYPES.keys)
423 424
  end

425 426
  # Runs code after a repository has been forked/imported.
  def after_import
427
    expire_content_cache
428 429 430
  end

  # Runs code after a new commit has been pushed.
431 432 433
  def after_push_commit(branch_name)
    expire_statistics_caches
    expire_branch_cache(branch_name)
Y
Yorick Peterse 已提交
434 435

    repository_event(:push_commit, branch: branch_name)
436 437 438 439
  end

  # Runs code after a new branch has been created.
  def after_create_branch
440
    expire_branches_cache
Y
Yorick Peterse 已提交
441 442

    repository_event(:push_branch)
443 444
  end

445 446 447
  # Runs code before removing an existing branch.
  def before_remove_branch
    expire_branches_cache
Y
Yorick Peterse 已提交
448 449

    repository_event(:remove_branch)
450 451
  end

452 453
  # Runs code after an existing branch has been removed.
  def after_remove_branch
454
    expire_branches_cache
455 456
  end

457
  def method_missing(m, *args, &block)
458 459
    if m == :lookup && !block_given?
      lookup_cache[m] ||= {}
460
      lookup_cache[m][args.join(":")] ||= raw_repository.__send__(m, *args, &block) # rubocop:disable GitlabSecurity/PublicSend
461
    else
462
      raw_repository.__send__(m, *args, &block) # rubocop:disable GitlabSecurity/PublicSend
463
    end
464 465
  end

466 467
  def respond_to_missing?(method, include_private = false)
    raw_repository.respond_to?(method, include_private) || super
468
  end
D
Dmitriy Zaporozhets 已提交
469 470

  def blob_at(sha, path)
471
    Blob.decorate(raw_repository.blob_at(sha, path), project)
D
Douwe Maan 已提交
472 473
  rescue Gitlab::Git::Repository::NoRepository
    nil
D
Dmitriy Zaporozhets 已提交
474
  end
475

476 477 478 479 480 481 482
  def root_ref
    if raw_repository
      raw_repository.root_ref
    else
      # When the repo does not exist we raise this error so no data is cached.
      raise Rugged::ReferenceError
    end
483
  end
484
  cache_method :root_ref
485

486
  # Gitaly migration: https://gitlab.com/gitlab-org/gitaly/issues/314
487
  def exists?
488
    return false unless full_path
489

490
    raw_repository.exists?
491 492 493
  end
  cache_method :exists?

D
Douwe Maan 已提交
494
  delegate :empty?, to: :raw_repository
495 496 497 498 499 500 501 502 503 504 505 506 507
  cache_method :empty?

  # The size of this repository in megabytes.
  def size
    exists? ? raw_repository.size : 0.0
  end
  cache_method :size, fallback: 0.0

  def commit_count
    root_ref ? raw_repository.commit_count(root_ref) : 0
  end
  cache_method :commit_count, fallback: 0

508
  def commit_count_for_ref(ref)
509
    return 0 unless exists?
510

511 512 513 514
    begin
      cache.fetch(:"commit_count_#{ref}") { raw_repository.commit_count(ref) }
    rescue Rugged::ReferenceError
      0
515 516 517
    end
  end

518
  delegate :branch_names, to: :raw_repository
519 520
  cache_method :branch_names, fallback: []

D
Douwe Maan 已提交
521
  delegate :tag_names, to: :raw_repository
522 523
  cache_method :tag_names, fallback: []

524
  delegate :branch_count, :tag_count, :has_visible_content?, to: :raw_repository
525 526
  cache_method :branch_count, fallback: 0
  cache_method :tag_count, fallback: 0
527
  cache_method :has_visible_content?, fallback: false
528 529

  def avatar
530 531 532 533 534
    # n+1: https://gitlab.com/gitlab-org/gitlab-ce/issues/38327
    Gitlab::GitalyClient.allow_n_plus_1_calls do
      if tree = file_on_head(:avatar)
        tree.path
      end
535 536
    end
  end
537
  cache_method :avatar
538

S
Sean McGivern 已提交
539 540 541 542 543 544 545 546 547 548
  def issue_template_names
    Gitlab::Template::IssueTemplate.dropdown_names(project)
  end
  cache_method :issue_template_names, fallback: []

  def merge_request_template_names
    Gitlab::Template::MergeRequestTemplate.dropdown_names(project)
  end
  cache_method :merge_request_template_names, fallback: []

549
  def readme
550 551
    if readme = tree(:head)&.readme
      ReadmeBlob.new(readme, self)
552
    end
553 554
  end

555
  def rendered_readme
T
Toon Claes 已提交
556
    MarkupHelper.markup_unsafe(readme.name, readme.data, project: project) if readme
557 558
  end
  cache_method :rendered_readme
559

560
  def contribution_guide
561
    file_on_head(:contributing)
562
  end
563
  cache_method :contribution_guide
564 565

  def changelog
566
    file_on_head(:changelog)
567
  end
568
  cache_method :changelog
569

570
  def license_blob
571
    file_on_head(:license)
572
  end
573
  cache_method :license_blob
Z
Zeger-Jan van de Weg 已提交
574

575
  def license_key
576
    return unless exists?
577

578
    Licensee.license(path).try(:key)
579
  end
580
  cache_method :license_key
581

D
Douwe Maan 已提交
582 583
  def license
    return unless license_key
584

585
    Licensee::License.new(license_key)
586
  end
587
  cache_method :license, memoize_only: true
588 589

  def gitignore
590
    file_on_head(:gitignore)
591
  end
592
  cache_method :gitignore
593 594

  def koding_yml
595
    file_on_head(:koding)
596
  end
597
  cache_method :koding_yml
598

599
  def gitlab_ci_yml
600
    file_on_head(:gitlab_ci)
601
  end
602
  cache_method :gitlab_ci_yml
603

604
  def head_commit
605 606 607 608
    @head_commit ||= commit(self.root_ref)
  end

  def head_tree
609 610 611
    if head_commit
      @head_tree ||= Tree.new(self, head_commit.sha, nil)
    end
612 613
  end

614
  def tree(sha = :head, path = nil, recursive: false)
615
    if sha == :head
616 617
      return unless head_commit

618 619 620 621 622
      if path.nil?
        return head_tree
      else
        sha = head_commit.sha
      end
623 624
    end

625
    Tree.new(self, sha, path, recursive: recursive)
626
  end
D
Dmitriy Zaporozhets 已提交
627 628

  def blob_at_branch(branch_name, path)
D
Dmitriy Zaporozhets 已提交
629
    last_commit = commit(branch_name)
D
Dmitriy Zaporozhets 已提交
630

D
Dmitriy Zaporozhets 已提交
631 632 633 634 635
    if last_commit
      blob_at(last_commit.sha, path)
    else
      nil
    end
D
Dmitriy Zaporozhets 已提交
636
  end
D
Dmitriy Zaporozhets 已提交
637

638
  def last_commit_for_path(sha, path)
639 640 641 642 643 644 645
    raw_repository.gitaly_migrate(:last_commit_for_path) do |is_enabled|
      if is_enabled
        last_commit_for_path_by_gitaly(sha, path)
      else
        last_commit_for_path_by_rugged(sha, path)
      end
    end
646
  end
647

H
Hiroyuki Sato 已提交
648 649
  def last_commit_id_for_path(sha, path)
    key = path.blank? ? "last_commit_id_for_path:#{sha}" : "last_commit_id_for_path:#{sha}:#{Digest::SHA1.hexdigest(path)}"
H
Hiroyuki Sato 已提交
650

H
Hiroyuki Sato 已提交
651
    cache.fetch(key) do
652 653 654 655 656 657 658
      raw_repository.gitaly_migrate(:last_commit_for_path) do |is_enabled|
        if is_enabled
          last_commit_for_path_by_gitaly(sha, path).id
        else
          last_commit_id_for_path_by_shelling_out(sha, path)
        end
      end
H
Hiroyuki Sato 已提交
659 660 661
    end
  end

662
  def next_branch(name, opts = {})
P
P.S.V.R 已提交
663 664 665
    branch_ids = self.branch_names.map do |n|
      next 1 if n == name
      result = n.match(/\A#{name}-([0-9]+)\z/)
666 667 668
      result[1].to_i if result
    end.compact

P
P.S.V.R 已提交
669
    highest_branch_id = branch_ids.max || 0
670

P
P.S.V.R 已提交
671 672 673
    return name if opts[:mild] && 0 == highest_branch_id

    "#{name}-#{highest_branch_id + 1}"
674 675
  end

676
  def branches_sorted_by(value)
677
    raw_repository.local_branches(sort_by: value)
678
  end
679

680 681 682
  def tags_sorted_by(value)
    case value
    when 'name'
683
      VersionSorter.rsort(tags) { |tag| tag.name }
684 685 686 687 688 689 690 691 692
    when 'updated_desc'
      tags_sorted_by_committed_date.reverse
    when 'updated_asc'
      tags_sorted_by_committed_date
    else
      tags
    end
  end

693
  def contributors
694
    commits = self.commits(nil, limit: 2000, offset: 0, skip_merges: true)
695

D
Dmitriy Zaporozhets 已提交
696
    commits.group_by(&:author_email).map do |email, commits|
697 698
      contributor = Gitlab::Contributor.new
      contributor.email = email
699

D
Dmitriy Zaporozhets 已提交
700
      commits.each do |commit|
701
        if contributor.name.blank?
D
Dmitriy Zaporozhets 已提交
702
          contributor.name = commit.author_name
703 704
        end

705
        contributor.commits += 1
706 707
      end

708 709
      contributor
    end
710
  end
D
Dmitriy Zaporozhets 已提交
711

712
  def refs_contains_sha(ref_type, sha)
713 714
    args = %W(#{ref_type} --contains #{sha})
    names = run_git(args).first
715 716 717 718 719 720 721 722 723 724 725 726 727

    if names.respond_to?(:split)
      names = names.split("\n").map(&:strip)

      names.each do |name|
        name.slice! '* '
      end

      names
    else
      []
    end
  end
H
Hannes Rosenögger 已提交
728

729 730 731
  def branch_names_contains(sha)
    refs_contains_sha('branch', sha)
  end
H
Hannes Rosenögger 已提交
732

733 734
  def tag_names_contains(sha)
    refs_contains_sha('tag', sha)
H
Hannes Rosenögger 已提交
735
  end
736

737
  def local_branches
738
    @local_branches ||= raw_repository.local_branches
739 740
  end

741 742
  alias_method :branches, :local_branches

743 744 745 746
  def tags
    @tags ||= raw_repository.tags
  end

D
Douwe Maan 已提交
747 748 749
  def create_dir(user, path, **options)
    options[:user] = user
    options[:actions] = [{ action: :create_dir, file_path: path }]
750

D
Douwe Maan 已提交
751
    multi_action(**options)
S
Stan Hu 已提交
752 753
  end

D
Douwe Maan 已提交
754 755 756
  def create_file(user, path, content, **options)
    options[:user] = user
    options[:actions] = [{ action: :create, file_path: path, content: content }]
757

D
Douwe Maan 已提交
758
    multi_action(**options)
S
Stan Hu 已提交
759
  end
760

D
Douwe Maan 已提交
761 762 763
  def update_file(user, path, content, **options)
    previous_path = options.delete(:previous_path)
    action = previous_path && previous_path != path ? :move : :update
764

D
Douwe Maan 已提交
765 766
    options[:user] = user
    options[:actions] = [{ action: action, file_path: path, previous_path: previous_path, content: content }]
767

D
Douwe Maan 已提交
768
    multi_action(**options)
769 770
  end

D
Douwe Maan 已提交
771 772 773
  def delete_file(user, path, **options)
    options[:user] = user
    options[:actions] = [{ action: :delete, file_path: path }]
774

D
Douwe Maan 已提交
775
    multi_action(**options)
776 777
  end

778 779
  def with_cache_hooks
    result = yield
780

781
    return unless result
782

783 784
    after_create if result.repo_created?
    after_create_branch if result.branch_created?
785

786 787 788 789 790 791 792 793 794
    result.newrev
  end

  def with_branch(user, *args)
    with_cache_hooks do
      Gitlab::Git::OperationService.new(user, raw_repository).with_branch(*args) do |start_commit|
        yield start_commit
      end
    end
795 796
  end

797
  # rubocop:disable Metrics/ParameterLists
L
Lin Jen-Shin 已提交
798
  def multi_action(
799
    user:, branch_name:, message:, actions:,
800
    author_email: nil, author_name: nil,
801
    start_branch_name: nil, start_project: project)
802

803 804
    with_branch(
      user,
805
      branch_name,
806
      start_branch_name: start_branch_name,
807
      start_repository: start_project.repository.raw_repository) do |start_commit|
808

809
      index = Gitlab::Git::Index.new(raw_repository)
810

811
      if start_commit
812
        index.read_tree(start_commit.rugged_commit.tree)
813 814 815
        parents = [start_commit.sha]
      else
        parents = []
M
Marc Siegfriedt 已提交
816 817
      end

818
      actions.each do |options|
819
        index.public_send(options.delete(:action), options) # rubocop:disable GitlabSecurity/PublicSend
M
Marc Siegfriedt 已提交
820 821 822
      end

      options = {
823
        tree: index.write_tree,
M
Marc Siegfriedt 已提交
824 825 826 827 828
        message: message,
        parents: parents
      }
      options.merge!(get_committer_and_author(user, email: author_email, name: author_name))

829
      create_commit(options)
M
Marc Siegfriedt 已提交
830 831
    end
  end
832
  # rubocop:enable Metrics/ParameterLists
M
Marc Siegfriedt 已提交
833

834 835
  def get_committer_and_author(user, email: nil, name: nil)
    committer = user_to_committer(user)
D
Douwe Maan 已提交
836
    author = Gitlab::Git.committer_hash(email: email, name: name) || committer
837

838
    {
839 840
      author: author,
      committer: committer
841 842 843 844 845 846 847 848 849 850 851 852 853 854
    }
  end

  def can_be_merged?(source_sha, target_branch)
    our_commit = rugged.branches[target_branch].target
    their_commit = rugged.lookup(source_sha)

    if our_commit && their_commit
      !rugged.merge_commits(our_commit, their_commit).conflicts?
    else
      false
    end
  end

855 856 857 858 859 860
  def merge(user, source_sha, merge_request, message)
    with_cache_hooks do
      raw_repository.merge(user, source_sha, merge_request.target_branch, message) do |commit_id|
        merge_request.update(in_progress_merge_commit_sha: commit_id)
        nil # Return value does not matter.
      end
861
    end
862 863
  end

864
  def ff_merge(user, source, target_branch, merge_request: nil)
865 866
    their_commit_id = commit(source)&.id
    raise 'Invalid merge source' if their_commit_id.nil?
867

868
    merge_request&.update(in_progress_merge_commit_sha: their_commit_id)
869

870
    with_cache_hooks { raw.ff_merge(user, their_commit_id, target_branch) }
871 872
  end

873
  def revert(
874
    user, commit, branch_name, message,
875
    start_branch_name: nil, start_project: project)
876

877 878 879 880 881 882 883 884 885
    with_cache_hooks do
      raw_repository.revert(
        user: user,
        commit: commit.raw,
        branch_name: branch_name,
        message: message,
        start_branch_name: start_branch_name,
        start_repository: start_project.repository.raw_repository
      )
886
    end
887 888
  end

889
  def cherry_pick(
890
    user, commit, branch_name, message,
891
    start_branch_name: nil, start_project: project)
P
P.S.V.R 已提交
892

893 894 895 896 897 898 899 900 901
    with_cache_hooks do
      raw_repository.cherry_pick(
        user: user,
        commit: commit.raw,
        branch_name: branch_name,
        message: message,
        start_branch_name: start_branch_name,
        start_repository: start_project.repository.raw_repository
      )
P
P.S.V.R 已提交
902 903 904
    end
  end

905 906 907 908
  def merged_to_root_ref?(branch_or_name, pre_loaded_merged_branches = nil)
    branch = Gitlab::Git::Branch.find(self, branch_or_name)

    if branch
909 910
      @root_ref_sha ||= commit(root_ref).sha
      same_head = branch.target == @root_ref_sha
911 912 913 914
      merged =
        if pre_loaded_merged_branches
          pre_loaded_merged_branches.include?(branch.name)
        else
915
          ancestor?(branch.target, @root_ref_sha)
916
        end
F
Florent (HP) 已提交
917

918
      !same_head && merged
F
Florent (HP) 已提交
919 920 921 922 923
    else
      nil
    end
  end

924 925
  delegate :merged_branch_names, to: :raw_repository

S
Stan Hu 已提交
926
  def merge_base(first_commit_id, second_commit_id)
927 928
    first_commit_id = commit(first_commit_id).try(:id) || first_commit_id
    second_commit_id = commit(second_commit_id).try(:id) || second_commit_id
S
Stan Hu 已提交
929
    rugged.merge_base(first_commit_id, second_commit_id)
D
Douwe Maan 已提交
930 931
  rescue Rugged::ReferenceError
    nil
S
Stan Hu 已提交
932 933
  end

934
  def ancestor?(ancestor_id, descendant_id)
935
    return false if ancestor_id.nil? || descendant_id.nil?
936

937 938
    Gitlab::GitalyClient.migrate(:is_ancestor) do |is_enabled|
      if is_enabled
939
        raw_repository.ancestor?(ancestor_id, descendant_id)
940
      else
J
Jacob Vosmaer 已提交
941
        rugged_is_ancestor?(ancestor_id, descendant_id)
942 943
      end
    end
944 945
  end

V
Valery Sizov 已提交
946 947 948
  def empty_repo?
    !exists? || !has_visible_content?
  end
949
  cache_method :empty_repo?, memoize_only: true
V
Valery Sizov 已提交
950 951 952

  def search_files_by_content(query, ref)
    return [] if empty_repo? || query.blank?
V
Valery Sizov 已提交
953

954
    offset = 2
955 956 957
    args = %W(grep -i -I -n --before-context #{offset} --after-context #{offset} -E -e #{Regexp.escape(query)} #{ref || root_ref})

    run_git(args).first.scrub.split(/^--$/)
958 959
  end

V
Valery Sizov 已提交
960 961
  def search_files_by_name(query, ref)
    return [] if empty_repo? || query.blank?
962

963 964 965
    args = %W(ls-tree --full-tree -r #{ref || root_ref} --name-status | #{Regexp.escape(query)})

    run_git(args).first.lines.map(&:strip)
966 967
  end

968 969
  def fetch_remote(remote, forced: false, ssh_auth: nil, no_tags: false)
    gitlab_shell.fetch_remote(raw_repository, remote, ssh_auth: ssh_auth, forced: forced, no_tags: no_tags)
970 971
  end

972 973
  def fetch_source_branch!(source_repository, source_branch, local_ref)
    raw_repository.fetch_source_branch!(source_repository.raw_repository, source_branch, local_ref)
974
  end
975

976 977
  def compare_source_branch(target_branch_name, source_repository, source_branch_name, straight:)
    raw_repository.compare_source_branch(target_branch_name, source_repository.raw_repository, source_branch_name, straight: straight)
978
  end
979

980
  def create_ref(ref, ref_path)
981
    raw_repository.write_ref(ref_path, ref)
982 983
  end

984 985 986 987 988
  def ls_files(ref)
    actual_ref = ref || root_ref
    raw_repository.ls_files(actual_ref)
  end

989 990 991 992
  def gitattribute(path, name)
    raw_repository.attributes(path)[name]
  end

993 994 995 996 997 998 999 1000 1001 1002
  def copy_gitattributes(ref)
    actual_ref = ref || root_ref
    begin
      raw_repository.copy_gitattributes(actual_ref)
      true
    rescue Gitlab::Git::Repository::InvalidRef
      false
    end
  end

1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013
  # Caches the supplied block both in a cache and in an instance variable.
  #
  # The cache key and instance variable are named the same way as the value of
  # the `key` argument.
  #
  # This method will return `nil` if the corresponding instance variable is also
  # set to `nil`. This ensures we don't keep yielding the block when it returns
  # `nil`.
  #
  # key - The name of the key to cache the data in.
  # fallback - A value to fall back to in the event of a Git error.
1014
  def cache_method_output(key, fallback: nil, memoize_only: false, &block)
1015
    ivar = cache_instance_variable_name(key)
1016

1017 1018 1019
    if instance_variable_defined?(ivar)
      instance_variable_get(ivar)
    else
1020 1021 1022 1023
      # If the repository doesn't exist and a fallback was specified we return
      # that value inmediately. This saves us Rugged/gRPC invocations.
      return fallback unless fallback.nil? || exists?

1024
      begin
1025 1026 1027 1028 1029 1030 1031
        value =
          if memoize_only
            yield
          else
            cache.fetch(key, &block)
          end
        instance_variable_set(ivar, value)
1032
      rescue Rugged::ReferenceError, Gitlab::Git::Repository::NoRepository
1033 1034 1035
        # Even if the above `#exists?` check passes these errors might still
        # occur (for example because of a non-existing HEAD). We want to
        # gracefully handle this and not cache anything
1036
        fallback
1037 1038 1039
      end
    end
  end
1040

1041 1042 1043
  def cache_instance_variable_name(key)
    :"@#{key.to_s.tr('?!', '')}"
  end
1044

1045 1046
  def file_on_head(type)
    if head = tree(:head)
D
Douwe Maan 已提交
1047 1048
      head.blobs.find do |blob|
        Gitlab::FileDetector.type_of(blob.path) == type
1049 1050 1051 1052
      end
    end
  end

D
Douwe Maan 已提交
1053 1054 1055 1056
  def route_map_for(sha)
    blob_data_at(sha, '.gitlab/route-map.yml')
  end

1057 1058
  def gitlab_ci_yml_for(sha, path = '.gitlab-ci.yml')
    blob_data_at(sha, path)
D
Douwe Maan 已提交
1059 1060
  end

1061 1062
  private

1063 1064 1065 1066 1067 1068 1069 1070 1071 1072 1073 1074
  # TODO Generice finder, later split this on finders by Ref or Oid
  # gitlab-org/gitlab-ce#39239
  def find_commit(oid_or_ref)
    commit = if oid_or_ref.is_a?(Gitlab::Git::Commit)
               oid_or_ref
             else
               Gitlab::Git::Commit.find(raw_repository, oid_or_ref)
             end

    ::Commit.new(commit, @project) if commit
  end

D
Douwe Maan 已提交
1075 1076
  def blob_data_at(sha, path)
    blob = blob_at(sha, path)
1077
    return unless blob
1078

1079
    blob.load_all_data!
1080
    blob.data
1081
  end
1082

1083
  def cache
1084 1085
    # TODO: should we use UUIDs here? We could move repositories without clearing this cache
    @cache ||= RepositoryCache.new(full_path, @project.id)
1086
  end
1087 1088

  def tags_sorted_by_committed_date
1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100
    tags.sort_by do |tag|
      # Annotated tags can point to any object (e.g. a blob), but generally
      # tags point to a commit. If we don't have a commit, then just default
      # to putting the tag at the end of the list.
      target = tag.dereferenced_target

      if target
        target.committed_date
      else
        Time.now
      end
    end
1101
  end
D
Douwe Maan 已提交
1102 1103

  def keep_around_ref_name(sha)
1104
    "refs/#{REF_KEEP_AROUND}/#{sha}"
D
Douwe Maan 已提交
1105
  end
Y
Yorick Peterse 已提交
1106 1107

  def repository_event(event, tags = {})
1108
    Gitlab::Metrics.add_event(event, { path: full_path }.merge(tags))
Y
Yorick Peterse 已提交
1109
  end
1110

1111 1112
  def last_commit_for_path_by_gitaly(sha, path)
    c = raw_repository.gitaly_commit_client.last_commit_for_path(sha, path)
1113
    commit_by(oid: c)
1114 1115 1116 1117
  end

  def last_commit_for_path_by_rugged(sha, path)
    sha = last_commit_id_for_path_by_shelling_out(sha, path)
1118
    commit_by(oid: sha)
1119 1120 1121
  end

  def last_commit_id_for_path_by_shelling_out(sha, path)
1122
    args = %W(rev-list --max-count=1 #{sha} -- #{path})
A
Andrew Newdigate 已提交
1123
    raw_repository.run_git_with_timeout(args, Gitlab::Git::Popen::FAST_GIT_PROCESS_TIMEOUT).first.strip
1124 1125
  end

1126 1127 1128
  def repository_storage_path
    @project.repository_storage_path
  end
1129

1130
  def initialize_raw_repository
1131
    Gitlab::Git::Repository.new(project.repository_storage, disk_path + '.git', Gitlab::GlRepository.gl_repository(project, is_wiki))
1132
  end
1133

1134 1135 1136 1137 1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152 1153
  def find_commits_by_message_by_shelling_out(query, ref, path, limit, offset)
    ref ||= root_ref

    args = %W(
      log #{ref} --pretty=%H --skip #{offset}
      --max-count #{limit} --grep=#{query} --regexp-ignore-case
    )
    args = args.concat(%W(-- #{path})) if path.present?

    git_log_results = run_git(args).first.lines

    git_log_results.map { |c| commit(c.chomp) }.compact
  end

  def find_commits_by_message_by_gitaly(query, ref, path, limit, offset)
    raw_repository
      .gitaly_commit_client
      .commits_by_message(query, revision: ref, path: path, limit: limit, offset: offset)
      .map { |c| commit(c) }
  end
1154
end