repository.rb 2.4 KB
Newer Older
1
class Repository
2 3
  include Gitlab::ShellAdapter

4 5 6 7
  attr_accessor :raw_repository

  def initialize(path_with_namespace, default_branch)
    @raw_repository = Gitlab::Git::Repository.new(path_with_namespace, default_branch)
8 9 10 11 12 13 14 15 16 17
  rescue Gitlab::Git::Repository::NoRepository
    nil
  end

  def exists?
    raw_repository
  end

  def empty?
    raw_repository.empty?
18 19 20 21 22 23 24 25 26 27
  end

  def commit(id = nil)
    commit = raw_repository.commit(id)
    commit = Commit.new(commit) if commit
    commit
  end

  def commits(ref, path = nil, limit = nil, offset = nil)
    commits = raw_repository.commits(ref, path, limit, offset)
28
    commits = Commit.decorate(commits) if commits.present?
29 30 31 32 33
    commits
  end

  def commits_between(target, source)
    commits = raw_repository.commits_between(target, source)
34
    commits = Commit.decorate(commits) if commits.present?
35 36 37
    commits
  end

38 39 40 41
  def rm_branch(branch_name)
    gitlab_shell.rm_branch(path_with_namespace, branch_name)
  end

42 43 44 45
  def rm_tag(tag_name)
    gitlab_shell.rm_tag(path_with_namespace, tag_name)
  end

46 47 48 49 50 51 52 53 54 55 56 57
  def round_commit_count
    if commit_count > 10000
      '10000+'
    elsif commit_count > 5000
      '5000+'
    elsif commit_count > 1000
      '1000+'
    else
      commit_count
    end
  end

58 59 60 61 62 63 64 65 66 67 68 69
  def branch_names
    Rails.cache.fetch(cache_key(:branch_names)) do
      raw_repository.branch_names
    end
  end

  def tag_names
    Rails.cache.fetch(cache_key(:tag_names)) do
      raw_repository.tag_names
    end
  end

70 71 72 73
  def commit_count
    Rails.cache.fetch(cache_key(:commit_count)) do
      raw_repository.raw.commit_count
    end
74 75
  end

76 77 78 79 80 81 82 83 84 85
  # Return repo size in megabytes
  # Cached in redis
  def size
    Rails.cache.fetch(cache_key(:size)) do
      raw_repository.size
    end
  end

  def expire_cache
    Rails.cache.delete(cache_key(:size))
86 87
    Rails.cache.delete(cache_key(:branch_names))
    Rails.cache.delete(cache_key(:tag_names))
88
    Rails.cache.delete(cache_key(:commit_count))
D
Dmitriy Zaporozhets 已提交
89 90 91 92
    Rails.cache.delete(cache_key(:graph_log))
  end

  def graph_log
93
    Rails.cache.fetch(cache_key(:graph_log)) do
D
Dmitriy Zaporozhets 已提交
94 95 96
      stats = Gitlab::Git::GitStats.new(raw, root_ref)
      stats.parsed_log
    end
97 98 99 100 101 102
  end

  def cache_key(type)
    "#{type}:#{path_with_namespace}"
  end

103 104 105 106
  def method_missing(m, *args, &block)
    raw_repository.send(m, *args, &block)
  end

107
  def respond_to?(method)
D
Dmitriy Zaporozhets 已提交
108
    return true if raw_repository.respond_to?(method)
109 110 111 112

    super
  end
end