gitlab-org--gitlab-foss/lib/gitlab/gitaly_client/conflict_files_stitcher.rb

52 lines
1.2 KiB
Ruby

# frozen_string_literal: true
module Gitlab
module GitalyClient
class ConflictFilesStitcher
include Enumerable
attr_reader :gitaly_repo
def initialize(rpc_response, gitaly_repo)
@rpc_response = rpc_response
@gitaly_repo = gitaly_repo
end
def each
current_file = nil
@rpc_response.each do |msg|
msg.files.each do |gitaly_file|
if gitaly_file.header
yield current_file if current_file
current_file = file_from_gitaly_header(gitaly_file.header)
else
current_file.raw_content = "#{current_file.raw_content}#{gitaly_file.content}"
end
end
end
yield current_file if current_file
end
private
def file_from_gitaly_header(header)
Gitlab::Git::Conflict::File.new(
Gitlab::GitalyClient::Util.git_repository(gitaly_repo),
header.commit_oid,
conflict_from_gitaly_file_header(header),
''
)
end
def conflict_from_gitaly_file_header(header)
{
ours: { path: header.our_path, mode: header.our_mode },
theirs: { path: header.their_path }
}
end
end
end
end