debian-mirror-gitlab/app/models/merge_request_diff.rb

206 lines
4.7 KiB
Ruby
Raw Normal View History

2014-09-02 18:07:02 +05:30
# == Schema Information
#
# Table name: merge_request_diffs
#
# id :integer not null, primary key
2015-04-26 12:48:37 +05:30
# state :string(255)
2014-09-02 18:07:02 +05:30
# st_commits :text
# st_diffs :text
# merge_request_id :integer not null
# created_at :datetime
# updated_at :datetime
#
require Rails.root.join("app/models/commit")
class MergeRequestDiff < ActiveRecord::Base
2015-04-26 12:48:37 +05:30
include Sortable
2015-09-25 12:07:36 +05:30
# Prevent store of diff if commits amount more then 500
COMMITS_SAFE_SIZE = 500
2014-09-02 18:07:02 +05:30
2015-11-26 14:37:03 +05:30
attr_reader :commits, :diffs, :diffs_no_whitespace
2014-09-02 18:07:02 +05:30
belongs_to :merge_request
delegate :target_branch, :source_branch, to: :merge_request, prefix: nil
state_machine :state, initial: :empty do
state :collected
state :timeout
state :overflow_commits_safe_size
state :overflow_diff_files_limit
state :overflow_diff_lines_limit
end
serialize :st_commits
serialize :st_diffs
after_create :reload_content
def reload_content
reload_commits
reload_diffs
end
def diffs
@diffs ||= (load_diffs(st_diffs) || [])
end
2015-11-26 14:37:03 +05:30
def diffs_no_whitespace
# Get latest sha of branch from source project
source_sha = merge_request.source_project.commit(source_branch).sha
compare_result = Gitlab::CompareResult.new(
Gitlab::Git::Compare.new(
merge_request.target_project.repository.raw_repository,
merge_request.target_branch,
source_sha,
), { ignore_whitespace_change: true }
)
@diffs_no_whitespace ||= load_diffs(dump_commits(compare_result.diffs))
end
2014-09-02 18:07:02 +05:30
def commits
@commits ||= load_commits(st_commits || [])
end
def last_commit
commits.first
end
2015-10-24 18:46:33 +05:30
def first_commit
commits.last
end
2016-01-29 22:53:50 +05:30
def base_commit
return nil unless self.base_commit_sha
merge_request.target_project.commit(self.base_commit_sha)
end
2014-09-02 18:07:02 +05:30
def last_commit_short_sha
2015-04-26 12:48:37 +05:30
@last_commit_short_sha ||= last_commit.short_id
2014-09-02 18:07:02 +05:30
end
private
def dump_commits(commits)
commits.map(&:to_hash)
end
def load_commits(array)
2015-09-11 14:41:01 +05:30
array.map { |hash| Commit.new(Gitlab::Git::Commit.new(hash), merge_request.source_project) }
2014-09-02 18:07:02 +05:30
end
def dump_diffs(diffs)
if diffs.respond_to?(:map)
diffs.map(&:to_hash)
end
end
def load_diffs(raw)
if raw.respond_to?(:map)
raw.map { |hash| Gitlab::Git::Diff.new(hash) }
end
end
# Collect array of Git::Commit objects
# between target and source branches
def unmerged_commits
commits = compare_result.commits
if commits.present?
2015-09-11 14:41:01 +05:30
commits = Commit.decorate(commits, merge_request.source_project).
2014-09-02 18:07:02 +05:30
sort_by(&:created_at).
reverse
end
commits
end
# Reload all commits related to current merge request from repo
# and save it as array of hashes in st_commits db field
def reload_commits
commit_objects = unmerged_commits
if commit_objects.present?
self.st_commits = dump_commits(commit_objects)
end
save
end
# Reload diffs between branches related to current merge request from repo
# and save it as array of hashes in st_diffs db field
def reload_diffs
new_diffs = []
if commits.size.zero?
self.state = :empty
elsif commits.size > COMMITS_SAFE_SIZE
self.state = :overflow_commits_safe_size
else
new_diffs = unmerged_diffs
end
if new_diffs.any?
if new_diffs.size > Commit::DIFF_HARD_LIMIT_FILES
self.state = :overflow_diff_files_limit
2015-10-24 18:46:33 +05:30
new_diffs = new_diffs.first(Commit::DIFF_HARD_LIMIT_LINES)
2014-09-02 18:07:02 +05:30
end
if new_diffs.sum { |diff| diff.diff.lines.count } > Commit::DIFF_HARD_LIMIT_LINES
self.state = :overflow_diff_lines_limit
2015-10-24 18:46:33 +05:30
new_diffs = new_diffs.first(Commit::DIFF_HARD_LIMIT_LINES)
2014-09-02 18:07:02 +05:30
end
end
if new_diffs.present?
new_diffs = dump_commits(new_diffs)
self.state = :collected
end
self.st_diffs = new_diffs
2016-01-29 22:53:50 +05:30
self.base_commit_sha = merge_request.target_project.commit(target_branch).try(:sha)
2014-09-02 18:07:02 +05:30
self.save
end
# Collect array of Git::Diff objects
# between target and source branches
def unmerged_diffs
2015-10-24 18:46:33 +05:30
compare_result.diffs || []
rescue Gitlab::Git::Diff::TimeoutError
2014-09-02 18:07:02 +05:30
self.state = :timeout
2015-10-24 18:46:33 +05:30
[]
2014-09-02 18:07:02 +05:30
end
def repository
merge_request.target_project.repository
end
private
def compare_result
2015-09-25 12:07:36 +05:30
@compare_result ||=
begin
# Update ref for merge request
merge_request.fetch_ref
# Get latest sha of branch from source project
2015-10-24 18:46:33 +05:30
source_commit = merge_request.source_project.commit(source_branch)
source_sha = source_commit.try(:sha)
2015-09-25 12:07:36 +05:30
Gitlab::CompareResult.new(
Gitlab::Git::Compare.new(
merge_request.target_project.repository.raw_repository,
merge_request.target_branch,
source_sha,
)
)
end
2014-09-02 18:07:02 +05:30
end
end