rosa-build/app/models/pull_request.rb

278 lines
8.3 KiB
Ruby

class PullRequest < ActiveRecord::Base
STATUSES = [
STATUS_OPEN = 'open',
STATUS_READY = 'ready',
STATUS_ALREADY = 'already',
STATUS_BLOCKED = 'blocked',
STATUS_MERGED = 'merged',
STATUS_CLOSED = 'closed'
]
belongs_to :issue,
autosave: true,
dependent: :destroy,
touch: true,
validate: true
belongs_to :to_project,
class_name: 'Project',
foreign_key: 'to_project_id'
belongs_to :from_project,
class_name: 'Project',
foreign_key: 'from_project_id'
delegate :user,
:user_id,
:title,
:body,
:serial_id,
:assignee,
:status,
:to_param,
:created_at,
:updated_at,
:comments,
:status=,
to: :issue, allow_nil: true
validates :from_project, :to_project, presence: true
validate :uniq_merge, if: ->(pull) { pull.to_project.present? }
validates_each :from_ref, :to_ref do |record, attr, value|
check_ref record, attr, value
end
before_create :clean_dir
before_create :set_add_data
after_destroy :clean_dir
accepts_nested_attributes_for :issue
attr_accessible :issue_attributes, :to_ref, :from_ref
scope :needed_checking, -> { includes(:issue).where(issues: { status: [STATUS_OPEN, STATUS_BLOCKED, STATUS_READY] }) }
scope :not_closed_or_merged, -> { needed_checking }
scope :closed_or_merged, -> { where(issues: { status: [STATUS_CLOSED, STATUS_MERGED] }) }
state_machine :status, initial: :open do
event :ready do
transition [:ready, :open, :blocked] => :ready
end
event :already do
transition [:ready, :open, :blocked] => :already
end
event :block do
transition [:ready, :open, :blocked] => :blocked
end
event :merging do
transition ready: :merged
end
event :close do
transition [:ready, :open, :blocked] => :closed
end
event :reopen do
transition closed: :open
end
end
def update_relations(old_from_project_name = nil)
FileUtils.mv path(old_from_project_name), path, force: true if old_from_project_name
return unless Dir.exists?(path)
Dir.chdir(path) do
system 'git', 'remote', 'set-url', 'origin', to_project.path
system 'git', 'remote', 'set-url', 'head', from_project.path if cross_pull?
end
end
later :update_relations, queue: :middle
def cross_pull?
from_project_id != to_project_id
end
def check(do_transaction = true)
if do_transaction && !valid?
issue.set_close nil
issue.save(validate: false) # FIXME remove this hack
return false
end
res = merge
new_status = case res
when /Already up-to-date/
'already'
when /Merge made by/
system("cd #{path} && git reset --hard HEAD^") # remove merge commit
'ready'
when /Automatic merge failed/
system("cd #{path} && git reset --hard HEAD") # clean git index
'block'
else
raise res
end
if do_transaction
new_status == 'already' ? (ready; merging) : send(new_status)
self.update_inline_comments
else
self.status = new_status == 'block' ? STATUS_BLOCKED : new_status
end
end
def merge!(who)
return false unless can_merging?
Dir.chdir(path) do
old_commit = to_project.repo.commits(to_ref).first
commit = repo.commits(to_ref).first
system "git config user.name \"#{who.uname}\" && git config user.email \"#{who.email}\""
res = merge
if commit.id != repo.commits(to_ref).first.id
res2 = %x(export GL_ID=user-#{who.id} GL_REPO_NAME=#{to_project.path} && git push origin HEAD)
system("git reset --hard HEAD^") # for diff maybe FIXME
if old_commit.id == to_project.repo.commits(to_ref).first.id
raise "merge result pull_request #{id}: #{$?.exitstatus}; #{res2}; #{res}"
end
set_user_and_time who
merging
else # Try to catch no merge errors
raise "merge result pull_request #{id}: #{res}"
end
end
end
def path(suffix = from_project_name)
last_part = [id, from_project_owner_uname, suffix].compact.join('-')
File.join(APP_CONFIG['git_path'], "#{new_record? ? 'temp_' : ''}pull_requests", to_project.owner.uname, to_project.name, last_part)
end
def from_branch
cross_pull? ? "head_#{from_ref}" : from_ref
end
def common_ancestor
return @common_ancestor if @common_ancestor
base_commit = repo.commits(to_ref).first
@common_ancestor = repo.commit(repo.git.merge_base({}, base_commit, from_commit)) || base_commit
end
alias_method :to_commit, :common_ancestor
def diff_stats
@diff_stats ||= repo.diff_stats(to_commit.id, from_commit.id)
end
# FIXME maybe move to warpc/grit?
def diff
@diff ||= Grit::Commit.diff(repo, to_commit.id, from_commit.id)
end
def set_user_and_time user
issue.closed_at = Time.now.utc
issue.closer = user
end
def self.check_ref(record, attr, value)
project = attr == :from_ref ? record.from_project : record.to_project
return if project.blank?
if record.to_project.repo.branches.count > 0
record.errors.add attr, I18n.t('projects.pull_requests.wrong_ref') unless project.repo.branches_and_tags.map(&:name).include?(value)
else
record.errors.add attr, I18n.t('projects.pull_requests.empty_repo')
end
end
def uniq_merge
if to_project && to_project.pull_requests.needed_checking
.where(from_project_id: from_project_id,
to_ref: to_ref, from_ref: from_ref)
.where('pull_requests.id <> :id or :id is null', id: id).count > 0
errors.add(:base_branch, I18n.t('projects.pull_requests.duplicate', from_ref: from_ref))
end
end
def repo
@repo ||= Grit::Repo.new(path)
end
def from_commit
repo.commits(from_branch).first
end
protected
def merge
clone
message = "Merge pull request ##{serial_id} from #{from_project_owner_uname}/#{from_project_name}:#{from_ref}\r\n #{title}"
%x(cd #{path} && git checkout #{to_ref.shellescape} && git merge --no-ff #{from_branch.shellescape} -m #{message.shellescape})
end
def clone
return if from_project.nil?
git = Grit::Git.new(path)
if new_record? || !git.exist?
#~ FileUtils.mkdir_p(path)
#~ system("git clone --local --no-hardlinks #{to_project.path} #{path}")
options = {bare: false, shared: false, branch: to_ref} # shared?
`rm -rf #{path}`
git.fs_mkdir('..')
git.clone(options, to_project.path, path)
if cross_pull?
Dir.chdir(path) do
system 'git', 'remote', 'add', 'head', from_project.path
end
end
#clean # Need testing
end
Dir.chdir(path) do
system 'git', 'tag', '-d', from_ref, to_ref
system 'git fetch --tags && git fetch --all'
tags, head = repo.tags.map(&:name), to_project == from_project ? 'origin' : 'head'
system 'git', 'checkout', to_ref
unless tags.include? to_ref
system 'git', 'reset', '--hard', "origin/#{to_ref}"
end
unless tags.include? from_ref
system 'git', 'branch', '-D', from_branch
system 'git', 'fetch', head, "+#{from_ref}:#{from_branch}"
system 'git', 'checkout', to_ref
end
end
end
def clean
Dir.chdir(path) do
to_project.repo.branches.each {|branch| system 'git', 'checkout', branch.name}
system 'git', 'checkout', to_ref
to_project.repo.branches.each do |branch|
system 'git', 'branch', '-D', branch.name unless [to_ref, from_branch].include? branch.name
end
to_project.repo.tags.each do |tag|
system 'git', 'tag', '-d', tag.name unless [to_ref, from_branch].include? tag.name
end
end
end
def clean_dir
FileUtils.rm_rf path
end
def update_inline_comments
self.comments.each do |c|
if c.data.present? # maybe need add new column 'actual'?
c.actual_inline_comment? diff, true
c.save
end
end
end
def set_add_data
self.from_project_owner_uname = from_project.owner.uname
self.from_project_name = from_project.name
end
end