##// END OF EJS Templates
merge-checks: added more detailed information about why merge is not possible....
marcink -
r1341:611f8936 default
parent child Browse files
Show More
@@ -1,1398 +1,1409 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2012-2017 RhodeCode GmbH
3 # Copyright (C) 2012-2017 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 from collections import namedtuple
26 from collections import namedtuple
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
31
32 from pylons.i18n.translation import _
32 from pylons.i18n.translation import _
33 from pylons.i18n.translation import lazy_ugettext
33 from pylons.i18n.translation import lazy_ugettext
34 from sqlalchemy import or_
34 from sqlalchemy import or_
35
35
36 from rhodecode.lib import helpers as h, hooks_utils, diffs
36 from rhodecode.lib import helpers as h, hooks_utils, diffs
37 from rhodecode.lib.compat import OrderedDict
37 from rhodecode.lib.compat import OrderedDict
38 from rhodecode.lib.hooks_daemon import prepare_callback_daemon
38 from rhodecode.lib.hooks_daemon import prepare_callback_daemon
39 from rhodecode.lib.markup_renderer import (
39 from rhodecode.lib.markup_renderer import (
40 DEFAULT_COMMENTS_RENDERER, RstTemplateRenderer)
40 DEFAULT_COMMENTS_RENDERER, RstTemplateRenderer)
41 from rhodecode.lib.utils import action_logger
41 from rhodecode.lib.utils import action_logger
42 from rhodecode.lib.utils2 import safe_unicode, safe_str, md5_safe
42 from rhodecode.lib.utils2 import safe_unicode, safe_str, md5_safe
43 from rhodecode.lib.vcs.backends.base import (
43 from rhodecode.lib.vcs.backends.base import (
44 Reference, MergeResponse, MergeFailureReason, UpdateFailureReason)
44 Reference, MergeResponse, MergeFailureReason, UpdateFailureReason)
45 from rhodecode.lib.vcs.conf import settings as vcs_settings
45 from rhodecode.lib.vcs.conf import settings as vcs_settings
46 from rhodecode.lib.vcs.exceptions import (
46 from rhodecode.lib.vcs.exceptions import (
47 CommitDoesNotExistError, EmptyRepositoryError)
47 CommitDoesNotExistError, EmptyRepositoryError)
48 from rhodecode.model import BaseModel
48 from rhodecode.model import BaseModel
49 from rhodecode.model.changeset_status import ChangesetStatusModel
49 from rhodecode.model.changeset_status import ChangesetStatusModel
50 from rhodecode.model.comment import CommentsModel
50 from rhodecode.model.comment import CommentsModel
51 from rhodecode.model.db import (
51 from rhodecode.model.db import (
52 PullRequest, PullRequestReviewers, ChangesetStatus,
52 PullRequest, PullRequestReviewers, ChangesetStatus,
53 PullRequestVersion, ChangesetComment)
53 PullRequestVersion, ChangesetComment)
54 from rhodecode.model.meta import Session
54 from rhodecode.model.meta import Session
55 from rhodecode.model.notification import NotificationModel, \
55 from rhodecode.model.notification import NotificationModel, \
56 EmailNotificationModel
56 EmailNotificationModel
57 from rhodecode.model.scm import ScmModel
57 from rhodecode.model.scm import ScmModel
58 from rhodecode.model.settings import VcsSettingsModel
58 from rhodecode.model.settings import VcsSettingsModel
59
59
60
60
61 log = logging.getLogger(__name__)
61 log = logging.getLogger(__name__)
62
62
63
63
64 # Data structure to hold the response data when updating commits during a pull
64 # Data structure to hold the response data when updating commits during a pull
65 # request update.
65 # request update.
66 UpdateResponse = namedtuple(
66 UpdateResponse = namedtuple(
67 'UpdateResponse', 'executed, reason, new, old, changes')
67 'UpdateResponse', 'executed, reason, new, old, changes')
68
68
69
69
70 class PullRequestModel(BaseModel):
70 class PullRequestModel(BaseModel):
71
71
72 cls = PullRequest
72 cls = PullRequest
73
73
74 DIFF_CONTEXT = 3
74 DIFF_CONTEXT = 3
75
75
76 MERGE_STATUS_MESSAGES = {
76 MERGE_STATUS_MESSAGES = {
77 MergeFailureReason.NONE: lazy_ugettext(
77 MergeFailureReason.NONE: lazy_ugettext(
78 'This pull request can be automatically merged.'),
78 'This pull request can be automatically merged.'),
79 MergeFailureReason.UNKNOWN: lazy_ugettext(
79 MergeFailureReason.UNKNOWN: lazy_ugettext(
80 'This pull request cannot be merged because of an unhandled'
80 'This pull request cannot be merged because of an unhandled'
81 ' exception.'),
81 ' exception.'),
82 MergeFailureReason.MERGE_FAILED: lazy_ugettext(
82 MergeFailureReason.MERGE_FAILED: lazy_ugettext(
83 'This pull request cannot be merged because of conflicts.'),
83 'This pull request cannot be merged because of conflicts.'),
84 MergeFailureReason.PUSH_FAILED: lazy_ugettext(
84 MergeFailureReason.PUSH_FAILED: lazy_ugettext(
85 'This pull request could not be merged because push to target'
85 'This pull request could not be merged because push to target'
86 ' failed.'),
86 ' failed.'),
87 MergeFailureReason.TARGET_IS_NOT_HEAD: lazy_ugettext(
87 MergeFailureReason.TARGET_IS_NOT_HEAD: lazy_ugettext(
88 'This pull request cannot be merged because the target is not a'
88 'This pull request cannot be merged because the target is not a'
89 ' head.'),
89 ' head.'),
90 MergeFailureReason.HG_SOURCE_HAS_MORE_BRANCHES: lazy_ugettext(
90 MergeFailureReason.HG_SOURCE_HAS_MORE_BRANCHES: lazy_ugettext(
91 'This pull request cannot be merged because the source contains'
91 'This pull request cannot be merged because the source contains'
92 ' more branches than the target.'),
92 ' more branches than the target.'),
93 MergeFailureReason.HG_TARGET_HAS_MULTIPLE_HEADS: lazy_ugettext(
93 MergeFailureReason.HG_TARGET_HAS_MULTIPLE_HEADS: lazy_ugettext(
94 'This pull request cannot be merged because the target has'
94 'This pull request cannot be merged because the target has'
95 ' multiple heads.'),
95 ' multiple heads.'),
96 MergeFailureReason.TARGET_IS_LOCKED: lazy_ugettext(
96 MergeFailureReason.TARGET_IS_LOCKED: lazy_ugettext(
97 'This pull request cannot be merged because the target repository'
97 'This pull request cannot be merged because the target repository'
98 ' is locked.'),
98 ' is locked.'),
99 MergeFailureReason._DEPRECATED_MISSING_COMMIT: lazy_ugettext(
99 MergeFailureReason._DEPRECATED_MISSING_COMMIT: lazy_ugettext(
100 'This pull request cannot be merged because the target or the '
100 'This pull request cannot be merged because the target or the '
101 'source reference is missing.'),
101 'source reference is missing.'),
102 MergeFailureReason.MISSING_TARGET_REF: lazy_ugettext(
102 MergeFailureReason.MISSING_TARGET_REF: lazy_ugettext(
103 'This pull request cannot be merged because the target '
103 'This pull request cannot be merged because the target '
104 'reference is missing.'),
104 'reference is missing.'),
105 MergeFailureReason.MISSING_SOURCE_REF: lazy_ugettext(
105 MergeFailureReason.MISSING_SOURCE_REF: lazy_ugettext(
106 'This pull request cannot be merged because the source '
106 'This pull request cannot be merged because the source '
107 'reference is missing.'),
107 'reference is missing.'),
108 MergeFailureReason.SUBREPO_MERGE_FAILED: lazy_ugettext(
108 MergeFailureReason.SUBREPO_MERGE_FAILED: lazy_ugettext(
109 'This pull request cannot be merged because of conflicts related '
109 'This pull request cannot be merged because of conflicts related '
110 'to sub repositories.'),
110 'to sub repositories.'),
111 }
111 }
112
112
113 UPDATE_STATUS_MESSAGES = {
113 UPDATE_STATUS_MESSAGES = {
114 UpdateFailureReason.NONE: lazy_ugettext(
114 UpdateFailureReason.NONE: lazy_ugettext(
115 'Pull request update successful.'),
115 'Pull request update successful.'),
116 UpdateFailureReason.UNKNOWN: lazy_ugettext(
116 UpdateFailureReason.UNKNOWN: lazy_ugettext(
117 'Pull request update failed because of an unknown error.'),
117 'Pull request update failed because of an unknown error.'),
118 UpdateFailureReason.NO_CHANGE: lazy_ugettext(
118 UpdateFailureReason.NO_CHANGE: lazy_ugettext(
119 'No update needed because the source reference is already '
119 'No update needed because the source reference is already '
120 'up to date.'),
120 'up to date.'),
121 UpdateFailureReason.WRONG_REF_TPYE: lazy_ugettext(
121 UpdateFailureReason.WRONG_REF_TPYE: lazy_ugettext(
122 'Pull request cannot be updated because the reference type is '
122 'Pull request cannot be updated because the reference type is '
123 'not supported for an update.'),
123 'not supported for an update.'),
124 UpdateFailureReason.MISSING_TARGET_REF: lazy_ugettext(
124 UpdateFailureReason.MISSING_TARGET_REF: lazy_ugettext(
125 'This pull request cannot be updated because the target '
125 'This pull request cannot be updated because the target '
126 'reference is missing.'),
126 'reference is missing.'),
127 UpdateFailureReason.MISSING_SOURCE_REF: lazy_ugettext(
127 UpdateFailureReason.MISSING_SOURCE_REF: lazy_ugettext(
128 'This pull request cannot be updated because the source '
128 'This pull request cannot be updated because the source '
129 'reference is missing.'),
129 'reference is missing.'),
130 }
130 }
131
131
132 def __get_pull_request(self, pull_request):
132 def __get_pull_request(self, pull_request):
133 return self._get_instance((
133 return self._get_instance((
134 PullRequest, PullRequestVersion), pull_request)
134 PullRequest, PullRequestVersion), pull_request)
135
135
136 def _check_perms(self, perms, pull_request, user, api=False):
136 def _check_perms(self, perms, pull_request, user, api=False):
137 if not api:
137 if not api:
138 return h.HasRepoPermissionAny(*perms)(
138 return h.HasRepoPermissionAny(*perms)(
139 user=user, repo_name=pull_request.target_repo.repo_name)
139 user=user, repo_name=pull_request.target_repo.repo_name)
140 else:
140 else:
141 return h.HasRepoPermissionAnyApi(*perms)(
141 return h.HasRepoPermissionAnyApi(*perms)(
142 user=user, repo_name=pull_request.target_repo.repo_name)
142 user=user, repo_name=pull_request.target_repo.repo_name)
143
143
144 def check_user_read(self, pull_request, user, api=False):
144 def check_user_read(self, pull_request, user, api=False):
145 _perms = ('repository.admin', 'repository.write', 'repository.read',)
145 _perms = ('repository.admin', 'repository.write', 'repository.read',)
146 return self._check_perms(_perms, pull_request, user, api)
146 return self._check_perms(_perms, pull_request, user, api)
147
147
148 def check_user_merge(self, pull_request, user, api=False):
148 def check_user_merge(self, pull_request, user, api=False):
149 _perms = ('repository.admin', 'repository.write', 'hg.admin',)
149 _perms = ('repository.admin', 'repository.write', 'hg.admin',)
150 return self._check_perms(_perms, pull_request, user, api)
150 return self._check_perms(_perms, pull_request, user, api)
151
151
152 def check_user_update(self, pull_request, user, api=False):
152 def check_user_update(self, pull_request, user, api=False):
153 owner = user.user_id == pull_request.user_id
153 owner = user.user_id == pull_request.user_id
154 return self.check_user_merge(pull_request, user, api) or owner
154 return self.check_user_merge(pull_request, user, api) or owner
155
155
156 def check_user_delete(self, pull_request, user):
156 def check_user_delete(self, pull_request, user):
157 owner = user.user_id == pull_request.user_id
157 owner = user.user_id == pull_request.user_id
158 _perms = ('repository.admin')
158 _perms = ('repository.admin')
159 return self._check_perms(_perms, pull_request, user) or owner
159 return self._check_perms(_perms, pull_request, user) or owner
160
160
161 def check_user_change_status(self, pull_request, user, api=False):
161 def check_user_change_status(self, pull_request, user, api=False):
162 reviewer = user.user_id in [x.user_id for x in
162 reviewer = user.user_id in [x.user_id for x in
163 pull_request.reviewers]
163 pull_request.reviewers]
164 return self.check_user_update(pull_request, user, api) or reviewer
164 return self.check_user_update(pull_request, user, api) or reviewer
165
165
166 def get(self, pull_request):
166 def get(self, pull_request):
167 return self.__get_pull_request(pull_request)
167 return self.__get_pull_request(pull_request)
168
168
169 def _prepare_get_all_query(self, repo_name, source=False, statuses=None,
169 def _prepare_get_all_query(self, repo_name, source=False, statuses=None,
170 opened_by=None, order_by=None,
170 opened_by=None, order_by=None,
171 order_dir='desc'):
171 order_dir='desc'):
172 repo = None
172 repo = None
173 if repo_name:
173 if repo_name:
174 repo = self._get_repo(repo_name)
174 repo = self._get_repo(repo_name)
175
175
176 q = PullRequest.query()
176 q = PullRequest.query()
177
177
178 # source or target
178 # source or target
179 if repo and source:
179 if repo and source:
180 q = q.filter(PullRequest.source_repo == repo)
180 q = q.filter(PullRequest.source_repo == repo)
181 elif repo:
181 elif repo:
182 q = q.filter(PullRequest.target_repo == repo)
182 q = q.filter(PullRequest.target_repo == repo)
183
183
184 # closed,opened
184 # closed,opened
185 if statuses:
185 if statuses:
186 q = q.filter(PullRequest.status.in_(statuses))
186 q = q.filter(PullRequest.status.in_(statuses))
187
187
188 # opened by filter
188 # opened by filter
189 if opened_by:
189 if opened_by:
190 q = q.filter(PullRequest.user_id.in_(opened_by))
190 q = q.filter(PullRequest.user_id.in_(opened_by))
191
191
192 if order_by:
192 if order_by:
193 order_map = {
193 order_map = {
194 'name_raw': PullRequest.pull_request_id,
194 'name_raw': PullRequest.pull_request_id,
195 'title': PullRequest.title,
195 'title': PullRequest.title,
196 'updated_on_raw': PullRequest.updated_on,
196 'updated_on_raw': PullRequest.updated_on,
197 'target_repo': PullRequest.target_repo_id
197 'target_repo': PullRequest.target_repo_id
198 }
198 }
199 if order_dir == 'asc':
199 if order_dir == 'asc':
200 q = q.order_by(order_map[order_by].asc())
200 q = q.order_by(order_map[order_by].asc())
201 else:
201 else:
202 q = q.order_by(order_map[order_by].desc())
202 q = q.order_by(order_map[order_by].desc())
203
203
204 return q
204 return q
205
205
206 def count_all(self, repo_name, source=False, statuses=None,
206 def count_all(self, repo_name, source=False, statuses=None,
207 opened_by=None):
207 opened_by=None):
208 """
208 """
209 Count the number of pull requests for a specific repository.
209 Count the number of pull requests for a specific repository.
210
210
211 :param repo_name: target or source repo
211 :param repo_name: target or source repo
212 :param source: boolean flag to specify if repo_name refers to source
212 :param source: boolean flag to specify if repo_name refers to source
213 :param statuses: list of pull request statuses
213 :param statuses: list of pull request statuses
214 :param opened_by: author user of the pull request
214 :param opened_by: author user of the pull request
215 :returns: int number of pull requests
215 :returns: int number of pull requests
216 """
216 """
217 q = self._prepare_get_all_query(
217 q = self._prepare_get_all_query(
218 repo_name, source=source, statuses=statuses, opened_by=opened_by)
218 repo_name, source=source, statuses=statuses, opened_by=opened_by)
219
219
220 return q.count()
220 return q.count()
221
221
222 def get_all(self, repo_name, source=False, statuses=None, opened_by=None,
222 def get_all(self, repo_name, source=False, statuses=None, opened_by=None,
223 offset=0, length=None, order_by=None, order_dir='desc'):
223 offset=0, length=None, order_by=None, order_dir='desc'):
224 """
224 """
225 Get all pull requests for a specific repository.
225 Get all pull requests for a specific repository.
226
226
227 :param repo_name: target or source repo
227 :param repo_name: target or source repo
228 :param source: boolean flag to specify if repo_name refers to source
228 :param source: boolean flag to specify if repo_name refers to source
229 :param statuses: list of pull request statuses
229 :param statuses: list of pull request statuses
230 :param opened_by: author user of the pull request
230 :param opened_by: author user of the pull request
231 :param offset: pagination offset
231 :param offset: pagination offset
232 :param length: length of returned list
232 :param length: length of returned list
233 :param order_by: order of the returned list
233 :param order_by: order of the returned list
234 :param order_dir: 'asc' or 'desc' ordering direction
234 :param order_dir: 'asc' or 'desc' ordering direction
235 :returns: list of pull requests
235 :returns: list of pull requests
236 """
236 """
237 q = self._prepare_get_all_query(
237 q = self._prepare_get_all_query(
238 repo_name, source=source, statuses=statuses, opened_by=opened_by,
238 repo_name, source=source, statuses=statuses, opened_by=opened_by,
239 order_by=order_by, order_dir=order_dir)
239 order_by=order_by, order_dir=order_dir)
240
240
241 if length:
241 if length:
242 pull_requests = q.limit(length).offset(offset).all()
242 pull_requests = q.limit(length).offset(offset).all()
243 else:
243 else:
244 pull_requests = q.all()
244 pull_requests = q.all()
245
245
246 return pull_requests
246 return pull_requests
247
247
248 def count_awaiting_review(self, repo_name, source=False, statuses=None,
248 def count_awaiting_review(self, repo_name, source=False, statuses=None,
249 opened_by=None):
249 opened_by=None):
250 """
250 """
251 Count the number of pull requests for a specific repository that are
251 Count the number of pull requests for a specific repository that are
252 awaiting review.
252 awaiting review.
253
253
254 :param repo_name: target or source repo
254 :param repo_name: target or source repo
255 :param source: boolean flag to specify if repo_name refers to source
255 :param source: boolean flag to specify if repo_name refers to source
256 :param statuses: list of pull request statuses
256 :param statuses: list of pull request statuses
257 :param opened_by: author user of the pull request
257 :param opened_by: author user of the pull request
258 :returns: int number of pull requests
258 :returns: int number of pull requests
259 """
259 """
260 pull_requests = self.get_awaiting_review(
260 pull_requests = self.get_awaiting_review(
261 repo_name, source=source, statuses=statuses, opened_by=opened_by)
261 repo_name, source=source, statuses=statuses, opened_by=opened_by)
262
262
263 return len(pull_requests)
263 return len(pull_requests)
264
264
265 def get_awaiting_review(self, repo_name, source=False, statuses=None,
265 def get_awaiting_review(self, repo_name, source=False, statuses=None,
266 opened_by=None, offset=0, length=None,
266 opened_by=None, offset=0, length=None,
267 order_by=None, order_dir='desc'):
267 order_by=None, order_dir='desc'):
268 """
268 """
269 Get all pull requests for a specific repository that are awaiting
269 Get all pull requests for a specific repository that are awaiting
270 review.
270 review.
271
271
272 :param repo_name: target or source repo
272 :param repo_name: target or source repo
273 :param source: boolean flag to specify if repo_name refers to source
273 :param source: boolean flag to specify if repo_name refers to source
274 :param statuses: list of pull request statuses
274 :param statuses: list of pull request statuses
275 :param opened_by: author user of the pull request
275 :param opened_by: author user of the pull request
276 :param offset: pagination offset
276 :param offset: pagination offset
277 :param length: length of returned list
277 :param length: length of returned list
278 :param order_by: order of the returned list
278 :param order_by: order of the returned list
279 :param order_dir: 'asc' or 'desc' ordering direction
279 :param order_dir: 'asc' or 'desc' ordering direction
280 :returns: list of pull requests
280 :returns: list of pull requests
281 """
281 """
282 pull_requests = self.get_all(
282 pull_requests = self.get_all(
283 repo_name, source=source, statuses=statuses, opened_by=opened_by,
283 repo_name, source=source, statuses=statuses, opened_by=opened_by,
284 order_by=order_by, order_dir=order_dir)
284 order_by=order_by, order_dir=order_dir)
285
285
286 _filtered_pull_requests = []
286 _filtered_pull_requests = []
287 for pr in pull_requests:
287 for pr in pull_requests:
288 status = pr.calculated_review_status()
288 status = pr.calculated_review_status()
289 if status in [ChangesetStatus.STATUS_NOT_REVIEWED,
289 if status in [ChangesetStatus.STATUS_NOT_REVIEWED,
290 ChangesetStatus.STATUS_UNDER_REVIEW]:
290 ChangesetStatus.STATUS_UNDER_REVIEW]:
291 _filtered_pull_requests.append(pr)
291 _filtered_pull_requests.append(pr)
292 if length:
292 if length:
293 return _filtered_pull_requests[offset:offset+length]
293 return _filtered_pull_requests[offset:offset+length]
294 else:
294 else:
295 return _filtered_pull_requests
295 return _filtered_pull_requests
296
296
297 def count_awaiting_my_review(self, repo_name, source=False, statuses=None,
297 def count_awaiting_my_review(self, repo_name, source=False, statuses=None,
298 opened_by=None, user_id=None):
298 opened_by=None, user_id=None):
299 """
299 """
300 Count the number of pull requests for a specific repository that are
300 Count the number of pull requests for a specific repository that are
301 awaiting review from a specific user.
301 awaiting review from a specific user.
302
302
303 :param repo_name: target or source repo
303 :param repo_name: target or source repo
304 :param source: boolean flag to specify if repo_name refers to source
304 :param source: boolean flag to specify if repo_name refers to source
305 :param statuses: list of pull request statuses
305 :param statuses: list of pull request statuses
306 :param opened_by: author user of the pull request
306 :param opened_by: author user of the pull request
307 :param user_id: reviewer user of the pull request
307 :param user_id: reviewer user of the pull request
308 :returns: int number of pull requests
308 :returns: int number of pull requests
309 """
309 """
310 pull_requests = self.get_awaiting_my_review(
310 pull_requests = self.get_awaiting_my_review(
311 repo_name, source=source, statuses=statuses, opened_by=opened_by,
311 repo_name, source=source, statuses=statuses, opened_by=opened_by,
312 user_id=user_id)
312 user_id=user_id)
313
313
314 return len(pull_requests)
314 return len(pull_requests)
315
315
316 def get_awaiting_my_review(self, repo_name, source=False, statuses=None,
316 def get_awaiting_my_review(self, repo_name, source=False, statuses=None,
317 opened_by=None, user_id=None, offset=0,
317 opened_by=None, user_id=None, offset=0,
318 length=None, order_by=None, order_dir='desc'):
318 length=None, order_by=None, order_dir='desc'):
319 """
319 """
320 Get all pull requests for a specific repository that are awaiting
320 Get all pull requests for a specific repository that are awaiting
321 review from a specific user.
321 review from a specific user.
322
322
323 :param repo_name: target or source repo
323 :param repo_name: target or source repo
324 :param source: boolean flag to specify if repo_name refers to source
324 :param source: boolean flag to specify if repo_name refers to source
325 :param statuses: list of pull request statuses
325 :param statuses: list of pull request statuses
326 :param opened_by: author user of the pull request
326 :param opened_by: author user of the pull request
327 :param user_id: reviewer user of the pull request
327 :param user_id: reviewer user of the pull request
328 :param offset: pagination offset
328 :param offset: pagination offset
329 :param length: length of returned list
329 :param length: length of returned list
330 :param order_by: order of the returned list
330 :param order_by: order of the returned list
331 :param order_dir: 'asc' or 'desc' ordering direction
331 :param order_dir: 'asc' or 'desc' ordering direction
332 :returns: list of pull requests
332 :returns: list of pull requests
333 """
333 """
334 pull_requests = self.get_all(
334 pull_requests = self.get_all(
335 repo_name, source=source, statuses=statuses, opened_by=opened_by,
335 repo_name, source=source, statuses=statuses, opened_by=opened_by,
336 order_by=order_by, order_dir=order_dir)
336 order_by=order_by, order_dir=order_dir)
337
337
338 _my = PullRequestModel().get_not_reviewed(user_id)
338 _my = PullRequestModel().get_not_reviewed(user_id)
339 my_participation = []
339 my_participation = []
340 for pr in pull_requests:
340 for pr in pull_requests:
341 if pr in _my:
341 if pr in _my:
342 my_participation.append(pr)
342 my_participation.append(pr)
343 _filtered_pull_requests = my_participation
343 _filtered_pull_requests = my_participation
344 if length:
344 if length:
345 return _filtered_pull_requests[offset:offset+length]
345 return _filtered_pull_requests[offset:offset+length]
346 else:
346 else:
347 return _filtered_pull_requests
347 return _filtered_pull_requests
348
348
349 def get_not_reviewed(self, user_id):
349 def get_not_reviewed(self, user_id):
350 return [
350 return [
351 x.pull_request for x in PullRequestReviewers.query().filter(
351 x.pull_request for x in PullRequestReviewers.query().filter(
352 PullRequestReviewers.user_id == user_id).all()
352 PullRequestReviewers.user_id == user_id).all()
353 ]
353 ]
354
354
355 def _prepare_participating_query(self, user_id=None, statuses=None,
355 def _prepare_participating_query(self, user_id=None, statuses=None,
356 order_by=None, order_dir='desc'):
356 order_by=None, order_dir='desc'):
357 q = PullRequest.query()
357 q = PullRequest.query()
358 if user_id:
358 if user_id:
359 reviewers_subquery = Session().query(
359 reviewers_subquery = Session().query(
360 PullRequestReviewers.pull_request_id).filter(
360 PullRequestReviewers.pull_request_id).filter(
361 PullRequestReviewers.user_id == user_id).subquery()
361 PullRequestReviewers.user_id == user_id).subquery()
362 user_filter= or_(
362 user_filter= or_(
363 PullRequest.user_id == user_id,
363 PullRequest.user_id == user_id,
364 PullRequest.pull_request_id.in_(reviewers_subquery)
364 PullRequest.pull_request_id.in_(reviewers_subquery)
365 )
365 )
366 q = PullRequest.query().filter(user_filter)
366 q = PullRequest.query().filter(user_filter)
367
367
368 # closed,opened
368 # closed,opened
369 if statuses:
369 if statuses:
370 q = q.filter(PullRequest.status.in_(statuses))
370 q = q.filter(PullRequest.status.in_(statuses))
371
371
372 if order_by:
372 if order_by:
373 order_map = {
373 order_map = {
374 'name_raw': PullRequest.pull_request_id,
374 'name_raw': PullRequest.pull_request_id,
375 'title': PullRequest.title,
375 'title': PullRequest.title,
376 'updated_on_raw': PullRequest.updated_on,
376 'updated_on_raw': PullRequest.updated_on,
377 'target_repo': PullRequest.target_repo_id
377 'target_repo': PullRequest.target_repo_id
378 }
378 }
379 if order_dir == 'asc':
379 if order_dir == 'asc':
380 q = q.order_by(order_map[order_by].asc())
380 q = q.order_by(order_map[order_by].asc())
381 else:
381 else:
382 q = q.order_by(order_map[order_by].desc())
382 q = q.order_by(order_map[order_by].desc())
383
383
384 return q
384 return q
385
385
386 def count_im_participating_in(self, user_id=None, statuses=None):
386 def count_im_participating_in(self, user_id=None, statuses=None):
387 q = self._prepare_participating_query(user_id, statuses=statuses)
387 q = self._prepare_participating_query(user_id, statuses=statuses)
388 return q.count()
388 return q.count()
389
389
390 def get_im_participating_in(
390 def get_im_participating_in(
391 self, user_id=None, statuses=None, offset=0,
391 self, user_id=None, statuses=None, offset=0,
392 length=None, order_by=None, order_dir='desc'):
392 length=None, order_by=None, order_dir='desc'):
393 """
393 """
394 Get all Pull requests that i'm participating in, or i have opened
394 Get all Pull requests that i'm participating in, or i have opened
395 """
395 """
396
396
397 q = self._prepare_participating_query(
397 q = self._prepare_participating_query(
398 user_id, statuses=statuses, order_by=order_by,
398 user_id, statuses=statuses, order_by=order_by,
399 order_dir=order_dir)
399 order_dir=order_dir)
400
400
401 if length:
401 if length:
402 pull_requests = q.limit(length).offset(offset).all()
402 pull_requests = q.limit(length).offset(offset).all()
403 else:
403 else:
404 pull_requests = q.all()
404 pull_requests = q.all()
405
405
406 return pull_requests
406 return pull_requests
407
407
408 def get_versions(self, pull_request):
408 def get_versions(self, pull_request):
409 """
409 """
410 returns version of pull request sorted by ID descending
410 returns version of pull request sorted by ID descending
411 """
411 """
412 return PullRequestVersion.query()\
412 return PullRequestVersion.query()\
413 .filter(PullRequestVersion.pull_request == pull_request)\
413 .filter(PullRequestVersion.pull_request == pull_request)\
414 .order_by(PullRequestVersion.pull_request_version_id.asc())\
414 .order_by(PullRequestVersion.pull_request_version_id.asc())\
415 .all()
415 .all()
416
416
417 def create(self, created_by, source_repo, source_ref, target_repo,
417 def create(self, created_by, source_repo, source_ref, target_repo,
418 target_ref, revisions, reviewers, title, description=None):
418 target_ref, revisions, reviewers, title, description=None):
419 created_by_user = self._get_user(created_by)
419 created_by_user = self._get_user(created_by)
420 source_repo = self._get_repo(source_repo)
420 source_repo = self._get_repo(source_repo)
421 target_repo = self._get_repo(target_repo)
421 target_repo = self._get_repo(target_repo)
422
422
423 pull_request = PullRequest()
423 pull_request = PullRequest()
424 pull_request.source_repo = source_repo
424 pull_request.source_repo = source_repo
425 pull_request.source_ref = source_ref
425 pull_request.source_ref = source_ref
426 pull_request.target_repo = target_repo
426 pull_request.target_repo = target_repo
427 pull_request.target_ref = target_ref
427 pull_request.target_ref = target_ref
428 pull_request.revisions = revisions
428 pull_request.revisions = revisions
429 pull_request.title = title
429 pull_request.title = title
430 pull_request.description = description
430 pull_request.description = description
431 pull_request.author = created_by_user
431 pull_request.author = created_by_user
432
432
433 Session().add(pull_request)
433 Session().add(pull_request)
434 Session().flush()
434 Session().flush()
435
435
436 reviewer_ids = set()
436 reviewer_ids = set()
437 # members / reviewers
437 # members / reviewers
438 for reviewer_object in reviewers:
438 for reviewer_object in reviewers:
439 if isinstance(reviewer_object, tuple):
439 if isinstance(reviewer_object, tuple):
440 user_id, reasons = reviewer_object
440 user_id, reasons = reviewer_object
441 else:
441 else:
442 user_id, reasons = reviewer_object, []
442 user_id, reasons = reviewer_object, []
443
443
444 user = self._get_user(user_id)
444 user = self._get_user(user_id)
445 reviewer_ids.add(user.user_id)
445 reviewer_ids.add(user.user_id)
446
446
447 reviewer = PullRequestReviewers(user, pull_request, reasons)
447 reviewer = PullRequestReviewers(user, pull_request, reasons)
448 Session().add(reviewer)
448 Session().add(reviewer)
449
449
450 # Set approval status to "Under Review" for all commits which are
450 # Set approval status to "Under Review" for all commits which are
451 # part of this pull request.
451 # part of this pull request.
452 ChangesetStatusModel().set_status(
452 ChangesetStatusModel().set_status(
453 repo=target_repo,
453 repo=target_repo,
454 status=ChangesetStatus.STATUS_UNDER_REVIEW,
454 status=ChangesetStatus.STATUS_UNDER_REVIEW,
455 user=created_by_user,
455 user=created_by_user,
456 pull_request=pull_request
456 pull_request=pull_request
457 )
457 )
458
458
459 self.notify_reviewers(pull_request, reviewer_ids)
459 self.notify_reviewers(pull_request, reviewer_ids)
460 self._trigger_pull_request_hook(
460 self._trigger_pull_request_hook(
461 pull_request, created_by_user, 'create')
461 pull_request, created_by_user, 'create')
462
462
463 return pull_request
463 return pull_request
464
464
465 def _trigger_pull_request_hook(self, pull_request, user, action):
465 def _trigger_pull_request_hook(self, pull_request, user, action):
466 pull_request = self.__get_pull_request(pull_request)
466 pull_request = self.__get_pull_request(pull_request)
467 target_scm = pull_request.target_repo.scm_instance()
467 target_scm = pull_request.target_repo.scm_instance()
468 if action == 'create':
468 if action == 'create':
469 trigger_hook = hooks_utils.trigger_log_create_pull_request_hook
469 trigger_hook = hooks_utils.trigger_log_create_pull_request_hook
470 elif action == 'merge':
470 elif action == 'merge':
471 trigger_hook = hooks_utils.trigger_log_merge_pull_request_hook
471 trigger_hook = hooks_utils.trigger_log_merge_pull_request_hook
472 elif action == 'close':
472 elif action == 'close':
473 trigger_hook = hooks_utils.trigger_log_close_pull_request_hook
473 trigger_hook = hooks_utils.trigger_log_close_pull_request_hook
474 elif action == 'review_status_change':
474 elif action == 'review_status_change':
475 trigger_hook = hooks_utils.trigger_log_review_pull_request_hook
475 trigger_hook = hooks_utils.trigger_log_review_pull_request_hook
476 elif action == 'update':
476 elif action == 'update':
477 trigger_hook = hooks_utils.trigger_log_update_pull_request_hook
477 trigger_hook = hooks_utils.trigger_log_update_pull_request_hook
478 else:
478 else:
479 return
479 return
480
480
481 trigger_hook(
481 trigger_hook(
482 username=user.username,
482 username=user.username,
483 repo_name=pull_request.target_repo.repo_name,
483 repo_name=pull_request.target_repo.repo_name,
484 repo_alias=target_scm.alias,
484 repo_alias=target_scm.alias,
485 pull_request=pull_request)
485 pull_request=pull_request)
486
486
487 def _get_commit_ids(self, pull_request):
487 def _get_commit_ids(self, pull_request):
488 """
488 """
489 Return the commit ids of the merged pull request.
489 Return the commit ids of the merged pull request.
490
490
491 This method is not dealing correctly yet with the lack of autoupdates
491 This method is not dealing correctly yet with the lack of autoupdates
492 nor with the implicit target updates.
492 nor with the implicit target updates.
493 For example: if a commit in the source repo is already in the target it
493 For example: if a commit in the source repo is already in the target it
494 will be reported anyways.
494 will be reported anyways.
495 """
495 """
496 merge_rev = pull_request.merge_rev
496 merge_rev = pull_request.merge_rev
497 if merge_rev is None:
497 if merge_rev is None:
498 raise ValueError('This pull request was not merged yet')
498 raise ValueError('This pull request was not merged yet')
499
499
500 commit_ids = list(pull_request.revisions)
500 commit_ids = list(pull_request.revisions)
501 if merge_rev not in commit_ids:
501 if merge_rev not in commit_ids:
502 commit_ids.append(merge_rev)
502 commit_ids.append(merge_rev)
503
503
504 return commit_ids
504 return commit_ids
505
505
506 def merge(self, pull_request, user, extras):
506 def merge(self, pull_request, user, extras):
507 log.debug("Merging pull request %s", pull_request.pull_request_id)
507 log.debug("Merging pull request %s", pull_request.pull_request_id)
508 merge_state = self._merge_pull_request(pull_request, user, extras)
508 merge_state = self._merge_pull_request(pull_request, user, extras)
509 if merge_state.executed:
509 if merge_state.executed:
510 log.debug(
510 log.debug(
511 "Merge was successful, updating the pull request comments.")
511 "Merge was successful, updating the pull request comments.")
512 self._comment_and_close_pr(pull_request, user, merge_state)
512 self._comment_and_close_pr(pull_request, user, merge_state)
513 self._log_action('user_merged_pull_request', user, pull_request)
513 self._log_action('user_merged_pull_request', user, pull_request)
514 else:
514 else:
515 log.warn("Merge failed, not updating the pull request.")
515 log.warn("Merge failed, not updating the pull request.")
516 return merge_state
516 return merge_state
517
517
518 def _merge_pull_request(self, pull_request, user, extras):
518 def _merge_pull_request(self, pull_request, user, extras):
519 target_vcs = pull_request.target_repo.scm_instance()
519 target_vcs = pull_request.target_repo.scm_instance()
520 source_vcs = pull_request.source_repo.scm_instance()
520 source_vcs = pull_request.source_repo.scm_instance()
521 target_ref = self._refresh_reference(
521 target_ref = self._refresh_reference(
522 pull_request.target_ref_parts, target_vcs)
522 pull_request.target_ref_parts, target_vcs)
523
523
524 message = _(
524 message = _(
525 'Merge pull request #%(pr_id)s from '
525 'Merge pull request #%(pr_id)s from '
526 '%(source_repo)s %(source_ref_name)s\n\n %(pr_title)s') % {
526 '%(source_repo)s %(source_ref_name)s\n\n %(pr_title)s') % {
527 'pr_id': pull_request.pull_request_id,
527 'pr_id': pull_request.pull_request_id,
528 'source_repo': source_vcs.name,
528 'source_repo': source_vcs.name,
529 'source_ref_name': pull_request.source_ref_parts.name,
529 'source_ref_name': pull_request.source_ref_parts.name,
530 'pr_title': pull_request.title
530 'pr_title': pull_request.title
531 }
531 }
532
532
533 workspace_id = self._workspace_id(pull_request)
533 workspace_id = self._workspace_id(pull_request)
534 use_rebase = self._use_rebase_for_merging(pull_request)
534 use_rebase = self._use_rebase_for_merging(pull_request)
535
535
536 callback_daemon, extras = prepare_callback_daemon(
536 callback_daemon, extras = prepare_callback_daemon(
537 extras, protocol=vcs_settings.HOOKS_PROTOCOL,
537 extras, protocol=vcs_settings.HOOKS_PROTOCOL,
538 use_direct_calls=vcs_settings.HOOKS_DIRECT_CALLS)
538 use_direct_calls=vcs_settings.HOOKS_DIRECT_CALLS)
539
539
540 with callback_daemon:
540 with callback_daemon:
541 # TODO: johbo: Implement a clean way to run a config_override
541 # TODO: johbo: Implement a clean way to run a config_override
542 # for a single call.
542 # for a single call.
543 target_vcs.config.set(
543 target_vcs.config.set(
544 'rhodecode', 'RC_SCM_DATA', json.dumps(extras))
544 'rhodecode', 'RC_SCM_DATA', json.dumps(extras))
545 merge_state = target_vcs.merge(
545 merge_state = target_vcs.merge(
546 target_ref, source_vcs, pull_request.source_ref_parts,
546 target_ref, source_vcs, pull_request.source_ref_parts,
547 workspace_id, user_name=user.username,
547 workspace_id, user_name=user.username,
548 user_email=user.email, message=message, use_rebase=use_rebase)
548 user_email=user.email, message=message, use_rebase=use_rebase)
549 return merge_state
549 return merge_state
550
550
551 def _comment_and_close_pr(self, pull_request, user, merge_state):
551 def _comment_and_close_pr(self, pull_request, user, merge_state):
552 pull_request.merge_rev = merge_state.merge_ref.commit_id
552 pull_request.merge_rev = merge_state.merge_ref.commit_id
553 pull_request.updated_on = datetime.datetime.now()
553 pull_request.updated_on = datetime.datetime.now()
554
554
555 CommentsModel().create(
555 CommentsModel().create(
556 text=unicode(_('Pull request merged and closed')),
556 text=unicode(_('Pull request merged and closed')),
557 repo=pull_request.target_repo.repo_id,
557 repo=pull_request.target_repo.repo_id,
558 user=user.user_id,
558 user=user.user_id,
559 pull_request=pull_request.pull_request_id,
559 pull_request=pull_request.pull_request_id,
560 f_path=None,
560 f_path=None,
561 line_no=None,
561 line_no=None,
562 closing_pr=True
562 closing_pr=True
563 )
563 )
564
564
565 Session().add(pull_request)
565 Session().add(pull_request)
566 Session().flush()
566 Session().flush()
567 # TODO: paris: replace invalidation with less radical solution
567 # TODO: paris: replace invalidation with less radical solution
568 ScmModel().mark_for_invalidation(
568 ScmModel().mark_for_invalidation(
569 pull_request.target_repo.repo_name)
569 pull_request.target_repo.repo_name)
570 self._trigger_pull_request_hook(pull_request, user, 'merge')
570 self._trigger_pull_request_hook(pull_request, user, 'merge')
571
571
572 def has_valid_update_type(self, pull_request):
572 def has_valid_update_type(self, pull_request):
573 source_ref_type = pull_request.source_ref_parts.type
573 source_ref_type = pull_request.source_ref_parts.type
574 return source_ref_type in ['book', 'branch', 'tag']
574 return source_ref_type in ['book', 'branch', 'tag']
575
575
576 def update_commits(self, pull_request):
576 def update_commits(self, pull_request):
577 """
577 """
578 Get the updated list of commits for the pull request
578 Get the updated list of commits for the pull request
579 and return the new pull request version and the list
579 and return the new pull request version and the list
580 of commits processed by this update action
580 of commits processed by this update action
581 """
581 """
582 pull_request = self.__get_pull_request(pull_request)
582 pull_request = self.__get_pull_request(pull_request)
583 source_ref_type = pull_request.source_ref_parts.type
583 source_ref_type = pull_request.source_ref_parts.type
584 source_ref_name = pull_request.source_ref_parts.name
584 source_ref_name = pull_request.source_ref_parts.name
585 source_ref_id = pull_request.source_ref_parts.commit_id
585 source_ref_id = pull_request.source_ref_parts.commit_id
586
586
587 if not self.has_valid_update_type(pull_request):
587 if not self.has_valid_update_type(pull_request):
588 log.debug(
588 log.debug(
589 "Skipping update of pull request %s due to ref type: %s",
589 "Skipping update of pull request %s due to ref type: %s",
590 pull_request, source_ref_type)
590 pull_request, source_ref_type)
591 return UpdateResponse(
591 return UpdateResponse(
592 executed=False,
592 executed=False,
593 reason=UpdateFailureReason.WRONG_REF_TPYE,
593 reason=UpdateFailureReason.WRONG_REF_TPYE,
594 old=pull_request, new=None, changes=None)
594 old=pull_request, new=None, changes=None)
595
595
596 source_repo = pull_request.source_repo.scm_instance()
596 source_repo = pull_request.source_repo.scm_instance()
597 try:
597 try:
598 source_commit = source_repo.get_commit(commit_id=source_ref_name)
598 source_commit = source_repo.get_commit(commit_id=source_ref_name)
599 except CommitDoesNotExistError:
599 except CommitDoesNotExistError:
600 return UpdateResponse(
600 return UpdateResponse(
601 executed=False,
601 executed=False,
602 reason=UpdateFailureReason.MISSING_SOURCE_REF,
602 reason=UpdateFailureReason.MISSING_SOURCE_REF,
603 old=pull_request, new=None, changes=None)
603 old=pull_request, new=None, changes=None)
604
604
605 if source_ref_id == source_commit.raw_id:
605 if source_ref_id == source_commit.raw_id:
606 log.debug("Nothing changed in pull request %s", pull_request)
606 log.debug("Nothing changed in pull request %s", pull_request)
607 return UpdateResponse(
607 return UpdateResponse(
608 executed=False,
608 executed=False,
609 reason=UpdateFailureReason.NO_CHANGE,
609 reason=UpdateFailureReason.NO_CHANGE,
610 old=pull_request, new=None, changes=None)
610 old=pull_request, new=None, changes=None)
611
611
612 # Finally there is a need for an update
612 # Finally there is a need for an update
613 pull_request_version = self._create_version_from_snapshot(pull_request)
613 pull_request_version = self._create_version_from_snapshot(pull_request)
614 self._link_comments_to_version(pull_request_version)
614 self._link_comments_to_version(pull_request_version)
615
615
616 target_ref_type = pull_request.target_ref_parts.type
616 target_ref_type = pull_request.target_ref_parts.type
617 target_ref_name = pull_request.target_ref_parts.name
617 target_ref_name = pull_request.target_ref_parts.name
618 target_ref_id = pull_request.target_ref_parts.commit_id
618 target_ref_id = pull_request.target_ref_parts.commit_id
619 target_repo = pull_request.target_repo.scm_instance()
619 target_repo = pull_request.target_repo.scm_instance()
620
620
621 try:
621 try:
622 if target_ref_type in ('tag', 'branch', 'book'):
622 if target_ref_type in ('tag', 'branch', 'book'):
623 target_commit = target_repo.get_commit(target_ref_name)
623 target_commit = target_repo.get_commit(target_ref_name)
624 else:
624 else:
625 target_commit = target_repo.get_commit(target_ref_id)
625 target_commit = target_repo.get_commit(target_ref_id)
626 except CommitDoesNotExistError:
626 except CommitDoesNotExistError:
627 return UpdateResponse(
627 return UpdateResponse(
628 executed=False,
628 executed=False,
629 reason=UpdateFailureReason.MISSING_TARGET_REF,
629 reason=UpdateFailureReason.MISSING_TARGET_REF,
630 old=pull_request, new=None, changes=None)
630 old=pull_request, new=None, changes=None)
631
631
632 # re-compute commit ids
632 # re-compute commit ids
633 old_commit_ids = set(pull_request.revisions)
633 old_commit_ids = set(pull_request.revisions)
634 pre_load = ["author", "branch", "date", "message"]
634 pre_load = ["author", "branch", "date", "message"]
635 commit_ranges = target_repo.compare(
635 commit_ranges = target_repo.compare(
636 target_commit.raw_id, source_commit.raw_id, source_repo, merge=True,
636 target_commit.raw_id, source_commit.raw_id, source_repo, merge=True,
637 pre_load=pre_load)
637 pre_load=pre_load)
638
638
639 ancestor = target_repo.get_common_ancestor(
639 ancestor = target_repo.get_common_ancestor(
640 target_commit.raw_id, source_commit.raw_id, source_repo)
640 target_commit.raw_id, source_commit.raw_id, source_repo)
641
641
642 pull_request.source_ref = '%s:%s:%s' % (
642 pull_request.source_ref = '%s:%s:%s' % (
643 source_ref_type, source_ref_name, source_commit.raw_id)
643 source_ref_type, source_ref_name, source_commit.raw_id)
644 pull_request.target_ref = '%s:%s:%s' % (
644 pull_request.target_ref = '%s:%s:%s' % (
645 target_ref_type, target_ref_name, ancestor)
645 target_ref_type, target_ref_name, ancestor)
646 pull_request.revisions = [
646 pull_request.revisions = [
647 commit.raw_id for commit in reversed(commit_ranges)]
647 commit.raw_id for commit in reversed(commit_ranges)]
648 pull_request.updated_on = datetime.datetime.now()
648 pull_request.updated_on = datetime.datetime.now()
649 Session().add(pull_request)
649 Session().add(pull_request)
650 new_commit_ids = set(pull_request.revisions)
650 new_commit_ids = set(pull_request.revisions)
651
651
652 changes = self._calculate_commit_id_changes(
652 changes = self._calculate_commit_id_changes(
653 old_commit_ids, new_commit_ids)
653 old_commit_ids, new_commit_ids)
654
654
655 old_diff_data, new_diff_data = self._generate_update_diffs(
655 old_diff_data, new_diff_data = self._generate_update_diffs(
656 pull_request, pull_request_version)
656 pull_request, pull_request_version)
657
657
658 CommentsModel().outdate_comments(
658 CommentsModel().outdate_comments(
659 pull_request, old_diff_data=old_diff_data,
659 pull_request, old_diff_data=old_diff_data,
660 new_diff_data=new_diff_data)
660 new_diff_data=new_diff_data)
661
661
662 file_changes = self._calculate_file_changes(
662 file_changes = self._calculate_file_changes(
663 old_diff_data, new_diff_data)
663 old_diff_data, new_diff_data)
664
664
665 # Add an automatic comment to the pull request
665 # Add an automatic comment to the pull request
666 update_comment = CommentsModel().create(
666 update_comment = CommentsModel().create(
667 text=self._render_update_message(changes, file_changes),
667 text=self._render_update_message(changes, file_changes),
668 repo=pull_request.target_repo,
668 repo=pull_request.target_repo,
669 user=pull_request.author,
669 user=pull_request.author,
670 pull_request=pull_request,
670 pull_request=pull_request,
671 send_email=False, renderer=DEFAULT_COMMENTS_RENDERER)
671 send_email=False, renderer=DEFAULT_COMMENTS_RENDERER)
672
672
673 # Update status to "Under Review" for added commits
673 # Update status to "Under Review" for added commits
674 for commit_id in changes.added:
674 for commit_id in changes.added:
675 ChangesetStatusModel().set_status(
675 ChangesetStatusModel().set_status(
676 repo=pull_request.source_repo,
676 repo=pull_request.source_repo,
677 status=ChangesetStatus.STATUS_UNDER_REVIEW,
677 status=ChangesetStatus.STATUS_UNDER_REVIEW,
678 comment=update_comment,
678 comment=update_comment,
679 user=pull_request.author,
679 user=pull_request.author,
680 pull_request=pull_request,
680 pull_request=pull_request,
681 revision=commit_id)
681 revision=commit_id)
682
682
683 log.debug(
683 log.debug(
684 'Updated pull request %s, added_ids: %s, common_ids: %s, '
684 'Updated pull request %s, added_ids: %s, common_ids: %s, '
685 'removed_ids: %s', pull_request.pull_request_id,
685 'removed_ids: %s', pull_request.pull_request_id,
686 changes.added, changes.common, changes.removed)
686 changes.added, changes.common, changes.removed)
687 log.debug('Updated pull request with the following file changes: %s',
687 log.debug('Updated pull request with the following file changes: %s',
688 file_changes)
688 file_changes)
689
689
690 log.info(
690 log.info(
691 "Updated pull request %s from commit %s to commit %s, "
691 "Updated pull request %s from commit %s to commit %s, "
692 "stored new version %s of this pull request.",
692 "stored new version %s of this pull request.",
693 pull_request.pull_request_id, source_ref_id,
693 pull_request.pull_request_id, source_ref_id,
694 pull_request.source_ref_parts.commit_id,
694 pull_request.source_ref_parts.commit_id,
695 pull_request_version.pull_request_version_id)
695 pull_request_version.pull_request_version_id)
696 Session().commit()
696 Session().commit()
697 self._trigger_pull_request_hook(pull_request, pull_request.author,
697 self._trigger_pull_request_hook(pull_request, pull_request.author,
698 'update')
698 'update')
699
699
700 return UpdateResponse(
700 return UpdateResponse(
701 executed=True, reason=UpdateFailureReason.NONE,
701 executed=True, reason=UpdateFailureReason.NONE,
702 old=pull_request, new=pull_request_version, changes=changes)
702 old=pull_request, new=pull_request_version, changes=changes)
703
703
704 def _create_version_from_snapshot(self, pull_request):
704 def _create_version_from_snapshot(self, pull_request):
705 version = PullRequestVersion()
705 version = PullRequestVersion()
706 version.title = pull_request.title
706 version.title = pull_request.title
707 version.description = pull_request.description
707 version.description = pull_request.description
708 version.status = pull_request.status
708 version.status = pull_request.status
709 version.created_on = datetime.datetime.now()
709 version.created_on = datetime.datetime.now()
710 version.updated_on = pull_request.updated_on
710 version.updated_on = pull_request.updated_on
711 version.user_id = pull_request.user_id
711 version.user_id = pull_request.user_id
712 version.source_repo = pull_request.source_repo
712 version.source_repo = pull_request.source_repo
713 version.source_ref = pull_request.source_ref
713 version.source_ref = pull_request.source_ref
714 version.target_repo = pull_request.target_repo
714 version.target_repo = pull_request.target_repo
715 version.target_ref = pull_request.target_ref
715 version.target_ref = pull_request.target_ref
716
716
717 version._last_merge_source_rev = pull_request._last_merge_source_rev
717 version._last_merge_source_rev = pull_request._last_merge_source_rev
718 version._last_merge_target_rev = pull_request._last_merge_target_rev
718 version._last_merge_target_rev = pull_request._last_merge_target_rev
719 version._last_merge_status = pull_request._last_merge_status
719 version._last_merge_status = pull_request._last_merge_status
720 version.shadow_merge_ref = pull_request.shadow_merge_ref
720 version.shadow_merge_ref = pull_request.shadow_merge_ref
721 version.merge_rev = pull_request.merge_rev
721 version.merge_rev = pull_request.merge_rev
722
722
723 version.revisions = pull_request.revisions
723 version.revisions = pull_request.revisions
724 version.pull_request = pull_request
724 version.pull_request = pull_request
725 Session().add(version)
725 Session().add(version)
726 Session().flush()
726 Session().flush()
727
727
728 return version
728 return version
729
729
730 def _generate_update_diffs(self, pull_request, pull_request_version):
730 def _generate_update_diffs(self, pull_request, pull_request_version):
731 diff_context = (
731 diff_context = (
732 self.DIFF_CONTEXT +
732 self.DIFF_CONTEXT +
733 CommentsModel.needed_extra_diff_context())
733 CommentsModel.needed_extra_diff_context())
734 old_diff = self._get_diff_from_pr_or_version(
734 old_diff = self._get_diff_from_pr_or_version(
735 pull_request_version, context=diff_context)
735 pull_request_version, context=diff_context)
736 new_diff = self._get_diff_from_pr_or_version(
736 new_diff = self._get_diff_from_pr_or_version(
737 pull_request, context=diff_context)
737 pull_request, context=diff_context)
738
738
739 old_diff_data = diffs.DiffProcessor(old_diff)
739 old_diff_data = diffs.DiffProcessor(old_diff)
740 old_diff_data.prepare()
740 old_diff_data.prepare()
741 new_diff_data = diffs.DiffProcessor(new_diff)
741 new_diff_data = diffs.DiffProcessor(new_diff)
742 new_diff_data.prepare()
742 new_diff_data.prepare()
743
743
744 return old_diff_data, new_diff_data
744 return old_diff_data, new_diff_data
745
745
746 def _link_comments_to_version(self, pull_request_version):
746 def _link_comments_to_version(self, pull_request_version):
747 """
747 """
748 Link all unlinked comments of this pull request to the given version.
748 Link all unlinked comments of this pull request to the given version.
749
749
750 :param pull_request_version: The `PullRequestVersion` to which
750 :param pull_request_version: The `PullRequestVersion` to which
751 the comments shall be linked.
751 the comments shall be linked.
752
752
753 """
753 """
754 pull_request = pull_request_version.pull_request
754 pull_request = pull_request_version.pull_request
755 comments = ChangesetComment.query().filter(
755 comments = ChangesetComment.query().filter(
756 # TODO: johbo: Should we query for the repo at all here?
756 # TODO: johbo: Should we query for the repo at all here?
757 # Pending decision on how comments of PRs are to be related
757 # Pending decision on how comments of PRs are to be related
758 # to either the source repo, the target repo or no repo at all.
758 # to either the source repo, the target repo or no repo at all.
759 ChangesetComment.repo_id == pull_request.target_repo.repo_id,
759 ChangesetComment.repo_id == pull_request.target_repo.repo_id,
760 ChangesetComment.pull_request == pull_request,
760 ChangesetComment.pull_request == pull_request,
761 ChangesetComment.pull_request_version == None)
761 ChangesetComment.pull_request_version == None)
762
762
763 # TODO: johbo: Find out why this breaks if it is done in a bulk
763 # TODO: johbo: Find out why this breaks if it is done in a bulk
764 # operation.
764 # operation.
765 for comment in comments:
765 for comment in comments:
766 comment.pull_request_version_id = (
766 comment.pull_request_version_id = (
767 pull_request_version.pull_request_version_id)
767 pull_request_version.pull_request_version_id)
768 Session().add(comment)
768 Session().add(comment)
769
769
770 def _calculate_commit_id_changes(self, old_ids, new_ids):
770 def _calculate_commit_id_changes(self, old_ids, new_ids):
771 added = new_ids.difference(old_ids)
771 added = new_ids.difference(old_ids)
772 common = old_ids.intersection(new_ids)
772 common = old_ids.intersection(new_ids)
773 removed = old_ids.difference(new_ids)
773 removed = old_ids.difference(new_ids)
774 return ChangeTuple(added, common, removed)
774 return ChangeTuple(added, common, removed)
775
775
776 def _calculate_file_changes(self, old_diff_data, new_diff_data):
776 def _calculate_file_changes(self, old_diff_data, new_diff_data):
777
777
778 old_files = OrderedDict()
778 old_files = OrderedDict()
779 for diff_data in old_diff_data.parsed_diff:
779 for diff_data in old_diff_data.parsed_diff:
780 old_files[diff_data['filename']] = md5_safe(diff_data['raw_diff'])
780 old_files[diff_data['filename']] = md5_safe(diff_data['raw_diff'])
781
781
782 added_files = []
782 added_files = []
783 modified_files = []
783 modified_files = []
784 removed_files = []
784 removed_files = []
785 for diff_data in new_diff_data.parsed_diff:
785 for diff_data in new_diff_data.parsed_diff:
786 new_filename = diff_data['filename']
786 new_filename = diff_data['filename']
787 new_hash = md5_safe(diff_data['raw_diff'])
787 new_hash = md5_safe(diff_data['raw_diff'])
788
788
789 old_hash = old_files.get(new_filename)
789 old_hash = old_files.get(new_filename)
790 if not old_hash:
790 if not old_hash:
791 # file is not present in old diff, means it's added
791 # file is not present in old diff, means it's added
792 added_files.append(new_filename)
792 added_files.append(new_filename)
793 else:
793 else:
794 if new_hash != old_hash:
794 if new_hash != old_hash:
795 modified_files.append(new_filename)
795 modified_files.append(new_filename)
796 # now remove a file from old, since we have seen it already
796 # now remove a file from old, since we have seen it already
797 del old_files[new_filename]
797 del old_files[new_filename]
798
798
799 # removed files is when there are present in old, but not in NEW,
799 # removed files is when there are present in old, but not in NEW,
800 # since we remove old files that are present in new diff, left-overs
800 # since we remove old files that are present in new diff, left-overs
801 # if any should be the removed files
801 # if any should be the removed files
802 removed_files.extend(old_files.keys())
802 removed_files.extend(old_files.keys())
803
803
804 return FileChangeTuple(added_files, modified_files, removed_files)
804 return FileChangeTuple(added_files, modified_files, removed_files)
805
805
806 def _render_update_message(self, changes, file_changes):
806 def _render_update_message(self, changes, file_changes):
807 """
807 """
808 render the message using DEFAULT_COMMENTS_RENDERER (RST renderer),
808 render the message using DEFAULT_COMMENTS_RENDERER (RST renderer),
809 so it's always looking the same disregarding on which default
809 so it's always looking the same disregarding on which default
810 renderer system is using.
810 renderer system is using.
811
811
812 :param changes: changes named tuple
812 :param changes: changes named tuple
813 :param file_changes: file changes named tuple
813 :param file_changes: file changes named tuple
814
814
815 """
815 """
816 new_status = ChangesetStatus.get_status_lbl(
816 new_status = ChangesetStatus.get_status_lbl(
817 ChangesetStatus.STATUS_UNDER_REVIEW)
817 ChangesetStatus.STATUS_UNDER_REVIEW)
818
818
819 changed_files = (
819 changed_files = (
820 file_changes.added + file_changes.modified + file_changes.removed)
820 file_changes.added + file_changes.modified + file_changes.removed)
821
821
822 params = {
822 params = {
823 'under_review_label': new_status,
823 'under_review_label': new_status,
824 'added_commits': changes.added,
824 'added_commits': changes.added,
825 'removed_commits': changes.removed,
825 'removed_commits': changes.removed,
826 'changed_files': changed_files,
826 'changed_files': changed_files,
827 'added_files': file_changes.added,
827 'added_files': file_changes.added,
828 'modified_files': file_changes.modified,
828 'modified_files': file_changes.modified,
829 'removed_files': file_changes.removed,
829 'removed_files': file_changes.removed,
830 }
830 }
831 renderer = RstTemplateRenderer()
831 renderer = RstTemplateRenderer()
832 return renderer.render('pull_request_update.mako', **params)
832 return renderer.render('pull_request_update.mako', **params)
833
833
834 def edit(self, pull_request, title, description):
834 def edit(self, pull_request, title, description):
835 pull_request = self.__get_pull_request(pull_request)
835 pull_request = self.__get_pull_request(pull_request)
836 if pull_request.is_closed():
836 if pull_request.is_closed():
837 raise ValueError('This pull request is closed')
837 raise ValueError('This pull request is closed')
838 if title:
838 if title:
839 pull_request.title = title
839 pull_request.title = title
840 pull_request.description = description
840 pull_request.description = description
841 pull_request.updated_on = datetime.datetime.now()
841 pull_request.updated_on = datetime.datetime.now()
842 Session().add(pull_request)
842 Session().add(pull_request)
843
843
844 def update_reviewers(self, pull_request, reviewer_data):
844 def update_reviewers(self, pull_request, reviewer_data):
845 """
845 """
846 Update the reviewers in the pull request
846 Update the reviewers in the pull request
847
847
848 :param pull_request: the pr to update
848 :param pull_request: the pr to update
849 :param reviewer_data: list of tuples [(user, ['reason1', 'reason2'])]
849 :param reviewer_data: list of tuples [(user, ['reason1', 'reason2'])]
850 """
850 """
851
851
852 reviewers_reasons = {}
852 reviewers_reasons = {}
853 for user_id, reasons in reviewer_data:
853 for user_id, reasons in reviewer_data:
854 if isinstance(user_id, (int, basestring)):
854 if isinstance(user_id, (int, basestring)):
855 user_id = self._get_user(user_id).user_id
855 user_id = self._get_user(user_id).user_id
856 reviewers_reasons[user_id] = reasons
856 reviewers_reasons[user_id] = reasons
857
857
858 reviewers_ids = set(reviewers_reasons.keys())
858 reviewers_ids = set(reviewers_reasons.keys())
859 pull_request = self.__get_pull_request(pull_request)
859 pull_request = self.__get_pull_request(pull_request)
860 current_reviewers = PullRequestReviewers.query()\
860 current_reviewers = PullRequestReviewers.query()\
861 .filter(PullRequestReviewers.pull_request ==
861 .filter(PullRequestReviewers.pull_request ==
862 pull_request).all()
862 pull_request).all()
863 current_reviewers_ids = set([x.user.user_id for x in current_reviewers])
863 current_reviewers_ids = set([x.user.user_id for x in current_reviewers])
864
864
865 ids_to_add = reviewers_ids.difference(current_reviewers_ids)
865 ids_to_add = reviewers_ids.difference(current_reviewers_ids)
866 ids_to_remove = current_reviewers_ids.difference(reviewers_ids)
866 ids_to_remove = current_reviewers_ids.difference(reviewers_ids)
867
867
868 log.debug("Adding %s reviewers", ids_to_add)
868 log.debug("Adding %s reviewers", ids_to_add)
869 log.debug("Removing %s reviewers", ids_to_remove)
869 log.debug("Removing %s reviewers", ids_to_remove)
870 changed = False
870 changed = False
871 for uid in ids_to_add:
871 for uid in ids_to_add:
872 changed = True
872 changed = True
873 _usr = self._get_user(uid)
873 _usr = self._get_user(uid)
874 reasons = reviewers_reasons[uid]
874 reasons = reviewers_reasons[uid]
875 reviewer = PullRequestReviewers(_usr, pull_request, reasons)
875 reviewer = PullRequestReviewers(_usr, pull_request, reasons)
876 Session().add(reviewer)
876 Session().add(reviewer)
877
877
878 self.notify_reviewers(pull_request, ids_to_add)
878 self.notify_reviewers(pull_request, ids_to_add)
879
879
880 for uid in ids_to_remove:
880 for uid in ids_to_remove:
881 changed = True
881 changed = True
882 reviewer = PullRequestReviewers.query()\
882 reviewer = PullRequestReviewers.query()\
883 .filter(PullRequestReviewers.user_id == uid,
883 .filter(PullRequestReviewers.user_id == uid,
884 PullRequestReviewers.pull_request == pull_request)\
884 PullRequestReviewers.pull_request == pull_request)\
885 .scalar()
885 .scalar()
886 if reviewer:
886 if reviewer:
887 Session().delete(reviewer)
887 Session().delete(reviewer)
888 if changed:
888 if changed:
889 pull_request.updated_on = datetime.datetime.now()
889 pull_request.updated_on = datetime.datetime.now()
890 Session().add(pull_request)
890 Session().add(pull_request)
891
891
892 return ids_to_add, ids_to_remove
892 return ids_to_add, ids_to_remove
893
893
894 def get_url(self, pull_request):
894 def get_url(self, pull_request):
895 return h.url('pullrequest_show',
895 return h.url('pullrequest_show',
896 repo_name=safe_str(pull_request.target_repo.repo_name),
896 repo_name=safe_str(pull_request.target_repo.repo_name),
897 pull_request_id=pull_request.pull_request_id,
897 pull_request_id=pull_request.pull_request_id,
898 qualified=True)
898 qualified=True)
899
899
900 def get_shadow_clone_url(self, pull_request):
900 def get_shadow_clone_url(self, pull_request):
901 """
901 """
902 Returns qualified url pointing to the shadow repository. If this pull
902 Returns qualified url pointing to the shadow repository. If this pull
903 request is closed there is no shadow repository and ``None`` will be
903 request is closed there is no shadow repository and ``None`` will be
904 returned.
904 returned.
905 """
905 """
906 if pull_request.is_closed():
906 if pull_request.is_closed():
907 return None
907 return None
908 else:
908 else:
909 pr_url = urllib.unquote(self.get_url(pull_request))
909 pr_url = urllib.unquote(self.get_url(pull_request))
910 return safe_unicode('{pr_url}/repository'.format(pr_url=pr_url))
910 return safe_unicode('{pr_url}/repository'.format(pr_url=pr_url))
911
911
912 def notify_reviewers(self, pull_request, reviewers_ids):
912 def notify_reviewers(self, pull_request, reviewers_ids):
913 # notification to reviewers
913 # notification to reviewers
914 if not reviewers_ids:
914 if not reviewers_ids:
915 return
915 return
916
916
917 pull_request_obj = pull_request
917 pull_request_obj = pull_request
918 # get the current participants of this pull request
918 # get the current participants of this pull request
919 recipients = reviewers_ids
919 recipients = reviewers_ids
920 notification_type = EmailNotificationModel.TYPE_PULL_REQUEST
920 notification_type = EmailNotificationModel.TYPE_PULL_REQUEST
921
921
922 pr_source_repo = pull_request_obj.source_repo
922 pr_source_repo = pull_request_obj.source_repo
923 pr_target_repo = pull_request_obj.target_repo
923 pr_target_repo = pull_request_obj.target_repo
924
924
925 pr_url = h.url(
925 pr_url = h.url(
926 'pullrequest_show',
926 'pullrequest_show',
927 repo_name=pr_target_repo.repo_name,
927 repo_name=pr_target_repo.repo_name,
928 pull_request_id=pull_request_obj.pull_request_id,
928 pull_request_id=pull_request_obj.pull_request_id,
929 qualified=True,)
929 qualified=True,)
930
930
931 # set some variables for email notification
931 # set some variables for email notification
932 pr_target_repo_url = h.url(
932 pr_target_repo_url = h.url(
933 'summary_home',
933 'summary_home',
934 repo_name=pr_target_repo.repo_name,
934 repo_name=pr_target_repo.repo_name,
935 qualified=True)
935 qualified=True)
936
936
937 pr_source_repo_url = h.url(
937 pr_source_repo_url = h.url(
938 'summary_home',
938 'summary_home',
939 repo_name=pr_source_repo.repo_name,
939 repo_name=pr_source_repo.repo_name,
940 qualified=True)
940 qualified=True)
941
941
942 # pull request specifics
942 # pull request specifics
943 pull_request_commits = [
943 pull_request_commits = [
944 (x.raw_id, x.message)
944 (x.raw_id, x.message)
945 for x in map(pr_source_repo.get_commit, pull_request.revisions)]
945 for x in map(pr_source_repo.get_commit, pull_request.revisions)]
946
946
947 kwargs = {
947 kwargs = {
948 'user': pull_request.author,
948 'user': pull_request.author,
949 'pull_request': pull_request_obj,
949 'pull_request': pull_request_obj,
950 'pull_request_commits': pull_request_commits,
950 'pull_request_commits': pull_request_commits,
951
951
952 'pull_request_target_repo': pr_target_repo,
952 'pull_request_target_repo': pr_target_repo,
953 'pull_request_target_repo_url': pr_target_repo_url,
953 'pull_request_target_repo_url': pr_target_repo_url,
954
954
955 'pull_request_source_repo': pr_source_repo,
955 'pull_request_source_repo': pr_source_repo,
956 'pull_request_source_repo_url': pr_source_repo_url,
956 'pull_request_source_repo_url': pr_source_repo_url,
957
957
958 'pull_request_url': pr_url,
958 'pull_request_url': pr_url,
959 }
959 }
960
960
961 # pre-generate the subject for notification itself
961 # pre-generate the subject for notification itself
962 (subject,
962 (subject,
963 _h, _e, # we don't care about those
963 _h, _e, # we don't care about those
964 body_plaintext) = EmailNotificationModel().render_email(
964 body_plaintext) = EmailNotificationModel().render_email(
965 notification_type, **kwargs)
965 notification_type, **kwargs)
966
966
967 # create notification objects, and emails
967 # create notification objects, and emails
968 NotificationModel().create(
968 NotificationModel().create(
969 created_by=pull_request.author,
969 created_by=pull_request.author,
970 notification_subject=subject,
970 notification_subject=subject,
971 notification_body=body_plaintext,
971 notification_body=body_plaintext,
972 notification_type=notification_type,
972 notification_type=notification_type,
973 recipients=recipients,
973 recipients=recipients,
974 email_kwargs=kwargs,
974 email_kwargs=kwargs,
975 )
975 )
976
976
977 def delete(self, pull_request):
977 def delete(self, pull_request):
978 pull_request = self.__get_pull_request(pull_request)
978 pull_request = self.__get_pull_request(pull_request)
979 self._cleanup_merge_workspace(pull_request)
979 self._cleanup_merge_workspace(pull_request)
980 Session().delete(pull_request)
980 Session().delete(pull_request)
981
981
982 def close_pull_request(self, pull_request, user):
982 def close_pull_request(self, pull_request, user):
983 pull_request = self.__get_pull_request(pull_request)
983 pull_request = self.__get_pull_request(pull_request)
984 self._cleanup_merge_workspace(pull_request)
984 self._cleanup_merge_workspace(pull_request)
985 pull_request.status = PullRequest.STATUS_CLOSED
985 pull_request.status = PullRequest.STATUS_CLOSED
986 pull_request.updated_on = datetime.datetime.now()
986 pull_request.updated_on = datetime.datetime.now()
987 Session().add(pull_request)
987 Session().add(pull_request)
988 self._trigger_pull_request_hook(
988 self._trigger_pull_request_hook(
989 pull_request, pull_request.author, 'close')
989 pull_request, pull_request.author, 'close')
990 self._log_action('user_closed_pull_request', user, pull_request)
990 self._log_action('user_closed_pull_request', user, pull_request)
991
991
992 def close_pull_request_with_comment(self, pull_request, user, repo,
992 def close_pull_request_with_comment(self, pull_request, user, repo,
993 message=None):
993 message=None):
994 status = ChangesetStatus.STATUS_REJECTED
994 status = ChangesetStatus.STATUS_REJECTED
995
995
996 if not message:
996 if not message:
997 message = (
997 message = (
998 _('Status change %(transition_icon)s %(status)s') % {
998 _('Status change %(transition_icon)s %(status)s') % {
999 'transition_icon': '>',
999 'transition_icon': '>',
1000 'status': ChangesetStatus.get_status_lbl(status)})
1000 'status': ChangesetStatus.get_status_lbl(status)})
1001
1001
1002 internal_message = _('Closing with') + ' ' + message
1002 internal_message = _('Closing with') + ' ' + message
1003
1003
1004 comm = CommentsModel().create(
1004 comm = CommentsModel().create(
1005 text=internal_message,
1005 text=internal_message,
1006 repo=repo.repo_id,
1006 repo=repo.repo_id,
1007 user=user.user_id,
1007 user=user.user_id,
1008 pull_request=pull_request.pull_request_id,
1008 pull_request=pull_request.pull_request_id,
1009 f_path=None,
1009 f_path=None,
1010 line_no=None,
1010 line_no=None,
1011 status_change=ChangesetStatus.get_status_lbl(status),
1011 status_change=ChangesetStatus.get_status_lbl(status),
1012 status_change_type=status,
1012 status_change_type=status,
1013 closing_pr=True
1013 closing_pr=True
1014 )
1014 )
1015
1015
1016 ChangesetStatusModel().set_status(
1016 ChangesetStatusModel().set_status(
1017 repo.repo_id,
1017 repo.repo_id,
1018 status,
1018 status,
1019 user.user_id,
1019 user.user_id,
1020 comm,
1020 comm,
1021 pull_request=pull_request.pull_request_id
1021 pull_request=pull_request.pull_request_id
1022 )
1022 )
1023 Session().flush()
1023 Session().flush()
1024
1024
1025 PullRequestModel().close_pull_request(
1025 PullRequestModel().close_pull_request(
1026 pull_request.pull_request_id, user)
1026 pull_request.pull_request_id, user)
1027
1027
1028 def merge_status(self, pull_request):
1028 def merge_status(self, pull_request):
1029 if not self._is_merge_enabled(pull_request):
1029 if not self._is_merge_enabled(pull_request):
1030 return False, _('Server-side pull request merging is disabled.')
1030 return False, _('Server-side pull request merging is disabled.')
1031 if pull_request.is_closed():
1031 if pull_request.is_closed():
1032 return False, _('This pull request is closed.')
1032 return False, _('This pull request is closed.')
1033 merge_possible, msg = self._check_repo_requirements(
1033 merge_possible, msg = self._check_repo_requirements(
1034 target=pull_request.target_repo, source=pull_request.source_repo)
1034 target=pull_request.target_repo, source=pull_request.source_repo)
1035 if not merge_possible:
1035 if not merge_possible:
1036 return merge_possible, msg
1036 return merge_possible, msg
1037
1037
1038 try:
1038 try:
1039 resp = self._try_merge(pull_request)
1039 resp = self._try_merge(pull_request)
1040 log.debug("Merge response: %s", resp)
1040 log.debug("Merge response: %s", resp)
1041 status = resp.possible, self.merge_status_message(
1041 status = resp.possible, self.merge_status_message(
1042 resp.failure_reason)
1042 resp.failure_reason)
1043 except NotImplementedError:
1043 except NotImplementedError:
1044 status = False, _('Pull request merging is not supported.')
1044 status = False, _('Pull request merging is not supported.')
1045
1045
1046 return status
1046 return status
1047
1047
1048 def _check_repo_requirements(self, target, source):
1048 def _check_repo_requirements(self, target, source):
1049 """
1049 """
1050 Check if `target` and `source` have compatible requirements.
1050 Check if `target` and `source` have compatible requirements.
1051
1051
1052 Currently this is just checking for largefiles.
1052 Currently this is just checking for largefiles.
1053 """
1053 """
1054 target_has_largefiles = self._has_largefiles(target)
1054 target_has_largefiles = self._has_largefiles(target)
1055 source_has_largefiles = self._has_largefiles(source)
1055 source_has_largefiles = self._has_largefiles(source)
1056 merge_possible = True
1056 merge_possible = True
1057 message = u''
1057 message = u''
1058
1058
1059 if target_has_largefiles != source_has_largefiles:
1059 if target_has_largefiles != source_has_largefiles:
1060 merge_possible = False
1060 merge_possible = False
1061 if source_has_largefiles:
1061 if source_has_largefiles:
1062 message = _(
1062 message = _(
1063 'Target repository large files support is disabled.')
1063 'Target repository large files support is disabled.')
1064 else:
1064 else:
1065 message = _(
1065 message = _(
1066 'Source repository large files support is disabled.')
1066 'Source repository large files support is disabled.')
1067
1067
1068 return merge_possible, message
1068 return merge_possible, message
1069
1069
1070 def _has_largefiles(self, repo):
1070 def _has_largefiles(self, repo):
1071 largefiles_ui = VcsSettingsModel(repo=repo).get_ui_settings(
1071 largefiles_ui = VcsSettingsModel(repo=repo).get_ui_settings(
1072 'extensions', 'largefiles')
1072 'extensions', 'largefiles')
1073 return largefiles_ui and largefiles_ui[0].active
1073 return largefiles_ui and largefiles_ui[0].active
1074
1074
1075 def _try_merge(self, pull_request):
1075 def _try_merge(self, pull_request):
1076 """
1076 """
1077 Try to merge the pull request and return the merge status.
1077 Try to merge the pull request and return the merge status.
1078 """
1078 """
1079 log.debug(
1079 log.debug(
1080 "Trying out if the pull request %s can be merged.",
1080 "Trying out if the pull request %s can be merged.",
1081 pull_request.pull_request_id)
1081 pull_request.pull_request_id)
1082 target_vcs = pull_request.target_repo.scm_instance()
1082 target_vcs = pull_request.target_repo.scm_instance()
1083
1083
1084 # Refresh the target reference.
1084 # Refresh the target reference.
1085 try:
1085 try:
1086 target_ref = self._refresh_reference(
1086 target_ref = self._refresh_reference(
1087 pull_request.target_ref_parts, target_vcs)
1087 pull_request.target_ref_parts, target_vcs)
1088 except CommitDoesNotExistError:
1088 except CommitDoesNotExistError:
1089 merge_state = MergeResponse(
1089 merge_state = MergeResponse(
1090 False, False, None, MergeFailureReason.MISSING_TARGET_REF)
1090 False, False, None, MergeFailureReason.MISSING_TARGET_REF)
1091 return merge_state
1091 return merge_state
1092
1092
1093 target_locked = pull_request.target_repo.locked
1093 target_locked = pull_request.target_repo.locked
1094 if target_locked and target_locked[0]:
1094 if target_locked and target_locked[0]:
1095 log.debug("The target repository is locked.")
1095 log.debug("The target repository is locked.")
1096 merge_state = MergeResponse(
1096 merge_state = MergeResponse(
1097 False, False, None, MergeFailureReason.TARGET_IS_LOCKED)
1097 False, False, None, MergeFailureReason.TARGET_IS_LOCKED)
1098 elif self._needs_merge_state_refresh(pull_request, target_ref):
1098 elif self._needs_merge_state_refresh(pull_request, target_ref):
1099 log.debug("Refreshing the merge status of the repository.")
1099 log.debug("Refreshing the merge status of the repository.")
1100 merge_state = self._refresh_merge_state(
1100 merge_state = self._refresh_merge_state(
1101 pull_request, target_vcs, target_ref)
1101 pull_request, target_vcs, target_ref)
1102 else:
1102 else:
1103 possible = pull_request.\
1103 possible = pull_request.\
1104 _last_merge_status == MergeFailureReason.NONE
1104 _last_merge_status == MergeFailureReason.NONE
1105 merge_state = MergeResponse(
1105 merge_state = MergeResponse(
1106 possible, False, None, pull_request._last_merge_status)
1106 possible, False, None, pull_request._last_merge_status)
1107
1107
1108 return merge_state
1108 return merge_state
1109
1109
1110 def _refresh_reference(self, reference, vcs_repository):
1110 def _refresh_reference(self, reference, vcs_repository):
1111 if reference.type in ('branch', 'book'):
1111 if reference.type in ('branch', 'book'):
1112 name_or_id = reference.name
1112 name_or_id = reference.name
1113 else:
1113 else:
1114 name_or_id = reference.commit_id
1114 name_or_id = reference.commit_id
1115 refreshed_commit = vcs_repository.get_commit(name_or_id)
1115 refreshed_commit = vcs_repository.get_commit(name_or_id)
1116 refreshed_reference = Reference(
1116 refreshed_reference = Reference(
1117 reference.type, reference.name, refreshed_commit.raw_id)
1117 reference.type, reference.name, refreshed_commit.raw_id)
1118 return refreshed_reference
1118 return refreshed_reference
1119
1119
1120 def _needs_merge_state_refresh(self, pull_request, target_reference):
1120 def _needs_merge_state_refresh(self, pull_request, target_reference):
1121 return not(
1121 return not(
1122 pull_request.revisions and
1122 pull_request.revisions and
1123 pull_request.revisions[0] == pull_request._last_merge_source_rev and
1123 pull_request.revisions[0] == pull_request._last_merge_source_rev and
1124 target_reference.commit_id == pull_request._last_merge_target_rev)
1124 target_reference.commit_id == pull_request._last_merge_target_rev)
1125
1125
1126 def _refresh_merge_state(self, pull_request, target_vcs, target_reference):
1126 def _refresh_merge_state(self, pull_request, target_vcs, target_reference):
1127 workspace_id = self._workspace_id(pull_request)
1127 workspace_id = self._workspace_id(pull_request)
1128 source_vcs = pull_request.source_repo.scm_instance()
1128 source_vcs = pull_request.source_repo.scm_instance()
1129 use_rebase = self._use_rebase_for_merging(pull_request)
1129 use_rebase = self._use_rebase_for_merging(pull_request)
1130 merge_state = target_vcs.merge(
1130 merge_state = target_vcs.merge(
1131 target_reference, source_vcs, pull_request.source_ref_parts,
1131 target_reference, source_vcs, pull_request.source_ref_parts,
1132 workspace_id, dry_run=True, use_rebase=use_rebase)
1132 workspace_id, dry_run=True, use_rebase=use_rebase)
1133
1133
1134 # Do not store the response if there was an unknown error.
1134 # Do not store the response if there was an unknown error.
1135 if merge_state.failure_reason != MergeFailureReason.UNKNOWN:
1135 if merge_state.failure_reason != MergeFailureReason.UNKNOWN:
1136 pull_request._last_merge_source_rev = \
1136 pull_request._last_merge_source_rev = \
1137 pull_request.source_ref_parts.commit_id
1137 pull_request.source_ref_parts.commit_id
1138 pull_request._last_merge_target_rev = target_reference.commit_id
1138 pull_request._last_merge_target_rev = target_reference.commit_id
1139 pull_request._last_merge_status = merge_state.failure_reason
1139 pull_request._last_merge_status = merge_state.failure_reason
1140 pull_request.shadow_merge_ref = merge_state.merge_ref
1140 pull_request.shadow_merge_ref = merge_state.merge_ref
1141 Session().add(pull_request)
1141 Session().add(pull_request)
1142 Session().commit()
1142 Session().commit()
1143
1143
1144 return merge_state
1144 return merge_state
1145
1145
1146 def _workspace_id(self, pull_request):
1146 def _workspace_id(self, pull_request):
1147 workspace_id = 'pr-%s' % pull_request.pull_request_id
1147 workspace_id = 'pr-%s' % pull_request.pull_request_id
1148 return workspace_id
1148 return workspace_id
1149
1149
1150 def merge_status_message(self, status_code):
1150 def merge_status_message(self, status_code):
1151 """
1151 """
1152 Return a human friendly error message for the given merge status code.
1152 Return a human friendly error message for the given merge status code.
1153 """
1153 """
1154 return self.MERGE_STATUS_MESSAGES[status_code]
1154 return self.MERGE_STATUS_MESSAGES[status_code]
1155
1155
1156 def generate_repo_data(self, repo, commit_id=None, branch=None,
1156 def generate_repo_data(self, repo, commit_id=None, branch=None,
1157 bookmark=None):
1157 bookmark=None):
1158 all_refs, selected_ref = \
1158 all_refs, selected_ref = \
1159 self._get_repo_pullrequest_sources(
1159 self._get_repo_pullrequest_sources(
1160 repo.scm_instance(), commit_id=commit_id,
1160 repo.scm_instance(), commit_id=commit_id,
1161 branch=branch, bookmark=bookmark)
1161 branch=branch, bookmark=bookmark)
1162
1162
1163 refs_select2 = []
1163 refs_select2 = []
1164 for element in all_refs:
1164 for element in all_refs:
1165 children = [{'id': x[0], 'text': x[1]} for x in element[0]]
1165 children = [{'id': x[0], 'text': x[1]} for x in element[0]]
1166 refs_select2.append({'text': element[1], 'children': children})
1166 refs_select2.append({'text': element[1], 'children': children})
1167
1167
1168 return {
1168 return {
1169 'user': {
1169 'user': {
1170 'user_id': repo.user.user_id,
1170 'user_id': repo.user.user_id,
1171 'username': repo.user.username,
1171 'username': repo.user.username,
1172 'firstname': repo.user.firstname,
1172 'firstname': repo.user.firstname,
1173 'lastname': repo.user.lastname,
1173 'lastname': repo.user.lastname,
1174 'gravatar_link': h.gravatar_url(repo.user.email, 14),
1174 'gravatar_link': h.gravatar_url(repo.user.email, 14),
1175 },
1175 },
1176 'description': h.chop_at_smart(repo.description, '\n'),
1176 'description': h.chop_at_smart(repo.description, '\n'),
1177 'refs': {
1177 'refs': {
1178 'all_refs': all_refs,
1178 'all_refs': all_refs,
1179 'selected_ref': selected_ref,
1179 'selected_ref': selected_ref,
1180 'select2_refs': refs_select2
1180 'select2_refs': refs_select2
1181 }
1181 }
1182 }
1182 }
1183
1183
1184 def generate_pullrequest_title(self, source, source_ref, target):
1184 def generate_pullrequest_title(self, source, source_ref, target):
1185 return u'{source}#{at_ref} to {target}'.format(
1185 return u'{source}#{at_ref} to {target}'.format(
1186 source=source,
1186 source=source,
1187 at_ref=source_ref,
1187 at_ref=source_ref,
1188 target=target,
1188 target=target,
1189 )
1189 )
1190
1190
1191 def _cleanup_merge_workspace(self, pull_request):
1191 def _cleanup_merge_workspace(self, pull_request):
1192 # Merging related cleanup
1192 # Merging related cleanup
1193 target_scm = pull_request.target_repo.scm_instance()
1193 target_scm = pull_request.target_repo.scm_instance()
1194 workspace_id = 'pr-%s' % pull_request.pull_request_id
1194 workspace_id = 'pr-%s' % pull_request.pull_request_id
1195
1195
1196 try:
1196 try:
1197 target_scm.cleanup_merge_workspace(workspace_id)
1197 target_scm.cleanup_merge_workspace(workspace_id)
1198 except NotImplementedError:
1198 except NotImplementedError:
1199 pass
1199 pass
1200
1200
1201 def _get_repo_pullrequest_sources(
1201 def _get_repo_pullrequest_sources(
1202 self, repo, commit_id=None, branch=None, bookmark=None):
1202 self, repo, commit_id=None, branch=None, bookmark=None):
1203 """
1203 """
1204 Return a structure with repo's interesting commits, suitable for
1204 Return a structure with repo's interesting commits, suitable for
1205 the selectors in pullrequest controller
1205 the selectors in pullrequest controller
1206
1206
1207 :param commit_id: a commit that must be in the list somehow
1207 :param commit_id: a commit that must be in the list somehow
1208 and selected by default
1208 and selected by default
1209 :param branch: a branch that must be in the list and selected
1209 :param branch: a branch that must be in the list and selected
1210 by default - even if closed
1210 by default - even if closed
1211 :param bookmark: a bookmark that must be in the list and selected
1211 :param bookmark: a bookmark that must be in the list and selected
1212 """
1212 """
1213
1213
1214 commit_id = safe_str(commit_id) if commit_id else None
1214 commit_id = safe_str(commit_id) if commit_id else None
1215 branch = safe_str(branch) if branch else None
1215 branch = safe_str(branch) if branch else None
1216 bookmark = safe_str(bookmark) if bookmark else None
1216 bookmark = safe_str(bookmark) if bookmark else None
1217
1217
1218 selected = None
1218 selected = None
1219
1219
1220 # order matters: first source that has commit_id in it will be selected
1220 # order matters: first source that has commit_id in it will be selected
1221 sources = []
1221 sources = []
1222 sources.append(('book', repo.bookmarks.items(), _('Bookmarks'), bookmark))
1222 sources.append(('book', repo.bookmarks.items(), _('Bookmarks'), bookmark))
1223 sources.append(('branch', repo.branches.items(), _('Branches'), branch))
1223 sources.append(('branch', repo.branches.items(), _('Branches'), branch))
1224
1224
1225 if commit_id:
1225 if commit_id:
1226 ref_commit = (h.short_id(commit_id), commit_id)
1226 ref_commit = (h.short_id(commit_id), commit_id)
1227 sources.append(('rev', [ref_commit], _('Commit IDs'), commit_id))
1227 sources.append(('rev', [ref_commit], _('Commit IDs'), commit_id))
1228
1228
1229 sources.append(
1229 sources.append(
1230 ('branch', repo.branches_closed.items(), _('Closed Branches'), branch),
1230 ('branch', repo.branches_closed.items(), _('Closed Branches'), branch),
1231 )
1231 )
1232
1232
1233 groups = []
1233 groups = []
1234 for group_key, ref_list, group_name, match in sources:
1234 for group_key, ref_list, group_name, match in sources:
1235 group_refs = []
1235 group_refs = []
1236 for ref_name, ref_id in ref_list:
1236 for ref_name, ref_id in ref_list:
1237 ref_key = '%s:%s:%s' % (group_key, ref_name, ref_id)
1237 ref_key = '%s:%s:%s' % (group_key, ref_name, ref_id)
1238 group_refs.append((ref_key, ref_name))
1238 group_refs.append((ref_key, ref_name))
1239
1239
1240 if not selected:
1240 if not selected:
1241 if set([commit_id, match]) & set([ref_id, ref_name]):
1241 if set([commit_id, match]) & set([ref_id, ref_name]):
1242 selected = ref_key
1242 selected = ref_key
1243
1243
1244 if group_refs:
1244 if group_refs:
1245 groups.append((group_refs, group_name))
1245 groups.append((group_refs, group_name))
1246
1246
1247 if not selected:
1247 if not selected:
1248 ref = commit_id or branch or bookmark
1248 ref = commit_id or branch or bookmark
1249 if ref:
1249 if ref:
1250 raise CommitDoesNotExistError(
1250 raise CommitDoesNotExistError(
1251 'No commit refs could be found matching: %s' % ref)
1251 'No commit refs could be found matching: %s' % ref)
1252 elif repo.DEFAULT_BRANCH_NAME in repo.branches:
1252 elif repo.DEFAULT_BRANCH_NAME in repo.branches:
1253 selected = 'branch:%s:%s' % (
1253 selected = 'branch:%s:%s' % (
1254 repo.DEFAULT_BRANCH_NAME,
1254 repo.DEFAULT_BRANCH_NAME,
1255 repo.branches[repo.DEFAULT_BRANCH_NAME]
1255 repo.branches[repo.DEFAULT_BRANCH_NAME]
1256 )
1256 )
1257 elif repo.commit_ids:
1257 elif repo.commit_ids:
1258 rev = repo.commit_ids[0]
1258 rev = repo.commit_ids[0]
1259 selected = 'rev:%s:%s' % (rev, rev)
1259 selected = 'rev:%s:%s' % (rev, rev)
1260 else:
1260 else:
1261 raise EmptyRepositoryError()
1261 raise EmptyRepositoryError()
1262 return groups, selected
1262 return groups, selected
1263
1263
1264 def get_diff(self, pull_request, context=DIFF_CONTEXT):
1264 def get_diff(self, pull_request, context=DIFF_CONTEXT):
1265 pull_request = self.__get_pull_request(pull_request)
1265 pull_request = self.__get_pull_request(pull_request)
1266 return self._get_diff_from_pr_or_version(pull_request, context=context)
1266 return self._get_diff_from_pr_or_version(pull_request, context=context)
1267
1267
1268 def _get_diff_from_pr_or_version(self, pr_or_version, context):
1268 def _get_diff_from_pr_or_version(self, pr_or_version, context):
1269 source_repo = pr_or_version.source_repo
1269 source_repo = pr_or_version.source_repo
1270
1270
1271 # we swap org/other ref since we run a simple diff on one repo
1271 # we swap org/other ref since we run a simple diff on one repo
1272 target_ref_id = pr_or_version.target_ref_parts.commit_id
1272 target_ref_id = pr_or_version.target_ref_parts.commit_id
1273 source_ref_id = pr_or_version.source_ref_parts.commit_id
1273 source_ref_id = pr_or_version.source_ref_parts.commit_id
1274 target_commit = source_repo.get_commit(
1274 target_commit = source_repo.get_commit(
1275 commit_id=safe_str(target_ref_id))
1275 commit_id=safe_str(target_ref_id))
1276 source_commit = source_repo.get_commit(commit_id=safe_str(source_ref_id))
1276 source_commit = source_repo.get_commit(commit_id=safe_str(source_ref_id))
1277 vcs_repo = source_repo.scm_instance()
1277 vcs_repo = source_repo.scm_instance()
1278
1278
1279 # TODO: johbo: In the context of an update, we cannot reach
1279 # TODO: johbo: In the context of an update, we cannot reach
1280 # the old commit anymore with our normal mechanisms. It needs
1280 # the old commit anymore with our normal mechanisms. It needs
1281 # some sort of special support in the vcs layer to avoid this
1281 # some sort of special support in the vcs layer to avoid this
1282 # workaround.
1282 # workaround.
1283 if (source_commit.raw_id == vcs_repo.EMPTY_COMMIT_ID and
1283 if (source_commit.raw_id == vcs_repo.EMPTY_COMMIT_ID and
1284 vcs_repo.alias == 'git'):
1284 vcs_repo.alias == 'git'):
1285 source_commit.raw_id = safe_str(source_ref_id)
1285 source_commit.raw_id = safe_str(source_ref_id)
1286
1286
1287 log.debug('calculating diff between '
1287 log.debug('calculating diff between '
1288 'source_ref:%s and target_ref:%s for repo `%s`',
1288 'source_ref:%s and target_ref:%s for repo `%s`',
1289 target_ref_id, source_ref_id,
1289 target_ref_id, source_ref_id,
1290 safe_unicode(vcs_repo.path))
1290 safe_unicode(vcs_repo.path))
1291
1291
1292 vcs_diff = vcs_repo.get_diff(
1292 vcs_diff = vcs_repo.get_diff(
1293 commit1=target_commit, commit2=source_commit, context=context)
1293 commit1=target_commit, commit2=source_commit, context=context)
1294 return vcs_diff
1294 return vcs_diff
1295
1295
1296 def _is_merge_enabled(self, pull_request):
1296 def _is_merge_enabled(self, pull_request):
1297 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1297 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1298 settings = settings_model.get_general_settings()
1298 settings = settings_model.get_general_settings()
1299 return settings.get('rhodecode_pr_merge_enabled', False)
1299 return settings.get('rhodecode_pr_merge_enabled', False)
1300
1300
1301 def _use_rebase_for_merging(self, pull_request):
1301 def _use_rebase_for_merging(self, pull_request):
1302 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1302 settings_model = VcsSettingsModel(repo=pull_request.target_repo)
1303 settings = settings_model.get_general_settings()
1303 settings = settings_model.get_general_settings()
1304 return settings.get('rhodecode_hg_use_rebase_for_merging', False)
1304 return settings.get('rhodecode_hg_use_rebase_for_merging', False)
1305
1305
1306 def _log_action(self, action, user, pull_request):
1306 def _log_action(self, action, user, pull_request):
1307 action_logger(
1307 action_logger(
1308 user,
1308 user,
1309 '{action}:{pr_id}'.format(
1309 '{action}:{pr_id}'.format(
1310 action=action, pr_id=pull_request.pull_request_id),
1310 action=action, pr_id=pull_request.pull_request_id),
1311 pull_request.target_repo)
1311 pull_request.target_repo)
1312
1312
1313
1313
1314 class MergeCheck(object):
1314 class MergeCheck(object):
1315 """
1315 """
1316 Perform Merge Checks and returns a check object which stores information
1316 Perform Merge Checks and returns a check object which stores information
1317 about merge errors, and merge conditions
1317 about merge errors, and merge conditions
1318 """
1318 """
1319 TODO_CHECK = 'todo'
1320 PERM_CHECK = 'perm'
1321 REVIEW_CHECK = 'review'
1322 MERGE_CHECK = 'merge'
1319
1323
1320 def __init__(self):
1324 def __init__(self):
1321 self.merge_possible = None
1325 self.merge_possible = None
1322 self.merge_msg = ''
1326 self.merge_msg = ''
1323 self.failed = None
1327 self.failed = None
1324 self.errors = []
1328 self.errors = []
1329 self.error_details = OrderedDict()
1325
1330
1326 def push_error(self, error_type, message):
1331 def push_error(self, error_type, message, error_key, details):
1327 self.failed = True
1332 self.failed = True
1328 self.errors.append([error_type, message])
1333 self.errors.append([error_type, message])
1334 self.error_details[error_key] = dict(
1335 details=details,
1336 error_type=error_type,
1337 message=message
1338 )
1329
1339
1330 @classmethod
1340 @classmethod
1331 def validate(cls, pull_request, user, fail_early=False, translator=None):
1341 def validate(cls, pull_request, user, fail_early=False, translator=None):
1332 # if migrated to pyramid...
1342 # if migrated to pyramid...
1333 # _ = lambda: translator or _ # use passed in translator if any
1343 # _ = lambda: translator or _ # use passed in translator if any
1334
1344
1335 merge_check = cls()
1345 merge_check = cls()
1336
1346
1337 # permissions
1347 # permissions
1338 user_allowed_to_merge = PullRequestModel().check_user_merge(
1348 user_allowed_to_merge = PullRequestModel().check_user_merge(
1339 pull_request, user)
1349 pull_request, user)
1340 if not user_allowed_to_merge:
1350 if not user_allowed_to_merge:
1341 log.debug("MergeCheck: cannot merge, approval is pending.")
1351 log.debug("MergeCheck: cannot merge, approval is pending.")
1342
1352
1343 msg = _('User `{}` not allowed to perform merge').format(user)
1353 msg = _('User `{}` not allowed to perform merge').format(user)
1344 merge_check.push_error('error', msg)
1354 merge_check.push_error('error', msg, cls.PERM_CHECK, user.username)
1345 if fail_early:
1355 if fail_early:
1346 return merge_check
1356 return merge_check
1347
1357
1348 # review status
1358 # review status
1349 review_status = pull_request.calculated_review_status()
1359 review_status = pull_request.calculated_review_status()
1350 status_approved = review_status == ChangesetStatus.STATUS_APPROVED
1360 status_approved = review_status == ChangesetStatus.STATUS_APPROVED
1351 if not status_approved:
1361 if not status_approved:
1352 log.debug("MergeCheck: cannot merge, approval is pending.")
1362 log.debug("MergeCheck: cannot merge, approval is pending.")
1353
1363
1354 msg = _('Pull request reviewer approval is pending.')
1364 msg = _('Pull request reviewer approval is pending.')
1355
1365
1356 merge_check.push_error('warning', msg)
1366 merge_check.push_error(
1367 'warning', msg, cls.REVIEW_CHECK, review_status)
1357
1368
1358 if fail_early:
1369 if fail_early:
1359 return merge_check
1370 return merge_check
1360
1371
1361 # left over TODOs
1372 # left over TODOs
1362 todos = CommentsModel().get_unresolved_todos(pull_request)
1373 todos = CommentsModel().get_unresolved_todos(pull_request)
1363 if todos:
1374 if todos:
1364 log.debug("MergeCheck: cannot merge, {} "
1375 log.debug("MergeCheck: cannot merge, {} "
1365 "unresolved todos left.".format(len(todos)))
1376 "unresolved todos left.".format(len(todos)))
1366
1377
1367 if len(todos) == 1:
1378 if len(todos) == 1:
1368 msg = _('Cannot merge, {} TODO still not resolved.').format(
1379 msg = _('Cannot merge, {} TODO still not resolved.').format(
1369 len(todos))
1380 len(todos))
1370 else:
1381 else:
1371 msg = _('Cannot merge, {} TODOs still not resolved.').format(
1382 msg = _('Cannot merge, {} TODOs still not resolved.').format(
1372 len(todos))
1383 len(todos))
1373
1384
1374 merge_check.push_error('warning', msg)
1385 merge_check.push_error('warning', msg, cls.TODO_CHECK, todos)
1375
1386
1376 if fail_early:
1387 if fail_early:
1377 return merge_check
1388 return merge_check
1378
1389
1379 # merge possible
1390 # merge possible
1380 merge_status, msg = PullRequestModel().merge_status(pull_request)
1391 merge_status, msg = PullRequestModel().merge_status(pull_request)
1381 merge_check.merge_possible = merge_status
1392 merge_check.merge_possible = merge_status
1382 merge_check.merge_msg = msg
1393 merge_check.merge_msg = msg
1383 if not merge_status:
1394 if not merge_status:
1384 log.debug(
1395 log.debug(
1385 "MergeCheck: cannot merge, pull request merge not possible.")
1396 "MergeCheck: cannot merge, pull request merge not possible.")
1386 merge_check.push_error('warning', msg)
1397 merge_check.push_error('warning', msg, cls.MERGE_CHECK, None)
1387
1398
1388 if fail_early:
1399 if fail_early:
1389 return merge_check
1400 return merge_check
1390
1401
1391 return merge_check
1402 return merge_check
1392
1403
1393
1404
1394 ChangeTuple = namedtuple('ChangeTuple',
1405 ChangeTuple = namedtuple('ChangeTuple',
1395 ['added', 'common', 'removed'])
1406 ['added', 'common', 'removed'])
1396
1407
1397 FileChangeTuple = namedtuple('FileChangeTuple',
1408 FileChangeTuple = namedtuple('FileChangeTuple',
1398 ['added', 'modified', 'removed'])
1409 ['added', 'modified', 'removed'])
General Comments 0
You need to be logged in to leave comments. Login now