|
|
# -*- coding: utf-8 -*-
|
|
|
"""
|
|
|
rhodecode.controllers.pullrequests
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
|
|
pull requests controller for rhodecode for initializing pull requests
|
|
|
|
|
|
:created_on: May 7, 2012
|
|
|
:author: marcink
|
|
|
:copyright: (C) 2010-2012 Marcin Kuzminski <marcin@python-works.com>
|
|
|
:license: GPLv3, see COPYING for more details.
|
|
|
"""
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
# (at your option) any later version.
|
|
|
#
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
# GNU General Public License for more details.
|
|
|
#
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
import logging
|
|
|
import traceback
|
|
|
|
|
|
from webob.exc import HTTPNotFound, HTTPForbidden
|
|
|
from collections import defaultdict
|
|
|
from itertools import groupby
|
|
|
|
|
|
from pylons import request, response, session, tmpl_context as c, url
|
|
|
from pylons.controllers.util import abort, redirect
|
|
|
from pylons.i18n.translation import _
|
|
|
from pylons.decorators import jsonify
|
|
|
|
|
|
from rhodecode.lib.base import BaseRepoController, render
|
|
|
from rhodecode.lib.auth import LoginRequired, HasRepoPermissionAnyDecorator
|
|
|
from rhodecode.lib import helpers as h
|
|
|
from rhodecode.lib import diffs
|
|
|
from rhodecode.lib.utils import action_logger
|
|
|
from rhodecode.model.db import User, PullRequest, ChangesetStatus,\
|
|
|
ChangesetComment
|
|
|
from rhodecode.model.pull_request import PullRequestModel
|
|
|
from rhodecode.model.meta import Session
|
|
|
from rhodecode.model.repo import RepoModel
|
|
|
from rhodecode.model.comment import ChangesetCommentsModel
|
|
|
from rhodecode.model.changeset_status import ChangesetStatusModel
|
|
|
|
|
|
log = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
class PullrequestsController(BaseRepoController):
|
|
|
|
|
|
@LoginRequired()
|
|
|
@HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
|
|
|
'repository.admin')
|
|
|
def __before__(self):
|
|
|
super(PullrequestsController, self).__before__()
|
|
|
|
|
|
def _get_repo_refs(self, repo):
|
|
|
hist_l = []
|
|
|
|
|
|
branches_group = ([('branch:%s:%s' % (k, v), k) for
|
|
|
k, v in repo.branches.iteritems()], _("Branches"))
|
|
|
bookmarks_group = ([('book:%s:%s' % (k, v), k) for
|
|
|
k, v in repo.bookmarks.iteritems()], _("Bookmarks"))
|
|
|
tags_group = ([('tag:%s:%s' % (k, v), k) for
|
|
|
k, v in repo.tags.iteritems()], _("Tags"))
|
|
|
|
|
|
hist_l.append(bookmarks_group)
|
|
|
hist_l.append(branches_group)
|
|
|
hist_l.append(tags_group)
|
|
|
|
|
|
return hist_l
|
|
|
|
|
|
def show_all(self, repo_name):
|
|
|
c.pull_requests = PullRequestModel().get_all(repo_name)
|
|
|
c.repo_name = repo_name
|
|
|
return render('/pullrequests/pullrequest_show_all.html')
|
|
|
|
|
|
def index(self):
|
|
|
org_repo = c.rhodecode_db_repo
|
|
|
|
|
|
if org_repo.scm_instance.alias != 'hg':
|
|
|
log.error('Review not available for GIT REPOS')
|
|
|
raise HTTPNotFound
|
|
|
|
|
|
c.org_refs = self._get_repo_refs(c.rhodecode_repo)
|
|
|
c.org_repos = []
|
|
|
c.other_repos = []
|
|
|
c.org_repos.append((org_repo.repo_name, '%s/%s' % (
|
|
|
org_repo.user.username, c.repo_name))
|
|
|
)
|
|
|
|
|
|
c.other_refs = c.org_refs
|
|
|
c.other_repos.extend(c.org_repos)
|
|
|
c.default_pull_request = org_repo.repo_name
|
|
|
#gather forks and add to this list
|
|
|
for fork in org_repo.forks:
|
|
|
c.other_repos.append((fork.repo_name, '%s/%s' % (
|
|
|
fork.user.username, fork.repo_name))
|
|
|
)
|
|
|
#add parents of this fork also
|
|
|
if org_repo.parent:
|
|
|
c.default_pull_request = org_repo.parent.repo_name
|
|
|
c.other_repos.append((org_repo.parent.repo_name, '%s/%s' % (
|
|
|
org_repo.parent.user.username,
|
|
|
org_repo.parent.repo_name))
|
|
|
)
|
|
|
|
|
|
c.review_members = []
|
|
|
c.available_members = []
|
|
|
for u in User.query().filter(User.username != 'default').all():
|
|
|
uname = u.username
|
|
|
if org_repo.user == u:
|
|
|
uname = _('%s (owner)' % u.username)
|
|
|
# auto add owner to pull-request recipients
|
|
|
c.review_members.append([u.user_id, uname])
|
|
|
c.available_members.append([u.user_id, uname])
|
|
|
return render('/pullrequests/pullrequest.html')
|
|
|
|
|
|
def create(self, repo_name):
|
|
|
req_p = request.POST
|
|
|
org_repo = req_p['org_repo']
|
|
|
org_ref = req_p['org_ref']
|
|
|
other_repo = req_p['other_repo']
|
|
|
other_ref = req_p['other_ref']
|
|
|
revisions = req_p.getall('revisions')
|
|
|
reviewers = req_p.getall('review_members')
|
|
|
#TODO: wrap this into a FORM !!!
|
|
|
|
|
|
title = req_p['pullrequest_title']
|
|
|
description = req_p['pullrequest_desc']
|
|
|
|
|
|
try:
|
|
|
model = PullRequestModel()
|
|
|
model.create(self.rhodecode_user.user_id, org_repo,
|
|
|
org_ref, other_repo, other_ref, revisions,
|
|
|
reviewers, title, description)
|
|
|
Session.commit()
|
|
|
h.flash(_('Pull request send'), category='success')
|
|
|
except Exception:
|
|
|
raise
|
|
|
h.flash(_('Error occured during sending pull request'),
|
|
|
category='error')
|
|
|
log.error(traceback.format_exc())
|
|
|
|
|
|
return redirect(url('changelog_home', repo_name=repo_name))
|
|
|
|
|
|
def _load_compare_data(self, pull_request):
|
|
|
"""
|
|
|
Load context data needed for generating compare diff
|
|
|
|
|
|
:param pull_request:
|
|
|
:type pull_request:
|
|
|
"""
|
|
|
|
|
|
org_repo = pull_request.org_repo
|
|
|
org_ref_type, org_ref_, org_ref = pull_request.org_ref.split(':')
|
|
|
other_repo = pull_request.other_repo
|
|
|
other_ref_type, other_ref, other_ref_ = pull_request.other_ref.split(':')
|
|
|
|
|
|
org_ref = (org_ref_type, org_ref)
|
|
|
other_ref = (other_ref_type, other_ref)
|
|
|
|
|
|
c.org_repo = org_repo
|
|
|
c.other_repo = other_repo
|
|
|
|
|
|
c.cs_ranges, discovery_data = PullRequestModel().get_compare_data(
|
|
|
org_repo, org_ref, other_repo, other_ref
|
|
|
)
|
|
|
|
|
|
c.statuses = c.rhodecode_db_repo.statuses([x.raw_id for x in
|
|
|
c.cs_ranges])
|
|
|
# defines that we need hidden inputs with changesets
|
|
|
c.as_form = request.GET.get('as_form', False)
|
|
|
|
|
|
c.org_ref = org_ref[1]
|
|
|
c.other_ref = other_ref[1]
|
|
|
# diff needs to have swapped org with other to generate proper diff
|
|
|
_diff = diffs.differ(other_repo, other_ref, org_repo, org_ref,
|
|
|
discovery_data)
|
|
|
diff_processor = diffs.DiffProcessor(_diff, format='gitdiff')
|
|
|
_parsed = diff_processor.prepare()
|
|
|
|
|
|
c.files = []
|
|
|
c.changes = {}
|
|
|
|
|
|
for f in _parsed:
|
|
|
fid = h.FID('', f['filename'])
|
|
|
c.files.append([fid, f['operation'], f['filename'], f['stats']])
|
|
|
diff = diff_processor.as_html(enable_comments=True,
|
|
|
diff_lines=[f])
|
|
|
c.changes[fid] = [f['operation'], f['filename'], diff]
|
|
|
|
|
|
def show(self, repo_name, pull_request_id):
|
|
|
repo_model = RepoModel()
|
|
|
c.users_array = repo_model.get_users_js()
|
|
|
c.users_groups_array = repo_model.get_users_groups_js()
|
|
|
c.pull_request = PullRequest.get(pull_request_id)
|
|
|
|
|
|
# valid ID
|
|
|
if not c.pull_request:
|
|
|
raise HTTPNotFound
|
|
|
cc_model = ChangesetCommentsModel()
|
|
|
cs_model = ChangesetStatusModel()
|
|
|
_cs_statuses = cs_model.get_statuses(c.pull_request.org_repo,
|
|
|
pull_request=c.pull_request,
|
|
|
with_revisions=True)
|
|
|
|
|
|
cs_statuses = defaultdict(list)
|
|
|
for st in _cs_statuses:
|
|
|
cs_statuses[st.author.username] += [st]
|
|
|
|
|
|
c.pull_request_reviewers = []
|
|
|
for o in c.pull_request.reviewers:
|
|
|
st = cs_statuses.get(o.user.username, None)
|
|
|
if st:
|
|
|
sorter = lambda k: k.version
|
|
|
st = [(x, list(y)[0])
|
|
|
for x, y in (groupby(sorted(st, key=sorter), sorter))]
|
|
|
c.pull_request_reviewers.append([o.user, st])
|
|
|
|
|
|
# pull_requests repo_name we opened it against
|
|
|
# ie. other_repo must match
|
|
|
if repo_name != c.pull_request.other_repo.repo_name:
|
|
|
raise HTTPNotFound
|
|
|
|
|
|
# load compare data into template context
|
|
|
self._load_compare_data(c.pull_request)
|
|
|
|
|
|
# inline comments
|
|
|
c.inline_cnt = 0
|
|
|
c.inline_comments = cc_model.get_inline_comments(
|
|
|
c.rhodecode_db_repo.repo_id,
|
|
|
pull_request=pull_request_id)
|
|
|
# count inline comments
|
|
|
for __, lines in c.inline_comments:
|
|
|
for comments in lines.values():
|
|
|
c.inline_cnt += len(comments)
|
|
|
# comments
|
|
|
c.comments = cc_model.get_comments(c.rhodecode_db_repo.repo_id,
|
|
|
pull_request=pull_request_id)
|
|
|
|
|
|
# changeset(pull-request) status
|
|
|
c.current_changeset_status = cs_model.calculate_status(
|
|
|
c.pull_request_reviewers
|
|
|
)
|
|
|
c.changeset_statuses = ChangesetStatus.STATUSES
|
|
|
c.target_repo = c.pull_request.org_repo.repo_name
|
|
|
return render('/pullrequests/pullrequest_show.html')
|
|
|
|
|
|
@jsonify
|
|
|
def comment(self, repo_name, pull_request_id):
|
|
|
|
|
|
status = request.POST.get('changeset_status')
|
|
|
change_status = request.POST.get('change_changeset_status')
|
|
|
|
|
|
comm = ChangesetCommentsModel().create(
|
|
|
text=request.POST.get('text'),
|
|
|
repo_id=c.rhodecode_db_repo.repo_id,
|
|
|
user_id=c.rhodecode_user.user_id,
|
|
|
pull_request=pull_request_id,
|
|
|
f_path=request.POST.get('f_path'),
|
|
|
line_no=request.POST.get('line'),
|
|
|
status_change=(ChangesetStatus.get_status_lbl(status)
|
|
|
if status and change_status else None)
|
|
|
)
|
|
|
|
|
|
# get status if set !
|
|
|
if status and change_status:
|
|
|
ChangesetStatusModel().set_status(
|
|
|
c.rhodecode_db_repo.repo_id,
|
|
|
status,
|
|
|
c.rhodecode_user.user_id,
|
|
|
comm,
|
|
|
pull_request=pull_request_id
|
|
|
)
|
|
|
action_logger(self.rhodecode_user,
|
|
|
'user_commented_pull_request:%s' % pull_request_id,
|
|
|
c.rhodecode_db_repo, self.ip_addr, self.sa)
|
|
|
|
|
|
Session.commit()
|
|
|
|
|
|
if not request.environ.get('HTTP_X_PARTIAL_XHR'):
|
|
|
return redirect(h.url('pullrequest_show', repo_name=repo_name,
|
|
|
pull_request_id=pull_request_id))
|
|
|
|
|
|
data = {
|
|
|
'target_id': h.safeid(h.safe_unicode(request.POST.get('f_path'))),
|
|
|
}
|
|
|
if comm:
|
|
|
c.co = comm
|
|
|
data.update(comm.get_dict())
|
|
|
data.update({'rendered_text':
|
|
|
render('changeset/changeset_comment_block.html')})
|
|
|
|
|
|
return data
|
|
|
|
|
|
@jsonify
|
|
|
def delete_comment(self, repo_name, comment_id):
|
|
|
co = ChangesetComment.get(comment_id)
|
|
|
owner = lambda: co.author.user_id == c.rhodecode_user.user_id
|
|
|
if h.HasPermissionAny('hg.admin', 'repository.admin')() or owner:
|
|
|
ChangesetCommentsModel().delete(comment=co)
|
|
|
Session.commit()
|
|
|
return True
|
|
|
else:
|
|
|
raise HTTPForbidden()
|
