##// END OF EJS Templates
pull-requests: fixed a bug in removal of multiple reviewers
marcink -
r3575:a4085180 stable
parent child Browse files
Show More
@@ -1,1717 +1,1728 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2012-2019 RhodeCode GmbH
3 # Copyright (C) 2012-2019 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21
21
22 """
22 """
23 pull request model for RhodeCode
23 pull request model for RhodeCode
24 """
24 """
25
25
26
26
27 import json
27 import json
28 import logging
28 import logging
29 import datetime
29 import datetime
30 import urllib
30 import urllib
31 import collections
31 import collections
32
32
33 from pyramid import compat
33 from pyramid import compat
34 from pyramid.threadlocal import get_current_request
34 from pyramid.threadlocal import get_current_request
35
35
36 from rhodecode import events
36 from rhodecode import events
37 from rhodecode.translation import lazy_ugettext
37 from rhodecode.translation import lazy_ugettext
38 from rhodecode.lib import helpers as h, hooks_utils, diffs
38 from rhodecode.lib import helpers as h, hooks_utils, diffs
39 from rhodecode.lib import audit_logger
39 from rhodecode.lib import audit_logger
40 from rhodecode.lib.compat import OrderedDict
40 from rhodecode.lib.compat import OrderedDict
41 from rhodecode.lib.hooks_daemon import prepare_callback_daemon
41 from rhodecode.lib.hooks_daemon import prepare_callback_daemon
42 from rhodecode.lib.markup_renderer import (
42 from rhodecode.lib.markup_renderer import (
43 DEFAULT_COMMENTS_RENDERER, RstTemplateRenderer)
43 DEFAULT_COMMENTS_RENDERER, RstTemplateRenderer)
44 from rhodecode.lib.utils2 import safe_unicode, safe_str, md5_safe
44 from rhodecode.lib.utils2 import safe_unicode, safe_str, md5_safe
45 from rhodecode.lib.vcs.backends.base import (
45 from rhodecode.lib.vcs.backends.base import (
46 Reference, MergeResponse, MergeFailureReason, UpdateFailureReason)
46 Reference, MergeResponse, MergeFailureReason, UpdateFailureReason)
47 from rhodecode.lib.vcs.conf import settings as vcs_settings
47 from rhodecode.lib.vcs.conf import settings as vcs_settings
48 from rhodecode.lib.vcs.exceptions import (
48 from rhodecode.lib.vcs.exceptions import (
49 CommitDoesNotExistError, EmptyRepositoryError)
49 CommitDoesNotExistError, EmptyRepositoryError)
50 from rhodecode.model import BaseModel
50 from rhodecode.model import BaseModel
51 from rhodecode.model.changeset_status import ChangesetStatusModel
51 from rhodecode.model.changeset_status import ChangesetStatusModel
52 from rhodecode.model.comment import CommentsModel
52 from rhodecode.model.comment import CommentsModel
53 from rhodecode.model.db import (
53 from rhodecode.model.db import (
54 or_, PullRequest, PullRequestReviewers, ChangesetStatus,
54 or_, PullRequest, PullRequestReviewers, ChangesetStatus,
55 PullRequestVersion, ChangesetComment, Repository, RepoReviewRule)
55 PullRequestVersion, ChangesetComment, Repository, RepoReviewRule)
56 from rhodecode.model.meta import Session
56 from rhodecode.model.meta import Session
57 from rhodecode.model.notification import NotificationModel, \
57 from rhodecode.model.notification import NotificationModel, \
58 EmailNotificationModel
58 EmailNotificationModel
59 from rhodecode.model.scm import ScmModel
59 from rhodecode.model.scm import ScmModel
60 from rhodecode.model.settings import VcsSettingsModel
60 from rhodecode.model.settings import VcsSettingsModel
61
61
62
62
63 log = logging.getLogger(__name__)
63 log = logging.getLogger(__name__)
64
64
65
65
66 # Data structure to hold the response data when updating commits during a pull
66 # Data structure to hold the response data when updating commits during a pull
67 # request update.
67 # request update.
68 UpdateResponse = collections.namedtuple('UpdateResponse', [
68 UpdateResponse = collections.namedtuple('UpdateResponse', [
69 'executed', 'reason', 'new', 'old', 'changes',
69 'executed', 'reason', 'new', 'old', 'changes',
70 'source_changed', 'target_changed'])
70 'source_changed', 'target_changed'])
71
71
72
72
73 class PullRequestModel(BaseModel):
73 class PullRequestModel(BaseModel):
74
74
75 cls = PullRequest
75 cls = PullRequest
76
76
77 DIFF_CONTEXT = diffs.DEFAULT_CONTEXT
77 DIFF_CONTEXT = diffs.DEFAULT_CONTEXT
78
78
79 UPDATE_STATUS_MESSAGES = {
79 UPDATE_STATUS_MESSAGES = {
80 UpdateFailureReason.NONE: lazy_ugettext(
80 UpdateFailureReason.NONE: lazy_ugettext(
81 'Pull request update successful.'),
81 'Pull request update successful.'),
82 UpdateFailureReason.UNKNOWN: lazy_ugettext(
82 UpdateFailureReason.UNKNOWN: lazy_ugettext(
83 'Pull request update failed because of an unknown error.'),
83 'Pull request update failed because of an unknown error.'),
84 UpdateFailureReason.NO_CHANGE: lazy_ugettext(
84 UpdateFailureReason.NO_CHANGE: lazy_ugettext(
85 'No update needed because the source and target have not changed.'),
85 'No update needed because the source and target have not changed.'),
86 UpdateFailureReason.WRONG_REF_TYPE: lazy_ugettext(
86 UpdateFailureReason.WRONG_REF_TYPE: lazy_ugettext(
87 'Pull request cannot be updated because the reference type is '
87 'Pull request cannot be updated because the reference type is '
88 'not supported for an update. Only Branch, Tag or Bookmark is allowed.'),
88 'not supported for an update. Only Branch, Tag or Bookmark is allowed.'),
89 UpdateFailureReason.MISSING_TARGET_REF: lazy_ugettext(
89 UpdateFailureReason.MISSING_TARGET_REF: lazy_ugettext(
90 'This pull request cannot be updated because the target '
90 'This pull request cannot be updated because the target '
91 'reference is missing.'),
91 'reference is missing.'),
92 UpdateFailureReason.MISSING_SOURCE_REF: lazy_ugettext(
92 UpdateFailureReason.MISSING_SOURCE_REF: lazy_ugettext(
93 'This pull request cannot be updated because the source '
93 'This pull request cannot be updated because the source '
94 'reference is missing.'),
94 'reference is missing.'),
95 }
95 }
96 REF_TYPES = ['bookmark', 'book', 'tag', 'branch']
96 REF_TYPES = ['bookmark', 'book', 'tag', 'branch']
97 UPDATABLE_REF_TYPES = ['bookmark', 'book', 'branch']
97 UPDATABLE_REF_TYPES = ['bookmark', 'book', 'branch']
98
98
99 def __get_pull_request(self, pull_request):
99 def __get_pull_request(self, pull_request):
100 return self._get_instance((
100 return self._get_instance((
101 PullRequest, PullRequestVersion), pull_request)
101 PullRequest, PullRequestVersion), pull_request)
102
102
103 def _check_perms(self, perms, pull_request, user, api=False):
103 def _check_perms(self, perms, pull_request, user, api=False):
104 if not api:
104 if not api:
105 return h.HasRepoPermissionAny(*perms)(
105 return h.HasRepoPermissionAny(*perms)(
106 user=user, repo_name=pull_request.target_repo.repo_name)
106 user=user, repo_name=pull_request.target_repo.repo_name)
107 else:
107 else:
108 return h.HasRepoPermissionAnyApi(*perms)(
108 return h.HasRepoPermissionAnyApi(*perms)(
109 user=user, repo_name=pull_request.target_repo.repo_name)
109 user=user, repo_name=pull_request.target_repo.repo_name)
110
110
111 def check_user_read(self, pull_request, user, api=False):
111 def check_user_read(self, pull_request, user, api=False):
112 _perms = ('repository.admin', 'repository.write', 'repository.read',)
112 _perms = ('repository.admin', 'repository.write', 'repository.read',)
113 return self._check_perms(_perms, pull_request, user, api)
113 return self._check_perms(_perms, pull_request, user, api)
114
114
115 def check_user_merge(self, pull_request, user, api=False):
115 def check_user_merge(self, pull_request, user, api=False):
116 _perms = ('repository.admin', 'repository.write', 'hg.admin',)
116 _perms = ('repository.admin', 'repository.write', 'hg.admin',)
117 return self._check_perms(_perms, pull_request, user, api)
117 return self._check_perms(_perms, pull_request, user, api)
118
118
119 def check_user_update(self, pull_request, user, api=False):
119 def check_user_update(self, pull_request, user, api=False):
120 owner = user.user_id == pull_request.user_id
120 owner = user.user_id == pull_request.user_id
121 return self.check_user_merge(pull_request, user, api) or owner
121 return self.check_user_merge(pull_request, user, api) or owner
122
122
123 def check_user_delete(self, pull_request, user):
123 def check_user_delete(self, pull_request, user):
124 owner = user.user_id == pull_request.user_id
124 owner = user.user_id == pull_request.user_id
125 _perms = ('repository.admin',)
125 _perms = ('repository.admin',)
126 return self._check_perms(_perms, pull_request, user) or owner
126 return self._check_perms(_perms, pull_request, user) or owner
127
127
128 def check_user_change_status(self, pull_request, user, api=False):
128 def check_user_change_status(self, pull_request, user, api=False):
129 reviewer = user.user_id in [x.user_id for x in
129 reviewer = user.user_id in [x.user_id for x in
130 pull_request.reviewers]
130 pull_request.reviewers]
131 return self.check_user_update(pull_request, user, api) or reviewer
131 return self.check_user_update(pull_request, user, api) or reviewer
132
132
133 def check_user_comment(self, pull_request, user):
133 def check_user_comment(self, pull_request, user):
134 owner = user.user_id == pull_request.user_id
134 owner = user.user_id == pull_request.user_id
135 return self.check_user_read(pull_request, user) or owner
135 return self.check_user_read(pull_request, user) or owner
136
136
137 def get(self, pull_request):
137 def get(self, pull_request):
138 return self.__get_pull_request(pull_request)
138 return self.__get_pull_request(pull_request)
139
139
140 def _prepare_get_all_query(self, repo_name, source=False, statuses=None,
140 def _prepare_get_all_query(self, repo_name, source=False, statuses=None,
141 opened_by=None, order_by=None,
141 opened_by=None, order_by=None,
142 order_dir='desc', only_created=True):
142 order_dir='desc', only_created=True):
143 repo = None
143 repo = None
144 if repo_name:
144 if repo_name:
145 repo = self._get_repo(repo_name)
145 repo = self._get_repo(repo_name)
146
146
147 q = PullRequest.query()
147 q = PullRequest.query()
148
148
149 # source or target
149 # source or target
150 if repo and source:
150 if repo and source:
151 q = q.filter(PullRequest.source_repo == repo)
151 q = q.filter(PullRequest.source_repo == repo)
152 elif repo:
152 elif repo:
153 q = q.filter(PullRequest.target_repo == repo)
153 q = q.filter(PullRequest.target_repo == repo)
154
154
155 # closed,opened
155 # closed,opened
156 if statuses:
156 if statuses:
157 q = q.filter(PullRequest.status.in_(statuses))
157 q = q.filter(PullRequest.status.in_(statuses))
158
158
159 # opened by filter
159 # opened by filter
160 if opened_by:
160 if opened_by:
161 q = q.filter(PullRequest.user_id.in_(opened_by))
161 q = q.filter(PullRequest.user_id.in_(opened_by))
162
162
163 # only get those that are in "created" state
163 # only get those that are in "created" state
164 if only_created:
164 if only_created:
165 q = q.filter(PullRequest.pull_request_state == PullRequest.STATE_CREATED)
165 q = q.filter(PullRequest.pull_request_state == PullRequest.STATE_CREATED)
166
166
167 if order_by:
167 if order_by:
168 order_map = {
168 order_map = {
169 'name_raw': PullRequest.pull_request_id,
169 'name_raw': PullRequest.pull_request_id,
170 'id': PullRequest.pull_request_id,
170 'id': PullRequest.pull_request_id,
171 'title': PullRequest.title,
171 'title': PullRequest.title,
172 'updated_on_raw': PullRequest.updated_on,
172 'updated_on_raw': PullRequest.updated_on,
173 'target_repo': PullRequest.target_repo_id
173 'target_repo': PullRequest.target_repo_id
174 }
174 }
175 if order_dir == 'asc':
175 if order_dir == 'asc':
176 q = q.order_by(order_map[order_by].asc())
176 q = q.order_by(order_map[order_by].asc())
177 else:
177 else:
178 q = q.order_by(order_map[order_by].desc())
178 q = q.order_by(order_map[order_by].desc())
179
179
180 return q
180 return q
181
181
182 def count_all(self, repo_name, source=False, statuses=None,
182 def count_all(self, repo_name, source=False, statuses=None,
183 opened_by=None):
183 opened_by=None):
184 """
184 """
185 Count the number of pull requests for a specific repository.
185 Count the number of pull requests for a specific repository.
186
186
187 :param repo_name: target or source repo
187 :param repo_name: target or source repo
188 :param source: boolean flag to specify if repo_name refers to source
188 :param source: boolean flag to specify if repo_name refers to source
189 :param statuses: list of pull request statuses
189 :param statuses: list of pull request statuses
190 :param opened_by: author user of the pull request
190 :param opened_by: author user of the pull request
191 :returns: int number of pull requests
191 :returns: int number of pull requests
192 """
192 """
193 q = self._prepare_get_all_query(
193 q = self._prepare_get_all_query(
194 repo_name, source=source, statuses=statuses, opened_by=opened_by)
194 repo_name, source=source, statuses=statuses, opened_by=opened_by)
195
195
196 return q.count()
196 return q.count()
197
197
198 def get_all(self, repo_name, source=False, statuses=None, opened_by=None,
198 def get_all(self, repo_name, source=False, statuses=None, opened_by=None,
199 offset=0, length=None, order_by=None, order_dir='desc'):
199 offset=0, length=None, order_by=None, order_dir='desc'):
200 """
200 """
201 Get all pull requests for a specific repository.
201 Get all pull requests for a specific repository.
202
202
203 :param repo_name: target or source repo
203 :param repo_name: target or source repo
204 :param source: boolean flag to specify if repo_name refers to source
204 :param source: boolean flag to specify if repo_name refers to source
205 :param statuses: list of pull request statuses
205 :param statuses: list of pull request statuses
206 :param opened_by: author user of the pull request
206 :param opened_by: author user of the pull request
207 :param offset: pagination offset
207 :param offset: pagination offset
208 :param length: length of returned list
208 :param length: length of returned list
209 :param order_by: order of the returned list
209 :param order_by: order of the returned list
210 :param order_dir: 'asc' or 'desc' ordering direction
210 :param order_dir: 'asc' or 'desc' ordering direction
211 :returns: list of pull requests
211 :returns: list of pull requests
212 """
212 """
213 q = self._prepare_get_all_query(
213 q = self._prepare_get_all_query(
214 repo_name, source=source, statuses=statuses, opened_by=opened_by,
214 repo_name, source=source, statuses=statuses, opened_by=opened_by,
215 order_by=order_by, order_dir=order_dir)
215 order_by=order_by, order_dir=order_dir)
216
216
217 if length:
217 if length:
218 pull_requests = q.limit(length).offset(offset).all()
218 pull_requests = q.limit(length).offset(offset).all()
219 else:
219 else:
220 pull_requests = q.all()
220 pull_requests = q.all()
221
221
222 return pull_requests
222 return pull_requests
223
223
224 def count_awaiting_review(self, repo_name, source=False, statuses=None,
224 def count_awaiting_review(self, repo_name, source=False, statuses=None,
225 opened_by=None):
225 opened_by=None):
226 """
226 """
227 Count the number of pull requests for a specific repository that are
227 Count the number of pull requests for a specific repository that are
228 awaiting review.
228 awaiting review.
229
229
230 :param repo_name: target or source repo
230 :param repo_name: target or source repo
231 :param source: boolean flag to specify if repo_name refers to source
231 :param source: boolean flag to specify if repo_name refers to source
232 :param statuses: list of pull request statuses
232 :param statuses: list of pull request statuses
233 :param opened_by: author user of the pull request
233 :param opened_by: author user of the pull request
234 :returns: int number of pull requests
234 :returns: int number of pull requests
235 """
235 """
236 pull_requests = self.get_awaiting_review(
236 pull_requests = self.get_awaiting_review(
237 repo_name, source=source, statuses=statuses, opened_by=opened_by)
237 repo_name, source=source, statuses=statuses, opened_by=opened_by)
238
238
239 return len(pull_requests)
239 return len(pull_requests)
240
240
241 def get_awaiting_review(self, repo_name, source=False, statuses=None,
241 def get_awaiting_review(self, repo_name, source=False, statuses=None,
242 opened_by=None, offset=0, length=None,
242 opened_by=None, offset=0, length=None,
243 order_by=None, order_dir='desc'):
243 order_by=None, order_dir='desc'):
244 """
244 """
245 Get all pull requests for a specific repository that are awaiting
245 Get all pull requests for a specific repository that are awaiting
246 review.
246 review.
247
247
248 :param repo_name: target or source repo
248 :param repo_name: target or source repo
249 :param source: boolean flag to specify if repo_name refers to source
249 :param source: boolean flag to specify if repo_name refers to source
250 :param statuses: list of pull request statuses
250 :param statuses: list of pull request statuses
251 :param opened_by: author user of the pull request
251 :param opened_by: author user of the pull request
252 :param offset: pagination offset
252 :param offset: pagination offset
253 :param length: length of returned list
253 :param length: length of returned list
254 :param order_by: order of the returned list
254 :param order_by: order of the returned list
255 :param order_dir: 'asc' or 'desc' ordering direction
255 :param order_dir: 'asc' or 'desc' ordering direction
256 :returns: list of pull requests
256 :returns: list of pull requests
257 """
257 """
258 pull_requests = self.get_all(
258 pull_requests = self.get_all(
259 repo_name, source=source, statuses=statuses, opened_by=opened_by,
259 repo_name, source=source, statuses=statuses, opened_by=opened_by,
260 order_by=order_by, order_dir=order_dir)
260 order_by=order_by, order_dir=order_dir)
261
261
262 _filtered_pull_requests = []
262 _filtered_pull_requests = []
263 for pr in pull_requests:
263 for pr in pull_requests:
264 status = pr.calculated_review_status()
264 status = pr.calculated_review_status()
265 if status in [ChangesetStatus.STATUS_NOT_REVIEWED,
265 if status in [ChangesetStatus.STATUS_NOT_REVIEWED,
266 ChangesetStatus.STATUS_UNDER_REVIEW]:
266 ChangesetStatus.STATUS_UNDER_REVIEW]:
267 _filtered_pull_requests.append(pr)
267 _filtered_pull_requests.append(pr)
268 if length:
268 if length:
269 return _filtered_pull_requests[offset:offset+length]
269 return _filtered_pull_requests[offset:offset+length]
270 else:
270 else:
271 return _filtered_pull_requests
271 return _filtered_pull_requests
272
272
273 def count_awaiting_my_review(self, repo_name, source=False, statuses=None,
273 def count_awaiting_my_review(self, repo_name, source=False, statuses=None,
274 opened_by=None, user_id=None):
274 opened_by=None, user_id=None):
275 """
275 """
276 Count the number of pull requests for a specific repository that are
276 Count the number of pull requests for a specific repository that are
277 awaiting review from a specific user.
277 awaiting review from a specific user.
278
278
279 :param repo_name: target or source repo
279 :param repo_name: target or source repo
280 :param source: boolean flag to specify if repo_name refers to source
280 :param source: boolean flag to specify if repo_name refers to source
281 :param statuses: list of pull request statuses
281 :param statuses: list of pull request statuses
282 :param opened_by: author user of the pull request
282 :param opened_by: author user of the pull request
283 :param user_id: reviewer user of the pull request
283 :param user_id: reviewer user of the pull request
284 :returns: int number of pull requests
284 :returns: int number of pull requests
285 """
285 """
286 pull_requests = self.get_awaiting_my_review(
286 pull_requests = self.get_awaiting_my_review(
287 repo_name, source=source, statuses=statuses, opened_by=opened_by,
287 repo_name, source=source, statuses=statuses, opened_by=opened_by,
288 user_id=user_id)
288 user_id=user_id)
289
289
290 return len(pull_requests)
290 return len(pull_requests)
291
291
292 def get_awaiting_my_review(self, repo_name, source=False, statuses=None,
292 def get_awaiting_my_review(self, repo_name, source=False, statuses=None,
293 opened_by=None, user_id=None, offset=0,
293 opened_by=None, user_id=None, offset=0,
294 length=None, order_by=None, order_dir='desc'):
294 length=None, order_by=None, order_dir='desc'):
295 """
295 """
296 Get all pull requests for a specific repository that are awaiting
296 Get all pull requests for a specific repository that are awaiting
297 review from a specific user.
297 review from a specific user.
298
298
299 :param repo_name: target or source repo
299 :param repo_name: target or source repo
300 :param source: boolean flag to specify if repo_name refers to source
300 :param source: boolean flag to specify if repo_name refers to source
301 :param statuses: list of pull request statuses
301 :param statuses: list of pull request statuses
302 :param opened_by: author user of the pull request
302 :param opened_by: author user of the pull request
303 :param user_id: reviewer user of the pull request
303 :param user_id: reviewer user of the pull request
304 :param offset: pagination offset
304 :param offset: pagination offset
305 :param length: length of returned list
305 :param length: length of returned list
306 :param order_by: order of the returned list
306 :param order_by: order of the returned list
307 :param order_dir: 'asc' or 'desc' ordering direction
307 :param order_dir: 'asc' or 'desc' ordering direction
308 :returns: list of pull requests
308 :returns: list of pull requests
309 """
309 """
310 pull_requests = self.get_all(
310 pull_requests = self.get_all(
311 repo_name, source=source, statuses=statuses, opened_by=opened_by,
311 repo_name, source=source, statuses=statuses, opened_by=opened_by,
312 order_by=order_by, order_dir=order_dir)
312 order_by=order_by, order_dir=order_dir)
313
313
314 _my = PullRequestModel().get_not_reviewed(user_id)
314 _my = PullRequestModel().get_not_reviewed(user_id)
315 my_participation = []
315 my_participation = []
316 for pr in pull_requests:
316 for pr in pull_requests:
317 if pr in _my:
317 if pr in _my:
318 my_participation.append(pr)
318 my_participation.append(pr)
319 _filtered_pull_requests = my_participation
319 _filtered_pull_requests = my_participation
320 if length:
320 if length:
321 return _filtered_pull_requests[offset:offset+length]
321 return _filtered_pull_requests[offset:offset+length]
322 else:
322 else:
323 return _filtered_pull_requests
323 return _filtered_pull_requests
324
324
325 def get_not_reviewed(self, user_id):
325 def get_not_reviewed(self, user_id):
326 return [
326 return [
327 x.pull_request for x in PullRequestReviewers.query().filter(
327 x.pull_request for x in PullRequestReviewers.query().filter(
328 PullRequestReviewers.user_id == user_id).all()
328 PullRequestReviewers.user_id == user_id).all()
329 ]
329 ]
330
330
331 def _prepare_participating_query(self, user_id=None, statuses=None,
331 def _prepare_participating_query(self, user_id=None, statuses=None,
332 order_by=None, order_dir='desc'):
332 order_by=None, order_dir='desc'):
333 q = PullRequest.query()
333 q = PullRequest.query()
334 if user_id:
334 if user_id:
335 reviewers_subquery = Session().query(
335 reviewers_subquery = Session().query(
336 PullRequestReviewers.pull_request_id).filter(
336 PullRequestReviewers.pull_request_id).filter(
337 PullRequestReviewers.user_id == user_id).subquery()
337 PullRequestReviewers.user_id == user_id).subquery()
338 user_filter = or_(
338 user_filter = or_(
339 PullRequest.user_id == user_id,
339 PullRequest.user_id == user_id,
340 PullRequest.pull_request_id.in_(reviewers_subquery)
340 PullRequest.pull_request_id.in_(reviewers_subquery)
341 )
341 )
342 q = PullRequest.query().filter(user_filter)
342 q = PullRequest.query().filter(user_filter)
343
343
344 # closed,opened
344 # closed,opened
345 if statuses:
345 if statuses:
346 q = q.filter(PullRequest.status.in_(statuses))
346 q = q.filter(PullRequest.status.in_(statuses))
347
347
348 if order_by:
348 if order_by:
349 order_map = {
349 order_map = {
350 'name_raw': PullRequest.pull_request_id,
350 'name_raw': PullRequest.pull_request_id,
351 'title': PullRequest.title,
351 'title': PullRequest.title,
352 'updated_on_raw': PullRequest.updated_on,
352 'updated_on_raw': PullRequest.updated_on,
353 'target_repo': PullRequest.target_repo_id
353 'target_repo': PullRequest.target_repo_id
354 }
354 }
355 if order_dir == 'asc':
355 if order_dir == 'asc':
356 q = q.order_by(order_map[order_by].asc())
356 q = q.order_by(order_map[order_by].asc())
357 else:
357 else:
358 q = q.order_by(order_map[order_by].desc())
358 q = q.order_by(order_map[order_by].desc())
359
359
360 return q
360 return q
361
361
362 def count_im_participating_in(self, user_id=None, statuses=None):
362 def count_im_participating_in(self, user_id=None, statuses=None):
363 q = self._prepare_participating_query(user_id, statuses=statuses)
363 q = self._prepare_participating_query(user_id, statuses=statuses)
364 return q.count()
364 return q.count()
365
365
366 def get_im_participating_in(
366 def get_im_participating_in(
367 self, user_id=None, statuses=None, offset=0,
367 self, user_id=None, statuses=None, offset=0,
368 length=None, order_by=None, order_dir='desc'):
368 length=None, order_by=None, order_dir='desc'):
369 """
369 """
370 Get all Pull requests that i'm participating in, or i have opened
370 Get all Pull requests that i'm participating in, or i have opened
371 """
371 """
372
372
373 q = self._prepare_participating_query(
373 q = self._prepare_participating_query(
374 user_id, statuses=statuses, order_by=order_by,
374 user_id, statuses=statuses, order_by=order_by,
375 order_dir=order_dir)
375 order_dir=order_dir)
376
376
377 if length:
377 if length:
378 pull_requests = q.limit(length).offset(offset).all()
378 pull_requests = q.limit(length).offset(offset).all()
379 else:
379 else:
380 pull_requests = q.all()
380 pull_requests = q.all()
381
381
382 return pull_requests
382 return pull_requests
383
383
384 def get_versions(self, pull_request):
384 def get_versions(self, pull_request):
385 """
385 """
386 returns version of pull request sorted by ID descending
386 returns version of pull request sorted by ID descending
387 """
387 """
388 return PullRequestVersion.query()\
388 return PullRequestVersion.query()\
389 .filter(PullRequestVersion.pull_request == pull_request)\
389 .filter(PullRequestVersion.pull_request == pull_request)\
390 .order_by(PullRequestVersion.pull_request_version_id.asc())\
390 .order_by(PullRequestVersion.pull_request_version_id.asc())\
391 .all()
391 .all()
392
392
393 def get_pr_version(self, pull_request_id, version=None):
393 def get_pr_version(self, pull_request_id, version=None):
394 at_version = None
394 at_version = None
395
395
396 if version and version == 'latest':
396 if version and version == 'latest':
397 pull_request_ver = PullRequest.get(pull_request_id)
397 pull_request_ver = PullRequest.get(pull_request_id)
398 pull_request_obj = pull_request_ver
398 pull_request_obj = pull_request_ver
399 _org_pull_request_obj = pull_request_obj
399 _org_pull_request_obj = pull_request_obj
400 at_version = 'latest'
400 at_version = 'latest'
401 elif version:
401 elif version:
402 pull_request_ver = PullRequestVersion.get_or_404(version)
402 pull_request_ver = PullRequestVersion.get_or_404(version)
403 pull_request_obj = pull_request_ver
403 pull_request_obj = pull_request_ver
404 _org_pull_request_obj = pull_request_ver.pull_request
404 _org_pull_request_obj = pull_request_ver.pull_request
405 at_version = pull_request_ver.pull_request_version_id
405 at_version = pull_request_ver.pull_request_version_id
406 else:
406 else:
407 _org_pull_request_obj = pull_request_obj = PullRequest.get_or_404(
407 _org_pull_request_obj = pull_request_obj = PullRequest.get_or_404(
408 pull_request_id)
408 pull_request_id)
409
409
410 pull_request_display_obj = PullRequest.get_pr_display_object(
410 pull_request_display_obj = PullRequest.get_pr_display_object(
411 pull_request_obj, _org_pull_request_obj)
411 pull_request_obj, _org_pull_request_obj)
412
412
413 return _org_pull_request_obj, pull_request_obj, \
413 return _org_pull_request_obj, pull_request_obj, \
414 pull_request_display_obj, at_version
414 pull_request_display_obj, at_version
415
415
416 def create(self, created_by, source_repo, source_ref, target_repo,
416 def create(self, created_by, source_repo, source_ref, target_repo,
417 target_ref, revisions, reviewers, title, description=None,
417 target_ref, revisions, reviewers, title, description=None,
418 description_renderer=None,
418 description_renderer=None,
419 reviewer_data=None, translator=None, auth_user=None):
419 reviewer_data=None, translator=None, auth_user=None):
420 translator = translator or get_current_request().translate
420 translator = translator or get_current_request().translate
421
421
422 created_by_user = self._get_user(created_by)
422 created_by_user = self._get_user(created_by)
423 auth_user = auth_user or created_by_user.AuthUser()
423 auth_user = auth_user or created_by_user.AuthUser()
424 source_repo = self._get_repo(source_repo)
424 source_repo = self._get_repo(source_repo)
425 target_repo = self._get_repo(target_repo)
425 target_repo = self._get_repo(target_repo)
426
426
427 pull_request = PullRequest()
427 pull_request = PullRequest()
428 pull_request.source_repo = source_repo
428 pull_request.source_repo = source_repo
429 pull_request.source_ref = source_ref
429 pull_request.source_ref = source_ref
430 pull_request.target_repo = target_repo
430 pull_request.target_repo = target_repo
431 pull_request.target_ref = target_ref
431 pull_request.target_ref = target_ref
432 pull_request.revisions = revisions
432 pull_request.revisions = revisions
433 pull_request.title = title
433 pull_request.title = title
434 pull_request.description = description
434 pull_request.description = description
435 pull_request.description_renderer = description_renderer
435 pull_request.description_renderer = description_renderer
436 pull_request.author = created_by_user
436 pull_request.author = created_by_user
437 pull_request.reviewer_data = reviewer_data
437 pull_request.reviewer_data = reviewer_data
438 pull_request.pull_request_state = pull_request.STATE_CREATING
438 pull_request.pull_request_state = pull_request.STATE_CREATING
439 Session().add(pull_request)
439 Session().add(pull_request)
440 Session().flush()
440 Session().flush()
441
441
442 reviewer_ids = set()
442 reviewer_ids = set()
443 # members / reviewers
443 # members / reviewers
444 for reviewer_object in reviewers:
444 for reviewer_object in reviewers:
445 user_id, reasons, mandatory, rules = reviewer_object
445 user_id, reasons, mandatory, rules = reviewer_object
446 user = self._get_user(user_id)
446 user = self._get_user(user_id)
447
447
448 # skip duplicates
448 # skip duplicates
449 if user.user_id in reviewer_ids:
449 if user.user_id in reviewer_ids:
450 continue
450 continue
451
451
452 reviewer_ids.add(user.user_id)
452 reviewer_ids.add(user.user_id)
453
453
454 reviewer = PullRequestReviewers()
454 reviewer = PullRequestReviewers()
455 reviewer.user = user
455 reviewer.user = user
456 reviewer.pull_request = pull_request
456 reviewer.pull_request = pull_request
457 reviewer.reasons = reasons
457 reviewer.reasons = reasons
458 reviewer.mandatory = mandatory
458 reviewer.mandatory = mandatory
459
459
460 # NOTE(marcink): pick only first rule for now
460 # NOTE(marcink): pick only first rule for now
461 rule_id = list(rules)[0] if rules else None
461 rule_id = list(rules)[0] if rules else None
462 rule = RepoReviewRule.get(rule_id) if rule_id else None
462 rule = RepoReviewRule.get(rule_id) if rule_id else None
463 if rule:
463 if rule:
464 review_group = rule.user_group_vote_rule(user_id)
464 review_group = rule.user_group_vote_rule(user_id)
465 # we check if this particular reviewer is member of a voting group
465 # we check if this particular reviewer is member of a voting group
466 if review_group:
466 if review_group:
467 # NOTE(marcink):
467 # NOTE(marcink):
468 # can be that user is member of more but we pick the first same,
468 # can be that user is member of more but we pick the first same,
469 # same as default reviewers algo
469 # same as default reviewers algo
470 review_group = review_group[0]
470 review_group = review_group[0]
471
471
472 rule_data = {
472 rule_data = {
473 'rule_name':
473 'rule_name':
474 rule.review_rule_name,
474 rule.review_rule_name,
475 'rule_user_group_entry_id':
475 'rule_user_group_entry_id':
476 review_group.repo_review_rule_users_group_id,
476 review_group.repo_review_rule_users_group_id,
477 'rule_user_group_name':
477 'rule_user_group_name':
478 review_group.users_group.users_group_name,
478 review_group.users_group.users_group_name,
479 'rule_user_group_members':
479 'rule_user_group_members':
480 [x.user.username for x in review_group.users_group.members],
480 [x.user.username for x in review_group.users_group.members],
481 'rule_user_group_members_id':
481 'rule_user_group_members_id':
482 [x.user.user_id for x in review_group.users_group.members],
482 [x.user.user_id for x in review_group.users_group.members],
483 }
483 }
484 # e.g {'vote_rule': -1, 'mandatory': True}
484 # e.g {'vote_rule': -1, 'mandatory': True}
485 rule_data.update(review_group.rule_data())
485 rule_data.update(review_group.rule_data())
486
486
487 reviewer.rule_data = rule_data
487 reviewer.rule_data = rule_data
488
488
489 Session().add(reviewer)
489 Session().add(reviewer)
490 Session().flush()
490 Session().flush()
491
491
492 # Set approval status to "Under Review" for all commits which are
492 # Set approval status to "Under Review" for all commits which are
493 # part of this pull request.
493 # part of this pull request.
494 ChangesetStatusModel().set_status(
494 ChangesetStatusModel().set_status(
495 repo=target_repo,
495 repo=target_repo,
496 status=ChangesetStatus.STATUS_UNDER_REVIEW,
496 status=ChangesetStatus.STATUS_UNDER_REVIEW,
497 user=created_by_user,
497 user=created_by_user,
498 pull_request=pull_request
498 pull_request=pull_request
499 )
499 )
500 # we commit early at this point. This has to do with a fact
500 # we commit early at this point. This has to do with a fact
501 # that before queries do some row-locking. And because of that
501 # that before queries do some row-locking. And because of that
502 # we need to commit and finish transaction before below validate call
502 # we need to commit and finish transaction before below validate call
503 # that for large repos could be long resulting in long row locks
503 # that for large repos could be long resulting in long row locks
504 Session().commit()
504 Session().commit()
505
505
506 # prepare workspace, and run initial merge simulation. Set state during that
506 # prepare workspace, and run initial merge simulation. Set state during that
507 # operation
507 # operation
508 pull_request = PullRequest.get(pull_request.pull_request_id)
508 pull_request = PullRequest.get(pull_request.pull_request_id)
509
509
510 # set as merging, for simulation, and if finished to created so we mark
510 # set as merging, for simulation, and if finished to created so we mark
511 # simulation is working fine
511 # simulation is working fine
512 with pull_request.set_state(PullRequest.STATE_MERGING,
512 with pull_request.set_state(PullRequest.STATE_MERGING,
513 final_state=PullRequest.STATE_CREATED):
513 final_state=PullRequest.STATE_CREATED):
514 MergeCheck.validate(
514 MergeCheck.validate(
515 pull_request, auth_user=auth_user, translator=translator)
515 pull_request, auth_user=auth_user, translator=translator)
516
516
517 self.notify_reviewers(pull_request, reviewer_ids)
517 self.notify_reviewers(pull_request, reviewer_ids)
518 self.trigger_pull_request_hook(
518 self.trigger_pull_request_hook(
519 pull_request, created_by_user, 'create')
519 pull_request, created_by_user, 'create')
520
520
521 creation_data = pull_request.get_api_data(with_merge_state=False)
521 creation_data = pull_request.get_api_data(with_merge_state=False)
522 self._log_audit_action(
522 self._log_audit_action(
523 'repo.pull_request.create', {'data': creation_data},
523 'repo.pull_request.create', {'data': creation_data},
524 auth_user, pull_request)
524 auth_user, pull_request)
525
525
526 return pull_request
526 return pull_request
527
527
528 def trigger_pull_request_hook(self, pull_request, user, action, data=None):
528 def trigger_pull_request_hook(self, pull_request, user, action, data=None):
529 pull_request = self.__get_pull_request(pull_request)
529 pull_request = self.__get_pull_request(pull_request)
530 target_scm = pull_request.target_repo.scm_instance()
530 target_scm = pull_request.target_repo.scm_instance()
531 if action == 'create':
531 if action == 'create':
532 trigger_hook = hooks_utils.trigger_log_create_pull_request_hook
532 trigger_hook = hooks_utils.trigger_log_create_pull_request_hook
533 elif action == 'merge':
533 elif action == 'merge':
534 trigger_hook = hooks_utils.trigger_log_merge_pull_request_hook
534 trigger_hook = hooks_utils.trigger_log_merge_pull_request_hook
535 elif action == 'close':
535 elif action == 'close':
536 trigger_hook = hooks_utils.trigger_log_close_pull_request_hook
536 trigger_hook = hooks_utils.trigger_log_close_pull_request_hook
537 elif action == 'review_status_change':
537 elif action == 'review_status_change':
538 trigger_hook = hooks_utils.trigger_log_review_pull_request_hook
538 trigger_hook = hooks_utils.trigger_log_review_pull_request_hook
539 elif action == 'update':
539 elif action == 'update':
540 trigger_hook = hooks_utils.trigger_log_update_pull_request_hook
540 trigger_hook = hooks_utils.trigger_log_update_pull_request_hook
541 elif action == 'comment':
541 elif action == 'comment':
542 # dummy hook ! for comment. We want this function to handle all cases
542 # dummy hook ! for comment. We want this function to handle all cases
543 def trigger_hook(*args, **kwargs):
543 def trigger_hook(*args, **kwargs):
544 pass
544 pass
545 comment = data['comment']
545 comment = data['comment']
546 events.trigger(events.PullRequestCommentEvent(pull_request, comment))
546 events.trigger(events.PullRequestCommentEvent(pull_request, comment))
547 else:
547 else:
548 return
548 return
549
549
550 trigger_hook(
550 trigger_hook(
551 username=user.username,
551 username=user.username,
552 repo_name=pull_request.target_repo.repo_name,
552 repo_name=pull_request.target_repo.repo_name,
553 repo_alias=target_scm.alias,
553 repo_alias=target_scm.alias,
554 pull_request=pull_request,
554 pull_request=pull_request,
555 data=data)
555 data=data)
556
556
557 def _get_commit_ids(self, pull_request):
557 def _get_commit_ids(self, pull_request):
558 """
558 """
559 Return the commit ids of the merged pull request.
559 Return the commit ids of the merged pull request.
560
560
561 This method is not dealing correctly yet with the lack of autoupdates
561 This method is not dealing correctly yet with the lack of autoupdates
562 nor with the implicit target updates.
562 nor with the implicit target updates.
563 For example: if a commit in the source repo is already in the target it
563 For example: if a commit in the source repo is already in the target it
564 will be reported anyways.
564 will be reported anyways.
565 """
565 """
566 merge_rev = pull_request.merge_rev
566 merge_rev = pull_request.merge_rev
567 if merge_rev is None:
567 if merge_rev is None:
568 raise ValueError('This pull request was not merged yet')
568 raise ValueError('This pull request was not merged yet')
569
569
570 commit_ids = list(pull_request.revisions)
570 commit_ids = list(pull_request.revisions)
571 if merge_rev not in commit_ids:
571 if merge_rev not in commit_ids:
572 commit_ids.append(merge_rev)
572 commit_ids.append(merge_rev)
573
573
574 return commit_ids
574 return commit_ids
575
575
576 def merge_repo(self, pull_request, user, extras):
576 def merge_repo(self, pull_request, user, extras):
577 log.debug("Merging pull request %s", pull_request.pull_request_id)
577 log.debug("Merging pull request %s", pull_request.pull_request_id)
578 extras['user_agent'] = 'internal-merge'
578 extras['user_agent'] = 'internal-merge'
579 merge_state = self._merge_pull_request(pull_request, user, extras)
579 merge_state = self._merge_pull_request(pull_request, user, extras)
580 if merge_state.executed:
580 if merge_state.executed:
581 log.debug("Merge was successful, updating the pull request comments.")
581 log.debug("Merge was successful, updating the pull request comments.")
582 self._comment_and_close_pr(pull_request, user, merge_state)
582 self._comment_and_close_pr(pull_request, user, merge_state)
583
583
584 self._log_audit_action(
584 self._log_audit_action(
585 'repo.pull_request.merge',
585 'repo.pull_request.merge',
586 {'merge_state': merge_state.__dict__},
586 {'merge_state': merge_state.__dict__},
587 user, pull_request)
587 user, pull_request)
588
588
589 else:
589 else:
590 log.warn("Merge failed, not updating the pull request.")
590 log.warn("Merge failed, not updating the pull request.")
591 return merge_state
591 return merge_state
592
592
593 def _merge_pull_request(self, pull_request, user, extras, merge_msg=None):
593 def _merge_pull_request(self, pull_request, user, extras, merge_msg=None):
594 target_vcs = pull_request.target_repo.scm_instance()
594 target_vcs = pull_request.target_repo.scm_instance()
595 source_vcs = pull_request.source_repo.scm_instance()
595 source_vcs = pull_request.source_repo.scm_instance()
596
596
597 message = safe_unicode(merge_msg or vcs_settings.MERGE_MESSAGE_TMPL).format(
597 message = safe_unicode(merge_msg or vcs_settings.MERGE_MESSAGE_TMPL).format(
598 pr_id=pull_request.pull_request_id,
598 pr_id=pull_request.pull_request_id,
599 pr_title=pull_request.title,
599 pr_title=pull_request.title,
600 source_repo=source_vcs.name,
600 source_repo=source_vcs.name,
601 source_ref_name=pull_request.source_ref_parts.name,
601 source_ref_name=pull_request.source_ref_parts.name,
602 target_repo=target_vcs.name,
602 target_repo=target_vcs.name,
603 target_ref_name=pull_request.target_ref_parts.name,
603 target_ref_name=pull_request.target_ref_parts.name,
604 )
604 )
605
605
606 workspace_id = self._workspace_id(pull_request)
606 workspace_id = self._workspace_id(pull_request)
607 repo_id = pull_request.target_repo.repo_id
607 repo_id = pull_request.target_repo.repo_id
608 use_rebase = self._use_rebase_for_merging(pull_request)
608 use_rebase = self._use_rebase_for_merging(pull_request)
609 close_branch = self._close_branch_before_merging(pull_request)
609 close_branch = self._close_branch_before_merging(pull_request)
610
610
611 target_ref = self._refresh_reference(
611 target_ref = self._refresh_reference(
612 pull_request.target_ref_parts, target_vcs)
612 pull_request.target_ref_parts, target_vcs)
613
613
614 callback_daemon, extras = prepare_callback_daemon(
614 callback_daemon, extras = prepare_callback_daemon(
615 extras, protocol=vcs_settings.HOOKS_PROTOCOL,
615 extras, protocol=vcs_settings.HOOKS_PROTOCOL,
616 host=vcs_settings.HOOKS_HOST,
616 host=vcs_settings.HOOKS_HOST,
617 use_direct_calls=vcs_settings.HOOKS_DIRECT_CALLS)
617 use_direct_calls=vcs_settings.HOOKS_DIRECT_CALLS)
618
618
619 with callback_daemon:
619 with callback_daemon:
620 # TODO: johbo: Implement a clean way to run a config_override
620 # TODO: johbo: Implement a clean way to run a config_override
621 # for a single call.
621 # for a single call.
622 target_vcs.config.set(
622 target_vcs.config.set(
623 'rhodecode', 'RC_SCM_DATA', json.dumps(extras))
623 'rhodecode', 'RC_SCM_DATA', json.dumps(extras))
624
624
625 user_name = user.short_contact
625 user_name = user.short_contact
626 merge_state = target_vcs.merge(
626 merge_state = target_vcs.merge(
627 repo_id, workspace_id, target_ref, source_vcs,
627 repo_id, workspace_id, target_ref, source_vcs,
628 pull_request.source_ref_parts,
628 pull_request.source_ref_parts,
629 user_name=user_name, user_email=user.email,
629 user_name=user_name, user_email=user.email,
630 message=message, use_rebase=use_rebase,
630 message=message, use_rebase=use_rebase,
631 close_branch=close_branch)
631 close_branch=close_branch)
632 return merge_state
632 return merge_state
633
633
634 def _comment_and_close_pr(self, pull_request, user, merge_state, close_msg=None):
634 def _comment_and_close_pr(self, pull_request, user, merge_state, close_msg=None):
635 pull_request.merge_rev = merge_state.merge_ref.commit_id
635 pull_request.merge_rev = merge_state.merge_ref.commit_id
636 pull_request.updated_on = datetime.datetime.now()
636 pull_request.updated_on = datetime.datetime.now()
637 close_msg = close_msg or 'Pull request merged and closed'
637 close_msg = close_msg or 'Pull request merged and closed'
638
638
639 CommentsModel().create(
639 CommentsModel().create(
640 text=safe_unicode(close_msg),
640 text=safe_unicode(close_msg),
641 repo=pull_request.target_repo.repo_id,
641 repo=pull_request.target_repo.repo_id,
642 user=user.user_id,
642 user=user.user_id,
643 pull_request=pull_request.pull_request_id,
643 pull_request=pull_request.pull_request_id,
644 f_path=None,
644 f_path=None,
645 line_no=None,
645 line_no=None,
646 closing_pr=True
646 closing_pr=True
647 )
647 )
648
648
649 Session().add(pull_request)
649 Session().add(pull_request)
650 Session().flush()
650 Session().flush()
651 # TODO: paris: replace invalidation with less radical solution
651 # TODO: paris: replace invalidation with less radical solution
652 ScmModel().mark_for_invalidation(
652 ScmModel().mark_for_invalidation(
653 pull_request.target_repo.repo_name)
653 pull_request.target_repo.repo_name)
654 self.trigger_pull_request_hook(pull_request, user, 'merge')
654 self.trigger_pull_request_hook(pull_request, user, 'merge')
655
655
656 def has_valid_update_type(self, pull_request):
656 def has_valid_update_type(self, pull_request):
657 source_ref_type = pull_request.source_ref_parts.type
657 source_ref_type = pull_request.source_ref_parts.type
658 return source_ref_type in self.REF_TYPES
658 return source_ref_type in self.REF_TYPES
659
659
660 def update_commits(self, pull_request):
660 def update_commits(self, pull_request):
661 """
661 """
662 Get the updated list of commits for the pull request
662 Get the updated list of commits for the pull request
663 and return the new pull request version and the list
663 and return the new pull request version and the list
664 of commits processed by this update action
664 of commits processed by this update action
665 """
665 """
666 pull_request = self.__get_pull_request(pull_request)
666 pull_request = self.__get_pull_request(pull_request)
667 source_ref_type = pull_request.source_ref_parts.type
667 source_ref_type = pull_request.source_ref_parts.type
668 source_ref_name = pull_request.source_ref_parts.name
668 source_ref_name = pull_request.source_ref_parts.name
669 source_ref_id = pull_request.source_ref_parts.commit_id
669 source_ref_id = pull_request.source_ref_parts.commit_id
670
670
671 target_ref_type = pull_request.target_ref_parts.type
671 target_ref_type = pull_request.target_ref_parts.type
672 target_ref_name = pull_request.target_ref_parts.name
672 target_ref_name = pull_request.target_ref_parts.name
673 target_ref_id = pull_request.target_ref_parts.commit_id
673 target_ref_id = pull_request.target_ref_parts.commit_id
674
674
675 if not self.has_valid_update_type(pull_request):
675 if not self.has_valid_update_type(pull_request):
676 log.debug("Skipping update of pull request %s due to ref type: %s",
676 log.debug("Skipping update of pull request %s due to ref type: %s",
677 pull_request, source_ref_type)
677 pull_request, source_ref_type)
678 return UpdateResponse(
678 return UpdateResponse(
679 executed=False,
679 executed=False,
680 reason=UpdateFailureReason.WRONG_REF_TYPE,
680 reason=UpdateFailureReason.WRONG_REF_TYPE,
681 old=pull_request, new=None, changes=None,
681 old=pull_request, new=None, changes=None,
682 source_changed=False, target_changed=False)
682 source_changed=False, target_changed=False)
683
683
684 # source repo
684 # source repo
685 source_repo = pull_request.source_repo.scm_instance()
685 source_repo = pull_request.source_repo.scm_instance()
686 try:
686 try:
687 source_commit = source_repo.get_commit(commit_id=source_ref_name)
687 source_commit = source_repo.get_commit(commit_id=source_ref_name)
688 except CommitDoesNotExistError:
688 except CommitDoesNotExistError:
689 return UpdateResponse(
689 return UpdateResponse(
690 executed=False,
690 executed=False,
691 reason=UpdateFailureReason.MISSING_SOURCE_REF,
691 reason=UpdateFailureReason.MISSING_SOURCE_REF,
692 old=pull_request, new=None, changes=None,
692 old=pull_request, new=None, changes=None,
693 source_changed=False, target_changed=False)
693 source_changed=False, target_changed=False)
694
694
695 source_changed = source_ref_id != source_commit.raw_id
695 source_changed = source_ref_id != source_commit.raw_id
696
696
697 # target repo
697 # target repo
698 target_repo = pull_request.target_repo.scm_instance()
698 target_repo = pull_request.target_repo.scm_instance()
699 try:
699 try:
700 target_commit = target_repo.get_commit(commit_id=target_ref_name)
700 target_commit = target_repo.get_commit(commit_id=target_ref_name)
701 except CommitDoesNotExistError:
701 except CommitDoesNotExistError:
702 return UpdateResponse(
702 return UpdateResponse(
703 executed=False,
703 executed=False,
704 reason=UpdateFailureReason.MISSING_TARGET_REF,
704 reason=UpdateFailureReason.MISSING_TARGET_REF,
705 old=pull_request, new=None, changes=None,
705 old=pull_request, new=None, changes=None,
706 source_changed=False, target_changed=False)
706 source_changed=False, target_changed=False)
707 target_changed = target_ref_id != target_commit.raw_id
707 target_changed = target_ref_id != target_commit.raw_id
708
708
709 if not (source_changed or target_changed):
709 if not (source_changed or target_changed):
710 log.debug("Nothing changed in pull request %s", pull_request)
710 log.debug("Nothing changed in pull request %s", pull_request)
711 return UpdateResponse(
711 return UpdateResponse(
712 executed=False,
712 executed=False,
713 reason=UpdateFailureReason.NO_CHANGE,
713 reason=UpdateFailureReason.NO_CHANGE,
714 old=pull_request, new=None, changes=None,
714 old=pull_request, new=None, changes=None,
715 source_changed=target_changed, target_changed=source_changed)
715 source_changed=target_changed, target_changed=source_changed)
716
716
717 change_in_found = 'target repo' if target_changed else 'source repo'
717 change_in_found = 'target repo' if target_changed else 'source repo'
718 log.debug('Updating pull request because of change in %s detected',
718 log.debug('Updating pull request because of change in %s detected',
719 change_in_found)
719 change_in_found)
720
720
721 # Finally there is a need for an update, in case of source change
721 # Finally there is a need for an update, in case of source change
722 # we create a new version, else just an update
722 # we create a new version, else just an update
723 if source_changed:
723 if source_changed:
724 pull_request_version = self._create_version_from_snapshot(pull_request)
724 pull_request_version = self._create_version_from_snapshot(pull_request)
725 self._link_comments_to_version(pull_request_version)
725 self._link_comments_to_version(pull_request_version)
726 else:
726 else:
727 try:
727 try:
728 ver = pull_request.versions[-1]
728 ver = pull_request.versions[-1]
729 except IndexError:
729 except IndexError:
730 ver = None
730 ver = None
731
731
732 pull_request.pull_request_version_id = \
732 pull_request.pull_request_version_id = \
733 ver.pull_request_version_id if ver else None
733 ver.pull_request_version_id if ver else None
734 pull_request_version = pull_request
734 pull_request_version = pull_request
735
735
736 try:
736 try:
737 if target_ref_type in self.REF_TYPES:
737 if target_ref_type in self.REF_TYPES:
738 target_commit = target_repo.get_commit(target_ref_name)
738 target_commit = target_repo.get_commit(target_ref_name)
739 else:
739 else:
740 target_commit = target_repo.get_commit(target_ref_id)
740 target_commit = target_repo.get_commit(target_ref_id)
741 except CommitDoesNotExistError:
741 except CommitDoesNotExistError:
742 return UpdateResponse(
742 return UpdateResponse(
743 executed=False,
743 executed=False,
744 reason=UpdateFailureReason.MISSING_TARGET_REF,
744 reason=UpdateFailureReason.MISSING_TARGET_REF,
745 old=pull_request, new=None, changes=None,
745 old=pull_request, new=None, changes=None,
746 source_changed=source_changed, target_changed=target_changed)
746 source_changed=source_changed, target_changed=target_changed)
747
747
748 # re-compute commit ids
748 # re-compute commit ids
749 old_commit_ids = pull_request.revisions
749 old_commit_ids = pull_request.revisions
750 pre_load = ["author", "branch", "date", "message"]
750 pre_load = ["author", "branch", "date", "message"]
751 commit_ranges = target_repo.compare(
751 commit_ranges = target_repo.compare(
752 target_commit.raw_id, source_commit.raw_id, source_repo, merge=True,
752 target_commit.raw_id, source_commit.raw_id, source_repo, merge=True,
753 pre_load=pre_load)
753 pre_load=pre_load)
754
754
755 ancestor = target_repo.get_common_ancestor(
755 ancestor = target_repo.get_common_ancestor(
756 target_commit.raw_id, source_commit.raw_id, source_repo)
756 target_commit.raw_id, source_commit.raw_id, source_repo)
757
757
758 pull_request.source_ref = '%s:%s:%s' % (
758 pull_request.source_ref = '%s:%s:%s' % (
759 source_ref_type, source_ref_name, source_commit.raw_id)
759 source_ref_type, source_ref_name, source_commit.raw_id)
760 pull_request.target_ref = '%s:%s:%s' % (
760 pull_request.target_ref = '%s:%s:%s' % (
761 target_ref_type, target_ref_name, ancestor)
761 target_ref_type, target_ref_name, ancestor)
762
762
763 pull_request.revisions = [
763 pull_request.revisions = [
764 commit.raw_id for commit in reversed(commit_ranges)]
764 commit.raw_id for commit in reversed(commit_ranges)]
765 pull_request.updated_on = datetime.datetime.now()
765 pull_request.updated_on = datetime.datetime.now()
766 Session().add(pull_request)
766 Session().add(pull_request)
767 new_commit_ids = pull_request.revisions
767 new_commit_ids = pull_request.revisions
768
768
769 old_diff_data, new_diff_data = self._generate_update_diffs(
769 old_diff_data, new_diff_data = self._generate_update_diffs(
770 pull_request, pull_request_version)
770 pull_request, pull_request_version)
771
771
772 # calculate commit and file changes
772 # calculate commit and file changes
773 changes = self._calculate_commit_id_changes(
773 changes = self._calculate_commit_id_changes(
774 old_commit_ids, new_commit_ids)
774 old_commit_ids, new_commit_ids)
775 file_changes = self._calculate_file_changes(
775 file_changes = self._calculate_file_changes(
776 old_diff_data, new_diff_data)
776 old_diff_data, new_diff_data)
777
777
778 # set comments as outdated if DIFFS changed
778 # set comments as outdated if DIFFS changed
779 CommentsModel().outdate_comments(
779 CommentsModel().outdate_comments(
780 pull_request, old_diff_data=old_diff_data,
780 pull_request, old_diff_data=old_diff_data,
781 new_diff_data=new_diff_data)
781 new_diff_data=new_diff_data)
782
782
783 commit_changes = (changes.added or changes.removed)
783 commit_changes = (changes.added or changes.removed)
784 file_node_changes = (
784 file_node_changes = (
785 file_changes.added or file_changes.modified or file_changes.removed)
785 file_changes.added or file_changes.modified or file_changes.removed)
786 pr_has_changes = commit_changes or file_node_changes
786 pr_has_changes = commit_changes or file_node_changes
787
787
788 # Add an automatic comment to the pull request, in case
788 # Add an automatic comment to the pull request, in case
789 # anything has changed
789 # anything has changed
790 if pr_has_changes:
790 if pr_has_changes:
791 update_comment = CommentsModel().create(
791 update_comment = CommentsModel().create(
792 text=self._render_update_message(changes, file_changes),
792 text=self._render_update_message(changes, file_changes),
793 repo=pull_request.target_repo,
793 repo=pull_request.target_repo,
794 user=pull_request.author,
794 user=pull_request.author,
795 pull_request=pull_request,
795 pull_request=pull_request,
796 send_email=False, renderer=DEFAULT_COMMENTS_RENDERER)
796 send_email=False, renderer=DEFAULT_COMMENTS_RENDERER)
797
797
798 # Update status to "Under Review" for added commits
798 # Update status to "Under Review" for added commits
799 for commit_id in changes.added:
799 for commit_id in changes.added:
800 ChangesetStatusModel().set_status(
800 ChangesetStatusModel().set_status(
801 repo=pull_request.source_repo,
801 repo=pull_request.source_repo,
802 status=ChangesetStatus.STATUS_UNDER_REVIEW,
802 status=ChangesetStatus.STATUS_UNDER_REVIEW,
803 comment=update_comment,
803 comment=update_comment,
804 user=pull_request.author,
804 user=pull_request.author,
805 pull_request=pull_request,
805 pull_request=pull_request,
806 revision=commit_id)
806 revision=commit_id)
807
807
808 log.debug(
808 log.debug(
809 'Updated pull request %s, added_ids: %s, common_ids: %s, '
809 'Updated pull request %s, added_ids: %s, common_ids: %s, '
810 'removed_ids: %s', pull_request.pull_request_id,
810 'removed_ids: %s', pull_request.pull_request_id,
811 changes.added, changes.common, changes.removed)
811 changes.added, changes.common, changes.removed)
812 log.debug(
812 log.debug(
813 'Updated pull request with the following file changes: %s',
813 'Updated pull request with the following file changes: %s',
814 file_changes)
814 file_changes)
815
815
816 log.info(
816 log.info(
817 "Updated pull request %s from commit %s to commit %s, "
817 "Updated pull request %s from commit %s to commit %s, "
818 "stored new version %s of this pull request.",
818 "stored new version %s of this pull request.",
819 pull_request.pull_request_id, source_ref_id,
819 pull_request.pull_request_id, source_ref_id,
820 pull_request.source_ref_parts.commit_id,
820 pull_request.source_ref_parts.commit_id,
821 pull_request_version.pull_request_version_id)
821 pull_request_version.pull_request_version_id)
822 Session().commit()
822 Session().commit()
823 self.trigger_pull_request_hook(pull_request, pull_request.author, 'update')
823 self.trigger_pull_request_hook(pull_request, pull_request.author, 'update')
824
824
825 return UpdateResponse(
825 return UpdateResponse(
826 executed=True, reason=UpdateFailureReason.NONE,
826 executed=True, reason=UpdateFailureReason.NONE,
827 old=pull_request, new=pull_request_version, changes=changes,
827 old=pull_request, new=pull_request_version, changes=changes,
828 source_changed=source_changed, target_changed=target_changed)
828 source_changed=source_changed, target_changed=target_changed)
829
829
830 def _create_version_from_snapshot(self, pull_request):
830 def _create_version_from_snapshot(self, pull_request):
831 version = PullRequestVersion()
831 version = PullRequestVersion()
832 version.title = pull_request.title
832 version.title = pull_request.title
833 version.description = pull_request.description
833 version.description = pull_request.description
834 version.status = pull_request.status
834 version.status = pull_request.status
835 version.pull_request_state = pull_request.pull_request_state
835 version.pull_request_state = pull_request.pull_request_state
836 version.created_on = datetime.datetime.now()
836 version.created_on = datetime.datetime.now()
837 version.updated_on = pull_request.updated_on
837 version.updated_on = pull_request.updated_on
838 version.user_id = pull_request.user_id
838 version.user_id = pull_request.user_id
839 version.source_repo = pull_request.source_repo
839 version.source_repo = pull_request.source_repo
840 version.source_ref = pull_request.source_ref
840 version.source_ref = pull_request.source_ref
841 version.target_repo = pull_request.target_repo
841 version.target_repo = pull_request.target_repo
842 version.target_ref = pull_request.target_ref
842 version.target_ref = pull_request.target_ref
843
843
844 version._last_merge_source_rev = pull_request._last_merge_source_rev
844 version._last_merge_source_rev = pull_request._last_merge_source_rev
845 version._last_merge_target_rev = pull_request._last_merge_target_rev
845 version._last_merge_target_rev = pull_request._last_merge_target_rev
846 version.last_merge_status = pull_request.last_merge_status
846 version.last_merge_status = pull_request.last_merge_status
847 version.shadow_merge_ref = pull_request.shadow_merge_ref
847 version.shadow_merge_ref = pull_request.shadow_merge_ref
848 version.merge_rev = pull_request.merge_rev
848 version.merge_rev = pull_request.merge_rev
849 version.reviewer_data = pull_request.reviewer_data
849 version.reviewer_data = pull_request.reviewer_data
850
850
851 version.revisions = pull_request.revisions
851 version.revisions = pull_request.revisions
852 version.pull_request = pull_request
852 version.pull_request = pull_request
853 Session().add(version)
853 Session().add(version)
854 Session().flush()
854 Session().flush()
855
855
856 return version
856 return version
857
857
858 def _generate_update_diffs(self, pull_request, pull_request_version):
858 def _generate_update_diffs(self, pull_request, pull_request_version):
859
859
860 diff_context = (
860 diff_context = (
861 self.DIFF_CONTEXT +
861 self.DIFF_CONTEXT +
862 CommentsModel.needed_extra_diff_context())
862 CommentsModel.needed_extra_diff_context())
863 hide_whitespace_changes = False
863 hide_whitespace_changes = False
864 source_repo = pull_request_version.source_repo
864 source_repo = pull_request_version.source_repo
865 source_ref_id = pull_request_version.source_ref_parts.commit_id
865 source_ref_id = pull_request_version.source_ref_parts.commit_id
866 target_ref_id = pull_request_version.target_ref_parts.commit_id
866 target_ref_id = pull_request_version.target_ref_parts.commit_id
867 old_diff = self._get_diff_from_pr_or_version(
867 old_diff = self._get_diff_from_pr_or_version(
868 source_repo, source_ref_id, target_ref_id,
868 source_repo, source_ref_id, target_ref_id,
869 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
869 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
870
870
871 source_repo = pull_request.source_repo
871 source_repo = pull_request.source_repo
872 source_ref_id = pull_request.source_ref_parts.commit_id
872 source_ref_id = pull_request.source_ref_parts.commit_id
873 target_ref_id = pull_request.target_ref_parts.commit_id
873 target_ref_id = pull_request.target_ref_parts.commit_id
874
874
875 new_diff = self._get_diff_from_pr_or_version(
875 new_diff = self._get_diff_from_pr_or_version(
876 source_repo, source_ref_id, target_ref_id,
876 source_repo, source_ref_id, target_ref_id,
877 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
877 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
878
878
879 old_diff_data = diffs.DiffProcessor(old_diff)
879 old_diff_data = diffs.DiffProcessor(old_diff)
880 old_diff_data.prepare()
880 old_diff_data.prepare()
881 new_diff_data = diffs.DiffProcessor(new_diff)
881 new_diff_data = diffs.DiffProcessor(new_diff)
882 new_diff_data.prepare()
882 new_diff_data.prepare()
883
883
884 return old_diff_data, new_diff_data
884 return old_diff_data, new_diff_data
885
885
886 def _link_comments_to_version(self, pull_request_version):
886 def _link_comments_to_version(self, pull_request_version):
887 """
887 """
888 Link all unlinked comments of this pull request to the given version.
888 Link all unlinked comments of this pull request to the given version.
889
889
890 :param pull_request_version: The `PullRequestVersion` to which
890 :param pull_request_version: The `PullRequestVersion` to which
891 the comments shall be linked.
891 the comments shall be linked.
892
892
893 """
893 """
894 pull_request = pull_request_version.pull_request
894 pull_request = pull_request_version.pull_request
895 comments = ChangesetComment.query()\
895 comments = ChangesetComment.query()\
896 .filter(
896 .filter(
897 # TODO: johbo: Should we query for the repo at all here?
897 # TODO: johbo: Should we query for the repo at all here?
898 # Pending decision on how comments of PRs are to be related
898 # Pending decision on how comments of PRs are to be related
899 # to either the source repo, the target repo or no repo at all.
899 # to either the source repo, the target repo or no repo at all.
900 ChangesetComment.repo_id == pull_request.target_repo.repo_id,
900 ChangesetComment.repo_id == pull_request.target_repo.repo_id,
901 ChangesetComment.pull_request == pull_request,
901 ChangesetComment.pull_request == pull_request,
902 ChangesetComment.pull_request_version == None)\
902 ChangesetComment.pull_request_version == None)\
903 .order_by(ChangesetComment.comment_id.asc())
903 .order_by(ChangesetComment.comment_id.asc())
904
904
905 # TODO: johbo: Find out why this breaks if it is done in a bulk
905 # TODO: johbo: Find out why this breaks if it is done in a bulk
906 # operation.
906 # operation.
907 for comment in comments:
907 for comment in comments:
908 comment.pull_request_version_id = (
908 comment.pull_request_version_id = (
909 pull_request_version.pull_request_version_id)
909 pull_request_version.pull_request_version_id)
910 Session().add(comment)
910 Session().add(comment)
911
911
912 def _calculate_commit_id_changes(self, old_ids, new_ids):
912 def _calculate_commit_id_changes(self, old_ids, new_ids):
913 added = [x for x in new_ids if x not in old_ids]
913 added = [x for x in new_ids if x not in old_ids]
914 common = [x for x in new_ids if x in old_ids]
914 common = [x for x in new_ids if x in old_ids]
915 removed = [x for x in old_ids if x not in new_ids]
915 removed = [x for x in old_ids if x not in new_ids]
916 total = new_ids
916 total = new_ids
917 return ChangeTuple(added, common, removed, total)
917 return ChangeTuple(added, common, removed, total)
918
918
919 def _calculate_file_changes(self, old_diff_data, new_diff_data):
919 def _calculate_file_changes(self, old_diff_data, new_diff_data):
920
920
921 old_files = OrderedDict()
921 old_files = OrderedDict()
922 for diff_data in old_diff_data.parsed_diff:
922 for diff_data in old_diff_data.parsed_diff:
923 old_files[diff_data['filename']] = md5_safe(diff_data['raw_diff'])
923 old_files[diff_data['filename']] = md5_safe(diff_data['raw_diff'])
924
924
925 added_files = []
925 added_files = []
926 modified_files = []
926 modified_files = []
927 removed_files = []
927 removed_files = []
928 for diff_data in new_diff_data.parsed_diff:
928 for diff_data in new_diff_data.parsed_diff:
929 new_filename = diff_data['filename']
929 new_filename = diff_data['filename']
930 new_hash = md5_safe(diff_data['raw_diff'])
930 new_hash = md5_safe(diff_data['raw_diff'])
931
931
932 old_hash = old_files.get(new_filename)
932 old_hash = old_files.get(new_filename)
933 if not old_hash:
933 if not old_hash:
934 # file is not present in old diff, means it's added
934 # file is not present in old diff, means it's added
935 added_files.append(new_filename)
935 added_files.append(new_filename)
936 else:
936 else:
937 if new_hash != old_hash:
937 if new_hash != old_hash:
938 modified_files.append(new_filename)
938 modified_files.append(new_filename)
939 # now remove a file from old, since we have seen it already
939 # now remove a file from old, since we have seen it already
940 del old_files[new_filename]
940 del old_files[new_filename]
941
941
942 # removed files is when there are present in old, but not in NEW,
942 # removed files is when there are present in old, but not in NEW,
943 # since we remove old files that are present in new diff, left-overs
943 # since we remove old files that are present in new diff, left-overs
944 # if any should be the removed files
944 # if any should be the removed files
945 removed_files.extend(old_files.keys())
945 removed_files.extend(old_files.keys())
946
946
947 return FileChangeTuple(added_files, modified_files, removed_files)
947 return FileChangeTuple(added_files, modified_files, removed_files)
948
948
949 def _render_update_message(self, changes, file_changes):
949 def _render_update_message(self, changes, file_changes):
950 """
950 """
951 render the message using DEFAULT_COMMENTS_RENDERER (RST renderer),
951 render the message using DEFAULT_COMMENTS_RENDERER (RST renderer),
952 so it's always looking the same disregarding on which default
952 so it's always looking the same disregarding on which default
953 renderer system is using.
953 renderer system is using.
954
954
955 :param changes: changes named tuple
955 :param changes: changes named tuple
956 :param file_changes: file changes named tuple
956 :param file_changes: file changes named tuple
957
957
958 """
958 """
959 new_status = ChangesetStatus.get_status_lbl(
959 new_status = ChangesetStatus.get_status_lbl(
960 ChangesetStatus.STATUS_UNDER_REVIEW)
960 ChangesetStatus.STATUS_UNDER_REVIEW)
961
961
962 changed_files = (
962 changed_files = (
963 file_changes.added + file_changes.modified + file_changes.removed)
963 file_changes.added + file_changes.modified + file_changes.removed)
964
964
965 params = {
965 params = {
966 'under_review_label': new_status,
966 'under_review_label': new_status,
967 'added_commits': changes.added,
967 'added_commits': changes.added,
968 'removed_commits': changes.removed,
968 'removed_commits': changes.removed,
969 'changed_files': changed_files,
969 'changed_files': changed_files,
970 'added_files': file_changes.added,
970 'added_files': file_changes.added,
971 'modified_files': file_changes.modified,
971 'modified_files': file_changes.modified,
972 'removed_files': file_changes.removed,
972 'removed_files': file_changes.removed,
973 }
973 }
974 renderer = RstTemplateRenderer()
974 renderer = RstTemplateRenderer()
975 return renderer.render('pull_request_update.mako', **params)
975 return renderer.render('pull_request_update.mako', **params)
976
976
977 def edit(self, pull_request, title, description, description_renderer, user):
977 def edit(self, pull_request, title, description, description_renderer, user):
978 pull_request = self.__get_pull_request(pull_request)
978 pull_request = self.__get_pull_request(pull_request)
979 old_data = pull_request.get_api_data(with_merge_state=False)
979 old_data = pull_request.get_api_data(with_merge_state=False)
980 if pull_request.is_closed():
980 if pull_request.is_closed():
981 raise ValueError('This pull request is closed')
981 raise ValueError('This pull request is closed')
982 if title:
982 if title:
983 pull_request.title = title
983 pull_request.title = title
984 pull_request.description = description
984 pull_request.description = description
985 pull_request.updated_on = datetime.datetime.now()
985 pull_request.updated_on = datetime.datetime.now()
986 pull_request.description_renderer = description_renderer
986 pull_request.description_renderer = description_renderer
987 Session().add(pull_request)
987 Session().add(pull_request)
988 self._log_audit_action(
988 self._log_audit_action(
989 'repo.pull_request.edit', {'old_data': old_data},
989 'repo.pull_request.edit', {'old_data': old_data},
990 user, pull_request)
990 user, pull_request)
991
991
992 def update_reviewers(self, pull_request, reviewer_data, user):
992 def update_reviewers(self, pull_request, reviewer_data, user):
993 """
993 """
994 Update the reviewers in the pull request
994 Update the reviewers in the pull request
995
995
996 :param pull_request: the pr to update
996 :param pull_request: the pr to update
997 :param reviewer_data: list of tuples
997 :param reviewer_data: list of tuples
998 [(user, ['reason1', 'reason2'], mandatory_flag, [rules])]
998 [(user, ['reason1', 'reason2'], mandatory_flag, [rules])]
999 """
999 """
1000 pull_request = self.__get_pull_request(pull_request)
1000 pull_request = self.__get_pull_request(pull_request)
1001 if pull_request.is_closed():
1001 if pull_request.is_closed():
1002 raise ValueError('This pull request is closed')
1002 raise ValueError('This pull request is closed')
1003
1003
1004 reviewers = {}
1004 reviewers = {}
1005 for user_id, reasons, mandatory, rules in reviewer_data:
1005 for user_id, reasons, mandatory, rules in reviewer_data:
1006 if isinstance(user_id, (int, compat.string_types)):
1006 if isinstance(user_id, (int, compat.string_types)):
1007 user_id = self._get_user(user_id).user_id
1007 user_id = self._get_user(user_id).user_id
1008 reviewers[user_id] = {
1008 reviewers[user_id] = {
1009 'reasons': reasons, 'mandatory': mandatory}
1009 'reasons': reasons, 'mandatory': mandatory}
1010
1010
1011 reviewers_ids = set(reviewers.keys())
1011 reviewers_ids = set(reviewers.keys())
1012 current_reviewers = PullRequestReviewers.query()\
1012 current_reviewers = PullRequestReviewers.query()\
1013 .filter(PullRequestReviewers.pull_request ==
1013 .filter(PullRequestReviewers.pull_request ==
1014 pull_request).all()
1014 pull_request).all()
1015 current_reviewers_ids = set([x.user.user_id for x in current_reviewers])
1015 current_reviewers_ids = set([x.user.user_id for x in current_reviewers])
1016
1016
1017 ids_to_add = reviewers_ids.difference(current_reviewers_ids)
1017 ids_to_add = reviewers_ids.difference(current_reviewers_ids)
1018 ids_to_remove = current_reviewers_ids.difference(reviewers_ids)
1018 ids_to_remove = current_reviewers_ids.difference(reviewers_ids)
1019
1019
1020 log.debug("Adding %s reviewers", ids_to_add)
1020 log.debug("Adding %s reviewers", ids_to_add)
1021 log.debug("Removing %s reviewers", ids_to_remove)
1021 log.debug("Removing %s reviewers", ids_to_remove)
1022 changed = False
1022 changed = False
1023 added_audit_reviewers = []
1024 removed_audit_reviewers = []
1025
1023 for uid in ids_to_add:
1026 for uid in ids_to_add:
1024 changed = True
1027 changed = True
1025 _usr = self._get_user(uid)
1028 _usr = self._get_user(uid)
1026 reviewer = PullRequestReviewers()
1029 reviewer = PullRequestReviewers()
1027 reviewer.user = _usr
1030 reviewer.user = _usr
1028 reviewer.pull_request = pull_request
1031 reviewer.pull_request = pull_request
1029 reviewer.reasons = reviewers[uid]['reasons']
1032 reviewer.reasons = reviewers[uid]['reasons']
1030 # NOTE(marcink): mandatory shouldn't be changed now
1033 # NOTE(marcink): mandatory shouldn't be changed now
1031 # reviewer.mandatory = reviewers[uid]['reasons']
1034 # reviewer.mandatory = reviewers[uid]['reasons']
1032 Session().add(reviewer)
1035 Session().add(reviewer)
1033 self._log_audit_action(
1036 added_audit_reviewers.append(reviewer.get_dict())
1034 'repo.pull_request.reviewer.add', {'data': reviewer.get_dict()},
1035 user, pull_request)
1036
1037
1037 for uid in ids_to_remove:
1038 for uid in ids_to_remove:
1038 changed = True
1039 changed = True
1040 # NOTE(marcink): we fetch "ALL" reviewers using .all(). This is an edge case
1041 # that prevents and fixes cases that we added the same reviewer twice.
1042 # this CAN happen due to the lack of DB checks
1039 reviewers = PullRequestReviewers.query()\
1043 reviewers = PullRequestReviewers.query()\
1040 .filter(PullRequestReviewers.user_id == uid,
1044 .filter(PullRequestReviewers.user_id == uid,
1041 PullRequestReviewers.pull_request == pull_request)\
1045 PullRequestReviewers.pull_request == pull_request)\
1042 .all()
1046 .all()
1043 # use .all() in case we accidentally added the same person twice
1047
1044 # this CAN happen due to the lack of DB checks
1045 for obj in reviewers:
1048 for obj in reviewers:
1046 old_data = obj.get_dict()
1049 added_audit_reviewers.append(obj.get_dict())
1047 Session().delete(obj)
1050 Session().delete(obj)
1048 self._log_audit_action(
1049 'repo.pull_request.reviewer.delete',
1050 {'old_data': old_data}, user, pull_request)
1051
1051
1052 if changed:
1052 if changed:
1053 Session().expire_all()
1053 pull_request.updated_on = datetime.datetime.now()
1054 pull_request.updated_on = datetime.datetime.now()
1054 Session().add(pull_request)
1055 Session().add(pull_request)
1055
1056
1057 # finally store audit logs
1058 for user_data in added_audit_reviewers:
1059 self._log_audit_action(
1060 'repo.pull_request.reviewer.add', {'data': user_data},
1061 user, pull_request)
1062 for user_data in removed_audit_reviewers:
1063 self._log_audit_action(
1064 'repo.pull_request.reviewer.delete', {'old_data': user_data},
1065 user, pull_request)
1066
1056 self.notify_reviewers(pull_request, ids_to_add)
1067 self.notify_reviewers(pull_request, ids_to_add)
1057 return ids_to_add, ids_to_remove
1068 return ids_to_add, ids_to_remove
1058
1069
1059 def get_url(self, pull_request, request=None, permalink=False):
1070 def get_url(self, pull_request, request=None, permalink=False):
1060 if not request:
1071 if not request:
1061 request = get_current_request()
1072 request = get_current_request()
1062
1073
1063 if permalink:
1074 if permalink:
1064 return request.route_url(
1075 return request.route_url(
1065 'pull_requests_global',
1076 'pull_requests_global',
1066 pull_request_id=pull_request.pull_request_id,)
1077 pull_request_id=pull_request.pull_request_id,)
1067 else:
1078 else:
1068 return request.route_url('pullrequest_show',
1079 return request.route_url('pullrequest_show',
1069 repo_name=safe_str(pull_request.target_repo.repo_name),
1080 repo_name=safe_str(pull_request.target_repo.repo_name),
1070 pull_request_id=pull_request.pull_request_id,)
1081 pull_request_id=pull_request.pull_request_id,)
1071
1082
1072 def get_shadow_clone_url(self, pull_request, request=None):
1083 def get_shadow_clone_url(self, pull_request, request=None):
1073 """
1084 """
1074 Returns qualified url pointing to the shadow repository. If this pull
1085 Returns qualified url pointing to the shadow repository. If this pull
1075 request is closed there is no shadow repository and ``None`` will be
1086 request is closed there is no shadow repository and ``None`` will be
1076 returned.
1087 returned.
1077 """
1088 """
1078 if pull_request.is_closed():
1089 if pull_request.is_closed():
1079 return None
1090 return None
1080 else:
1091 else:
1081 pr_url = urllib.unquote(self.get_url(pull_request, request=request))
1092 pr_url = urllib.unquote(self.get_url(pull_request, request=request))
1082 return safe_unicode('{pr_url}/repository'.format(pr_url=pr_url))
1093 return safe_unicode('{pr_url}/repository'.format(pr_url=pr_url))
1083
1094
1084 def notify_reviewers(self, pull_request, reviewers_ids):
1095 def notify_reviewers(self, pull_request, reviewers_ids):
1085 # notification to reviewers
1096 # notification to reviewers
1086 if not reviewers_ids:
1097 if not reviewers_ids:
1087 return
1098 return
1088
1099
1089 pull_request_obj = pull_request
1100 pull_request_obj = pull_request
1090 # get the current participants of this pull request
1101 # get the current participants of this pull request
1091 recipients = reviewers_ids
1102 recipients = reviewers_ids
1092 notification_type = EmailNotificationModel.TYPE_PULL_REQUEST
1103 notification_type = EmailNotificationModel.TYPE_PULL_REQUEST
1093
1104
1094 pr_source_repo = pull_request_obj.source_repo
1105 pr_source_repo = pull_request_obj.source_repo
1095 pr_target_repo = pull_request_obj.target_repo
1106 pr_target_repo = pull_request_obj.target_repo
1096
1107
1097 pr_url = h.route_url('pullrequest_show',
1108 pr_url = h.route_url('pullrequest_show',
1098 repo_name=pr_target_repo.repo_name,
1109 repo_name=pr_target_repo.repo_name,
1099 pull_request_id=pull_request_obj.pull_request_id,)
1110 pull_request_id=pull_request_obj.pull_request_id,)
1100
1111
1101 # set some variables for email notification
1112 # set some variables for email notification
1102 pr_target_repo_url = h.route_url(
1113 pr_target_repo_url = h.route_url(
1103 'repo_summary', repo_name=pr_target_repo.repo_name)
1114 'repo_summary', repo_name=pr_target_repo.repo_name)
1104
1115
1105 pr_source_repo_url = h.route_url(
1116 pr_source_repo_url = h.route_url(
1106 'repo_summary', repo_name=pr_source_repo.repo_name)
1117 'repo_summary', repo_name=pr_source_repo.repo_name)
1107
1118
1108 # pull request specifics
1119 # pull request specifics
1109 pull_request_commits = [
1120 pull_request_commits = [
1110 (x.raw_id, x.message)
1121 (x.raw_id, x.message)
1111 for x in map(pr_source_repo.get_commit, pull_request.revisions)]
1122 for x in map(pr_source_repo.get_commit, pull_request.revisions)]
1112
1123
1113 kwargs = {
1124 kwargs = {
1114 'user': pull_request.author,
1125 'user': pull_request.author,
1115 'pull_request': pull_request_obj,
1126 'pull_request': pull_request_obj,
1116 'pull_request_commits': pull_request_commits,
1127 'pull_request_commits': pull_request_commits,
1117
1128
1118 'pull_request_target_repo': pr_target_repo,
1129 'pull_request_target_repo': pr_target_repo,
1119 'pull_request_target_repo_url': pr_target_repo_url,
1130 'pull_request_target_repo_url': pr_target_repo_url,
1120
1131
1121 'pull_request_source_repo': pr_source_repo,
1132 'pull_request_source_repo': pr_source_repo,
1122 'pull_request_source_repo_url': pr_source_repo_url,
1133 'pull_request_source_repo_url': pr_source_repo_url,
1123
1134
1124 'pull_request_url': pr_url,
1135 'pull_request_url': pr_url,
1125 }
1136 }
1126
1137
1127 # pre-generate the subject for notification itself
1138 # pre-generate the subject for notification itself
1128 (subject,
1139 (subject,
1129 _h, _e, # we don't care about those
1140 _h, _e, # we don't care about those
1130 body_plaintext) = EmailNotificationModel().render_email(
1141 body_plaintext) = EmailNotificationModel().render_email(
1131 notification_type, **kwargs)
1142 notification_type, **kwargs)
1132
1143
1133 # create notification objects, and emails
1144 # create notification objects, and emails
1134 NotificationModel().create(
1145 NotificationModel().create(
1135 created_by=pull_request.author,
1146 created_by=pull_request.author,
1136 notification_subject=subject,
1147 notification_subject=subject,
1137 notification_body=body_plaintext,
1148 notification_body=body_plaintext,
1138 notification_type=notification_type,
1149 notification_type=notification_type,
1139 recipients=recipients,
1150 recipients=recipients,
1140 email_kwargs=kwargs,
1151 email_kwargs=kwargs,
1141 )
1152 )
1142
1153
1143 def delete(self, pull_request, user):
1154 def delete(self, pull_request, user):
1144 pull_request = self.__get_pull_request(pull_request)
1155 pull_request = self.__get_pull_request(pull_request)
1145 old_data = pull_request.get_api_data(with_merge_state=False)
1156 old_data = pull_request.get_api_data(with_merge_state=False)
1146 self._cleanup_merge_workspace(pull_request)
1157 self._cleanup_merge_workspace(pull_request)
1147 self._log_audit_action(
1158 self._log_audit_action(
1148 'repo.pull_request.delete', {'old_data': old_data},
1159 'repo.pull_request.delete', {'old_data': old_data},
1149 user, pull_request)
1160 user, pull_request)
1150 Session().delete(pull_request)
1161 Session().delete(pull_request)
1151
1162
1152 def close_pull_request(self, pull_request, user):
1163 def close_pull_request(self, pull_request, user):
1153 pull_request = self.__get_pull_request(pull_request)
1164 pull_request = self.__get_pull_request(pull_request)
1154 self._cleanup_merge_workspace(pull_request)
1165 self._cleanup_merge_workspace(pull_request)
1155 pull_request.status = PullRequest.STATUS_CLOSED
1166 pull_request.status = PullRequest.STATUS_CLOSED
1156 pull_request.updated_on = datetime.datetime.now()
1167 pull_request.updated_on = datetime.datetime.now()
1157 Session().add(pull_request)
1168 Session().add(pull_request)
1158 self.trigger_pull_request_hook(
1169 self.trigger_pull_request_hook(
1159 pull_request, pull_request.author, 'close')
1170 pull_request, pull_request.author, 'close')
1160
1171
1161 pr_data = pull_request.get_api_data(with_merge_state=False)
1172 pr_data = pull_request.get_api_data(with_merge_state=False)
1162 self._log_audit_action(
1173 self._log_audit_action(
1163 'repo.pull_request.close', {'data': pr_data}, user, pull_request)
1174 'repo.pull_request.close', {'data': pr_data}, user, pull_request)
1164
1175
1165 def close_pull_request_with_comment(
1176 def close_pull_request_with_comment(
1166 self, pull_request, user, repo, message=None, auth_user=None):
1177 self, pull_request, user, repo, message=None, auth_user=None):
1167
1178
1168 pull_request_review_status = pull_request.calculated_review_status()
1179 pull_request_review_status = pull_request.calculated_review_status()
1169
1180
1170 if pull_request_review_status == ChangesetStatus.STATUS_APPROVED:
1181 if pull_request_review_status == ChangesetStatus.STATUS_APPROVED:
1171 # approved only if we have voting consent
1182 # approved only if we have voting consent
1172 status = ChangesetStatus.STATUS_APPROVED
1183 status = ChangesetStatus.STATUS_APPROVED
1173 else:
1184 else:
1174 status = ChangesetStatus.STATUS_REJECTED
1185 status = ChangesetStatus.STATUS_REJECTED
1175 status_lbl = ChangesetStatus.get_status_lbl(status)
1186 status_lbl = ChangesetStatus.get_status_lbl(status)
1176
1187
1177 default_message = (
1188 default_message = (
1178 'Closing with status change {transition_icon} {status}.'
1189 'Closing with status change {transition_icon} {status}.'
1179 ).format(transition_icon='>', status=status_lbl)
1190 ).format(transition_icon='>', status=status_lbl)
1180 text = message or default_message
1191 text = message or default_message
1181
1192
1182 # create a comment, and link it to new status
1193 # create a comment, and link it to new status
1183 comment = CommentsModel().create(
1194 comment = CommentsModel().create(
1184 text=text,
1195 text=text,
1185 repo=repo.repo_id,
1196 repo=repo.repo_id,
1186 user=user.user_id,
1197 user=user.user_id,
1187 pull_request=pull_request.pull_request_id,
1198 pull_request=pull_request.pull_request_id,
1188 status_change=status_lbl,
1199 status_change=status_lbl,
1189 status_change_type=status,
1200 status_change_type=status,
1190 closing_pr=True,
1201 closing_pr=True,
1191 auth_user=auth_user,
1202 auth_user=auth_user,
1192 )
1203 )
1193
1204
1194 # calculate old status before we change it
1205 # calculate old status before we change it
1195 old_calculated_status = pull_request.calculated_review_status()
1206 old_calculated_status = pull_request.calculated_review_status()
1196 ChangesetStatusModel().set_status(
1207 ChangesetStatusModel().set_status(
1197 repo.repo_id,
1208 repo.repo_id,
1198 status,
1209 status,
1199 user.user_id,
1210 user.user_id,
1200 comment=comment,
1211 comment=comment,
1201 pull_request=pull_request.pull_request_id
1212 pull_request=pull_request.pull_request_id
1202 )
1213 )
1203
1214
1204 Session().flush()
1215 Session().flush()
1205 events.trigger(events.PullRequestCommentEvent(pull_request, comment))
1216 events.trigger(events.PullRequestCommentEvent(pull_request, comment))
1206 # we now calculate the status of pull request again, and based on that
1217 # we now calculate the status of pull request again, and based on that
1207 # calculation trigger status change. This might happen in cases
1218 # calculation trigger status change. This might happen in cases
1208 # that non-reviewer admin closes a pr, which means his vote doesn't
1219 # that non-reviewer admin closes a pr, which means his vote doesn't
1209 # change the status, while if he's a reviewer this might change it.
1220 # change the status, while if he's a reviewer this might change it.
1210 calculated_status = pull_request.calculated_review_status()
1221 calculated_status = pull_request.calculated_review_status()
1211 if old_calculated_status != calculated_status:
1222 if old_calculated_status != calculated_status:
1212 self.trigger_pull_request_hook(
1223 self.trigger_pull_request_hook(
1213 pull_request, user, 'review_status_change',
1224 pull_request, user, 'review_status_change',
1214 data={'status': calculated_status})
1225 data={'status': calculated_status})
1215
1226
1216 # finally close the PR
1227 # finally close the PR
1217 PullRequestModel().close_pull_request(
1228 PullRequestModel().close_pull_request(
1218 pull_request.pull_request_id, user)
1229 pull_request.pull_request_id, user)
1219
1230
1220 return comment, status
1231 return comment, status
1221
1232
1222 def merge_status(self, pull_request, translator=None,
1233 def merge_status(self, pull_request, translator=None,
1223 force_shadow_repo_refresh=False):
1234 force_shadow_repo_refresh=False):
1224 _ = translator or get_current_request().translate
1235 _ = translator or get_current_request().translate
1225
1236
1226 if not self._is_merge_enabled(pull_request):
1237 if not self._is_merge_enabled(pull_request):
1227 return False, _('Server-side pull request merging is disabled.')
1238 return False, _('Server-side pull request merging is disabled.')
1228 if pull_request.is_closed():
1239 if pull_request.is_closed():
1229 return False, _('This pull request is closed.')
1240 return False, _('This pull request is closed.')
1230 merge_possible, msg = self._check_repo_requirements(
1241 merge_possible, msg = self._check_repo_requirements(
1231 target=pull_request.target_repo, source=pull_request.source_repo,
1242 target=pull_request.target_repo, source=pull_request.source_repo,
1232 translator=_)
1243 translator=_)
1233 if not merge_possible:
1244 if not merge_possible:
1234 return merge_possible, msg
1245 return merge_possible, msg
1235
1246
1236 try:
1247 try:
1237 resp = self._try_merge(
1248 resp = self._try_merge(
1238 pull_request,
1249 pull_request,
1239 force_shadow_repo_refresh=force_shadow_repo_refresh)
1250 force_shadow_repo_refresh=force_shadow_repo_refresh)
1240 log.debug("Merge response: %s", resp)
1251 log.debug("Merge response: %s", resp)
1241 status = resp.possible, resp.merge_status_message
1252 status = resp.possible, resp.merge_status_message
1242 except NotImplementedError:
1253 except NotImplementedError:
1243 status = False, _('Pull request merging is not supported.')
1254 status = False, _('Pull request merging is not supported.')
1244
1255
1245 return status
1256 return status
1246
1257
1247 def _check_repo_requirements(self, target, source, translator):
1258 def _check_repo_requirements(self, target, source, translator):
1248 """
1259 """
1249 Check if `target` and `source` have compatible requirements.
1260 Check if `target` and `source` have compatible requirements.
1250
1261
1251 Currently this is just checking for largefiles.
1262 Currently this is just checking for largefiles.
1252 """
1263 """
1253 _ = translator
1264 _ = translator
1254 target_has_largefiles = self._has_largefiles(target)
1265 target_has_largefiles = self._has_largefiles(target)
1255 source_has_largefiles = self._has_largefiles(source)
1266 source_has_largefiles = self._has_largefiles(source)
1256 merge_possible = True
1267 merge_possible = True
1257 message = u''
1268 message = u''
1258
1269
1259 if target_has_largefiles != source_has_largefiles:
1270 if target_has_largefiles != source_has_largefiles:
1260 merge_possible = False
1271 merge_possible = False
1261 if source_has_largefiles:
1272 if source_has_largefiles:
1262 message = _(
1273 message = _(
1263 'Target repository large files support is disabled.')
1274 'Target repository large files support is disabled.')
1264 else:
1275 else:
1265 message = _(
1276 message = _(
1266 'Source repository large files support is disabled.')
1277 'Source repository large files support is disabled.')
1267
1278
1268 return merge_possible, message
1279 return merge_possible, message
1269
1280
1270 def _has_largefiles(self, repo):
1281 def _has_largefiles(self, repo):
1271 largefiles_ui = VcsSettingsModel(repo=repo).get_ui_settings(
1282 largefiles_ui = VcsSettingsModel(repo=repo).get_ui_settings(
1272 'extensions', 'largefiles')
1283 'extensions', 'largefiles')
1273 return largefiles_ui and largefiles_ui[0].active
1284 return largefiles_ui and largefiles_ui[0].active
1274
1285
1275 def _try_merge(self, pull_request, force_shadow_repo_refresh=False):
1286 def _try_merge(self, pull_request, force_shadow_repo_refresh=False):
1276 """
1287 """
1277 Try to merge the pull request and return the merge status.
1288 Try to merge the pull request and return the merge status.
1278 """
1289 """
1279 log.debug(
1290 log.debug(
1280 "Trying out if the pull request %s can be merged. Force_refresh=%s",
1291 "Trying out if the pull request %s can be merged. Force_refresh=%s",
1281 pull_request.pull_request_id, force_shadow_repo_refresh)
1292 pull_request.pull_request_id, force_shadow_repo_refresh)
1282 target_vcs = pull_request.target_repo.scm_instance()
1293 target_vcs = pull_request.target_repo.scm_instance()
1283 # Refresh the target reference.
1294 # Refresh the target reference.
1284 try:
1295 try:
1285 target_ref = self._refresh_reference(
1296 target_ref = self._refresh_reference(
1286 pull_request.target_ref_parts, target_vcs)
1297 pull_request.target_ref_parts, target_vcs)
1287 except CommitDoesNotExistError:
1298 except CommitDoesNotExistError:
1288 merge_state = MergeResponse(
1299 merge_state = MergeResponse(
1289 False, False, None, MergeFailureReason.MISSING_TARGET_REF,
1300 False, False, None, MergeFailureReason.MISSING_TARGET_REF,
1290 metadata={'target_ref': pull_request.target_ref_parts})
1301 metadata={'target_ref': pull_request.target_ref_parts})
1291 return merge_state
1302 return merge_state
1292
1303
1293 target_locked = pull_request.target_repo.locked
1304 target_locked = pull_request.target_repo.locked
1294 if target_locked and target_locked[0]:
1305 if target_locked and target_locked[0]:
1295 locked_by = 'user:{}'.format(target_locked[0])
1306 locked_by = 'user:{}'.format(target_locked[0])
1296 log.debug("The target repository is locked by %s.", locked_by)
1307 log.debug("The target repository is locked by %s.", locked_by)
1297 merge_state = MergeResponse(
1308 merge_state = MergeResponse(
1298 False, False, None, MergeFailureReason.TARGET_IS_LOCKED,
1309 False, False, None, MergeFailureReason.TARGET_IS_LOCKED,
1299 metadata={'locked_by': locked_by})
1310 metadata={'locked_by': locked_by})
1300 elif force_shadow_repo_refresh or self._needs_merge_state_refresh(
1311 elif force_shadow_repo_refresh or self._needs_merge_state_refresh(
1301 pull_request, target_ref):
1312 pull_request, target_ref):
1302 log.debug("Refreshing the merge status of the repository.")
1313 log.debug("Refreshing the merge status of the repository.")
1303 merge_state = self._refresh_merge_state(
1314 merge_state = self._refresh_merge_state(
1304 pull_request, target_vcs, target_ref)
1315 pull_request, target_vcs, target_ref)
1305 else:
1316 else:
1306 possible = pull_request.\
1317 possible = pull_request.\
1307 last_merge_status == MergeFailureReason.NONE
1318 last_merge_status == MergeFailureReason.NONE
1308 merge_state = MergeResponse(
1319 merge_state = MergeResponse(
1309 possible, False, None, pull_request.last_merge_status)
1320 possible, False, None, pull_request.last_merge_status)
1310
1321
1311 return merge_state
1322 return merge_state
1312
1323
1313 def _refresh_reference(self, reference, vcs_repository):
1324 def _refresh_reference(self, reference, vcs_repository):
1314 if reference.type in self.UPDATABLE_REF_TYPES:
1325 if reference.type in self.UPDATABLE_REF_TYPES:
1315 name_or_id = reference.name
1326 name_or_id = reference.name
1316 else:
1327 else:
1317 name_or_id = reference.commit_id
1328 name_or_id = reference.commit_id
1318 refreshed_commit = vcs_repository.get_commit(name_or_id)
1329 refreshed_commit = vcs_repository.get_commit(name_or_id)
1319 refreshed_reference = Reference(
1330 refreshed_reference = Reference(
1320 reference.type, reference.name, refreshed_commit.raw_id)
1331 reference.type, reference.name, refreshed_commit.raw_id)
1321 return refreshed_reference
1332 return refreshed_reference
1322
1333
1323 def _needs_merge_state_refresh(self, pull_request, target_reference):
1334 def _needs_merge_state_refresh(self, pull_request, target_reference):
1324 return not(
1335 return not(
1325 pull_request.revisions and
1336 pull_request.revisions and
1326 pull_request.revisions[0] == pull_request._last_merge_source_rev and
1337 pull_request.revisions[0] == pull_request._last_merge_source_rev and
1327 target_reference.commit_id == pull_request._last_merge_target_rev)
1338 target_reference.commit_id == pull_request._last_merge_target_rev)
1328
1339
1329 def _refresh_merge_state(self, pull_request, target_vcs, target_reference):
1340 def _refresh_merge_state(self, pull_request, target_vcs, target_reference):
1330 workspace_id = self._workspace_id(pull_request)
1341 workspace_id = self._workspace_id(pull_request)
1331 source_vcs = pull_request.source_repo.scm_instance()
1342 source_vcs = pull_request.source_repo.scm_instance()
1332 repo_id = pull_request.target_repo.repo_id
1343 repo_id = pull_request.target_repo.repo_id
1333 use_rebase = self._use_rebase_for_merging(pull_request)
1344 use_rebase = self._use_rebase_for_merging(pull_request)
1334 close_branch = self._close_branch_before_merging(pull_request)
1345 close_branch = self._close_branch_before_merging(pull_request)
1335 merge_state = target_vcs.merge(
1346 merge_state = target_vcs.merge(
1336 repo_id, workspace_id,
1347 repo_id, workspace_id,
1337 target_reference, source_vcs, pull_request.source_ref_parts,
1348 target_reference, source_vcs, pull_request.source_ref_parts,
1338 dry_run=True, use_rebase=use_rebase,
1349 dry_run=True, use_rebase=use_rebase,
1339 close_branch=close_branch)
1350 close_branch=close_branch)
1340
1351
1341 # Do not store the response if there was an unknown error.
1352 # Do not store the response if there was an unknown error.
1342 if merge_state.failure_reason != MergeFailureReason.UNKNOWN:
1353 if merge_state.failure_reason != MergeFailureReason.UNKNOWN:
1343 pull_request._last_merge_source_rev = \
1354 pull_request._last_merge_source_rev = \
1344 pull_request.source_ref_parts.commit_id
1355 pull_request.source_ref_parts.commit_id
1345 pull_request._last_merge_target_rev = target_reference.commit_id
1356 pull_request._last_merge_target_rev = target_reference.commit_id
1346 pull_request.last_merge_status = merge_state.failure_reason
1357 pull_request.last_merge_status = merge_state.failure_reason
1347 pull_request.shadow_merge_ref = merge_state.merge_ref
1358 pull_request.shadow_merge_ref = merge_state.merge_ref
1348 Session().add(pull_request)
1359 Session().add(pull_request)
1349 Session().commit()
1360 Session().commit()
1350
1361
1351 return merge_state
1362 return merge_state
1352
1363
1353 def _workspace_id(self, pull_request):
1364 def _workspace_id(self, pull_request):
1354 workspace_id = 'pr-%s' % pull_request.pull_request_id
1365 workspace_id = 'pr-%s' % pull_request.pull_request_id
1355 return workspace_id
1366 return workspace_id
1356
1367
1357 def generate_repo_data(self, repo, commit_id=None, branch=None,
1368 def generate_repo_data(self, repo, commit_id=None, branch=None,
1358 bookmark=None, translator=None):
1369 bookmark=None, translator=None):
1359 from rhodecode.model.repo import RepoModel
1370 from rhodecode.model.repo import RepoModel
1360
1371
1361 all_refs, selected_ref = \
1372 all_refs, selected_ref = \
1362 self._get_repo_pullrequest_sources(
1373 self._get_repo_pullrequest_sources(
1363 repo.scm_instance(), commit_id=commit_id,
1374 repo.scm_instance(), commit_id=commit_id,
1364 branch=branch, bookmark=bookmark, translator=translator)
1375 branch=branch, bookmark=bookmark, translator=translator)
1365
1376
1366 refs_select2 = []
1377 refs_select2 = []
1367 for element in all_refs:
1378 for element in all_refs:
1368 children = [{'id': x[0], 'text': x[1]} for x in element[0]]
1379 children = [{'id': x[0], 'text': x[1]} for x in element[0]]
1369 refs_select2.append({'text': element[1], 'children': children})
1380 refs_select2.append({'text': element[1], 'children': children})
1370
1381
1371 return {
1382 return {
1372 'user': {
1383 'user': {
1373 'user_id': repo.user.user_id,
1384 'user_id': repo.user.user_id,
1374 'username': repo.user.username,
1385 'username': repo.user.username,
1375 'firstname': repo.user.first_name,
1386 'firstname': repo.user.first_name,
1376 'lastname': repo.user.last_name,
1387 'lastname': repo.user.last_name,
1377 'gravatar_link': h.gravatar_url(repo.user.email, 14),
1388 'gravatar_link': h.gravatar_url(repo.user.email, 14),
1378 },
1389 },
1379 'name': repo.repo_name,
1390 'name': repo.repo_name,
1380 'link': RepoModel().get_url(repo),
1391 'link': RepoModel().get_url(repo),
1381 'description': h.chop_at_smart(repo.description_safe, '\n'),
1392 'description': h.chop_at_smart(repo.description_safe, '\n'),
1382 'refs': {
1393 'refs': {
1383 'all_refs': all_refs,
1394 'all_refs': all_refs,
1384 'selected_ref': selected_ref,
1395 'selected_ref': selected_ref,
1385 'select2_refs': refs_select2
1396 'select2_refs': refs_select2
1386 }
1397 }
1387 }
1398 }
1388
1399
1389 def generate_pullrequest_title(self, source, source_ref, target):
1400 def generate_pullrequest_title(self, source, source_ref, target):
1390 return u'{source}#{at_ref} to {target}'.format(
1401 return u'{source}#{at_ref} to {target}'.format(
1391 source=source,
1402 source=source,
1392 at_ref=source_ref,
1403 at_ref=source_ref,
1393 target=target,
1404 target=target,
1394 )
1405 )
1395
1406
1396 def _cleanup_merge_workspace(self, pull_request):
1407 def _cleanup_merge_workspace(self, pull_request):
1397 # Merging related cleanup
1408 # Merging related cleanup
1398 repo_id = pull_request.target_repo.repo_id
1409 repo_id = pull_request.target_repo.repo_id
1399 target_scm = pull_request.target_repo.scm_instance()
1410 target_scm = pull_request.target_repo.scm_instance()
1400 workspace_id = self._workspace_id(pull_request)
1411 workspace_id = self._workspace_id(pull_request)
1401
1412
1402 try:
1413 try:
1403 target_scm.cleanup_merge_workspace(repo_id, workspace_id)
1414 target_scm.cleanup_merge_workspace(repo_id, workspace_id)
1404 except NotImplementedError:
1415 except NotImplementedError:
1405 pass
1416 pass
1406
1417
1407 def _get_repo_pullrequest_sources(
1418 def _get_repo_pullrequest_sources(
1408 self, repo, commit_id=None, branch=None, bookmark=None,
1419 self, repo, commit_id=None, branch=None, bookmark=None,
1409 translator=None):
1420 translator=None):
1410 """
1421 """
1411 Return a structure with repo's interesting commits, suitable for
1422 Return a structure with repo's interesting commits, suitable for
1412 the selectors in pullrequest controller
1423 the selectors in pullrequest controller
1413
1424
1414 :param commit_id: a commit that must be in the list somehow
1425 :param commit_id: a commit that must be in the list somehow
1415 and selected by default
1426 and selected by default
1416 :param branch: a branch that must be in the list and selected
1427 :param branch: a branch that must be in the list and selected
1417 by default - even if closed
1428 by default - even if closed
1418 :param bookmark: a bookmark that must be in the list and selected
1429 :param bookmark: a bookmark that must be in the list and selected
1419 """
1430 """
1420 _ = translator or get_current_request().translate
1431 _ = translator or get_current_request().translate
1421
1432
1422 commit_id = safe_str(commit_id) if commit_id else None
1433 commit_id = safe_str(commit_id) if commit_id else None
1423 branch = safe_unicode(branch) if branch else None
1434 branch = safe_unicode(branch) if branch else None
1424 bookmark = safe_unicode(bookmark) if bookmark else None
1435 bookmark = safe_unicode(bookmark) if bookmark else None
1425
1436
1426 selected = None
1437 selected = None
1427
1438
1428 # order matters: first source that has commit_id in it will be selected
1439 # order matters: first source that has commit_id in it will be selected
1429 sources = []
1440 sources = []
1430 sources.append(('book', repo.bookmarks.items(), _('Bookmarks'), bookmark))
1441 sources.append(('book', repo.bookmarks.items(), _('Bookmarks'), bookmark))
1431 sources.append(('branch', repo.branches.items(), _('Branches'), branch))
1442 sources.append(('branch', repo.branches.items(), _('Branches'), branch))
1432
1443
1433 if commit_id:
1444 if commit_id:
1434 ref_commit = (h.short_id(commit_id), commit_id)
1445 ref_commit = (h.short_id(commit_id), commit_id)
1435 sources.append(('rev', [ref_commit], _('Commit IDs'), commit_id))
1446 sources.append(('rev', [ref_commit], _('Commit IDs'), commit_id))
1436
1447
1437 sources.append(
1448 sources.append(
1438 ('branch', repo.branches_closed.items(), _('Closed Branches'), branch),
1449 ('branch', repo.branches_closed.items(), _('Closed Branches'), branch),
1439 )
1450 )
1440
1451
1441 groups = []
1452 groups = []
1442
1453
1443 for group_key, ref_list, group_name, match in sources:
1454 for group_key, ref_list, group_name, match in sources:
1444 group_refs = []
1455 group_refs = []
1445 for ref_name, ref_id in ref_list:
1456 for ref_name, ref_id in ref_list:
1446 ref_key = u'{}:{}:{}'.format(group_key, ref_name, ref_id)
1457 ref_key = u'{}:{}:{}'.format(group_key, ref_name, ref_id)
1447 group_refs.append((ref_key, ref_name))
1458 group_refs.append((ref_key, ref_name))
1448
1459
1449 if not selected:
1460 if not selected:
1450 if set([commit_id, match]) & set([ref_id, ref_name]):
1461 if set([commit_id, match]) & set([ref_id, ref_name]):
1451 selected = ref_key
1462 selected = ref_key
1452
1463
1453 if group_refs:
1464 if group_refs:
1454 groups.append((group_refs, group_name))
1465 groups.append((group_refs, group_name))
1455
1466
1456 if not selected:
1467 if not selected:
1457 ref = commit_id or branch or bookmark
1468 ref = commit_id or branch or bookmark
1458 if ref:
1469 if ref:
1459 raise CommitDoesNotExistError(
1470 raise CommitDoesNotExistError(
1460 u'No commit refs could be found matching: {}'.format(ref))
1471 u'No commit refs could be found matching: {}'.format(ref))
1461 elif repo.DEFAULT_BRANCH_NAME in repo.branches:
1472 elif repo.DEFAULT_BRANCH_NAME in repo.branches:
1462 selected = u'branch:{}:{}'.format(
1473 selected = u'branch:{}:{}'.format(
1463 safe_unicode(repo.DEFAULT_BRANCH_NAME),
1474 safe_unicode(repo.DEFAULT_BRANCH_NAME),
1464 safe_unicode(repo.branches[repo.DEFAULT_BRANCH_NAME])
1475 safe_unicode(repo.branches[repo.DEFAULT_BRANCH_NAME])
1465 )
1476 )
1466 elif repo.commit_ids:
1477 elif repo.commit_ids:
1467 # make the user select in this case
1478 # make the user select in this case
1468 selected = None
1479 selected = None
1469 else:
1480 else:
1470 raise EmptyRepositoryError()
1481 raise EmptyRepositoryError()
1471 return groups, selected
1482 return groups, selected
1472
1483
1473 def get_diff(self, source_repo, source_ref_id, target_ref_id,
1484 def get_diff(self, source_repo, source_ref_id, target_ref_id,
1474 hide_whitespace_changes, diff_context):
1485 hide_whitespace_changes, diff_context):
1475
1486
1476 return self._get_diff_from_pr_or_version(
1487 return self._get_diff_from_pr_or_version(
1477 source_repo, source_ref_id, target_ref_id,
1488 source_repo, source_ref_id, target_ref_id,
1478 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
1489 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
1479
1490
1480 def _get_diff_from_pr_or_version(
1491 def _get_diff_from_pr_or_version(
1481 self, source_repo, source_ref_id, target_ref_id,
1492 self, source_repo, source_ref_id, target_ref_id,
1482 hide_whitespace_changes, diff_context):
1493 hide_whitespace_changes, diff_context):
1483
1494
1484 target_commit = source_repo.get_commit(
1495 target_commit = source_repo.get_commit(
1485 commit_id=safe_str(target_ref_id))
1496 commit_id=safe_str(target_ref_id))
1486 source_commit = source_repo.get_commit(
1497 source_commit = source_repo.get_commit(
1487 commit_id=safe_str(source_ref_id))
1498 commit_id=safe_str(source_ref_id))
1488 if isinstance(source_repo, Repository):
1499 if isinstance(source_repo, Repository):
1489 vcs_repo = source_repo.scm_instance()
1500 vcs_repo = source_repo.scm_instance()
1490 else:
1501 else:
1491 vcs_repo = source_repo
1502 vcs_repo = source_repo
1492
1503
1493 # TODO: johbo: In the context of an update, we cannot reach
1504 # TODO: johbo: In the context of an update, we cannot reach
1494 # the old commit anymore with our normal mechanisms. It needs
1505 # the old commit anymore with our normal mechanisms. It needs
1495 # some sort of special support in the vcs layer to avoid this
1506 # some sort of special support in the vcs layer to avoid this
1496 # workaround.
1507 # workaround.
1497 if (source_commit.raw_id == vcs_repo.EMPTY_COMMIT_ID and
1508 if (source_commit.raw_id == vcs_repo.EMPTY_COMMIT_ID and
1498 vcs_repo.alias == 'git'):
1509 vcs_repo.alias == 'git'):
1499 source_commit.raw_id = safe_str(source_ref_id)
1510 source_commit.raw_id = safe_str(source_ref_id)
1500
1511
1501 log.debug('calculating diff between '
1512 log.debug('calculating diff between '
1502 'source_ref:%s and target_ref:%s for repo `%s`',
1513 'source_ref:%s and target_ref:%s for repo `%s`',
1503 target_ref_id, source_ref_id,
1514 target_ref_id, source_ref_id,
1504 safe_unicode(vcs_repo.path))
1515 safe_unicode(vcs_repo.path))
1505
1516
1506 vcs_diff = vcs_repo.get_diff(
1517 vcs_diff = vcs_repo.get_diff(
1507 commit1=target_commit, commit2=source_commit,
1518 commit1=target_commit, commit2=source_commit,
1508 ignore_whitespace=hide_whitespace_changes, context=diff_context)
1519 ignore_whitespace=hide_whitespace_changes, context=diff_context)
1509 return vcs_diff
1520 return vcs_diff
1510
1521
1511 def _is_merge_enabled(self, pull_request):
1522 def _is_merge_enabled(self, pull_request):
1512 return self._get_general_setting(
1523 return self._get_general_setting(
1513 pull_request, 'rhodecode_pr_merge_enabled')
1524 pull_request, 'rhodecode_pr_merge_enabled')
1514
1525
1515 def _use_rebase_for_merging(self, pull_request):
1526 def _use_rebase_for_merging(self, pull_request):
1516 repo_type = pull_request.target_repo.repo_type
1527 repo_type = pull_request.target_repo.repo_type
1517 if repo_type == 'hg':
1528 if repo_type == 'hg':
1518 return self._get_general_setting(
1529 return self._get_general_setting(
1519 pull_request, 'rhodecode_hg_use_rebase_for_merging')
1530 pull_request, 'rhodecode_hg_use_rebase_for_merging')
1520 elif repo_type == 'git':
1531 elif repo_type == 'git':
1521 return self._get_general_setting(
1532 return self._get_general_setting(
1522 pull_request, 'rhodecode_git_use_rebase_for_merging')
1533 pull_request, 'rhodecode_git_use_rebase_for_merging')
1523
1534
1524 return False
1535 return False
1525
1536
1526 def _close_branch_before_merging(self, pull_request):
1537 def _close_branch_before_merging(self, pull_request):
1527 repo_type = pull_request.target_repo.repo_type
1538 repo_type = pull_request.target_repo.repo_type
1528 if repo_type == 'hg':
1539 if repo_type == 'hg':
1529 return self._get_general_setting(
1540 return self._get_general_setting(
1530 pull_request, 'rhodecode_hg_close_branch_before_merging')
1541 pull_request, 'rhodecode_hg_close_branch_before_merging')
1531 elif repo_type == 'git':
1542 elif repo_type == 'git':
1532 return self._get_general_setting(
1543 return self._get_general_setting(
1533 pull_request, 'rhodecode_git_close_branch_before_merging')
1544 pull_request, 'rhodecode_git_close_branch_before_merging')
1534
1545
1535 return False
1546 return False
1536
1547
1537 def _get_general_setting(self, pull_request, settings_key, default=False):
1548 def _get_general_setting(self, pull_request, settings_key, default=False):
1538 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1549 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1539 settings = settings_model.get_general_settings()
1550 settings = settings_model.get_general_settings()
1540 return settings.get(settings_key, default)
1551 return settings.get(settings_key, default)
1541
1552
1542 def _log_audit_action(self, action, action_data, user, pull_request):
1553 def _log_audit_action(self, action, action_data, user, pull_request):
1543 audit_logger.store(
1554 audit_logger.store(
1544 action=action,
1555 action=action,
1545 action_data=action_data,
1556 action_data=action_data,
1546 user=user,
1557 user=user,
1547 repo=pull_request.target_repo)
1558 repo=pull_request.target_repo)
1548
1559
1549 def get_reviewer_functions(self):
1560 def get_reviewer_functions(self):
1550 """
1561 """
1551 Fetches functions for validation and fetching default reviewers.
1562 Fetches functions for validation and fetching default reviewers.
1552 If available we use the EE package, else we fallback to CE
1563 If available we use the EE package, else we fallback to CE
1553 package functions
1564 package functions
1554 """
1565 """
1555 try:
1566 try:
1556 from rc_reviewers.utils import get_default_reviewers_data
1567 from rc_reviewers.utils import get_default_reviewers_data
1557 from rc_reviewers.utils import validate_default_reviewers
1568 from rc_reviewers.utils import validate_default_reviewers
1558 except ImportError:
1569 except ImportError:
1559 from rhodecode.apps.repository.utils import get_default_reviewers_data
1570 from rhodecode.apps.repository.utils import get_default_reviewers_data
1560 from rhodecode.apps.repository.utils import validate_default_reviewers
1571 from rhodecode.apps.repository.utils import validate_default_reviewers
1561
1572
1562 return get_default_reviewers_data, validate_default_reviewers
1573 return get_default_reviewers_data, validate_default_reviewers
1563
1574
1564
1575
1565 class MergeCheck(object):
1576 class MergeCheck(object):
1566 """
1577 """
1567 Perform Merge Checks and returns a check object which stores information
1578 Perform Merge Checks and returns a check object which stores information
1568 about merge errors, and merge conditions
1579 about merge errors, and merge conditions
1569 """
1580 """
1570 TODO_CHECK = 'todo'
1581 TODO_CHECK = 'todo'
1571 PERM_CHECK = 'perm'
1582 PERM_CHECK = 'perm'
1572 REVIEW_CHECK = 'review'
1583 REVIEW_CHECK = 'review'
1573 MERGE_CHECK = 'merge'
1584 MERGE_CHECK = 'merge'
1574
1585
1575 def __init__(self):
1586 def __init__(self):
1576 self.review_status = None
1587 self.review_status = None
1577 self.merge_possible = None
1588 self.merge_possible = None
1578 self.merge_msg = ''
1589 self.merge_msg = ''
1579 self.failed = None
1590 self.failed = None
1580 self.errors = []
1591 self.errors = []
1581 self.error_details = OrderedDict()
1592 self.error_details = OrderedDict()
1582
1593
1583 def push_error(self, error_type, message, error_key, details):
1594 def push_error(self, error_type, message, error_key, details):
1584 self.failed = True
1595 self.failed = True
1585 self.errors.append([error_type, message])
1596 self.errors.append([error_type, message])
1586 self.error_details[error_key] = dict(
1597 self.error_details[error_key] = dict(
1587 details=details,
1598 details=details,
1588 error_type=error_type,
1599 error_type=error_type,
1589 message=message
1600 message=message
1590 )
1601 )
1591
1602
1592 @classmethod
1603 @classmethod
1593 def validate(cls, pull_request, auth_user, translator, fail_early=False,
1604 def validate(cls, pull_request, auth_user, translator, fail_early=False,
1594 force_shadow_repo_refresh=False):
1605 force_shadow_repo_refresh=False):
1595 _ = translator
1606 _ = translator
1596 merge_check = cls()
1607 merge_check = cls()
1597
1608
1598 # permissions to merge
1609 # permissions to merge
1599 user_allowed_to_merge = PullRequestModel().check_user_merge(
1610 user_allowed_to_merge = PullRequestModel().check_user_merge(
1600 pull_request, auth_user)
1611 pull_request, auth_user)
1601 if not user_allowed_to_merge:
1612 if not user_allowed_to_merge:
1602 log.debug("MergeCheck: cannot merge, approval is pending.")
1613 log.debug("MergeCheck: cannot merge, approval is pending.")
1603
1614
1604 msg = _('User `{}` not allowed to perform merge.').format(auth_user.username)
1615 msg = _('User `{}` not allowed to perform merge.').format(auth_user.username)
1605 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1616 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1606 if fail_early:
1617 if fail_early:
1607 return merge_check
1618 return merge_check
1608
1619
1609 # permission to merge into the target branch
1620 # permission to merge into the target branch
1610 target_commit_id = pull_request.target_ref_parts.commit_id
1621 target_commit_id = pull_request.target_ref_parts.commit_id
1611 if pull_request.target_ref_parts.type == 'branch':
1622 if pull_request.target_ref_parts.type == 'branch':
1612 branch_name = pull_request.target_ref_parts.name
1623 branch_name = pull_request.target_ref_parts.name
1613 else:
1624 else:
1614 # for mercurial we can always figure out the branch from the commit
1625 # for mercurial we can always figure out the branch from the commit
1615 # in case of bookmark
1626 # in case of bookmark
1616 target_commit = pull_request.target_repo.get_commit(target_commit_id)
1627 target_commit = pull_request.target_repo.get_commit(target_commit_id)
1617 branch_name = target_commit.branch
1628 branch_name = target_commit.branch
1618
1629
1619 rule, branch_perm = auth_user.get_rule_and_branch_permission(
1630 rule, branch_perm = auth_user.get_rule_and_branch_permission(
1620 pull_request.target_repo.repo_name, branch_name)
1631 pull_request.target_repo.repo_name, branch_name)
1621 if branch_perm and branch_perm == 'branch.none':
1632 if branch_perm and branch_perm == 'branch.none':
1622 msg = _('Target branch `{}` changes rejected by rule {}.').format(
1633 msg = _('Target branch `{}` changes rejected by rule {}.').format(
1623 branch_name, rule)
1634 branch_name, rule)
1624 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1635 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1625 if fail_early:
1636 if fail_early:
1626 return merge_check
1637 return merge_check
1627
1638
1628 # review status, must be always present
1639 # review status, must be always present
1629 review_status = pull_request.calculated_review_status()
1640 review_status = pull_request.calculated_review_status()
1630 merge_check.review_status = review_status
1641 merge_check.review_status = review_status
1631
1642
1632 status_approved = review_status == ChangesetStatus.STATUS_APPROVED
1643 status_approved = review_status == ChangesetStatus.STATUS_APPROVED
1633 if not status_approved:
1644 if not status_approved:
1634 log.debug("MergeCheck: cannot merge, approval is pending.")
1645 log.debug("MergeCheck: cannot merge, approval is pending.")
1635
1646
1636 msg = _('Pull request reviewer approval is pending.')
1647 msg = _('Pull request reviewer approval is pending.')
1637
1648
1638 merge_check.push_error('warning', msg, cls.REVIEW_CHECK, review_status)
1649 merge_check.push_error('warning', msg, cls.REVIEW_CHECK, review_status)
1639
1650
1640 if fail_early:
1651 if fail_early:
1641 return merge_check
1652 return merge_check
1642
1653
1643 # left over TODOs
1654 # left over TODOs
1644 todos = CommentsModel().get_pull_request_unresolved_todos(pull_request)
1655 todos = CommentsModel().get_pull_request_unresolved_todos(pull_request)
1645 if todos:
1656 if todos:
1646 log.debug("MergeCheck: cannot merge, {} "
1657 log.debug("MergeCheck: cannot merge, {} "
1647 "unresolved TODOs left.".format(len(todos)))
1658 "unresolved TODOs left.".format(len(todos)))
1648
1659
1649 if len(todos) == 1:
1660 if len(todos) == 1:
1650 msg = _('Cannot merge, {} TODO still not resolved.').format(
1661 msg = _('Cannot merge, {} TODO still not resolved.').format(
1651 len(todos))
1662 len(todos))
1652 else:
1663 else:
1653 msg = _('Cannot merge, {} TODOs still not resolved.').format(
1664 msg = _('Cannot merge, {} TODOs still not resolved.').format(
1654 len(todos))
1665 len(todos))
1655
1666
1656 merge_check.push_error('warning', msg, cls.TODO_CHECK, todos)
1667 merge_check.push_error('warning', msg, cls.TODO_CHECK, todos)
1657
1668
1658 if fail_early:
1669 if fail_early:
1659 return merge_check
1670 return merge_check
1660
1671
1661 # merge possible, here is the filesystem simulation + shadow repo
1672 # merge possible, here is the filesystem simulation + shadow repo
1662 merge_status, msg = PullRequestModel().merge_status(
1673 merge_status, msg = PullRequestModel().merge_status(
1663 pull_request, translator=translator,
1674 pull_request, translator=translator,
1664 force_shadow_repo_refresh=force_shadow_repo_refresh)
1675 force_shadow_repo_refresh=force_shadow_repo_refresh)
1665 merge_check.merge_possible = merge_status
1676 merge_check.merge_possible = merge_status
1666 merge_check.merge_msg = msg
1677 merge_check.merge_msg = msg
1667 if not merge_status:
1678 if not merge_status:
1668 log.debug("MergeCheck: cannot merge, pull request merge not possible.")
1679 log.debug("MergeCheck: cannot merge, pull request merge not possible.")
1669 merge_check.push_error('warning', msg, cls.MERGE_CHECK, None)
1680 merge_check.push_error('warning', msg, cls.MERGE_CHECK, None)
1670
1681
1671 if fail_early:
1682 if fail_early:
1672 return merge_check
1683 return merge_check
1673
1684
1674 log.debug('MergeCheck: is failed: %s', merge_check.failed)
1685 log.debug('MergeCheck: is failed: %s', merge_check.failed)
1675 return merge_check
1686 return merge_check
1676
1687
1677 @classmethod
1688 @classmethod
1678 def get_merge_conditions(cls, pull_request, translator):
1689 def get_merge_conditions(cls, pull_request, translator):
1679 _ = translator
1690 _ = translator
1680 merge_details = {}
1691 merge_details = {}
1681
1692
1682 model = PullRequestModel()
1693 model = PullRequestModel()
1683 use_rebase = model._use_rebase_for_merging(pull_request)
1694 use_rebase = model._use_rebase_for_merging(pull_request)
1684
1695
1685 if use_rebase:
1696 if use_rebase:
1686 merge_details['merge_strategy'] = dict(
1697 merge_details['merge_strategy'] = dict(
1687 details={},
1698 details={},
1688 message=_('Merge strategy: rebase')
1699 message=_('Merge strategy: rebase')
1689 )
1700 )
1690 else:
1701 else:
1691 merge_details['merge_strategy'] = dict(
1702 merge_details['merge_strategy'] = dict(
1692 details={},
1703 details={},
1693 message=_('Merge strategy: explicit merge commit')
1704 message=_('Merge strategy: explicit merge commit')
1694 )
1705 )
1695
1706
1696 close_branch = model._close_branch_before_merging(pull_request)
1707 close_branch = model._close_branch_before_merging(pull_request)
1697 if close_branch:
1708 if close_branch:
1698 repo_type = pull_request.target_repo.repo_type
1709 repo_type = pull_request.target_repo.repo_type
1699 close_msg = ''
1710 close_msg = ''
1700 if repo_type == 'hg':
1711 if repo_type == 'hg':
1701 close_msg = _('Source branch will be closed after merge.')
1712 close_msg = _('Source branch will be closed after merge.')
1702 elif repo_type == 'git':
1713 elif repo_type == 'git':
1703 close_msg = _('Source branch will be deleted after merge.')
1714 close_msg = _('Source branch will be deleted after merge.')
1704
1715
1705 merge_details['close_branch'] = dict(
1716 merge_details['close_branch'] = dict(
1706 details={},
1717 details={},
1707 message=close_msg
1718 message=close_msg
1708 )
1719 )
1709
1720
1710 return merge_details
1721 return merge_details
1711
1722
1712
1723
1713 ChangeTuple = collections.namedtuple(
1724 ChangeTuple = collections.namedtuple(
1714 'ChangeTuple', ['added', 'common', 'removed', 'total'])
1725 'ChangeTuple', ['added', 'common', 'removed', 'total'])
1715
1726
1716 FileChangeTuple = collections.namedtuple(
1727 FileChangeTuple = collections.namedtuple(
1717 'FileChangeTuple', ['added', 'modified', 'removed'])
1728 'FileChangeTuple', ['added', 'modified', 'removed'])
General Comments 0
You need to be logged in to leave comments. Login now