##// END OF EJS Templates
pull-requests: handle non-ascii branches from short branch selector via URL
marcink -
r3504:7963a3f6 stable
parent child Browse files
Show More
@@ -1,1716 +1,1717 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 for uid in ids_to_add:
1023 for uid in ids_to_add:
1024 changed = True
1024 changed = True
1025 _usr = self._get_user(uid)
1025 _usr = self._get_user(uid)
1026 reviewer = PullRequestReviewers()
1026 reviewer = PullRequestReviewers()
1027 reviewer.user = _usr
1027 reviewer.user = _usr
1028 reviewer.pull_request = pull_request
1028 reviewer.pull_request = pull_request
1029 reviewer.reasons = reviewers[uid]['reasons']
1029 reviewer.reasons = reviewers[uid]['reasons']
1030 # NOTE(marcink): mandatory shouldn't be changed now
1030 # NOTE(marcink): mandatory shouldn't be changed now
1031 # reviewer.mandatory = reviewers[uid]['reasons']
1031 # reviewer.mandatory = reviewers[uid]['reasons']
1032 Session().add(reviewer)
1032 Session().add(reviewer)
1033 self._log_audit_action(
1033 self._log_audit_action(
1034 'repo.pull_request.reviewer.add', {'data': reviewer.get_dict()},
1034 'repo.pull_request.reviewer.add', {'data': reviewer.get_dict()},
1035 user, pull_request)
1035 user, pull_request)
1036
1036
1037 for uid in ids_to_remove:
1037 for uid in ids_to_remove:
1038 changed = True
1038 changed = True
1039 reviewers = PullRequestReviewers.query()\
1039 reviewers = PullRequestReviewers.query()\
1040 .filter(PullRequestReviewers.user_id == uid,
1040 .filter(PullRequestReviewers.user_id == uid,
1041 PullRequestReviewers.pull_request == pull_request)\
1041 PullRequestReviewers.pull_request == pull_request)\
1042 .all()
1042 .all()
1043 # use .all() in case we accidentally added the same person twice
1043 # use .all() in case we accidentally added the same person twice
1044 # this CAN happen due to the lack of DB checks
1044 # this CAN happen due to the lack of DB checks
1045 for obj in reviewers:
1045 for obj in reviewers:
1046 old_data = obj.get_dict()
1046 old_data = obj.get_dict()
1047 Session().delete(obj)
1047 Session().delete(obj)
1048 self._log_audit_action(
1048 self._log_audit_action(
1049 'repo.pull_request.reviewer.delete',
1049 'repo.pull_request.reviewer.delete',
1050 {'old_data': old_data}, user, pull_request)
1050 {'old_data': old_data}, user, pull_request)
1051
1051
1052 if changed:
1052 if changed:
1053 pull_request.updated_on = datetime.datetime.now()
1053 pull_request.updated_on = datetime.datetime.now()
1054 Session().add(pull_request)
1054 Session().add(pull_request)
1055
1055
1056 self.notify_reviewers(pull_request, ids_to_add)
1056 self.notify_reviewers(pull_request, ids_to_add)
1057 return ids_to_add, ids_to_remove
1057 return ids_to_add, ids_to_remove
1058
1058
1059 def get_url(self, pull_request, request=None, permalink=False):
1059 def get_url(self, pull_request, request=None, permalink=False):
1060 if not request:
1060 if not request:
1061 request = get_current_request()
1061 request = get_current_request()
1062
1062
1063 if permalink:
1063 if permalink:
1064 return request.route_url(
1064 return request.route_url(
1065 'pull_requests_global',
1065 'pull_requests_global',
1066 pull_request_id=pull_request.pull_request_id,)
1066 pull_request_id=pull_request.pull_request_id,)
1067 else:
1067 else:
1068 return request.route_url('pullrequest_show',
1068 return request.route_url('pullrequest_show',
1069 repo_name=safe_str(pull_request.target_repo.repo_name),
1069 repo_name=safe_str(pull_request.target_repo.repo_name),
1070 pull_request_id=pull_request.pull_request_id,)
1070 pull_request_id=pull_request.pull_request_id,)
1071
1071
1072 def get_shadow_clone_url(self, pull_request, request=None):
1072 def get_shadow_clone_url(self, pull_request, request=None):
1073 """
1073 """
1074 Returns qualified url pointing to the shadow repository. If this pull
1074 Returns qualified url pointing to the shadow repository. If this pull
1075 request is closed there is no shadow repository and ``None`` will be
1075 request is closed there is no shadow repository and ``None`` will be
1076 returned.
1076 returned.
1077 """
1077 """
1078 if pull_request.is_closed():
1078 if pull_request.is_closed():
1079 return None
1079 return None
1080 else:
1080 else:
1081 pr_url = urllib.unquote(self.get_url(pull_request, request=request))
1081 pr_url = urllib.unquote(self.get_url(pull_request, request=request))
1082 return safe_unicode('{pr_url}/repository'.format(pr_url=pr_url))
1082 return safe_unicode('{pr_url}/repository'.format(pr_url=pr_url))
1083
1083
1084 def notify_reviewers(self, pull_request, reviewers_ids):
1084 def notify_reviewers(self, pull_request, reviewers_ids):
1085 # notification to reviewers
1085 # notification to reviewers
1086 if not reviewers_ids:
1086 if not reviewers_ids:
1087 return
1087 return
1088
1088
1089 pull_request_obj = pull_request
1089 pull_request_obj = pull_request
1090 # get the current participants of this pull request
1090 # get the current participants of this pull request
1091 recipients = reviewers_ids
1091 recipients = reviewers_ids
1092 notification_type = EmailNotificationModel.TYPE_PULL_REQUEST
1092 notification_type = EmailNotificationModel.TYPE_PULL_REQUEST
1093
1093
1094 pr_source_repo = pull_request_obj.source_repo
1094 pr_source_repo = pull_request_obj.source_repo
1095 pr_target_repo = pull_request_obj.target_repo
1095 pr_target_repo = pull_request_obj.target_repo
1096
1096
1097 pr_url = h.route_url('pullrequest_show',
1097 pr_url = h.route_url('pullrequest_show',
1098 repo_name=pr_target_repo.repo_name,
1098 repo_name=pr_target_repo.repo_name,
1099 pull_request_id=pull_request_obj.pull_request_id,)
1099 pull_request_id=pull_request_obj.pull_request_id,)
1100
1100
1101 # set some variables for email notification
1101 # set some variables for email notification
1102 pr_target_repo_url = h.route_url(
1102 pr_target_repo_url = h.route_url(
1103 'repo_summary', repo_name=pr_target_repo.repo_name)
1103 'repo_summary', repo_name=pr_target_repo.repo_name)
1104
1104
1105 pr_source_repo_url = h.route_url(
1105 pr_source_repo_url = h.route_url(
1106 'repo_summary', repo_name=pr_source_repo.repo_name)
1106 'repo_summary', repo_name=pr_source_repo.repo_name)
1107
1107
1108 # pull request specifics
1108 # pull request specifics
1109 pull_request_commits = [
1109 pull_request_commits = [
1110 (x.raw_id, x.message)
1110 (x.raw_id, x.message)
1111 for x in map(pr_source_repo.get_commit, pull_request.revisions)]
1111 for x in map(pr_source_repo.get_commit, pull_request.revisions)]
1112
1112
1113 kwargs = {
1113 kwargs = {
1114 'user': pull_request.author,
1114 'user': pull_request.author,
1115 'pull_request': pull_request_obj,
1115 'pull_request': pull_request_obj,
1116 'pull_request_commits': pull_request_commits,
1116 'pull_request_commits': pull_request_commits,
1117
1117
1118 'pull_request_target_repo': pr_target_repo,
1118 'pull_request_target_repo': pr_target_repo,
1119 'pull_request_target_repo_url': pr_target_repo_url,
1119 'pull_request_target_repo_url': pr_target_repo_url,
1120
1120
1121 'pull_request_source_repo': pr_source_repo,
1121 'pull_request_source_repo': pr_source_repo,
1122 'pull_request_source_repo_url': pr_source_repo_url,
1122 'pull_request_source_repo_url': pr_source_repo_url,
1123
1123
1124 'pull_request_url': pr_url,
1124 'pull_request_url': pr_url,
1125 }
1125 }
1126
1126
1127 # pre-generate the subject for notification itself
1127 # pre-generate the subject for notification itself
1128 (subject,
1128 (subject,
1129 _h, _e, # we don't care about those
1129 _h, _e, # we don't care about those
1130 body_plaintext) = EmailNotificationModel().render_email(
1130 body_plaintext) = EmailNotificationModel().render_email(
1131 notification_type, **kwargs)
1131 notification_type, **kwargs)
1132
1132
1133 # create notification objects, and emails
1133 # create notification objects, and emails
1134 NotificationModel().create(
1134 NotificationModel().create(
1135 created_by=pull_request.author,
1135 created_by=pull_request.author,
1136 notification_subject=subject,
1136 notification_subject=subject,
1137 notification_body=body_plaintext,
1137 notification_body=body_plaintext,
1138 notification_type=notification_type,
1138 notification_type=notification_type,
1139 recipients=recipients,
1139 recipients=recipients,
1140 email_kwargs=kwargs,
1140 email_kwargs=kwargs,
1141 )
1141 )
1142
1142
1143 def delete(self, pull_request, user):
1143 def delete(self, pull_request, user):
1144 pull_request = self.__get_pull_request(pull_request)
1144 pull_request = self.__get_pull_request(pull_request)
1145 old_data = pull_request.get_api_data(with_merge_state=False)
1145 old_data = pull_request.get_api_data(with_merge_state=False)
1146 self._cleanup_merge_workspace(pull_request)
1146 self._cleanup_merge_workspace(pull_request)
1147 self._log_audit_action(
1147 self._log_audit_action(
1148 'repo.pull_request.delete', {'old_data': old_data},
1148 'repo.pull_request.delete', {'old_data': old_data},
1149 user, pull_request)
1149 user, pull_request)
1150 Session().delete(pull_request)
1150 Session().delete(pull_request)
1151
1151
1152 def close_pull_request(self, pull_request, user):
1152 def close_pull_request(self, pull_request, user):
1153 pull_request = self.__get_pull_request(pull_request)
1153 pull_request = self.__get_pull_request(pull_request)
1154 self._cleanup_merge_workspace(pull_request)
1154 self._cleanup_merge_workspace(pull_request)
1155 pull_request.status = PullRequest.STATUS_CLOSED
1155 pull_request.status = PullRequest.STATUS_CLOSED
1156 pull_request.updated_on = datetime.datetime.now()
1156 pull_request.updated_on = datetime.datetime.now()
1157 Session().add(pull_request)
1157 Session().add(pull_request)
1158 self.trigger_pull_request_hook(
1158 self.trigger_pull_request_hook(
1159 pull_request, pull_request.author, 'close')
1159 pull_request, pull_request.author, 'close')
1160
1160
1161 pr_data = pull_request.get_api_data(with_merge_state=False)
1161 pr_data = pull_request.get_api_data(with_merge_state=False)
1162 self._log_audit_action(
1162 self._log_audit_action(
1163 'repo.pull_request.close', {'data': pr_data}, user, pull_request)
1163 'repo.pull_request.close', {'data': pr_data}, user, pull_request)
1164
1164
1165 def close_pull_request_with_comment(
1165 def close_pull_request_with_comment(
1166 self, pull_request, user, repo, message=None, auth_user=None):
1166 self, pull_request, user, repo, message=None, auth_user=None):
1167
1167
1168 pull_request_review_status = pull_request.calculated_review_status()
1168 pull_request_review_status = pull_request.calculated_review_status()
1169
1169
1170 if pull_request_review_status == ChangesetStatus.STATUS_APPROVED:
1170 if pull_request_review_status == ChangesetStatus.STATUS_APPROVED:
1171 # approved only if we have voting consent
1171 # approved only if we have voting consent
1172 status = ChangesetStatus.STATUS_APPROVED
1172 status = ChangesetStatus.STATUS_APPROVED
1173 else:
1173 else:
1174 status = ChangesetStatus.STATUS_REJECTED
1174 status = ChangesetStatus.STATUS_REJECTED
1175 status_lbl = ChangesetStatus.get_status_lbl(status)
1175 status_lbl = ChangesetStatus.get_status_lbl(status)
1176
1176
1177 default_message = (
1177 default_message = (
1178 'Closing with status change {transition_icon} {status}.'
1178 'Closing with status change {transition_icon} {status}.'
1179 ).format(transition_icon='>', status=status_lbl)
1179 ).format(transition_icon='>', status=status_lbl)
1180 text = message or default_message
1180 text = message or default_message
1181
1181
1182 # create a comment, and link it to new status
1182 # create a comment, and link it to new status
1183 comment = CommentsModel().create(
1183 comment = CommentsModel().create(
1184 text=text,
1184 text=text,
1185 repo=repo.repo_id,
1185 repo=repo.repo_id,
1186 user=user.user_id,
1186 user=user.user_id,
1187 pull_request=pull_request.pull_request_id,
1187 pull_request=pull_request.pull_request_id,
1188 status_change=status_lbl,
1188 status_change=status_lbl,
1189 status_change_type=status,
1189 status_change_type=status,
1190 closing_pr=True,
1190 closing_pr=True,
1191 auth_user=auth_user,
1191 auth_user=auth_user,
1192 )
1192 )
1193
1193
1194 # calculate old status before we change it
1194 # calculate old status before we change it
1195 old_calculated_status = pull_request.calculated_review_status()
1195 old_calculated_status = pull_request.calculated_review_status()
1196 ChangesetStatusModel().set_status(
1196 ChangesetStatusModel().set_status(
1197 repo.repo_id,
1197 repo.repo_id,
1198 status,
1198 status,
1199 user.user_id,
1199 user.user_id,
1200 comment=comment,
1200 comment=comment,
1201 pull_request=pull_request.pull_request_id
1201 pull_request=pull_request.pull_request_id
1202 )
1202 )
1203
1203
1204 Session().flush()
1204 Session().flush()
1205 events.trigger(events.PullRequestCommentEvent(pull_request, comment))
1205 events.trigger(events.PullRequestCommentEvent(pull_request, comment))
1206 # we now calculate the status of pull request again, and based on that
1206 # we now calculate the status of pull request again, and based on that
1207 # calculation trigger status change. This might happen in cases
1207 # calculation trigger status change. This might happen in cases
1208 # that non-reviewer admin closes a pr, which means his vote doesn't
1208 # 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.
1209 # change the status, while if he's a reviewer this might change it.
1210 calculated_status = pull_request.calculated_review_status()
1210 calculated_status = pull_request.calculated_review_status()
1211 if old_calculated_status != calculated_status:
1211 if old_calculated_status != calculated_status:
1212 self.trigger_pull_request_hook(
1212 self.trigger_pull_request_hook(
1213 pull_request, user, 'review_status_change',
1213 pull_request, user, 'review_status_change',
1214 data={'status': calculated_status})
1214 data={'status': calculated_status})
1215
1215
1216 # finally close the PR
1216 # finally close the PR
1217 PullRequestModel().close_pull_request(
1217 PullRequestModel().close_pull_request(
1218 pull_request.pull_request_id, user)
1218 pull_request.pull_request_id, user)
1219
1219
1220 return comment, status
1220 return comment, status
1221
1221
1222 def merge_status(self, pull_request, translator=None,
1222 def merge_status(self, pull_request, translator=None,
1223 force_shadow_repo_refresh=False):
1223 force_shadow_repo_refresh=False):
1224 _ = translator or get_current_request().translate
1224 _ = translator or get_current_request().translate
1225
1225
1226 if not self._is_merge_enabled(pull_request):
1226 if not self._is_merge_enabled(pull_request):
1227 return False, _('Server-side pull request merging is disabled.')
1227 return False, _('Server-side pull request merging is disabled.')
1228 if pull_request.is_closed():
1228 if pull_request.is_closed():
1229 return False, _('This pull request is closed.')
1229 return False, _('This pull request is closed.')
1230 merge_possible, msg = self._check_repo_requirements(
1230 merge_possible, msg = self._check_repo_requirements(
1231 target=pull_request.target_repo, source=pull_request.source_repo,
1231 target=pull_request.target_repo, source=pull_request.source_repo,
1232 translator=_)
1232 translator=_)
1233 if not merge_possible:
1233 if not merge_possible:
1234 return merge_possible, msg
1234 return merge_possible, msg
1235
1235
1236 try:
1236 try:
1237 resp = self._try_merge(
1237 resp = self._try_merge(
1238 pull_request,
1238 pull_request,
1239 force_shadow_repo_refresh=force_shadow_repo_refresh)
1239 force_shadow_repo_refresh=force_shadow_repo_refresh)
1240 log.debug("Merge response: %s", resp)
1240 log.debug("Merge response: %s", resp)
1241 status = resp.possible, resp.merge_status_message
1241 status = resp.possible, resp.merge_status_message
1242 except NotImplementedError:
1242 except NotImplementedError:
1243 status = False, _('Pull request merging is not supported.')
1243 status = False, _('Pull request merging is not supported.')
1244
1244
1245 return status
1245 return status
1246
1246
1247 def _check_repo_requirements(self, target, source, translator):
1247 def _check_repo_requirements(self, target, source, translator):
1248 """
1248 """
1249 Check if `target` and `source` have compatible requirements.
1249 Check if `target` and `source` have compatible requirements.
1250
1250
1251 Currently this is just checking for largefiles.
1251 Currently this is just checking for largefiles.
1252 """
1252 """
1253 _ = translator
1253 _ = translator
1254 target_has_largefiles = self._has_largefiles(target)
1254 target_has_largefiles = self._has_largefiles(target)
1255 source_has_largefiles = self._has_largefiles(source)
1255 source_has_largefiles = self._has_largefiles(source)
1256 merge_possible = True
1256 merge_possible = True
1257 message = u''
1257 message = u''
1258
1258
1259 if target_has_largefiles != source_has_largefiles:
1259 if target_has_largefiles != source_has_largefiles:
1260 merge_possible = False
1260 merge_possible = False
1261 if source_has_largefiles:
1261 if source_has_largefiles:
1262 message = _(
1262 message = _(
1263 'Target repository large files support is disabled.')
1263 'Target repository large files support is disabled.')
1264 else:
1264 else:
1265 message = _(
1265 message = _(
1266 'Source repository large files support is disabled.')
1266 'Source repository large files support is disabled.')
1267
1267
1268 return merge_possible, message
1268 return merge_possible, message
1269
1269
1270 def _has_largefiles(self, repo):
1270 def _has_largefiles(self, repo):
1271 largefiles_ui = VcsSettingsModel(repo=repo).get_ui_settings(
1271 largefiles_ui = VcsSettingsModel(repo=repo).get_ui_settings(
1272 'extensions', 'largefiles')
1272 'extensions', 'largefiles')
1273 return largefiles_ui and largefiles_ui[0].active
1273 return largefiles_ui and largefiles_ui[0].active
1274
1274
1275 def _try_merge(self, pull_request, force_shadow_repo_refresh=False):
1275 def _try_merge(self, pull_request, force_shadow_repo_refresh=False):
1276 """
1276 """
1277 Try to merge the pull request and return the merge status.
1277 Try to merge the pull request and return the merge status.
1278 """
1278 """
1279 log.debug(
1279 log.debug(
1280 "Trying out if the pull request %s can be merged. Force_refresh=%s",
1280 "Trying out if the pull request %s can be merged. Force_refresh=%s",
1281 pull_request.pull_request_id, force_shadow_repo_refresh)
1281 pull_request.pull_request_id, force_shadow_repo_refresh)
1282 target_vcs = pull_request.target_repo.scm_instance()
1282 target_vcs = pull_request.target_repo.scm_instance()
1283 # Refresh the target reference.
1283 # Refresh the target reference.
1284 try:
1284 try:
1285 target_ref = self._refresh_reference(
1285 target_ref = self._refresh_reference(
1286 pull_request.target_ref_parts, target_vcs)
1286 pull_request.target_ref_parts, target_vcs)
1287 except CommitDoesNotExistError:
1287 except CommitDoesNotExistError:
1288 merge_state = MergeResponse(
1288 merge_state = MergeResponse(
1289 False, False, None, MergeFailureReason.MISSING_TARGET_REF,
1289 False, False, None, MergeFailureReason.MISSING_TARGET_REF,
1290 metadata={'target_ref': pull_request.target_ref_parts})
1290 metadata={'target_ref': pull_request.target_ref_parts})
1291 return merge_state
1291 return merge_state
1292
1292
1293 target_locked = pull_request.target_repo.locked
1293 target_locked = pull_request.target_repo.locked
1294 if target_locked and target_locked[0]:
1294 if target_locked and target_locked[0]:
1295 locked_by = 'user:{}'.format(target_locked[0])
1295 locked_by = 'user:{}'.format(target_locked[0])
1296 log.debug("The target repository is locked by %s.", locked_by)
1296 log.debug("The target repository is locked by %s.", locked_by)
1297 merge_state = MergeResponse(
1297 merge_state = MergeResponse(
1298 False, False, None, MergeFailureReason.TARGET_IS_LOCKED,
1298 False, False, None, MergeFailureReason.TARGET_IS_LOCKED,
1299 metadata={'locked_by': locked_by})
1299 metadata={'locked_by': locked_by})
1300 elif force_shadow_repo_refresh or self._needs_merge_state_refresh(
1300 elif force_shadow_repo_refresh or self._needs_merge_state_refresh(
1301 pull_request, target_ref):
1301 pull_request, target_ref):
1302 log.debug("Refreshing the merge status of the repository.")
1302 log.debug("Refreshing the merge status of the repository.")
1303 merge_state = self._refresh_merge_state(
1303 merge_state = self._refresh_merge_state(
1304 pull_request, target_vcs, target_ref)
1304 pull_request, target_vcs, target_ref)
1305 else:
1305 else:
1306 possible = pull_request.\
1306 possible = pull_request.\
1307 last_merge_status == MergeFailureReason.NONE
1307 last_merge_status == MergeFailureReason.NONE
1308 merge_state = MergeResponse(
1308 merge_state = MergeResponse(
1309 possible, False, None, pull_request.last_merge_status)
1309 possible, False, None, pull_request.last_merge_status)
1310
1310
1311 return merge_state
1311 return merge_state
1312
1312
1313 def _refresh_reference(self, reference, vcs_repository):
1313 def _refresh_reference(self, reference, vcs_repository):
1314 if reference.type in self.UPDATABLE_REF_TYPES:
1314 if reference.type in self.UPDATABLE_REF_TYPES:
1315 name_or_id = reference.name
1315 name_or_id = reference.name
1316 else:
1316 else:
1317 name_or_id = reference.commit_id
1317 name_or_id = reference.commit_id
1318 refreshed_commit = vcs_repository.get_commit(name_or_id)
1318 refreshed_commit = vcs_repository.get_commit(name_or_id)
1319 refreshed_reference = Reference(
1319 refreshed_reference = Reference(
1320 reference.type, reference.name, refreshed_commit.raw_id)
1320 reference.type, reference.name, refreshed_commit.raw_id)
1321 return refreshed_reference
1321 return refreshed_reference
1322
1322
1323 def _needs_merge_state_refresh(self, pull_request, target_reference):
1323 def _needs_merge_state_refresh(self, pull_request, target_reference):
1324 return not(
1324 return not(
1325 pull_request.revisions and
1325 pull_request.revisions and
1326 pull_request.revisions[0] == pull_request._last_merge_source_rev and
1326 pull_request.revisions[0] == pull_request._last_merge_source_rev and
1327 target_reference.commit_id == pull_request._last_merge_target_rev)
1327 target_reference.commit_id == pull_request._last_merge_target_rev)
1328
1328
1329 def _refresh_merge_state(self, pull_request, target_vcs, target_reference):
1329 def _refresh_merge_state(self, pull_request, target_vcs, target_reference):
1330 workspace_id = self._workspace_id(pull_request)
1330 workspace_id = self._workspace_id(pull_request)
1331 source_vcs = pull_request.source_repo.scm_instance()
1331 source_vcs = pull_request.source_repo.scm_instance()
1332 repo_id = pull_request.target_repo.repo_id
1332 repo_id = pull_request.target_repo.repo_id
1333 use_rebase = self._use_rebase_for_merging(pull_request)
1333 use_rebase = self._use_rebase_for_merging(pull_request)
1334 close_branch = self._close_branch_before_merging(pull_request)
1334 close_branch = self._close_branch_before_merging(pull_request)
1335 merge_state = target_vcs.merge(
1335 merge_state = target_vcs.merge(
1336 repo_id, workspace_id,
1336 repo_id, workspace_id,
1337 target_reference, source_vcs, pull_request.source_ref_parts,
1337 target_reference, source_vcs, pull_request.source_ref_parts,
1338 dry_run=True, use_rebase=use_rebase,
1338 dry_run=True, use_rebase=use_rebase,
1339 close_branch=close_branch)
1339 close_branch=close_branch)
1340
1340
1341 # Do not store the response if there was an unknown error.
1341 # Do not store the response if there was an unknown error.
1342 if merge_state.failure_reason != MergeFailureReason.UNKNOWN:
1342 if merge_state.failure_reason != MergeFailureReason.UNKNOWN:
1343 pull_request._last_merge_source_rev = \
1343 pull_request._last_merge_source_rev = \
1344 pull_request.source_ref_parts.commit_id
1344 pull_request.source_ref_parts.commit_id
1345 pull_request._last_merge_target_rev = target_reference.commit_id
1345 pull_request._last_merge_target_rev = target_reference.commit_id
1346 pull_request.last_merge_status = merge_state.failure_reason
1346 pull_request.last_merge_status = merge_state.failure_reason
1347 pull_request.shadow_merge_ref = merge_state.merge_ref
1347 pull_request.shadow_merge_ref = merge_state.merge_ref
1348 Session().add(pull_request)
1348 Session().add(pull_request)
1349 Session().commit()
1349 Session().commit()
1350
1350
1351 return merge_state
1351 return merge_state
1352
1352
1353 def _workspace_id(self, pull_request):
1353 def _workspace_id(self, pull_request):
1354 workspace_id = 'pr-%s' % pull_request.pull_request_id
1354 workspace_id = 'pr-%s' % pull_request.pull_request_id
1355 return workspace_id
1355 return workspace_id
1356
1356
1357 def generate_repo_data(self, repo, commit_id=None, branch=None,
1357 def generate_repo_data(self, repo, commit_id=None, branch=None,
1358 bookmark=None, translator=None):
1358 bookmark=None, translator=None):
1359 from rhodecode.model.repo import RepoModel
1359 from rhodecode.model.repo import RepoModel
1360
1360
1361 all_refs, selected_ref = \
1361 all_refs, selected_ref = \
1362 self._get_repo_pullrequest_sources(
1362 self._get_repo_pullrequest_sources(
1363 repo.scm_instance(), commit_id=commit_id,
1363 repo.scm_instance(), commit_id=commit_id,
1364 branch=branch, bookmark=bookmark, translator=translator)
1364 branch=branch, bookmark=bookmark, translator=translator)
1365
1365
1366 refs_select2 = []
1366 refs_select2 = []
1367 for element in all_refs:
1367 for element in all_refs:
1368 children = [{'id': x[0], 'text': x[1]} for x in element[0]]
1368 children = [{'id': x[0], 'text': x[1]} for x in element[0]]
1369 refs_select2.append({'text': element[1], 'children': children})
1369 refs_select2.append({'text': element[1], 'children': children})
1370
1370
1371 return {
1371 return {
1372 'user': {
1372 'user': {
1373 'user_id': repo.user.user_id,
1373 'user_id': repo.user.user_id,
1374 'username': repo.user.username,
1374 'username': repo.user.username,
1375 'firstname': repo.user.first_name,
1375 'firstname': repo.user.first_name,
1376 'lastname': repo.user.last_name,
1376 'lastname': repo.user.last_name,
1377 'gravatar_link': h.gravatar_url(repo.user.email, 14),
1377 'gravatar_link': h.gravatar_url(repo.user.email, 14),
1378 },
1378 },
1379 'name': repo.repo_name,
1379 'name': repo.repo_name,
1380 'link': RepoModel().get_url(repo),
1380 'link': RepoModel().get_url(repo),
1381 'description': h.chop_at_smart(repo.description_safe, '\n'),
1381 'description': h.chop_at_smart(repo.description_safe, '\n'),
1382 'refs': {
1382 'refs': {
1383 'all_refs': all_refs,
1383 'all_refs': all_refs,
1384 'selected_ref': selected_ref,
1384 'selected_ref': selected_ref,
1385 'select2_refs': refs_select2
1385 'select2_refs': refs_select2
1386 }
1386 }
1387 }
1387 }
1388
1388
1389 def generate_pullrequest_title(self, source, source_ref, target):
1389 def generate_pullrequest_title(self, source, source_ref, target):
1390 return u'{source}#{at_ref} to {target}'.format(
1390 return u'{source}#{at_ref} to {target}'.format(
1391 source=source,
1391 source=source,
1392 at_ref=source_ref,
1392 at_ref=source_ref,
1393 target=target,
1393 target=target,
1394 )
1394 )
1395
1395
1396 def _cleanup_merge_workspace(self, pull_request):
1396 def _cleanup_merge_workspace(self, pull_request):
1397 # Merging related cleanup
1397 # Merging related cleanup
1398 repo_id = pull_request.target_repo.repo_id
1398 repo_id = pull_request.target_repo.repo_id
1399 target_scm = pull_request.target_repo.scm_instance()
1399 target_scm = pull_request.target_repo.scm_instance()
1400 workspace_id = self._workspace_id(pull_request)
1400 workspace_id = self._workspace_id(pull_request)
1401
1401
1402 try:
1402 try:
1403 target_scm.cleanup_merge_workspace(repo_id, workspace_id)
1403 target_scm.cleanup_merge_workspace(repo_id, workspace_id)
1404 except NotImplementedError:
1404 except NotImplementedError:
1405 pass
1405 pass
1406
1406
1407 def _get_repo_pullrequest_sources(
1407 def _get_repo_pullrequest_sources(
1408 self, repo, commit_id=None, branch=None, bookmark=None,
1408 self, repo, commit_id=None, branch=None, bookmark=None,
1409 translator=None):
1409 translator=None):
1410 """
1410 """
1411 Return a structure with repo's interesting commits, suitable for
1411 Return a structure with repo's interesting commits, suitable for
1412 the selectors in pullrequest controller
1412 the selectors in pullrequest controller
1413
1413
1414 :param commit_id: a commit that must be in the list somehow
1414 :param commit_id: a commit that must be in the list somehow
1415 and selected by default
1415 and selected by default
1416 :param branch: a branch that must be in the list and selected
1416 :param branch: a branch that must be in the list and selected
1417 by default - even if closed
1417 by default - even if closed
1418 :param bookmark: a bookmark that must be in the list and selected
1418 :param bookmark: a bookmark that must be in the list and selected
1419 """
1419 """
1420 _ = translator or get_current_request().translate
1420 _ = translator or get_current_request().translate
1421
1421
1422 commit_id = safe_str(commit_id) if commit_id else None
1422 commit_id = safe_str(commit_id) if commit_id else None
1423 branch = safe_str(branch) if branch else None
1423 branch = safe_unicode(branch) if branch else None
1424 bookmark = safe_str(bookmark) if bookmark else None
1424 bookmark = safe_unicode(bookmark) if bookmark else None
1425
1425
1426 selected = None
1426 selected = None
1427
1427
1428 # order matters: first source that has commit_id in it will be selected
1428 # order matters: first source that has commit_id in it will be selected
1429 sources = []
1429 sources = []
1430 sources.append(('book', repo.bookmarks.items(), _('Bookmarks'), bookmark))
1430 sources.append(('book', repo.bookmarks.items(), _('Bookmarks'), bookmark))
1431 sources.append(('branch', repo.branches.items(), _('Branches'), branch))
1431 sources.append(('branch', repo.branches.items(), _('Branches'), branch))
1432
1432
1433 if commit_id:
1433 if commit_id:
1434 ref_commit = (h.short_id(commit_id), commit_id)
1434 ref_commit = (h.short_id(commit_id), commit_id)
1435 sources.append(('rev', [ref_commit], _('Commit IDs'), commit_id))
1435 sources.append(('rev', [ref_commit], _('Commit IDs'), commit_id))
1436
1436
1437 sources.append(
1437 sources.append(
1438 ('branch', repo.branches_closed.items(), _('Closed Branches'), branch),
1438 ('branch', repo.branches_closed.items(), _('Closed Branches'), branch),
1439 )
1439 )
1440
1440
1441 groups = []
1441 groups = []
1442
1442 for group_key, ref_list, group_name, match in sources:
1443 for group_key, ref_list, group_name, match in sources:
1443 group_refs = []
1444 group_refs = []
1444 for ref_name, ref_id in ref_list:
1445 for ref_name, ref_id in ref_list:
1445 ref_key = '%s:%s:%s' % (group_key, ref_name, ref_id)
1446 ref_key = u'{}:{}:{}'.format(group_key, ref_name, ref_id)
1446 group_refs.append((ref_key, ref_name))
1447 group_refs.append((ref_key, ref_name))
1447
1448
1448 if not selected:
1449 if not selected:
1449 if set([commit_id, match]) & set([ref_id, ref_name]):
1450 if set([commit_id, match]) & set([ref_id, ref_name]):
1450 selected = ref_key
1451 selected = ref_key
1451
1452
1452 if group_refs:
1453 if group_refs:
1453 groups.append((group_refs, group_name))
1454 groups.append((group_refs, group_name))
1454
1455
1455 if not selected:
1456 if not selected:
1456 ref = commit_id or branch or bookmark
1457 ref = commit_id or branch or bookmark
1457 if ref:
1458 if ref:
1458 raise CommitDoesNotExistError(
1459 raise CommitDoesNotExistError(
1459 'No commit refs could be found matching: %s' % ref)
1460 u'No commit refs could be found matching: {}'.format(ref))
1460 elif repo.DEFAULT_BRANCH_NAME in repo.branches:
1461 elif repo.DEFAULT_BRANCH_NAME in repo.branches:
1461 selected = 'branch:%s:%s' % (
1462 selected = u'branch:{}:{}'.format(
1462 repo.DEFAULT_BRANCH_NAME,
1463 safe_unicode(repo.DEFAULT_BRANCH_NAME),
1463 repo.branches[repo.DEFAULT_BRANCH_NAME]
1464 safe_unicode(repo.branches[repo.DEFAULT_BRANCH_NAME])
1464 )
1465 )
1465 elif repo.commit_ids:
1466 elif repo.commit_ids:
1466 # make the user select in this case
1467 # make the user select in this case
1467 selected = None
1468 selected = None
1468 else:
1469 else:
1469 raise EmptyRepositoryError()
1470 raise EmptyRepositoryError()
1470 return groups, selected
1471 return groups, selected
1471
1472
1472 def get_diff(self, source_repo, source_ref_id, target_ref_id,
1473 def get_diff(self, source_repo, source_ref_id, target_ref_id,
1473 hide_whitespace_changes, diff_context):
1474 hide_whitespace_changes, diff_context):
1474
1475
1475 return self._get_diff_from_pr_or_version(
1476 return self._get_diff_from_pr_or_version(
1476 source_repo, source_ref_id, target_ref_id,
1477 source_repo, source_ref_id, target_ref_id,
1477 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
1478 hide_whitespace_changes=hide_whitespace_changes, diff_context=diff_context)
1478
1479
1479 def _get_diff_from_pr_or_version(
1480 def _get_diff_from_pr_or_version(
1480 self, source_repo, source_ref_id, target_ref_id,
1481 self, source_repo, source_ref_id, target_ref_id,
1481 hide_whitespace_changes, diff_context):
1482 hide_whitespace_changes, diff_context):
1482
1483
1483 target_commit = source_repo.get_commit(
1484 target_commit = source_repo.get_commit(
1484 commit_id=safe_str(target_ref_id))
1485 commit_id=safe_str(target_ref_id))
1485 source_commit = source_repo.get_commit(
1486 source_commit = source_repo.get_commit(
1486 commit_id=safe_str(source_ref_id))
1487 commit_id=safe_str(source_ref_id))
1487 if isinstance(source_repo, Repository):
1488 if isinstance(source_repo, Repository):
1488 vcs_repo = source_repo.scm_instance()
1489 vcs_repo = source_repo.scm_instance()
1489 else:
1490 else:
1490 vcs_repo = source_repo
1491 vcs_repo = source_repo
1491
1492
1492 # TODO: johbo: In the context of an update, we cannot reach
1493 # TODO: johbo: In the context of an update, we cannot reach
1493 # the old commit anymore with our normal mechanisms. It needs
1494 # the old commit anymore with our normal mechanisms. It needs
1494 # some sort of special support in the vcs layer to avoid this
1495 # some sort of special support in the vcs layer to avoid this
1495 # workaround.
1496 # workaround.
1496 if (source_commit.raw_id == vcs_repo.EMPTY_COMMIT_ID and
1497 if (source_commit.raw_id == vcs_repo.EMPTY_COMMIT_ID and
1497 vcs_repo.alias == 'git'):
1498 vcs_repo.alias == 'git'):
1498 source_commit.raw_id = safe_str(source_ref_id)
1499 source_commit.raw_id = safe_str(source_ref_id)
1499
1500
1500 log.debug('calculating diff between '
1501 log.debug('calculating diff between '
1501 'source_ref:%s and target_ref:%s for repo `%s`',
1502 'source_ref:%s and target_ref:%s for repo `%s`',
1502 target_ref_id, source_ref_id,
1503 target_ref_id, source_ref_id,
1503 safe_unicode(vcs_repo.path))
1504 safe_unicode(vcs_repo.path))
1504
1505
1505 vcs_diff = vcs_repo.get_diff(
1506 vcs_diff = vcs_repo.get_diff(
1506 commit1=target_commit, commit2=source_commit,
1507 commit1=target_commit, commit2=source_commit,
1507 ignore_whitespace=hide_whitespace_changes, context=diff_context)
1508 ignore_whitespace=hide_whitespace_changes, context=diff_context)
1508 return vcs_diff
1509 return vcs_diff
1509
1510
1510 def _is_merge_enabled(self, pull_request):
1511 def _is_merge_enabled(self, pull_request):
1511 return self._get_general_setting(
1512 return self._get_general_setting(
1512 pull_request, 'rhodecode_pr_merge_enabled')
1513 pull_request, 'rhodecode_pr_merge_enabled')
1513
1514
1514 def _use_rebase_for_merging(self, pull_request):
1515 def _use_rebase_for_merging(self, pull_request):
1515 repo_type = pull_request.target_repo.repo_type
1516 repo_type = pull_request.target_repo.repo_type
1516 if repo_type == 'hg':
1517 if repo_type == 'hg':
1517 return self._get_general_setting(
1518 return self._get_general_setting(
1518 pull_request, 'rhodecode_hg_use_rebase_for_merging')
1519 pull_request, 'rhodecode_hg_use_rebase_for_merging')
1519 elif repo_type == 'git':
1520 elif repo_type == 'git':
1520 return self._get_general_setting(
1521 return self._get_general_setting(
1521 pull_request, 'rhodecode_git_use_rebase_for_merging')
1522 pull_request, 'rhodecode_git_use_rebase_for_merging')
1522
1523
1523 return False
1524 return False
1524
1525
1525 def _close_branch_before_merging(self, pull_request):
1526 def _close_branch_before_merging(self, pull_request):
1526 repo_type = pull_request.target_repo.repo_type
1527 repo_type = pull_request.target_repo.repo_type
1527 if repo_type == 'hg':
1528 if repo_type == 'hg':
1528 return self._get_general_setting(
1529 return self._get_general_setting(
1529 pull_request, 'rhodecode_hg_close_branch_before_merging')
1530 pull_request, 'rhodecode_hg_close_branch_before_merging')
1530 elif repo_type == 'git':
1531 elif repo_type == 'git':
1531 return self._get_general_setting(
1532 return self._get_general_setting(
1532 pull_request, 'rhodecode_git_close_branch_before_merging')
1533 pull_request, 'rhodecode_git_close_branch_before_merging')
1533
1534
1534 return False
1535 return False
1535
1536
1536 def _get_general_setting(self, pull_request, settings_key, default=False):
1537 def _get_general_setting(self, pull_request, settings_key, default=False):
1537 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1538 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1538 settings = settings_model.get_general_settings()
1539 settings = settings_model.get_general_settings()
1539 return settings.get(settings_key, default)
1540 return settings.get(settings_key, default)
1540
1541
1541 def _log_audit_action(self, action, action_data, user, pull_request):
1542 def _log_audit_action(self, action, action_data, user, pull_request):
1542 audit_logger.store(
1543 audit_logger.store(
1543 action=action,
1544 action=action,
1544 action_data=action_data,
1545 action_data=action_data,
1545 user=user,
1546 user=user,
1546 repo=pull_request.target_repo)
1547 repo=pull_request.target_repo)
1547
1548
1548 def get_reviewer_functions(self):
1549 def get_reviewer_functions(self):
1549 """
1550 """
1550 Fetches functions for validation and fetching default reviewers.
1551 Fetches functions for validation and fetching default reviewers.
1551 If available we use the EE package, else we fallback to CE
1552 If available we use the EE package, else we fallback to CE
1552 package functions
1553 package functions
1553 """
1554 """
1554 try:
1555 try:
1555 from rc_reviewers.utils import get_default_reviewers_data
1556 from rc_reviewers.utils import get_default_reviewers_data
1556 from rc_reviewers.utils import validate_default_reviewers
1557 from rc_reviewers.utils import validate_default_reviewers
1557 except ImportError:
1558 except ImportError:
1558 from rhodecode.apps.repository.utils import get_default_reviewers_data
1559 from rhodecode.apps.repository.utils import get_default_reviewers_data
1559 from rhodecode.apps.repository.utils import validate_default_reviewers
1560 from rhodecode.apps.repository.utils import validate_default_reviewers
1560
1561
1561 return get_default_reviewers_data, validate_default_reviewers
1562 return get_default_reviewers_data, validate_default_reviewers
1562
1563
1563
1564
1564 class MergeCheck(object):
1565 class MergeCheck(object):
1565 """
1566 """
1566 Perform Merge Checks and returns a check object which stores information
1567 Perform Merge Checks and returns a check object which stores information
1567 about merge errors, and merge conditions
1568 about merge errors, and merge conditions
1568 """
1569 """
1569 TODO_CHECK = 'todo'
1570 TODO_CHECK = 'todo'
1570 PERM_CHECK = 'perm'
1571 PERM_CHECK = 'perm'
1571 REVIEW_CHECK = 'review'
1572 REVIEW_CHECK = 'review'
1572 MERGE_CHECK = 'merge'
1573 MERGE_CHECK = 'merge'
1573
1574
1574 def __init__(self):
1575 def __init__(self):
1575 self.review_status = None
1576 self.review_status = None
1576 self.merge_possible = None
1577 self.merge_possible = None
1577 self.merge_msg = ''
1578 self.merge_msg = ''
1578 self.failed = None
1579 self.failed = None
1579 self.errors = []
1580 self.errors = []
1580 self.error_details = OrderedDict()
1581 self.error_details = OrderedDict()
1581
1582
1582 def push_error(self, error_type, message, error_key, details):
1583 def push_error(self, error_type, message, error_key, details):
1583 self.failed = True
1584 self.failed = True
1584 self.errors.append([error_type, message])
1585 self.errors.append([error_type, message])
1585 self.error_details[error_key] = dict(
1586 self.error_details[error_key] = dict(
1586 details=details,
1587 details=details,
1587 error_type=error_type,
1588 error_type=error_type,
1588 message=message
1589 message=message
1589 )
1590 )
1590
1591
1591 @classmethod
1592 @classmethod
1592 def validate(cls, pull_request, auth_user, translator, fail_early=False,
1593 def validate(cls, pull_request, auth_user, translator, fail_early=False,
1593 force_shadow_repo_refresh=False):
1594 force_shadow_repo_refresh=False):
1594 _ = translator
1595 _ = translator
1595 merge_check = cls()
1596 merge_check = cls()
1596
1597
1597 # permissions to merge
1598 # permissions to merge
1598 user_allowed_to_merge = PullRequestModel().check_user_merge(
1599 user_allowed_to_merge = PullRequestModel().check_user_merge(
1599 pull_request, auth_user)
1600 pull_request, auth_user)
1600 if not user_allowed_to_merge:
1601 if not user_allowed_to_merge:
1601 log.debug("MergeCheck: cannot merge, approval is pending.")
1602 log.debug("MergeCheck: cannot merge, approval is pending.")
1602
1603
1603 msg = _('User `{}` not allowed to perform merge.').format(auth_user.username)
1604 msg = _('User `{}` not allowed to perform merge.').format(auth_user.username)
1604 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1605 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1605 if fail_early:
1606 if fail_early:
1606 return merge_check
1607 return merge_check
1607
1608
1608 # permission to merge into the target branch
1609 # permission to merge into the target branch
1609 target_commit_id = pull_request.target_ref_parts.commit_id
1610 target_commit_id = pull_request.target_ref_parts.commit_id
1610 if pull_request.target_ref_parts.type == 'branch':
1611 if pull_request.target_ref_parts.type == 'branch':
1611 branch_name = pull_request.target_ref_parts.name
1612 branch_name = pull_request.target_ref_parts.name
1612 else:
1613 else:
1613 # for mercurial we can always figure out the branch from the commit
1614 # for mercurial we can always figure out the branch from the commit
1614 # in case of bookmark
1615 # in case of bookmark
1615 target_commit = pull_request.target_repo.get_commit(target_commit_id)
1616 target_commit = pull_request.target_repo.get_commit(target_commit_id)
1616 branch_name = target_commit.branch
1617 branch_name = target_commit.branch
1617
1618
1618 rule, branch_perm = auth_user.get_rule_and_branch_permission(
1619 rule, branch_perm = auth_user.get_rule_and_branch_permission(
1619 pull_request.target_repo.repo_name, branch_name)
1620 pull_request.target_repo.repo_name, branch_name)
1620 if branch_perm and branch_perm == 'branch.none':
1621 if branch_perm and branch_perm == 'branch.none':
1621 msg = _('Target branch `{}` changes rejected by rule {}.').format(
1622 msg = _('Target branch `{}` changes rejected by rule {}.').format(
1622 branch_name, rule)
1623 branch_name, rule)
1623 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1624 merge_check.push_error('error', msg, cls.PERM_CHECK, auth_user.username)
1624 if fail_early:
1625 if fail_early:
1625 return merge_check
1626 return merge_check
1626
1627
1627 # review status, must be always present
1628 # review status, must be always present
1628 review_status = pull_request.calculated_review_status()
1629 review_status = pull_request.calculated_review_status()
1629 merge_check.review_status = review_status
1630 merge_check.review_status = review_status
1630
1631
1631 status_approved = review_status == ChangesetStatus.STATUS_APPROVED
1632 status_approved = review_status == ChangesetStatus.STATUS_APPROVED
1632 if not status_approved:
1633 if not status_approved:
1633 log.debug("MergeCheck: cannot merge, approval is pending.")
1634 log.debug("MergeCheck: cannot merge, approval is pending.")
1634
1635
1635 msg = _('Pull request reviewer approval is pending.')
1636 msg = _('Pull request reviewer approval is pending.')
1636
1637
1637 merge_check.push_error('warning', msg, cls.REVIEW_CHECK, review_status)
1638 merge_check.push_error('warning', msg, cls.REVIEW_CHECK, review_status)
1638
1639
1639 if fail_early:
1640 if fail_early:
1640 return merge_check
1641 return merge_check
1641
1642
1642 # left over TODOs
1643 # left over TODOs
1643 todos = CommentsModel().get_pull_request_unresolved_todos(pull_request)
1644 todos = CommentsModel().get_pull_request_unresolved_todos(pull_request)
1644 if todos:
1645 if todos:
1645 log.debug("MergeCheck: cannot merge, {} "
1646 log.debug("MergeCheck: cannot merge, {} "
1646 "unresolved TODOs left.".format(len(todos)))
1647 "unresolved TODOs left.".format(len(todos)))
1647
1648
1648 if len(todos) == 1:
1649 if len(todos) == 1:
1649 msg = _('Cannot merge, {} TODO still not resolved.').format(
1650 msg = _('Cannot merge, {} TODO still not resolved.').format(
1650 len(todos))
1651 len(todos))
1651 else:
1652 else:
1652 msg = _('Cannot merge, {} TODOs still not resolved.').format(
1653 msg = _('Cannot merge, {} TODOs still not resolved.').format(
1653 len(todos))
1654 len(todos))
1654
1655
1655 merge_check.push_error('warning', msg, cls.TODO_CHECK, todos)
1656 merge_check.push_error('warning', msg, cls.TODO_CHECK, todos)
1656
1657
1657 if fail_early:
1658 if fail_early:
1658 return merge_check
1659 return merge_check
1659
1660
1660 # merge possible, here is the filesystem simulation + shadow repo
1661 # merge possible, here is the filesystem simulation + shadow repo
1661 merge_status, msg = PullRequestModel().merge_status(
1662 merge_status, msg = PullRequestModel().merge_status(
1662 pull_request, translator=translator,
1663 pull_request, translator=translator,
1663 force_shadow_repo_refresh=force_shadow_repo_refresh)
1664 force_shadow_repo_refresh=force_shadow_repo_refresh)
1664 merge_check.merge_possible = merge_status
1665 merge_check.merge_possible = merge_status
1665 merge_check.merge_msg = msg
1666 merge_check.merge_msg = msg
1666 if not merge_status:
1667 if not merge_status:
1667 log.debug("MergeCheck: cannot merge, pull request merge not possible.")
1668 log.debug("MergeCheck: cannot merge, pull request merge not possible.")
1668 merge_check.push_error('warning', msg, cls.MERGE_CHECK, None)
1669 merge_check.push_error('warning', msg, cls.MERGE_CHECK, None)
1669
1670
1670 if fail_early:
1671 if fail_early:
1671 return merge_check
1672 return merge_check
1672
1673
1673 log.debug('MergeCheck: is failed: %s', merge_check.failed)
1674 log.debug('MergeCheck: is failed: %s', merge_check.failed)
1674 return merge_check
1675 return merge_check
1675
1676
1676 @classmethod
1677 @classmethod
1677 def get_merge_conditions(cls, pull_request, translator):
1678 def get_merge_conditions(cls, pull_request, translator):
1678 _ = translator
1679 _ = translator
1679 merge_details = {}
1680 merge_details = {}
1680
1681
1681 model = PullRequestModel()
1682 model = PullRequestModel()
1682 use_rebase = model._use_rebase_for_merging(pull_request)
1683 use_rebase = model._use_rebase_for_merging(pull_request)
1683
1684
1684 if use_rebase:
1685 if use_rebase:
1685 merge_details['merge_strategy'] = dict(
1686 merge_details['merge_strategy'] = dict(
1686 details={},
1687 details={},
1687 message=_('Merge strategy: rebase')
1688 message=_('Merge strategy: rebase')
1688 )
1689 )
1689 else:
1690 else:
1690 merge_details['merge_strategy'] = dict(
1691 merge_details['merge_strategy'] = dict(
1691 details={},
1692 details={},
1692 message=_('Merge strategy: explicit merge commit')
1693 message=_('Merge strategy: explicit merge commit')
1693 )
1694 )
1694
1695
1695 close_branch = model._close_branch_before_merging(pull_request)
1696 close_branch = model._close_branch_before_merging(pull_request)
1696 if close_branch:
1697 if close_branch:
1697 repo_type = pull_request.target_repo.repo_type
1698 repo_type = pull_request.target_repo.repo_type
1698 close_msg = ''
1699 close_msg = ''
1699 if repo_type == 'hg':
1700 if repo_type == 'hg':
1700 close_msg = _('Source branch will be closed after merge.')
1701 close_msg = _('Source branch will be closed after merge.')
1701 elif repo_type == 'git':
1702 elif repo_type == 'git':
1702 close_msg = _('Source branch will be deleted after merge.')
1703 close_msg = _('Source branch will be deleted after merge.')
1703
1704
1704 merge_details['close_branch'] = dict(
1705 merge_details['close_branch'] = dict(
1705 details={},
1706 details={},
1706 message=close_msg
1707 message=close_msg
1707 )
1708 )
1708
1709
1709 return merge_details
1710 return merge_details
1710
1711
1711
1712
1712 ChangeTuple = collections.namedtuple(
1713 ChangeTuple = collections.namedtuple(
1713 'ChangeTuple', ['added', 'common', 'removed', 'total'])
1714 'ChangeTuple', ['added', 'common', 'removed', 'total'])
1714
1715
1715 FileChangeTuple = collections.namedtuple(
1716 FileChangeTuple = collections.namedtuple(
1716 'FileChangeTuple', ['added', 'modified', 'removed'])
1717 'FileChangeTuple', ['added', 'modified', 'removed'])
General Comments 0
You need to be logged in to leave comments. Login now