700 lines
16 KiB
Ruby
700 lines
16 KiB
Ruby
require 'securerandom'
|
|
|
|
class Repository
|
|
class CommitError < StandardError; end
|
|
|
|
include Gitlab::ShellAdapter
|
|
|
|
attr_accessor :path_with_namespace, :project
|
|
|
|
def self.clean_old_archives
|
|
repository_downloads_path = Gitlab.config.gitlab.repository_downloads_path
|
|
|
|
return unless File.directory?(repository_downloads_path)
|
|
|
|
Gitlab::Popen.popen(%W(find #{repository_downloads_path} -not -path #{repository_downloads_path} -mmin +120 -delete))
|
|
end
|
|
|
|
def initialize(path_with_namespace, default_branch = nil, project = nil)
|
|
@path_with_namespace = path_with_namespace
|
|
@project = project
|
|
end
|
|
|
|
def raw_repository
|
|
return nil unless path_with_namespace
|
|
|
|
@raw_repository ||= begin
|
|
repo = Gitlab::Git::Repository.new(path_to_repo)
|
|
repo.autocrlf = :input
|
|
repo
|
|
rescue Gitlab::Git::Repository::NoRepository
|
|
nil
|
|
end
|
|
end
|
|
|
|
# Return absolute path to repository
|
|
def path_to_repo
|
|
@path_to_repo ||= File.expand_path(
|
|
File.join(Gitlab.config.gitlab_shell.repos_path, path_with_namespace + ".git")
|
|
)
|
|
end
|
|
|
|
def exists?
|
|
raw_repository
|
|
end
|
|
|
|
def empty?
|
|
raw_repository.empty?
|
|
end
|
|
|
|
#
|
|
# Git repository can contains some hidden refs like:
|
|
# /refs/notes/*
|
|
# /refs/git-as-svn/*
|
|
# /refs/pulls/*
|
|
# This refs by default not visible in project page and not cloned to client side.
|
|
#
|
|
# This method return true if repository contains some content visible in project page.
|
|
#
|
|
def has_visible_content?
|
|
raw_repository.branch_count > 0
|
|
end
|
|
|
|
def commit(id = 'HEAD')
|
|
return nil unless raw_repository
|
|
commit = Gitlab::Git::Commit.find(raw_repository, id)
|
|
commit = Commit.new(commit, @project) if commit
|
|
commit
|
|
rescue Rugged::OdbError
|
|
nil
|
|
end
|
|
|
|
def commits(ref, path = nil, limit = nil, offset = nil, skip_merges = false)
|
|
options = {
|
|
repo: raw_repository,
|
|
ref: ref,
|
|
path: path,
|
|
limit: limit,
|
|
offset: offset,
|
|
# --follow doesn't play well with --skip. See:
|
|
# https://gitlab.com/gitlab-org/gitlab-ce/issues/3574#note_3040520
|
|
follow: false
|
|
}
|
|
|
|
commits = Gitlab::Git::Commit.where(options)
|
|
commits = Commit.decorate(commits, @project) if commits.present?
|
|
commits
|
|
end
|
|
|
|
def commits_between(from, to)
|
|
commits = Gitlab::Git::Commit.between(raw_repository, from, to)
|
|
commits = Commit.decorate(commits, @project) if commits.present?
|
|
commits
|
|
end
|
|
|
|
def find_commits_by_message(query, ref = nil, path = nil, limit = 1000, offset = 0)
|
|
ref ||= root_ref
|
|
|
|
# Limited to 1000 commits for now, could be parameterized?
|
|
args = %W(#{Gitlab.config.git.bin_path} log #{ref} --pretty=%H --skip #{offset} --max-count #{limit} --grep=#{query})
|
|
args = args.concat(%W(-- #{path})) if path.present?
|
|
|
|
git_log_results = Gitlab::Popen.popen(args, path_to_repo).first.lines.map(&:chomp)
|
|
commits = git_log_results.map { |c| commit(c) }
|
|
commits
|
|
end
|
|
|
|
def find_branch(name)
|
|
raw_repository.branches.find { |branch| branch.name == name }
|
|
end
|
|
|
|
def find_tag(name)
|
|
raw_repository.tags.find { |tag| tag.name == name }
|
|
end
|
|
|
|
def add_branch(user, branch_name, target)
|
|
oldrev = Gitlab::Git::BLANK_SHA
|
|
ref = Gitlab::Git::BRANCH_REF_PREFIX + branch_name
|
|
target = commit(target).try(:id)
|
|
|
|
return false unless target
|
|
|
|
GitHooksService.new.execute(user, path_to_repo, oldrev, target, ref) do
|
|
rugged.branches.create(branch_name, target)
|
|
end
|
|
|
|
expire_branches_cache
|
|
find_branch(branch_name)
|
|
end
|
|
|
|
def add_tag(tag_name, ref, message = nil)
|
|
expire_tags_cache
|
|
|
|
gitlab_shell.add_tag(path_with_namespace, tag_name, ref, message)
|
|
end
|
|
|
|
def rm_branch(user, branch_name)
|
|
expire_branches_cache
|
|
|
|
branch = find_branch(branch_name)
|
|
oldrev = branch.try(:target)
|
|
newrev = Gitlab::Git::BLANK_SHA
|
|
ref = Gitlab::Git::BRANCH_REF_PREFIX + branch_name
|
|
|
|
GitHooksService.new.execute(user, path_to_repo, oldrev, newrev, ref) do
|
|
rugged.branches.delete(branch_name)
|
|
end
|
|
|
|
expire_branches_cache
|
|
true
|
|
end
|
|
|
|
def rm_tag(tag_name)
|
|
expire_tags_cache
|
|
|
|
gitlab_shell.rm_tag(path_with_namespace, tag_name)
|
|
end
|
|
|
|
def branch_names
|
|
cache.fetch(:branch_names) { raw_repository.branch_names }
|
|
end
|
|
|
|
def tag_names
|
|
cache.fetch(:tag_names) { raw_repository.tag_names }
|
|
end
|
|
|
|
def commit_count
|
|
cache.fetch(:commit_count) do
|
|
begin
|
|
raw_repository.commit_count(self.root_ref)
|
|
rescue
|
|
0
|
|
end
|
|
end
|
|
end
|
|
|
|
# Return repo size in megabytes
|
|
# Cached in redis
|
|
def size
|
|
cache.fetch(:size) { raw_repository.size }
|
|
end
|
|
|
|
def diverging_commit_counts(branch)
|
|
root_ref_hash = raw_repository.rev_parse_target(root_ref).oid
|
|
cache.fetch(:"diverging_commit_counts_#{branch.name}") do
|
|
# Rugged seems to throw a `ReferenceError` when given branch_names rather
|
|
# than SHA-1 hashes
|
|
number_commits_behind = commits_between(branch.target, root_ref_hash).size
|
|
number_commits_ahead = commits_between(root_ref_hash, branch.target).size
|
|
|
|
{ behind: number_commits_behind, ahead: number_commits_ahead }
|
|
end
|
|
end
|
|
|
|
def cache_keys
|
|
%i(size branch_names tag_names commit_count
|
|
readme version contribution_guide changelog license)
|
|
end
|
|
|
|
def branch_cache_keys
|
|
branches.map do |branch|
|
|
:"diverging_commit_counts_#{branch.name}"
|
|
end
|
|
end
|
|
|
|
def build_cache
|
|
cache_keys.each do |key|
|
|
unless cache.exist?(key)
|
|
send(key)
|
|
end
|
|
end
|
|
|
|
branches.each do |branch|
|
|
unless cache.exist?(:"diverging_commit_counts_#{branch.name}")
|
|
send(:diverging_commit_counts, branch)
|
|
end
|
|
end
|
|
end
|
|
|
|
def expire_tags_cache
|
|
cache.expire(:tag_names)
|
|
@tags = nil
|
|
end
|
|
|
|
def expire_branches_cache
|
|
cache.expire(:branch_names)
|
|
@branches = nil
|
|
end
|
|
|
|
def expire_cache
|
|
cache_keys.each do |key|
|
|
cache.expire(key)
|
|
end
|
|
|
|
expire_branch_cache
|
|
end
|
|
|
|
def expire_branch_cache
|
|
branches.each do |branch|
|
|
cache.expire(:"diverging_commit_counts_#{branch.name}")
|
|
end
|
|
end
|
|
|
|
def rebuild_cache
|
|
cache_keys.each do |key|
|
|
cache.expire(key)
|
|
send(key)
|
|
end
|
|
|
|
branches.each do |branch|
|
|
cache.expire(:"diverging_commit_counts_#{branch.name}")
|
|
diverging_commit_counts(branch)
|
|
end
|
|
end
|
|
|
|
def lookup_cache
|
|
@lookup_cache ||= {}
|
|
end
|
|
|
|
def expire_branch_names
|
|
cache.expire(:branch_names)
|
|
end
|
|
|
|
def method_missing(m, *args, &block)
|
|
if m == :lookup && !block_given?
|
|
lookup_cache[m] ||= {}
|
|
lookup_cache[m][args.join(":")] ||= raw_repository.send(m, *args, &block)
|
|
else
|
|
raw_repository.send(m, *args, &block)
|
|
end
|
|
end
|
|
|
|
def respond_to_missing?(method, include_private = false)
|
|
raw_repository.respond_to?(method, include_private) || super
|
|
end
|
|
|
|
def blob_at(sha, path)
|
|
unless Gitlab::Git.blank_ref?(sha)
|
|
Gitlab::Git::Blob.find(self, sha, path)
|
|
end
|
|
end
|
|
|
|
def blob_by_oid(oid)
|
|
Gitlab::Git::Blob.raw(self, oid)
|
|
end
|
|
|
|
def readme
|
|
cache.fetch(:readme) { tree(:head).readme }
|
|
end
|
|
|
|
def version
|
|
cache.fetch(:version) do
|
|
tree(:head).blobs.find do |file|
|
|
file.name.downcase == 'version'
|
|
end
|
|
end
|
|
end
|
|
|
|
def contribution_guide
|
|
cache.fetch(:contribution_guide) do
|
|
tree(:head).blobs.find do |file|
|
|
file.contributing?
|
|
end
|
|
end
|
|
end
|
|
|
|
def changelog
|
|
cache.fetch(:changelog) do
|
|
tree(:head).blobs.find do |file|
|
|
file.name =~ /\A(changelog|history)/i
|
|
end
|
|
end
|
|
end
|
|
|
|
def license
|
|
cache.fetch(:license) do
|
|
licenses = tree(:head).blobs.find_all do |file|
|
|
file.name =~ /\A(copying|license|licence)/i
|
|
end
|
|
|
|
preferences = [
|
|
/\Alicen[sc]e\z/i, # LICENSE, LICENCE
|
|
/\Alicen[sc]e\./i, # LICENSE.md, LICENSE.txt
|
|
/\Acopying\z/i, # COPYING
|
|
/\Acopying\.(?!lesser)/i, # COPYING.txt
|
|
/Acopying.lesser/i # COPYING.LESSER
|
|
]
|
|
|
|
license = nil
|
|
preferences.each do |r|
|
|
license = licenses.find { |l| l.name =~ r }
|
|
break if license
|
|
end
|
|
|
|
license
|
|
end
|
|
end
|
|
|
|
def head_commit
|
|
@head_commit ||= commit(self.root_ref)
|
|
end
|
|
|
|
def head_tree
|
|
@head_tree ||= Tree.new(self, head_commit.sha, nil)
|
|
end
|
|
|
|
def tree(sha = :head, path = nil)
|
|
if sha == :head
|
|
if path.nil?
|
|
return head_tree
|
|
else
|
|
sha = head_commit.sha
|
|
end
|
|
end
|
|
|
|
Tree.new(self, sha, path)
|
|
end
|
|
|
|
def blob_at_branch(branch_name, path)
|
|
last_commit = commit(branch_name)
|
|
|
|
if last_commit
|
|
blob_at(last_commit.sha, path)
|
|
else
|
|
nil
|
|
end
|
|
end
|
|
|
|
# Returns url for submodule
|
|
#
|
|
# Ex.
|
|
# @repository.submodule_url_for('master', 'rack')
|
|
# # => git@localhost:rack.git
|
|
#
|
|
def submodule_url_for(ref, path)
|
|
if submodules(ref).any?
|
|
submodule = submodules(ref)[path]
|
|
|
|
if submodule
|
|
submodule['url']
|
|
end
|
|
end
|
|
end
|
|
|
|
def last_commit_for_path(sha, path)
|
|
args = %W(#{Gitlab.config.git.bin_path} rev-list --max-count=1 #{sha} -- #{path})
|
|
sha = Gitlab::Popen.popen(args, path_to_repo).first.strip
|
|
commit(sha)
|
|
end
|
|
|
|
def next_patch_branch
|
|
patch_branch_ids = self.branch_names.map do |n|
|
|
result = n.match(/\Apatch-([0-9]+)\z/)
|
|
result[1].to_i if result
|
|
end.compact
|
|
|
|
highest_patch_branch_id = patch_branch_ids.max || 0
|
|
|
|
"patch-#{highest_patch_branch_id + 1}"
|
|
end
|
|
|
|
# Remove archives older than 2 hours
|
|
def branches_sorted_by(value)
|
|
case value
|
|
when 'recently_updated'
|
|
branches.sort do |a, b|
|
|
commit(b.target).committed_date <=> commit(a.target).committed_date
|
|
end
|
|
when 'last_updated'
|
|
branches.sort do |a, b|
|
|
commit(a.target).committed_date <=> commit(b.target).committed_date
|
|
end
|
|
else
|
|
branches
|
|
end
|
|
end
|
|
|
|
def contributors
|
|
commits = self.commits(nil, nil, 2000, 0, true)
|
|
|
|
commits.group_by(&:author_email).map do |email, commits|
|
|
contributor = Gitlab::Contributor.new
|
|
contributor.email = email
|
|
|
|
commits.each do |commit|
|
|
if contributor.name.blank?
|
|
contributor.name = commit.author_name
|
|
end
|
|
|
|
contributor.commits += 1
|
|
end
|
|
|
|
contributor
|
|
end
|
|
end
|
|
|
|
def blob_for_diff(commit, diff)
|
|
blob_at(commit.id, diff.file_path)
|
|
end
|
|
|
|
def prev_blob_for_diff(commit, diff)
|
|
if commit.parent_id
|
|
blob_at(commit.parent_id, diff.old_path)
|
|
end
|
|
end
|
|
|
|
def refs_contains_sha(ref_type, sha)
|
|
args = %W(#{Gitlab.config.git.bin_path} #{ref_type} --contains #{sha})
|
|
names = Gitlab::Popen.popen(args, path_to_repo).first
|
|
|
|
if names.respond_to?(:split)
|
|
names = names.split("\n").map(&:strip)
|
|
|
|
names.each do |name|
|
|
name.slice! '* '
|
|
end
|
|
|
|
names
|
|
else
|
|
[]
|
|
end
|
|
end
|
|
|
|
def branch_names_contains(sha)
|
|
refs_contains_sha('branch', sha)
|
|
end
|
|
|
|
def tag_names_contains(sha)
|
|
refs_contains_sha('tag', sha)
|
|
end
|
|
|
|
def branches
|
|
@branches ||= raw_repository.branches
|
|
end
|
|
|
|
def tags
|
|
@tags ||= raw_repository.tags
|
|
end
|
|
|
|
def root_ref
|
|
@root_ref ||= raw_repository.root_ref
|
|
end
|
|
|
|
def commit_dir(user, path, message, branch)
|
|
commit_with_hooks(user, branch) do |ref|
|
|
committer = user_to_committer(user)
|
|
options = {}
|
|
options[:committer] = committer
|
|
options[:author] = committer
|
|
|
|
options[:commit] = {
|
|
message: message,
|
|
branch: ref,
|
|
}
|
|
|
|
raw_repository.mkdir(path, options)
|
|
end
|
|
end
|
|
|
|
def commit_file(user, path, content, message, branch, update)
|
|
commit_with_hooks(user, branch) do |ref|
|
|
committer = user_to_committer(user)
|
|
options = {}
|
|
options[:committer] = committer
|
|
options[:author] = committer
|
|
options[:commit] = {
|
|
message: message,
|
|
branch: ref,
|
|
}
|
|
|
|
options[:file] = {
|
|
content: content,
|
|
path: path,
|
|
update: update
|
|
}
|
|
|
|
Gitlab::Git::Blob.commit(raw_repository, options)
|
|
end
|
|
end
|
|
|
|
def remove_file(user, path, message, branch)
|
|
commit_with_hooks(user, branch) do |ref|
|
|
committer = user_to_committer(user)
|
|
options = {}
|
|
options[:committer] = committer
|
|
options[:author] = committer
|
|
options[:commit] = {
|
|
message: message,
|
|
branch: ref
|
|
}
|
|
|
|
options[:file] = {
|
|
path: path
|
|
}
|
|
|
|
Gitlab::Git::Blob.remove(raw_repository, options)
|
|
end
|
|
end
|
|
|
|
def user_to_committer(user)
|
|
{
|
|
email: user.email,
|
|
name: user.name,
|
|
time: Time.now
|
|
}
|
|
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
|
|
|
|
def merge(user, source_sha, target_branch, options = {})
|
|
our_commit = rugged.branches[target_branch].target
|
|
their_commit = rugged.lookup(source_sha)
|
|
|
|
raise "Invalid merge target" if our_commit.nil?
|
|
raise "Invalid merge source" if their_commit.nil?
|
|
|
|
merge_index = rugged.merge_commits(our_commit, their_commit)
|
|
return false if merge_index.conflicts?
|
|
|
|
commit_with_hooks(user, target_branch) do |ref|
|
|
actual_options = options.merge(
|
|
parents: [our_commit, their_commit],
|
|
tree: merge_index.write_tree(rugged),
|
|
update_ref: ref
|
|
)
|
|
|
|
Rugged::Commit.create(rugged, actual_options)
|
|
end
|
|
end
|
|
|
|
def merged_to_root_ref?(branch_name)
|
|
branch_commit = commit(branch_name)
|
|
root_ref_commit = commit(root_ref)
|
|
|
|
if branch_commit
|
|
is_ancestor?(branch_commit.id, root_ref_commit.id)
|
|
else
|
|
nil
|
|
end
|
|
end
|
|
|
|
def merge_base(first_commit_id, second_commit_id)
|
|
rugged.merge_base(first_commit_id, second_commit_id)
|
|
rescue Rugged::ReferenceError
|
|
nil
|
|
end
|
|
|
|
def is_ancestor?(ancestor_id, descendant_id)
|
|
merge_base(ancestor_id, descendant_id) == ancestor_id
|
|
end
|
|
|
|
|
|
def search_files(query, ref)
|
|
offset = 2
|
|
args = %W(#{Gitlab.config.git.bin_path} grep -i -I -n --before-context #{offset} --after-context #{offset} -e #{query} #{ref || root_ref})
|
|
Gitlab::Popen.popen(args, path_to_repo).first.scrub.split(/^--$/)
|
|
end
|
|
|
|
def parse_search_result(result)
|
|
ref = nil
|
|
filename = nil
|
|
startline = 0
|
|
|
|
result.each_line.each_with_index do |line, index|
|
|
if line =~ /^.*:.*:\d+:/
|
|
ref, filename, startline = line.split(':')
|
|
startline = startline.to_i - index
|
|
break
|
|
end
|
|
end
|
|
|
|
data = ""
|
|
|
|
result.each_line do |line|
|
|
data << line.sub(ref, '').sub(filename, '').sub(/^:-\d+-/, '').sub(/^::\d+:/, '')
|
|
end
|
|
|
|
OpenStruct.new(
|
|
filename: filename,
|
|
ref: ref,
|
|
startline: startline,
|
|
data: data
|
|
)
|
|
end
|
|
|
|
def fetch_ref(source_path, source_ref, target_ref)
|
|
args = %W(#{Gitlab.config.git.bin_path} fetch -f #{source_path} #{source_ref}:#{target_ref})
|
|
Gitlab::Popen.popen(args, path_to_repo)
|
|
end
|
|
|
|
def with_tmp_ref(oldrev = nil)
|
|
random_string = SecureRandom.hex
|
|
tmp_ref = "refs/tmp/#{random_string}/head"
|
|
|
|
if oldrev && !Gitlab::Git.blank_ref?(oldrev)
|
|
rugged.references.create(tmp_ref, oldrev)
|
|
end
|
|
|
|
# Make commit in tmp ref
|
|
yield(tmp_ref)
|
|
ensure
|
|
rugged.references.delete(tmp_ref) rescue nil
|
|
end
|
|
|
|
def commit_with_hooks(current_user, branch)
|
|
oldrev = Gitlab::Git::BLANK_SHA
|
|
ref = Gitlab::Git::BRANCH_REF_PREFIX + branch
|
|
was_empty = empty?
|
|
|
|
unless was_empty
|
|
oldrev = find_branch(branch).target
|
|
end
|
|
|
|
with_tmp_ref(oldrev) do |tmp_ref|
|
|
# Make commit in tmp ref
|
|
newrev = yield(tmp_ref)
|
|
|
|
unless newrev
|
|
raise CommitError.new('Failed to create commit')
|
|
end
|
|
|
|
GitHooksService.new.execute(current_user, path_to_repo, oldrev, newrev, ref) do
|
|
if was_empty
|
|
# Create branch
|
|
rugged.references.create(ref, newrev)
|
|
else
|
|
# Update head
|
|
current_head = find_branch(branch).target
|
|
|
|
# Make sure target branch was not changed during pre-receive hook
|
|
if current_head == oldrev
|
|
rugged.references.update(ref, newrev)
|
|
else
|
|
raise CommitError.new('Commit was rejected because branch received new push')
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def ls_files(ref)
|
|
actual_ref = ref || root_ref
|
|
raw_repository.ls_files(actual_ref)
|
|
end
|
|
|
|
private
|
|
|
|
def cache
|
|
@cache ||= RepositoryCache.new(path_with_namespace)
|
|
end
|
|
end
|