pull_request_api.py
1111 lines
| 42.2 KiB
| text/x-python
|
PythonLexer
r5088 | # Copyright (C) 2011-2023 RhodeCode GmbH | |||
r1 | # | |||
# This program is free software: you can redistribute it and/or modify | ||||
# it under the terms of the GNU Affero General Public License, version 3 | ||||
# (only), as published by the Free Software Foundation. | ||||
# | ||||
# 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 Affero General Public License | ||||
# along with this program. If not, see <http://www.gnu.org/licenses/>. | ||||
# | ||||
# This program is dual-licensed. If you wish to learn more about the | ||||
# RhodeCode Enterprise Edition, including its added features, Support services, | ||||
# and proprietary license terms, please see https://rhodecode.com/licenses/ | ||||
import logging | ||||
r1769 | from rhodecode.api import jsonrpc_method, JSONRPCError, JSONRPCValidationError | |||
r1 | from rhodecode.api.utils import ( | |||
has_superadmin_permission, Optional, OAttr, get_repo_or_error, | ||||
get_pull_request_or_error, get_commit_or_error, get_user_or_error, | ||||
r3474 | validate_repo_permissions, resolve_ref_or_error, validate_set_owner_permissions) | |||
r4505 | from rhodecode.lib import channelstream | |||
r1 | from rhodecode.lib.auth import (HasRepoPermissionAnyApi) | |||
from rhodecode.lib.base import vcs_operation_context | ||||
from rhodecode.lib.utils2 import str2bool | ||||
r4519 | from rhodecode.lib.vcs.backends.base import unicode_to_reference | |||
r1 | from rhodecode.model.changeset_status import ChangesetStatusModel | |||
r1323 | from rhodecode.model.comment import CommentsModel | |||
r4519 | from rhodecode.model.db import ( | |||
Session, ChangesetStatus, ChangesetComment, PullRequest, PullRequestReviewers) | ||||
r1335 | from rhodecode.model.pull_request import PullRequestModel, MergeCheck | |||
r1 | from rhodecode.model.settings import SettingsModel | |||
r1769 | from rhodecode.model.validation_schema import Invalid | |||
r4440 | from rhodecode.model.validation_schema.schemas.reviewer_schema import ReviewerListSchema | |||
r1 | ||||
log = logging.getLogger(__name__) | ||||
@jsonrpc_method() | ||||
r3818 | def get_pull_request(request, apiuser, pullrequestid, repoid=Optional(None), | |||
merge_state=Optional(False)): | ||||
r1 | """ | |||
Get a pull request based on the given ID. | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
r2395 | :param repoid: Optional, repository name or repository ID from where | |||
the pull request was opened. | ||||
r1 | :type repoid: str or int | |||
:param pullrequestid: ID of the requested pull request. | ||||
:type pullrequestid: int | ||||
r3818 | :param merge_state: Optional calculate merge state for each repository. | |||
This could result in longer time to fetch the data | ||||
:type merge_state: bool | ||||
r1 | ||||
Example output: | ||||
.. code-block:: bash | ||||
"id": <id_given_in_input>, | ||||
"result": | ||||
{ | ||||
"pull_request_id": "<pull_request_id>", | ||||
"url": "<url>", | ||||
"title": "<title>", | ||||
"description": "<description>", | ||||
"status" : "<status>", | ||||
"created_on": "<date_time_created>", | ||||
"updated_on": "<date_time_updated>", | ||||
r4197 | "versions": "<number_or_versions_of_pr>", | |||
r1 | "commit_ids": [ | |||
... | ||||
"<commit_id>", | ||||
"<commit_id>", | ||||
... | ||||
], | ||||
"review_status": "<review_status>", | ||||
"mergeable": { | ||||
"status": "<bool>", | ||||
"message": "<message>", | ||||
}, | ||||
"source": { | ||||
"clone_url": "<clone_url>", | ||||
"repository": "<repository_name>", | ||||
"reference": | ||||
{ | ||||
"name": "<name>", | ||||
"type": "<type>", | ||||
"commit_id": "<commit_id>", | ||||
} | ||||
}, | ||||
"target": { | ||||
"clone_url": "<clone_url>", | ||||
"repository": "<repository_name>", | ||||
"reference": | ||||
{ | ||||
"name": "<name>", | ||||
"type": "<type>", | ||||
"commit_id": "<commit_id>", | ||||
} | ||||
}, | ||||
Martin Bornhold
|
r1054 | "merge": { | ||
Martin Bornhold
|
r893 | "clone_url": "<clone_url>", | ||
Martin Bornhold
|
r1054 | "reference": | ||
{ | ||||
"name": "<name>", | ||||
"type": "<type>", | ||||
"commit_id": "<commit_id>", | ||||
} | ||||
Martin Bornhold
|
r893 | }, | ||
r1 | "author": <user_obj>, | |||
"reviewers": [ | ||||
... | ||||
{ | ||||
"user": "<user_obj>", | ||||
"review_status": "<review_status>", | ||||
} | ||||
... | ||||
] | ||||
}, | ||||
"error": null | ||||
""" | ||||
r2395 | ||||
r1 | pull_request = get_pull_request_or_error(pullrequestid) | |||
r2395 | if Optional.extract(repoid): | |||
repo = get_repo_or_error(repoid) | ||||
else: | ||||
repo = pull_request.target_repo | ||||
r3371 | if not PullRequestModel().check_user_read(pull_request, apiuser, api=True): | |||
r2395 | raise JSONRPCError('repository `%s` or pull request `%s` ' | |||
'does not exist' % (repoid, pullrequestid)) | ||||
r3371 | ||||
# NOTE(marcink): only calculate and return merge state if the pr state is 'created' | ||||
# otherwise we can lock the repo on calculation of merge state while update/merge | ||||
# is happening. | ||||
r3818 | pr_created = pull_request.pull_request_state == pull_request.STATE_CREATED | |||
merge_state = Optional.extract(merge_state, binary=True) and pr_created | ||||
r3371 | data = pull_request.get_api_data(with_merge_state=merge_state) | |||
r1 | return data | |||
@jsonrpc_method() | ||||
r3445 | def get_pull_requests(request, apiuser, repoid, status=Optional('new'), | |||
r3818 | merge_state=Optional(False)): | |||
r1 | """ | |||
Get all pull requests from the repository specified in `repoid`. | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
r2395 | :param repoid: Optional repository name or repository ID. | |||
r1 | :type repoid: str or int | |||
:param status: Only return pull requests with the specified status. | ||||
Valid options are. | ||||
* ``new`` (default) | ||||
* ``open`` | ||||
* ``closed`` | ||||
:type status: str | ||||
r3445 | :param merge_state: Optional calculate merge state for each repository. | |||
This could result in longer time to fetch the data | ||||
:type merge_state: bool | ||||
r1 | ||||
Example output: | ||||
.. code-block:: bash | ||||
"id": <id_given_in_input>, | ||||
"result": | ||||
[ | ||||
... | ||||
{ | ||||
"pull_request_id": "<pull_request_id>", | ||||
"url": "<url>", | ||||
"title" : "<title>", | ||||
"description": "<description>", | ||||
"status": "<status>", | ||||
"created_on": "<date_time_created>", | ||||
"updated_on": "<date_time_updated>", | ||||
"commit_ids": [ | ||||
... | ||||
"<commit_id>", | ||||
"<commit_id>", | ||||
... | ||||
], | ||||
"review_status": "<review_status>", | ||||
"mergeable": { | ||||
"status": "<bool>", | ||||
"message: "<message>", | ||||
}, | ||||
"source": { | ||||
"clone_url": "<clone_url>", | ||||
"reference": | ||||
{ | ||||
"name": "<name>", | ||||
"type": "<type>", | ||||
"commit_id": "<commit_id>", | ||||
} | ||||
}, | ||||
"target": { | ||||
"clone_url": "<clone_url>", | ||||
"reference": | ||||
{ | ||||
"name": "<name>", | ||||
"type": "<type>", | ||||
"commit_id": "<commit_id>", | ||||
} | ||||
}, | ||||
Martin Bornhold
|
r1054 | "merge": { | ||
Martin Bornhold
|
r893 | "clone_url": "<clone_url>", | ||
Martin Bornhold
|
r1054 | "reference": | ||
{ | ||||
"name": "<name>", | ||||
"type": "<type>", | ||||
"commit_id": "<commit_id>", | ||||
} | ||||
Martin Bornhold
|
r893 | }, | ||
r1 | "author": <user_obj>, | |||
"reviewers": [ | ||||
... | ||||
{ | ||||
"user": "<user_obj>", | ||||
"review_status": "<review_status>", | ||||
} | ||||
... | ||||
] | ||||
} | ||||
... | ||||
], | ||||
"error": null | ||||
""" | ||||
repo = get_repo_or_error(repoid) | ||||
if not has_superadmin_permission(apiuser): | ||||
_perms = ( | ||||
'repository.admin', 'repository.write', 'repository.read',) | ||||
r1150 | validate_repo_permissions(apiuser, repoid, repo, _perms) | |||
r1 | ||||
status = Optional.extract(status) | ||||
r3445 | merge_state = Optional.extract(merge_state, binary=True) | |||
pull_requests = PullRequestModel().get_all(repo, statuses=[status], | ||||
order_by='id', order_dir='desc') | ||||
data = [pr.get_api_data(with_merge_state=merge_state) for pr in pull_requests] | ||||
r1 | return data | |||
@jsonrpc_method() | ||||
r1792 | def merge_pull_request( | |||
r2395 | request, apiuser, pullrequestid, repoid=Optional(None), | |||
r1792 | userid=Optional(OAttr('apiuser'))): | |||
r1 | """ | |||
Merge the pull request specified by `pullrequestid` into its target | ||||
repository. | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
r2395 | :param repoid: Optional, repository name or repository ID of the | |||
r1 | target repository to which the |pr| is to be merged. | |||
:type repoid: str or int | ||||
:param pullrequestid: ID of the pull request which shall be merged. | ||||
:type pullrequestid: int | ||||
:param userid: Merge the pull request as this user. | ||||
:type userid: Optional(str or int) | ||||
Example output: | ||||
.. code-block:: bash | ||||
r1603 | "id": <id_given_in_input>, | |||
"result": { | ||||
r3458 | "executed": "<bool>", | |||
"failure_reason": "<int>", | ||||
"merge_status_message": "<str>", | ||||
"merge_commit_id": "<merge_commit_id>", | ||||
"possible": "<bool>", | ||||
Martin Bornhold
|
r1055 | "merge_ref": { | ||
"commit_id": "<commit_id>", | ||||
"type": "<type>", | ||||
"name": "<name>" | ||||
} | ||||
r1 | }, | |||
r1603 | "error": null | |||
r1 | """ | |||
r2395 | pull_request = get_pull_request_or_error(pullrequestid) | |||
if Optional.extract(repoid): | ||||
repo = get_repo_or_error(repoid) | ||||
else: | ||||
repo = pull_request.target_repo | ||||
r3481 | auth_user = apiuser | |||
r4429 | ||||
r1 | if not isinstance(userid, Optional): | |||
r4429 | is_repo_admin = HasRepoPermissionAnyApi('repository.admin')( | |||
user=apiuser, repo_name=repo.repo_name) | ||||
if has_superadmin_permission(apiuser) or is_repo_admin: | ||||
r1 | apiuser = get_user_or_error(userid) | |||
r3481 | auth_user = apiuser.AuthUser() | |||
r1 | else: | |||
raise JSONRPCError('userid is not the same as your user') | ||||
r3371 | if pull_request.pull_request_state != PullRequest.STATE_CREATED: | |||
raise JSONRPCError( | ||||
'Operation forbidden because pull request is in state {}, ' | ||||
'only state {} is allowed.'.format( | ||||
pull_request.pull_request_state, PullRequest.STATE_CREATED)) | ||||
with pull_request.set_state(PullRequest.STATE_UPDATING): | ||||
r3481 | check = MergeCheck.validate(pull_request, auth_user=auth_user, | |||
translator=request.translate) | ||||
r1335 | merge_possible = not check.failed | |||
if not merge_possible: | ||||
r1759 | error_messages = [] | |||
for err_type, error_msg in check.errors: | ||||
error_msg = request.translate(error_msg) | ||||
error_messages.append(error_msg) | ||||
reasons = ','.join(error_messages) | ||||
r1 | raise JSONRPCError( | |||
r5092 | f'merge not possible for following reasons: {reasons}') | |||
r1 | ||||
target_repo = pull_request.target_repo | ||||
extras = vcs_operation_context( | ||||
request.environ, repo_name=target_repo.repo_name, | ||||
r3481 | username=auth_user.username, action='push', | |||
r1 | scm=target_repo.repo_type) | |||
r3371 | with pull_request.set_state(PullRequest.STATE_UPDATING): | |||
merge_response = PullRequestModel().merge_repo( | ||||
pull_request, apiuser, extras=extras) | ||||
Martin Bornhold
|
r1055 | if merge_response.executed: | ||
r3481 | PullRequestModel().close_pull_request(pull_request.pull_request_id, auth_user) | |||
r1 | ||||
r506 | Session().commit() | |||
Martin Bornhold
|
r1055 | |||
# In previous versions the merge response directly contained the merge | ||||
# commit id. It is now contained in the merge reference object. To be | ||||
# backwards compatible we have to extract it again. | ||||
r3339 | merge_response = merge_response.asdict() | |||
Martin Bornhold
|
r1055 | merge_response['merge_commit_id'] = merge_response['merge_ref'].commit_id | ||
return merge_response | ||||
r1 | ||||
@jsonrpc_method() | ||||
r2394 | def get_pull_request_comments( | |||
request, apiuser, pullrequestid, repoid=Optional(None)): | ||||
""" | ||||
Get all comments of pull request specified with the `pullrequestid` | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
:param repoid: Optional repository name or repository ID. | ||||
:type repoid: str or int | ||||
:param pullrequestid: The pull request ID. | ||||
:type pullrequestid: int | ||||
Example output: | ||||
.. code-block:: bash | ||||
id : <id_given_in_input> | ||||
result : [ | ||||
{ | ||||
"comment_author": { | ||||
"active": true, | ||||
"full_name_or_username": "Tom Gore", | ||||
"username": "admin" | ||||
}, | ||||
"comment_created_on": "2017-01-02T18:43:45.533", | ||||
"comment_f_path": null, | ||||
"comment_id": 25, | ||||
"comment_lineno": null, | ||||
"comment_status": { | ||||
"status": "under_review", | ||||
"status_lbl": "Under Review" | ||||
}, | ||||
"comment_text": "Example text", | ||||
"comment_type": null, | ||||
r4440 | "comment_last_version: 0, | |||
r4304 | "pull_request_version": null, | |||
"comment_commit_id": None, | ||||
"comment_pull_request_id": <pull_request_id> | ||||
r2394 | } | |||
], | ||||
error : null | ||||
""" | ||||
pull_request = get_pull_request_or_error(pullrequestid) | ||||
if Optional.extract(repoid): | ||||
repo = get_repo_or_error(repoid) | ||||
else: | ||||
repo = pull_request.target_repo | ||||
if not PullRequestModel().check_user_read( | ||||
pull_request, apiuser, api=True): | ||||
raise JSONRPCError('repository `%s` or pull request `%s` ' | ||||
'does not exist' % (repoid, pullrequestid)) | ||||
(pull_request_latest, | ||||
pull_request_at_ver, | ||||
pull_request_display_obj, | ||||
at_version) = PullRequestModel().get_pr_version( | ||||
pull_request.pull_request_id, version=None) | ||||
versions = pull_request_display_obj.versions() | ||||
ver_map = { | ||||
ver.pull_request_version_id: cnt | ||||
for cnt, ver in enumerate(versions, 1) | ||||
} | ||||
# GENERAL COMMENTS with versions # | ||||
q = CommentsModel()._all_general_comments_of_pull_request(pull_request) | ||||
q = q.order_by(ChangesetComment.comment_id.asc()) | ||||
general_comments = q.all() | ||||
# INLINE COMMENTS with versions # | ||||
q = CommentsModel()._all_inline_comments_of_pull_request(pull_request) | ||||
q = q.order_by(ChangesetComment.comment_id.asc()) | ||||
inline_comments = q.all() | ||||
data = [] | ||||
for comment in inline_comments + general_comments: | ||||
full_data = comment.get_api_data() | ||||
pr_version_id = None | ||||
if comment.pull_request_version_id: | ||||
pr_version_id = 'v{}'.format( | ||||
ver_map[comment.pull_request_version_id]) | ||||
# sanitize some entries | ||||
full_data['pull_request_version'] = pr_version_id | ||||
full_data['comment_author'] = { | ||||
'username': full_data['comment_author'].username, | ||||
'full_name_or_username': full_data['comment_author'].full_name_or_username, | ||||
'active': full_data['comment_author'].active, | ||||
} | ||||
if full_data['comment_status']: | ||||
full_data['comment_status'] = { | ||||
'status': full_data['comment_status'][0].status, | ||||
'status_lbl': full_data['comment_status'][0].status_lbl, | ||||
} | ||||
else: | ||||
full_data['comment_status'] = {} | ||||
data.append(full_data) | ||||
return data | ||||
@jsonrpc_method() | ||||
r1337 | def comment_pull_request( | |||
r2395 | request, apiuser, pullrequestid, repoid=Optional(None), | |||
message=Optional(None), commit_id=Optional(None), status=Optional(None), | ||||
r1337 | comment_type=Optional(ChangesetComment.COMMENT_TYPE_NOTE), | |||
r4049 | resolves_comment_id=Optional(None), extra_recipients=Optional([]), | |||
r4196 | userid=Optional(OAttr('apiuser')), send_email=Optional(True)): | |||
r1 | """ | |||
Comment on the pull request specified with the `pullrequestid`, | ||||
in the |repo| specified by the `repoid`, and optionally change the | ||||
review status. | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
r2395 | :param repoid: Optional repository name or repository ID. | |||
r1 | :type repoid: str or int | |||
:param pullrequestid: The pull request ID. | ||||
:type pullrequestid: int | ||||
r1269 | :param commit_id: Specify the commit_id for which to set a comment. If | |||
given commit_id is different than latest in the PR status | ||||
change won't be performed. | ||||
:type commit_id: str | ||||
r1337 | :param message: The text content of the comment. | |||
:type message: str | ||||
:param status: (**Optional**) Set the approval status of the pull | ||||
request. One of: 'not_reviewed', 'approved', 'rejected', | ||||
'under_review' | ||||
:type status: str | ||||
:param comment_type: Comment type, one of: 'note', 'todo' | ||||
:type comment_type: Optional(str), default: 'note' | ||||
r4049 | :param resolves_comment_id: id of comment which this one will resolve | |||
:type resolves_comment_id: Optional(int) | ||||
:param extra_recipients: list of user ids or usernames to add | ||||
notifications for this comment. Acts like a CC for notification | ||||
:type extra_recipients: Optional(list) | ||||
r1 | :param userid: Comment on the pull request as this user | |||
:type userid: Optional(str or int) | ||||
r4196 | :param send_email: Define if this comment should also send email notification | |||
:type send_email: Optional(bool) | ||||
r1 | ||||
Example output: | ||||
.. code-block:: bash | ||||
r1603 | id : <id_given_in_input> | |||
result : { | ||||
r1 | "pull_request_id": "<Integer>", | |||
r1269 | "comment_id": "<Integer>", | |||
"status": {"given": <given_status>, | ||||
"was_changed": <bool status_was_actually_changed> }, | ||||
r1603 | }, | |||
error : null | ||||
r1 | """ | |||
r4505 | _ = request.translate | |||
r2395 | pull_request = get_pull_request_or_error(pullrequestid) | |||
if Optional.extract(repoid): | ||||
repo = get_repo_or_error(repoid) | ||||
else: | ||||
repo = pull_request.target_repo | ||||
r4505 | db_repo_name = repo.repo_name | |||
r3481 | auth_user = apiuser | |||
r1 | if not isinstance(userid, Optional): | |||
r4429 | is_repo_admin = HasRepoPermissionAnyApi('repository.admin')( | |||
r4505 | user=apiuser, repo_name=db_repo_name) | |||
r4429 | if has_superadmin_permission(apiuser) or is_repo_admin: | |||
r1 | apiuser = get_user_or_error(userid) | |||
r3481 | auth_user = apiuser.AuthUser() | |||
r1 | else: | |||
raise JSONRPCError('userid is not the same as your user') | ||||
r3463 | if pull_request.is_closed(): | |||
r5047 | raise JSONRPCError(f'pull request `{pullrequestid}` comment failed, pull request is closed') | |||
r3463 | ||||
r1 | if not PullRequestModel().check_user_read( | |||
pull_request, apiuser, api=True): | ||||
r5095 | raise JSONRPCError(f'repository `{repoid}` does not exist') | |||
r1 | message = Optional.extract(message) | |||
status = Optional.extract(status) | ||||
r1269 | commit_id = Optional.extract(commit_id) | |||
r1337 | comment_type = Optional.extract(comment_type) | |||
r1338 | resolves_comment_id = Optional.extract(resolves_comment_id) | |||
r4049 | extra_recipients = Optional.extract(extra_recipients) | |||
r4196 | send_email = Optional.extract(send_email, binary=True) | |||
r1269 | ||||
r1 | if not message and not status: | |||
r1269 | raise JSONRPCError( | |||
'Both message and status parameters are missing. ' | ||||
'At least one is required.') | ||||
r1 | ||||
r5047 | if status and status not in (st[0] for st in ChangesetStatus.STATUSES): | |||
raise JSONRPCError(f'Unknown comment status: `{status}`') | ||||
r1269 | ||||
if commit_id and commit_id not in pull_request.revisions: | ||||
r5047 | raise JSONRPCError(f'Invalid commit_id `{commit_id}` for this pull request.') | |||
r1 | ||||
allowed_to_change_status = PullRequestModel().check_user_change_status( | ||||
pull_request, apiuser) | ||||
r1269 | ||||
# if commit_id is passed re-validated if user is allowed to change status | ||||
r5047 | # based on the latest commit_id from the PR | |||
r1269 | if commit_id: | |||
commit_idx = pull_request.revisions.index(commit_id) | ||||
if commit_idx != 0: | ||||
r5047 | log.warning('Resetting allowed_to_change_status = False because commit is NOT the latest in pull-request') | |||
r1269 | allowed_to_change_status = False | |||
r1338 | if resolves_comment_id: | |||
comment = ChangesetComment.get(resolves_comment_id) | ||||
if not comment: | ||||
r5047 | raise JSONRPCError(f'Invalid resolves_comment_id `{resolves_comment_id}` for this pull request.') | |||
r1338 | if comment.comment_type != ChangesetComment.COMMENT_TYPE_TODO: | |||
r5047 | raise JSONRPCError(f'Comment `{resolves_comment_id}` is wrong type for setting status to resolved.') | |||
r1338 | ||||
r1 | text = message | |||
r1269 | status_label = ChangesetStatus.get_status_lbl(status) | |||
r1 | if status and allowed_to_change_status: | |||
r1269 | st_message = ('Status change %(transition_icon)s %(status)s' | |||
% {'transition_icon': '>', 'status': status_label}) | ||||
r1 | text = message or st_message | |||
rc_config = SettingsModel().get_all_settings() | ||||
renderer = rc_config.get('rhodecode_markup_renderer', 'rst') | ||||
r1269 | ||||
status_change = status and allowed_to_change_status | ||||
r1323 | comment = CommentsModel().create( | |||
r1 | text=text, | |||
repo=pull_request.target_repo.repo_id, | ||||
user=apiuser.user_id, | ||||
pull_request=pull_request.pull_request_id, | ||||
f_path=None, | ||||
line_no=None, | ||||
r1269 | status_change=(status_label if status_change else None), | |||
status_change_type=(status if status_change else None), | ||||
r1 | closing_pr=False, | |||
r1337 | renderer=renderer, | |||
r1338 | comment_type=comment_type, | |||
r2728 | resolves_comment_id=resolves_comment_id, | |||
r4049 | auth_user=auth_user, | |||
r4196 | extra_recipients=extra_recipients, | |||
send_email=send_email | ||||
r1 | ) | |||
r5047 | ||||
r4519 | is_inline = comment.is_inline | |||
r1 | ||||
if allowed_to_change_status and status: | ||||
r3416 | old_calculated_status = pull_request.calculated_review_status() | |||
r1 | ChangesetStatusModel().set_status( | |||
pull_request.target_repo.repo_id, | ||||
status, | ||||
apiuser.user_id, | ||||
comment, | ||||
pull_request=pull_request.pull_request_id | ||||
) | ||||
Session().flush() | ||||
Session().commit() | ||||
r3416 | ||||
PullRequestModel().trigger_pull_request_hook( | ||||
pull_request, apiuser, 'comment', | ||||
data={'comment': comment}) | ||||
if allowed_to_change_status and status: | ||||
# we now calculate the status of pull request, and based on that | ||||
# calculation we set the commits status | ||||
calculated_status = pull_request.calculated_review_status() | ||||
if old_calculated_status != calculated_status: | ||||
PullRequestModel().trigger_pull_request_hook( | ||||
pull_request, apiuser, 'review_status_change', | ||||
data={'status': calculated_status}) | ||||
r1 | data = { | |||
'pull_request_id': pull_request.pull_request_id, | ||||
r1269 | 'comment_id': comment.comment_id if comment else None, | |||
'status': {'given': status, 'was_changed': status_change}, | ||||
r1 | } | |||
r4505 | ||||
comment_broadcast_channel = channelstream.comment_channel( | ||||
db_repo_name, pull_request_obj=pull_request) | ||||
comment_data = data | ||||
comment_type = 'inline' if is_inline else 'general' | ||||
channelstream.comment_channelstream_push( | ||||
request, comment_broadcast_channel, apiuser, | ||||
_('posted a new {} comment').format(comment_type), | ||||
comment_data=comment_data) | ||||
r1 | return data | |||
r5047 | ||||
r4519 | def _reviewers_validation(obj_list): | |||
schema = ReviewerListSchema() | ||||
try: | ||||
reviewer_objects = schema.deserialize(obj_list) | ||||
except Invalid as err: | ||||
raise JSONRPCValidationError(colander_exc=err) | ||||
# validate users | ||||
for reviewer_object in reviewer_objects: | ||||
user = get_user_or_error(reviewer_object['username']) | ||||
reviewer_object['user_id'] = user.user_id | ||||
return reviewer_objects | ||||
r1 | ||||
@jsonrpc_method() | ||||
def create_pull_request( | ||||
request, apiuser, source_repo, target_repo, source_ref, target_ref, | ||||
r3474 | owner=Optional(OAttr('apiuser')), title=Optional(''), description=Optional(''), | |||
r4519 | description_renderer=Optional(''), | |||
reviewers=Optional(None), observers=Optional(None)): | ||||
r1 | """ | |||
Creates a new pull request. | ||||
Accepts refs in the following formats: | ||||
* branch:<branch_name>:<sha> | ||||
* branch:<branch_name> | ||||
* bookmark:<bookmark_name>:<sha> (Mercurial only) | ||||
* bookmark:<bookmark_name> (Mercurial only) | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
:param source_repo: Set the source repository name. | ||||
:type source_repo: str | ||||
:param target_repo: Set the target repository name. | ||||
:type target_repo: str | ||||
:param source_ref: Set the source ref name. | ||||
:type source_ref: str | ||||
:param target_ref: Set the target ref name. | ||||
:type target_ref: str | ||||
r3474 | :param owner: user_id or username | |||
:type owner: Optional(str) | ||||
r2859 | :param title: Optionally Set the pull request title, it's generated otherwise | |||
r1 | :type title: str | |||
:param description: Set the pull request description. | ||||
:type description: Optional(str) | ||||
r2904 | :type description_renderer: Optional(str) | |||
:param description_renderer: Set pull request renderer for the description. | ||||
It should be 'rst', 'markdown' or 'plain'. If not give default | ||||
system renderer will be used | ||||
r1 | :param reviewers: Set the new pull request reviewers list. | |||
r1795 | Reviewer defined by review rules will be added automatically to the | |||
defined list. | ||||
r1 | :type reviewers: Optional(list) | |||
r873 | Accepts username strings or objects of the format: | |||
r1603 | ||||
r1795 | [{'username': 'nick', 'reasons': ['original author'], 'mandatory': <bool>}] | |||
r4519 | :param observers: Set the new pull request observers list. | |||
Reviewer defined by review rules will be added automatically to the | ||||
defined list. This feature is only available in RhodeCode EE | ||||
:type observers: Optional(list) | ||||
Accepts username strings or objects of the format: | ||||
[{'username': 'nick', 'reasons': ['original author']}] | ||||
r1 | """ | |||
r873 | ||||
r2873 | source_db_repo = get_repo_or_error(source_repo) | |||
r1795 | target_db_repo = get_repo_or_error(target_repo) | |||
r1 | if not has_superadmin_permission(apiuser): | |||
_perms = ('repository.admin', 'repository.write', 'repository.read',) | ||||
r1795 | validate_repo_permissions(apiuser, source_repo, source_db_repo, _perms) | |||
r1 | ||||
r3474 | owner = validate_set_owner_permissions(apiuser, owner) | |||
r1795 | full_source_ref = resolve_ref_or_error(source_ref, source_db_repo) | |||
full_target_ref = resolve_ref_or_error(target_ref, target_db_repo) | ||||
r2859 | ||||
r4519 | get_commit_or_error(full_source_ref, source_db_repo) | |||
get_commit_or_error(full_target_ref, target_db_repo) | ||||
r2859 | ||||
r873 | reviewer_objects = Optional.extract(reviewers) or [] | |||
r4519 | observer_objects = Optional.extract(observers) or [] | |||
r1795 | ||||
r2859 | # serialize and validate passed in given reviewers | |||
r1769 | if reviewer_objects: | |||
r4519 | reviewer_objects = _reviewers_validation(reviewer_objects) | |||
r1 | ||||
r4519 | if observer_objects: | |||
observer_objects = _reviewers_validation(reviewer_objects) | ||||
r1795 | ||||
r4500 | get_default_reviewers_data, validate_default_reviewers, validate_observers = \ | |||
r1795 | PullRequestModel().get_reviewer_functions() | |||
r4519 | source_ref_obj = unicode_to_reference(full_source_ref) | |||
target_ref_obj = unicode_to_reference(full_target_ref) | ||||
r2859 | # recalculate reviewers logic, to make sure we can validate this | |||
r4346 | default_reviewers_data = get_default_reviewers_data( | |||
r4510 | owner, | |||
r4519 | source_db_repo, | |||
source_ref_obj, | ||||
target_db_repo, | ||||
target_ref_obj, | ||||
r4510 | ) | |||
r1795 | ||||
r4519 | # now MERGE our given with the calculated from the default rules | |||
just_reviewers = [ | ||||
x for x in default_reviewers_data['reviewers'] | ||||
if x['role'] == PullRequestReviewers.ROLE_REVIEWER] | ||||
reviewer_objects = just_reviewers + reviewer_objects | ||||
r1795 | ||||
try: | ||||
r2859 | reviewers = validate_default_reviewers( | |||
r4346 | reviewer_objects, default_reviewers_data) | |||
r1795 | except ValueError as e: | |||
r5092 | raise JSONRPCError(f'Reviewers Validation: {e}') | |||
r1 | ||||
r4519 | # now MERGE our given with the calculated from the default rules | |||
just_observers = [ | ||||
x for x in default_reviewers_data['reviewers'] | ||||
if x['role'] == PullRequestReviewers.ROLE_OBSERVER] | ||||
observer_objects = just_observers + observer_objects | ||||
try: | ||||
observers = validate_observers( | ||||
observer_objects, default_reviewers_data) | ||||
except ValueError as e: | ||||
r5092 | raise JSONRPCError(f'Observer Validation: {e}') | |||
r4519 | ||||
r2859 | title = Optional.extract(title) | |||
if not title: | ||||
r4519 | title_source_ref = source_ref_obj.name | |||
r2859 | title = PullRequestModel().generate_pullrequest_title( | |||
source=source_repo, | ||||
source_ref=title_source_ref, | ||||
target=target_repo | ||||
) | ||||
r4346 | ||||
diff_info = default_reviewers_data['diff_info'] | ||||
common_ancestor_id = diff_info['ancestor'] | ||||
r4519 | # NOTE(marcink): reversed is consistent with how we open it in the WEB interface | |||
commits = [commit['commit_id'] for commit in reversed(diff_info['commits'])] | ||||
r4346 | ||||
if not common_ancestor_id: | ||||
r4519 | raise JSONRPCError('no common ancestor found between specified references') | |||
r4346 | ||||
if not commits: | ||||
r4519 | raise JSONRPCError('no commits found for merge between specified references') | |||
r4346 | ||||
# recalculate target ref based on ancestor | ||||
r4519 | full_target_ref = ':'.join((target_ref_obj.type, target_ref_obj.name, common_ancestor_id)) | |||
r4346 | ||||
r2904 | # fetch renderer, if set fallback to plain in case of PR | |||
rc_config = SettingsModel().get_all_settings() | ||||
default_system_renderer = rc_config.get('rhodecode_markup_renderer', 'plain') | ||||
r2859 | description = Optional.extract(description) | |||
r2904 | description_renderer = Optional.extract(description_renderer) or default_system_renderer | |||
r2859 | ||||
pull_request = PullRequestModel().create( | ||||
r3474 | created_by=owner.user_id, | |||
r1 | source_repo=source_repo, | |||
source_ref=full_source_ref, | ||||
target_repo=target_repo, | ||||
target_ref=full_target_ref, | ||||
r4346 | common_ancestor_id=common_ancestor_id, | |||
r4519 | revisions=commits, | |||
r1769 | reviewers=reviewers, | |||
r4519 | observers=observers, | |||
r1 | title=title, | |||
r2859 | description=description, | |||
r2904 | description_renderer=description_renderer, | |||
r4346 | reviewer_data=default_reviewers_data, | |||
r2788 | auth_user=apiuser | |||
r1 | ) | |||
Session().commit() | ||||
data = { | ||||
r5092 | 'msg': f'Created new pull request `{title}`', | |||
r1 | 'pull_request_id': pull_request.pull_request_id, | |||
} | ||||
return data | ||||
@jsonrpc_method() | ||||
def update_pull_request( | ||||
r2395 | request, apiuser, pullrequestid, repoid=Optional(None), | |||
r2904 | title=Optional(''), description=Optional(''), description_renderer=Optional(''), | |||
r4519 | reviewers=Optional(None), observers=Optional(None), update_commits=Optional(None)): | |||
r1 | """ | |||
Updates a pull request. | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
r2395 | :param repoid: Optional repository name or repository ID. | |||
r1 | :type repoid: str or int | |||
:param pullrequestid: The pull request ID. | ||||
:type pullrequestid: int | ||||
:param title: Set the pull request title. | ||||
:type title: str | ||||
:param description: Update pull request description. | ||||
:type description: Optional(str) | ||||
r2904 | :type description_renderer: Optional(str) | |||
:param description_renderer: Update pull request renderer for the description. | ||||
It should be 'rst', 'markdown' or 'plain' | ||||
r1 | :param reviewers: Update pull request reviewers list with new value. | |||
:type reviewers: Optional(list) | ||||
r1769 | Accepts username strings or objects of the format: | |||
r1795 | [{'username': 'nick', 'reasons': ['original author'], 'mandatory': <bool>}] | |||
r4519 | :param observers: Update pull request observers list with new value. | |||
:type observers: Optional(list) | ||||
Accepts username strings or objects of the format: | ||||
r1769 | ||||
r4519 | [{'username': 'nick', 'reasons': ['should be aware about this PR']}] | |||
r1 | :param update_commits: Trigger update of commits for this pull request | |||
:type: update_commits: Optional(bool) | ||||
Example output: | ||||
.. code-block:: bash | ||||
r1603 | id : <id_given_in_input> | |||
result : { | ||||
r1 | "msg": "Updated pull request `63`", | |||
"pull_request": <pull_request_object>, | ||||
"updated_reviewers": { | ||||
"added": [ | ||||
"username" | ||||
], | ||||
"removed": [] | ||||
}, | ||||
r4519 | "updated_observers": { | |||
"added": [ | ||||
"username" | ||||
], | ||||
"removed": [] | ||||
}, | ||||
r1 | "updated_commits": { | |||
"added": [ | ||||
"<sha1_hash>" | ||||
], | ||||
"common": [ | ||||
"<sha1_hash>", | ||||
"<sha1_hash>", | ||||
], | ||||
"removed": [] | ||||
} | ||||
} | ||||
r1603 | error : null | |||
r1 | """ | |||
pull_request = get_pull_request_or_error(pullrequestid) | ||||
r2395 | if Optional.extract(repoid): | |||
repo = get_repo_or_error(repoid) | ||||
else: | ||||
repo = pull_request.target_repo | ||||
r1 | if not PullRequestModel().check_user_update( | |||
pull_request, apiuser, api=True): | ||||
raise JSONRPCError( | ||||
r5092 | 'pull request `{}` update failed, no permission to update.'.format( | |||
pullrequestid)) | ||||
r1 | if pull_request.is_closed(): | |||
raise JSONRPCError( | ||||
r5092 | 'pull request `{}` update failed, pull request is closed'.format( | |||
pullrequestid)) | ||||
r1 | ||||
r873 | reviewer_objects = Optional.extract(reviewers) or [] | |||
r4519 | observer_objects = Optional.extract(observers) or [] | |||
r1 | ||||
title = Optional.extract(title) | ||||
description = Optional.extract(description) | ||||
r2904 | description_renderer = Optional.extract(description_renderer) | |||
r4505 | # Update title/description | |||
title_changed = False | ||||
r1 | if title or description: | |||
PullRequestModel().edit( | ||||
r2904 | pull_request, | |||
title or pull_request.title, | ||||
description or pull_request.description, | ||||
description_renderer or pull_request.description_renderer, | ||||
apiuser) | ||||
r1 | Session().commit() | |||
r4505 | title_changed = True | |||
r1 | ||||
commit_changes = {"added": [], "common": [], "removed": []} | ||||
r4505 | ||||
# Update commits | ||||
commits_changed = False | ||||
r1 | if str2bool(Optional.extract(update_commits)): | |||
r3371 | ||||
if pull_request.pull_request_state != PullRequest.STATE_CREATED: | ||||
raise JSONRPCError( | ||||
'Operation forbidden because pull request is in state {}, ' | ||||
'only state {} is allowed.'.format( | ||||
pull_request.pull_request_state, PullRequest.STATE_CREATED)) | ||||
with pull_request.set_state(PullRequest.STATE_UPDATING): | ||||
if PullRequestModel().has_valid_update_type(pull_request): | ||||
r4120 | db_user = apiuser.get_instance() | |||
update_response = PullRequestModel().update_commits( | ||||
pull_request, db_user) | ||||
r3371 | commit_changes = update_response.changes or commit_changes | |||
Session().commit() | ||||
r4505 | commits_changed = True | |||
r1 | ||||
r4505 | # Update reviewers | |||
r4519 | # serialize and validate passed in given reviewers | |||
if reviewer_objects: | ||||
reviewer_objects = _reviewers_validation(reviewer_objects) | ||||
if observer_objects: | ||||
observer_objects = _reviewers_validation(reviewer_objects) | ||||
# re-use stored rules | ||||
default_reviewers_data = pull_request.reviewer_data | ||||
__, validate_default_reviewers, validate_observers = \ | ||||
PullRequestModel().get_reviewer_functions() | ||||
if reviewer_objects: | ||||
try: | ||||
reviewers = validate_default_reviewers(reviewer_objects, default_reviewers_data) | ||||
except ValueError as e: | ||||
r5092 | raise JSONRPCError(f'Reviewers Validation: {e}') | |||
r4519 | else: | |||
reviewers = [] | ||||
if observer_objects: | ||||
try: | ||||
observers = validate_default_reviewers(reviewer_objects, default_reviewers_data) | ||||
except ValueError as e: | ||||
r5092 | raise JSONRPCError(f'Observer Validation: {e}') | |||
r4519 | else: | |||
observers = [] | ||||
r4505 | reviewers_changed = False | |||
r1 | reviewers_changes = {"added": [], "removed": []} | |||
r1769 | if reviewers: | |||
r3416 | old_calculated_status = pull_request.calculated_review_status() | |||
r1 | added_reviewers, removed_reviewers = \ | |||
r4519 | PullRequestModel().update_reviewers(pull_request, reviewers, apiuser.get_instance()) | |||
r1 | ||||
reviewers_changes['added'] = sorted( | ||||
[get_user_or_error(n).username for n in added_reviewers]) | ||||
reviewers_changes['removed'] = sorted( | ||||
[get_user_or_error(n).username for n in removed_reviewers]) | ||||
Session().commit() | ||||
r3416 | # trigger status changed if change in reviewers changes the status | |||
calculated_status = pull_request.calculated_review_status() | ||||
if old_calculated_status != calculated_status: | ||||
PullRequestModel().trigger_pull_request_hook( | ||||
pull_request, apiuser, 'review_status_change', | ||||
data={'status': calculated_status}) | ||||
r4505 | reviewers_changed = True | |||
observers_changed = False | ||||
r4519 | observers_changes = {"added": [], "removed": []} | |||
if observers: | ||||
added_observers, removed_observers = \ | ||||
PullRequestModel().update_observers(pull_request, observers, apiuser.get_instance()) | ||||
observers_changes['added'] = sorted( | ||||
[get_user_or_error(n).username for n in added_observers]) | ||||
observers_changes['removed'] = sorted( | ||||
[get_user_or_error(n).username for n in removed_observers]) | ||||
Session().commit() | ||||
reviewers_changed = True | ||||
r4505 | ||||
# push changed to channelstream | ||||
if commits_changed or reviewers_changed or observers_changed: | ||||
pr_broadcast_channel = channelstream.pr_channel(pull_request) | ||||
msg = 'Pull request was updated.' | ||||
channelstream.pr_update_channelstream_push( | ||||
request, pr_broadcast_channel, apiuser, msg) | ||||
r3416 | ||||
r1 | data = { | |||
r5092 | 'msg': f'Updated pull request `{pull_request.pull_request_id}`', | |||
r1 | 'pull_request': pull_request.get_api_data(), | |||
'updated_commits': commit_changes, | ||||
r4519 | 'updated_reviewers': reviewers_changes, | |||
'updated_observers': observers_changes, | ||||
r1 | } | |||
Martin Bornhold
|
r1077 | |||
r1 | return data | |||
r1792 | ||||
@jsonrpc_method() | ||||
def close_pull_request( | ||||
r2395 | request, apiuser, pullrequestid, repoid=Optional(None), | |||
r1792 | userid=Optional(OAttr('apiuser')), message=Optional('')): | |||
""" | ||||
Close the pull request specified by `pullrequestid`. | ||||
:param apiuser: This is filled automatically from the |authtoken|. | ||||
:type apiuser: AuthUser | ||||
:param repoid: Repository name or repository ID to which the pull | ||||
request belongs. | ||||
:type repoid: str or int | ||||
:param pullrequestid: ID of the pull request to be closed. | ||||
:type pullrequestid: int | ||||
:param userid: Close the pull request as this user. | ||||
:type userid: Optional(str or int) | ||||
:param message: Optional message to close the Pull Request with. If not | ||||
specified it will be generated automatically. | ||||
:type message: Optional(str) | ||||
Example output: | ||||
.. code-block:: bash | ||||
"id": <id_given_in_input>, | ||||
"result": { | ||||
"pull_request_id": "<int>", | ||||
"close_status": "<str:status_lbl>, | ||||
"closed": "<bool>" | ||||
}, | ||||
"error": null | ||||
""" | ||||
_ = request.translate | ||||
r2395 | pull_request = get_pull_request_or_error(pullrequestid) | |||
if Optional.extract(repoid): | ||||
repo = get_repo_or_error(repoid) | ||||
else: | ||||
repo = pull_request.target_repo | ||||
r4429 | is_repo_admin = HasRepoPermissionAnyApi('repository.admin')( | |||
user=apiuser, repo_name=repo.repo_name) | ||||
r1792 | if not isinstance(userid, Optional): | |||
r4429 | if has_superadmin_permission(apiuser) or is_repo_admin: | |||
r1792 | apiuser = get_user_or_error(userid) | |||
else: | ||||
raise JSONRPCError('userid is not the same as your user') | ||||
if pull_request.is_closed(): | ||||
raise JSONRPCError( | ||||
r5095 | f'pull request `{pullrequestid}` is already closed') | |||
r1792 | ||||
# only owner or admin or person with write permissions | ||||
allowed_to_close = PullRequestModel().check_user_update( | ||||
pull_request, apiuser, api=True) | ||||
if not allowed_to_close: | ||||
raise JSONRPCError( | ||||
r5092 | 'pull request `{}` close failed, no permission to close.'.format( | |||
pullrequestid)) | ||||
r1792 | ||||
# message we're using to close the PR, else it's automatically generated | ||||
message = Optional.extract(message) | ||||
# finally close the PR, with proper message comment | ||||
comment, status = PullRequestModel().close_pull_request_with_comment( | ||||
r3027 | pull_request, apiuser, repo, message=message, auth_user=apiuser) | |||
r1792 | status_lbl = ChangesetStatus.get_status_lbl(status) | |||
Session().commit() | ||||
data = { | ||||
'pull_request_id': pull_request.pull_request_id, | ||||
'close_status': status_lbl, | ||||
'closed': True, | ||||
} | ||||
return data | ||||