##// END OF EJS Templates
reviewers: added validation and default review rules via API calls....
dan -
r1795:4b1dac88 default
parent child Browse files
Show More
@@ -1,297 +1,297 b''
1 1 # -*- coding: utf-8 -*-
2 2
3 3 # Copyright (C) 2010-2017 RhodeCode GmbH
4 4 #
5 5 # This program is free software: you can redistribute it and/or modify
6 6 # it under the terms of the GNU Affero General Public License, version 3
7 7 # (only), as published by the Free Software Foundation.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU Affero General Public License
15 15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 16 #
17 17 # This program is dual-licensed. If you wish to learn more about the
18 18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20 20
21 21 import pytest
22 22
23 23 from rhodecode.model.db import User
24 24 from rhodecode.model.pull_request import PullRequestModel
25 25 from rhodecode.model.repo import RepoModel
26 26 from rhodecode.model.user import UserModel
27 27 from rhodecode.tests import TEST_USER_ADMIN_LOGIN, TEST_USER_REGULAR_LOGIN
28 28 from rhodecode.api.tests.utils import build_data, api_call, assert_error
29 29
30 30
31 31 @pytest.mark.usefixtures("testuser_api", "app")
32 32 class TestCreatePullRequestApi(object):
33 33 finalizers = []
34 34
35 35 def teardown_method(self, method):
36 36 if self.finalizers:
37 37 for finalizer in self.finalizers:
38 38 finalizer()
39 39 self.finalizers = []
40 40
41 41 def test_create_with_wrong_data(self):
42 42 required_data = {
43 43 'source_repo': 'tests/source_repo',
44 44 'target_repo': 'tests/target_repo',
45 45 'source_ref': 'branch:default:initial',
46 46 'target_ref': 'branch:default:new-feature',
47 47 'title': 'Test PR 1'
48 48 }
49 49 for key in required_data:
50 50 data = required_data.copy()
51 51 data.pop(key)
52 52 id_, params = build_data(
53 53 self.apikey, 'create_pull_request', **data)
54 54 response = api_call(self.app, params)
55 55
56 56 expected = 'Missing non optional `{}` arg in JSON DATA'.format(key)
57 57 assert_error(id_, expected, given=response.body)
58 58
59 59 @pytest.mark.backends("git", "hg")
60 60 def test_create_with_correct_data(self, backend):
61 61 data = self._prepare_data(backend)
62 62 RepoModel().revoke_user_permission(
63 63 self.source.repo_name, User.DEFAULT_USER)
64 64 id_, params = build_data(
65 65 self.apikey_regular, 'create_pull_request', **data)
66 66 response = api_call(self.app, params)
67 67 expected_message = "Created new pull request `{title}`".format(
68 68 title=data['title'])
69 69 result = response.json
70 70 assert result['result']['msg'] == expected_message
71 71 pull_request_id = result['result']['pull_request_id']
72 72 pull_request = PullRequestModel().get(pull_request_id)
73 73 assert pull_request.title == data['title']
74 74 assert pull_request.description == data['description']
75 75 assert pull_request.source_ref == data['source_ref']
76 76 assert pull_request.target_ref == data['target_ref']
77 77 assert pull_request.source_repo.repo_name == data['source_repo']
78 78 assert pull_request.target_repo.repo_name == data['target_repo']
79 79 assert pull_request.revisions == [self.commit_ids['change']]
80 assert pull_request.reviewers == []
80 assert len(pull_request.reviewers) == 1
81 81
82 82 @pytest.mark.backends("git", "hg")
83 83 def test_create_with_empty_description(self, backend):
84 84 data = self._prepare_data(backend)
85 85 data.pop('description')
86 86 id_, params = build_data(
87 87 self.apikey_regular, 'create_pull_request', **data)
88 88 response = api_call(self.app, params)
89 89 expected_message = "Created new pull request `{title}`".format(
90 90 title=data['title'])
91 91 result = response.json
92 92 assert result['result']['msg'] == expected_message
93 93 pull_request_id = result['result']['pull_request_id']
94 94 pull_request = PullRequestModel().get(pull_request_id)
95 95 assert pull_request.description == ''
96 96
97 97 @pytest.mark.backends("git", "hg")
98 98 def test_create_with_reviewers_specified_by_names(
99 99 self, backend, no_notifications):
100 100 data = self._prepare_data(backend)
101 101 reviewers = [
102 102 {'username': TEST_USER_REGULAR_LOGIN,
103 103 'reasons': ['added manually']},
104 104 {'username': TEST_USER_ADMIN_LOGIN,
105 105 'reasons': ['added manually']},
106 106 ]
107 107 data['reviewers'] = reviewers
108 108 id_, params = build_data(
109 109 self.apikey_regular, 'create_pull_request', **data)
110 110 response = api_call(self.app, params)
111 111
112 112 expected_message = "Created new pull request `{title}`".format(
113 113 title=data['title'])
114 114 result = response.json
115 115 assert result['result']['msg'] == expected_message
116 116 pull_request_id = result['result']['pull_request_id']
117 117 pull_request = PullRequestModel().get(pull_request_id)
118 118 actual_reviewers = [
119 119 {'username': r.user.username,
120 120 'reasons': ['added manually'],
121 121 } for r in pull_request.reviewers
122 122 ]
123 123 assert sorted(actual_reviewers) == sorted(reviewers)
124 124
125 125 @pytest.mark.backends("git", "hg")
126 126 def test_create_with_reviewers_specified_by_ids(
127 127 self, backend, no_notifications):
128 128 data = self._prepare_data(backend)
129 129 reviewers = [
130 130 {'username': UserModel().get_by_username(
131 131 TEST_USER_REGULAR_LOGIN).user_id,
132 132 'reasons': ['added manually']},
133 133 {'username': UserModel().get_by_username(
134 134 TEST_USER_ADMIN_LOGIN).user_id,
135 135 'reasons': ['added manually']},
136 136 ]
137 137
138 138 data['reviewers'] = reviewers
139 139 id_, params = build_data(
140 140 self.apikey_regular, 'create_pull_request', **data)
141 141 response = api_call(self.app, params)
142 142
143 143 expected_message = "Created new pull request `{title}`".format(
144 144 title=data['title'])
145 145 result = response.json
146 146 assert result['result']['msg'] == expected_message
147 147 pull_request_id = result['result']['pull_request_id']
148 148 pull_request = PullRequestModel().get(pull_request_id)
149 149 actual_reviewers = [
150 150 {'username': r.user.user_id,
151 151 'reasons': ['added manually'],
152 152 } for r in pull_request.reviewers
153 153 ]
154 154 assert sorted(actual_reviewers) == sorted(reviewers)
155 155
156 156 @pytest.mark.backends("git", "hg")
157 157 def test_create_fails_when_the_reviewer_is_not_found(self, backend):
158 158 data = self._prepare_data(backend)
159 159 data['reviewers'] = [{'username': 'somebody'}]
160 160 id_, params = build_data(
161 161 self.apikey_regular, 'create_pull_request', **data)
162 162 response = api_call(self.app, params)
163 163 expected_message = 'user `somebody` does not exist'
164 164 assert_error(id_, expected_message, given=response.body)
165 165
166 166 @pytest.mark.backends("git", "hg")
167 167 def test_cannot_create_with_reviewers_in_wrong_format(self, backend):
168 168 data = self._prepare_data(backend)
169 169 reviewers = ','.join([TEST_USER_REGULAR_LOGIN, TEST_USER_ADMIN_LOGIN])
170 170 data['reviewers'] = reviewers
171 171 id_, params = build_data(
172 172 self.apikey_regular, 'create_pull_request', **data)
173 173 response = api_call(self.app, params)
174 174 expected_message = {u'': '"test_regular,test_admin" is not iterable'}
175 175 assert_error(id_, expected_message, given=response.body)
176 176
177 177 @pytest.mark.backends("git", "hg")
178 178 def test_create_with_no_commit_hashes(self, backend):
179 179 data = self._prepare_data(backend)
180 180 expected_source_ref = data['source_ref']
181 181 expected_target_ref = data['target_ref']
182 182 data['source_ref'] = 'branch:{}'.format(backend.default_branch_name)
183 183 data['target_ref'] = 'branch:{}'.format(backend.default_branch_name)
184 184 id_, params = build_data(
185 185 self.apikey_regular, 'create_pull_request', **data)
186 186 response = api_call(self.app, params)
187 187 expected_message = "Created new pull request `{title}`".format(
188 188 title=data['title'])
189 189 result = response.json
190 190 assert result['result']['msg'] == expected_message
191 191 pull_request_id = result['result']['pull_request_id']
192 192 pull_request = PullRequestModel().get(pull_request_id)
193 193 assert pull_request.source_ref == expected_source_ref
194 194 assert pull_request.target_ref == expected_target_ref
195 195
196 196 @pytest.mark.backends("git", "hg")
197 197 @pytest.mark.parametrize("data_key", ["source_repo", "target_repo"])
198 198 def test_create_fails_with_wrong_repo(self, backend, data_key):
199 199 repo_name = 'fake-repo'
200 200 data = self._prepare_data(backend)
201 201 data[data_key] = repo_name
202 202 id_, params = build_data(
203 203 self.apikey_regular, 'create_pull_request', **data)
204 204 response = api_call(self.app, params)
205 205 expected_message = 'repository `{}` does not exist'.format(repo_name)
206 206 assert_error(id_, expected_message, given=response.body)
207 207
208 208 @pytest.mark.backends("git", "hg")
209 209 @pytest.mark.parametrize("data_key", ["source_ref", "target_ref"])
210 210 def test_create_fails_with_non_existing_branch(self, backend, data_key):
211 211 branch_name = 'test-branch'
212 212 data = self._prepare_data(backend)
213 213 data[data_key] = "branch:{}".format(branch_name)
214 214 id_, params = build_data(
215 215 self.apikey_regular, 'create_pull_request', **data)
216 216 response = api_call(self.app, params)
217 217 expected_message = 'The specified branch `{}` does not exist'.format(
218 218 branch_name)
219 219 assert_error(id_, expected_message, given=response.body)
220 220
221 221 @pytest.mark.backends("git", "hg")
222 222 @pytest.mark.parametrize("data_key", ["source_ref", "target_ref"])
223 223 def test_create_fails_with_ref_in_a_wrong_format(self, backend, data_key):
224 224 data = self._prepare_data(backend)
225 225 ref = 'stange-ref'
226 226 data[data_key] = ref
227 227 id_, params = build_data(
228 228 self.apikey_regular, 'create_pull_request', **data)
229 229 response = api_call(self.app, params)
230 230 expected_message = (
231 231 'Ref `{ref}` given in a wrong format. Please check the API'
232 232 ' documentation for more details'.format(ref=ref))
233 233 assert_error(id_, expected_message, given=response.body)
234 234
235 235 @pytest.mark.backends("git", "hg")
236 236 @pytest.mark.parametrize("data_key", ["source_ref", "target_ref"])
237 237 def test_create_fails_with_non_existing_ref(self, backend, data_key):
238 238 commit_id = 'aaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaaa10'
239 239 ref = self._get_full_ref(backend, commit_id)
240 240 data = self._prepare_data(backend)
241 241 data[data_key] = ref
242 242 id_, params = build_data(
243 243 self.apikey_regular, 'create_pull_request', **data)
244 244 response = api_call(self.app, params)
245 245 expected_message = 'Ref `{}` does not exist'.format(ref)
246 246 assert_error(id_, expected_message, given=response.body)
247 247
248 248 @pytest.mark.backends("git", "hg")
249 249 def test_create_fails_when_no_revisions(self, backend):
250 250 data = self._prepare_data(backend, source_head='initial')
251 251 id_, params = build_data(
252 252 self.apikey_regular, 'create_pull_request', **data)
253 253 response = api_call(self.app, params)
254 254 expected_message = 'no commits found'
255 255 assert_error(id_, expected_message, given=response.body)
256 256
257 257 @pytest.mark.backends("git", "hg")
258 258 def test_create_fails_when_no_permissions(self, backend):
259 259 data = self._prepare_data(backend)
260 260 RepoModel().revoke_user_permission(
261 261 self.source.repo_name, User.DEFAULT_USER)
262 262 RepoModel().revoke_user_permission(
263 263 self.source.repo_name, self.test_user)
264 264 id_, params = build_data(
265 265 self.apikey_regular, 'create_pull_request', **data)
266 266 response = api_call(self.app, params)
267 267 expected_message = 'repository `{}` does not exist'.format(
268 268 self.source.repo_name)
269 269 assert_error(id_, expected_message, given=response.body)
270 270
271 271 def _prepare_data(
272 272 self, backend, source_head='change', target_head='initial'):
273 273 commits = [
274 274 {'message': 'initial'},
275 275 {'message': 'change'},
276 276 {'message': 'new-feature', 'parents': ['initial']},
277 277 ]
278 278 self.commit_ids = backend.create_master_repo(commits)
279 279 self.source = backend.create_repo(heads=[source_head])
280 280 self.target = backend.create_repo(heads=[target_head])
281 281 data = {
282 282 'source_repo': self.source.repo_name,
283 283 'target_repo': self.target.repo_name,
284 284 'source_ref': self._get_full_ref(
285 285 backend, self.commit_ids[source_head]),
286 286 'target_ref': self._get_full_ref(
287 287 backend, self.commit_ids[target_head]),
288 288 'title': 'Test PR 1',
289 289 'description': 'Test'
290 290 }
291 291 RepoModel().grant_user_permission(
292 292 self.source.repo_name, self.TEST_USER_LOGIN, 'repository.read')
293 293 return data
294 294
295 295 def _get_full_ref(self, backend, commit_id):
296 296 return 'branch:{branch}:{commit_id}'.format(
297 297 branch=backend.default_branch_name, commit_id=commit_id)
@@ -1,748 +1,779 b''
1 1 # -*- coding: utf-8 -*-
2 2
3 3 # Copyright (C) 2011-2017 RhodeCode GmbH
4 4 #
5 5 # This program is free software: you can redistribute it and/or modify
6 6 # it under the terms of the GNU Affero General Public License, version 3
7 7 # (only), as published by the Free Software Foundation.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU Affero General Public License
15 15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 16 #
17 17 # This program is dual-licensed. If you wish to learn more about the
18 18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20 20
21 21
22 22 import logging
23 23
24 24 from rhodecode import events
25 25 from rhodecode.api import jsonrpc_method, JSONRPCError, JSONRPCValidationError
26 26 from rhodecode.api.utils import (
27 27 has_superadmin_permission, Optional, OAttr, get_repo_or_error,
28 28 get_pull_request_or_error, get_commit_or_error, get_user_or_error,
29 29 validate_repo_permissions, resolve_ref_or_error)
30 30 from rhodecode.lib.auth import (HasRepoPermissionAnyApi)
31 31 from rhodecode.lib.base import vcs_operation_context
32 32 from rhodecode.lib.utils2 import str2bool
33 33 from rhodecode.model.changeset_status import ChangesetStatusModel
34 34 from rhodecode.model.comment import CommentsModel
35 35 from rhodecode.model.db import Session, ChangesetStatus, ChangesetComment
36 36 from rhodecode.model.pull_request import PullRequestModel, MergeCheck
37 37 from rhodecode.model.settings import SettingsModel
38 38 from rhodecode.model.validation_schema import Invalid
39 39 from rhodecode.model.validation_schema.schemas.reviewer_schema import(
40 40 ReviewerListSchema)
41 41
42 42 log = logging.getLogger(__name__)
43 43
44 44
45 45 @jsonrpc_method()
46 46 def get_pull_request(request, apiuser, repoid, pullrequestid):
47 47 """
48 48 Get a pull request based on the given ID.
49 49
50 50 :param apiuser: This is filled automatically from the |authtoken|.
51 51 :type apiuser: AuthUser
52 52 :param repoid: Repository name or repository ID from where the pull
53 53 request was opened.
54 54 :type repoid: str or int
55 55 :param pullrequestid: ID of the requested pull request.
56 56 :type pullrequestid: int
57 57
58 58 Example output:
59 59
60 60 .. code-block:: bash
61 61
62 62 "id": <id_given_in_input>,
63 63 "result":
64 64 {
65 65 "pull_request_id": "<pull_request_id>",
66 66 "url": "<url>",
67 67 "title": "<title>",
68 68 "description": "<description>",
69 69 "status" : "<status>",
70 70 "created_on": "<date_time_created>",
71 71 "updated_on": "<date_time_updated>",
72 72 "commit_ids": [
73 73 ...
74 74 "<commit_id>",
75 75 "<commit_id>",
76 76 ...
77 77 ],
78 78 "review_status": "<review_status>",
79 79 "mergeable": {
80 80 "status": "<bool>",
81 81 "message": "<message>",
82 82 },
83 83 "source": {
84 84 "clone_url": "<clone_url>",
85 85 "repository": "<repository_name>",
86 86 "reference":
87 87 {
88 88 "name": "<name>",
89 89 "type": "<type>",
90 90 "commit_id": "<commit_id>",
91 91 }
92 92 },
93 93 "target": {
94 94 "clone_url": "<clone_url>",
95 95 "repository": "<repository_name>",
96 96 "reference":
97 97 {
98 98 "name": "<name>",
99 99 "type": "<type>",
100 100 "commit_id": "<commit_id>",
101 101 }
102 102 },
103 103 "merge": {
104 104 "clone_url": "<clone_url>",
105 105 "reference":
106 106 {
107 107 "name": "<name>",
108 108 "type": "<type>",
109 109 "commit_id": "<commit_id>",
110 110 }
111 111 },
112 112 "author": <user_obj>,
113 113 "reviewers": [
114 114 ...
115 115 {
116 116 "user": "<user_obj>",
117 117 "review_status": "<review_status>",
118 118 }
119 119 ...
120 120 ]
121 121 },
122 122 "error": null
123 123 """
124 124 get_repo_or_error(repoid)
125 125 pull_request = get_pull_request_or_error(pullrequestid)
126 126 if not PullRequestModel().check_user_read(
127 127 pull_request, apiuser, api=True):
128 128 raise JSONRPCError('repository `%s` does not exist' % (repoid,))
129 129 data = pull_request.get_api_data()
130 130 return data
131 131
132 132
133 133 @jsonrpc_method()
134 134 def get_pull_requests(request, apiuser, repoid, status=Optional('new')):
135 135 """
136 136 Get all pull requests from the repository specified in `repoid`.
137 137
138 138 :param apiuser: This is filled automatically from the |authtoken|.
139 139 :type apiuser: AuthUser
140 140 :param repoid: Repository name or repository ID.
141 141 :type repoid: str or int
142 142 :param status: Only return pull requests with the specified status.
143 143 Valid options are.
144 144 * ``new`` (default)
145 145 * ``open``
146 146 * ``closed``
147 147 :type status: str
148 148
149 149 Example output:
150 150
151 151 .. code-block:: bash
152 152
153 153 "id": <id_given_in_input>,
154 154 "result":
155 155 [
156 156 ...
157 157 {
158 158 "pull_request_id": "<pull_request_id>",
159 159 "url": "<url>",
160 160 "title" : "<title>",
161 161 "description": "<description>",
162 162 "status": "<status>",
163 163 "created_on": "<date_time_created>",
164 164 "updated_on": "<date_time_updated>",
165 165 "commit_ids": [
166 166 ...
167 167 "<commit_id>",
168 168 "<commit_id>",
169 169 ...
170 170 ],
171 171 "review_status": "<review_status>",
172 172 "mergeable": {
173 173 "status": "<bool>",
174 174 "message: "<message>",
175 175 },
176 176 "source": {
177 177 "clone_url": "<clone_url>",
178 178 "reference":
179 179 {
180 180 "name": "<name>",
181 181 "type": "<type>",
182 182 "commit_id": "<commit_id>",
183 183 }
184 184 },
185 185 "target": {
186 186 "clone_url": "<clone_url>",
187 187 "reference":
188 188 {
189 189 "name": "<name>",
190 190 "type": "<type>",
191 191 "commit_id": "<commit_id>",
192 192 }
193 193 },
194 194 "merge": {
195 195 "clone_url": "<clone_url>",
196 196 "reference":
197 197 {
198 198 "name": "<name>",
199 199 "type": "<type>",
200 200 "commit_id": "<commit_id>",
201 201 }
202 202 },
203 203 "author": <user_obj>,
204 204 "reviewers": [
205 205 ...
206 206 {
207 207 "user": "<user_obj>",
208 208 "review_status": "<review_status>",
209 209 }
210 210 ...
211 211 ]
212 212 }
213 213 ...
214 214 ],
215 215 "error": null
216 216
217 217 """
218 218 repo = get_repo_or_error(repoid)
219 219 if not has_superadmin_permission(apiuser):
220 220 _perms = (
221 221 'repository.admin', 'repository.write', 'repository.read',)
222 222 validate_repo_permissions(apiuser, repoid, repo, _perms)
223 223
224 224 status = Optional.extract(status)
225 225 pull_requests = PullRequestModel().get_all(repo, statuses=[status])
226 226 data = [pr.get_api_data() for pr in pull_requests]
227 227 return data
228 228
229 229
230 230 @jsonrpc_method()
231 231 def merge_pull_request(
232 232 request, apiuser, repoid, pullrequestid,
233 233 userid=Optional(OAttr('apiuser'))):
234 234 """
235 235 Merge the pull request specified by `pullrequestid` into its target
236 236 repository.
237 237
238 238 :param apiuser: This is filled automatically from the |authtoken|.
239 239 :type apiuser: AuthUser
240 240 :param repoid: The Repository name or repository ID of the
241 241 target repository to which the |pr| is to be merged.
242 242 :type repoid: str or int
243 243 :param pullrequestid: ID of the pull request which shall be merged.
244 244 :type pullrequestid: int
245 245 :param userid: Merge the pull request as this user.
246 246 :type userid: Optional(str or int)
247 247
248 248 Example output:
249 249
250 250 .. code-block:: bash
251 251
252 252 "id": <id_given_in_input>,
253 253 "result": {
254 254 "executed": "<bool>",
255 255 "failure_reason": "<int>",
256 256 "merge_commit_id": "<merge_commit_id>",
257 257 "possible": "<bool>",
258 258 "merge_ref": {
259 259 "commit_id": "<commit_id>",
260 260 "type": "<type>",
261 261 "name": "<name>"
262 262 }
263 263 },
264 264 "error": null
265 265 """
266 266 repo = get_repo_or_error(repoid)
267 267 if not isinstance(userid, Optional):
268 268 if (has_superadmin_permission(apiuser) or
269 269 HasRepoPermissionAnyApi('repository.admin')(
270 270 user=apiuser, repo_name=repo.repo_name)):
271 271 apiuser = get_user_or_error(userid)
272 272 else:
273 273 raise JSONRPCError('userid is not the same as your user')
274 274
275 275 pull_request = get_pull_request_or_error(pullrequestid)
276 276
277 277 check = MergeCheck.validate(pull_request, user=apiuser)
278 278 merge_possible = not check.failed
279 279
280 280 if not merge_possible:
281 281 error_messages = []
282 282 for err_type, error_msg in check.errors:
283 283 error_msg = request.translate(error_msg)
284 284 error_messages.append(error_msg)
285 285
286 286 reasons = ','.join(error_messages)
287 287 raise JSONRPCError(
288 288 'merge not possible for following reasons: {}'.format(reasons))
289 289
290 290 target_repo = pull_request.target_repo
291 291 extras = vcs_operation_context(
292 292 request.environ, repo_name=target_repo.repo_name,
293 293 username=apiuser.username, action='push',
294 294 scm=target_repo.repo_type)
295 295 merge_response = PullRequestModel().merge(
296 296 pull_request, apiuser, extras=extras)
297 297 if merge_response.executed:
298 298 PullRequestModel().close_pull_request(
299 299 pull_request.pull_request_id, apiuser)
300 300
301 301 Session().commit()
302 302
303 303 # In previous versions the merge response directly contained the merge
304 304 # commit id. It is now contained in the merge reference object. To be
305 305 # backwards compatible we have to extract it again.
306 306 merge_response = merge_response._asdict()
307 307 merge_response['merge_commit_id'] = merge_response['merge_ref'].commit_id
308 308
309 309 return merge_response
310 310
311 311
312 312 @jsonrpc_method()
313 313 def comment_pull_request(
314 314 request, apiuser, repoid, pullrequestid, message=Optional(None),
315 315 commit_id=Optional(None), status=Optional(None),
316 316 comment_type=Optional(ChangesetComment.COMMENT_TYPE_NOTE),
317 317 resolves_comment_id=Optional(None),
318 318 userid=Optional(OAttr('apiuser'))):
319 319 """
320 320 Comment on the pull request specified with the `pullrequestid`,
321 321 in the |repo| specified by the `repoid`, and optionally change the
322 322 review status.
323 323
324 324 :param apiuser: This is filled automatically from the |authtoken|.
325 325 :type apiuser: AuthUser
326 326 :param repoid: The repository name or repository ID.
327 327 :type repoid: str or int
328 328 :param pullrequestid: The pull request ID.
329 329 :type pullrequestid: int
330 330 :param commit_id: Specify the commit_id for which to set a comment. If
331 331 given commit_id is different than latest in the PR status
332 332 change won't be performed.
333 333 :type commit_id: str
334 334 :param message: The text content of the comment.
335 335 :type message: str
336 336 :param status: (**Optional**) Set the approval status of the pull
337 337 request. One of: 'not_reviewed', 'approved', 'rejected',
338 338 'under_review'
339 339 :type status: str
340 340 :param comment_type: Comment type, one of: 'note', 'todo'
341 341 :type comment_type: Optional(str), default: 'note'
342 342 :param userid: Comment on the pull request as this user
343 343 :type userid: Optional(str or int)
344 344
345 345 Example output:
346 346
347 347 .. code-block:: bash
348 348
349 349 id : <id_given_in_input>
350 350 result : {
351 351 "pull_request_id": "<Integer>",
352 352 "comment_id": "<Integer>",
353 353 "status": {"given": <given_status>,
354 354 "was_changed": <bool status_was_actually_changed> },
355 355 },
356 356 error : null
357 357 """
358 358 repo = get_repo_or_error(repoid)
359 359 if not isinstance(userid, Optional):
360 360 if (has_superadmin_permission(apiuser) or
361 361 HasRepoPermissionAnyApi('repository.admin')(
362 362 user=apiuser, repo_name=repo.repo_name)):
363 363 apiuser = get_user_or_error(userid)
364 364 else:
365 365 raise JSONRPCError('userid is not the same as your user')
366 366
367 367 pull_request = get_pull_request_or_error(pullrequestid)
368 368 if not PullRequestModel().check_user_read(
369 369 pull_request, apiuser, api=True):
370 370 raise JSONRPCError('repository `%s` does not exist' % (repoid,))
371 371 message = Optional.extract(message)
372 372 status = Optional.extract(status)
373 373 commit_id = Optional.extract(commit_id)
374 374 comment_type = Optional.extract(comment_type)
375 375 resolves_comment_id = Optional.extract(resolves_comment_id)
376 376
377 377 if not message and not status:
378 378 raise JSONRPCError(
379 379 'Both message and status parameters are missing. '
380 380 'At least one is required.')
381 381
382 382 if (status not in (st[0] for st in ChangesetStatus.STATUSES) and
383 383 status is not None):
384 384 raise JSONRPCError('Unknown comment status: `%s`' % status)
385 385
386 386 if commit_id and commit_id not in pull_request.revisions:
387 387 raise JSONRPCError(
388 388 'Invalid commit_id `%s` for this pull request.' % commit_id)
389 389
390 390 allowed_to_change_status = PullRequestModel().check_user_change_status(
391 391 pull_request, apiuser)
392 392
393 393 # if commit_id is passed re-validated if user is allowed to change status
394 394 # based on latest commit_id from the PR
395 395 if commit_id:
396 396 commit_idx = pull_request.revisions.index(commit_id)
397 397 if commit_idx != 0:
398 398 allowed_to_change_status = False
399 399
400 400 if resolves_comment_id:
401 401 comment = ChangesetComment.get(resolves_comment_id)
402 402 if not comment:
403 403 raise JSONRPCError(
404 404 'Invalid resolves_comment_id `%s` for this pull request.'
405 405 % resolves_comment_id)
406 406 if comment.comment_type != ChangesetComment.COMMENT_TYPE_TODO:
407 407 raise JSONRPCError(
408 408 'Comment `%s` is wrong type for setting status to resolved.'
409 409 % resolves_comment_id)
410 410
411 411 text = message
412 412 status_label = ChangesetStatus.get_status_lbl(status)
413 413 if status and allowed_to_change_status:
414 414 st_message = ('Status change %(transition_icon)s %(status)s'
415 415 % {'transition_icon': '>', 'status': status_label})
416 416 text = message or st_message
417 417
418 418 rc_config = SettingsModel().get_all_settings()
419 419 renderer = rc_config.get('rhodecode_markup_renderer', 'rst')
420 420
421 421 status_change = status and allowed_to_change_status
422 422 comment = CommentsModel().create(
423 423 text=text,
424 424 repo=pull_request.target_repo.repo_id,
425 425 user=apiuser.user_id,
426 426 pull_request=pull_request.pull_request_id,
427 427 f_path=None,
428 428 line_no=None,
429 429 status_change=(status_label if status_change else None),
430 430 status_change_type=(status if status_change else None),
431 431 closing_pr=False,
432 432 renderer=renderer,
433 433 comment_type=comment_type,
434 434 resolves_comment_id=resolves_comment_id
435 435 )
436 436
437 437 if allowed_to_change_status and status:
438 438 ChangesetStatusModel().set_status(
439 439 pull_request.target_repo.repo_id,
440 440 status,
441 441 apiuser.user_id,
442 442 comment,
443 443 pull_request=pull_request.pull_request_id
444 444 )
445 445 Session().flush()
446 446
447 447 Session().commit()
448 448 data = {
449 449 'pull_request_id': pull_request.pull_request_id,
450 450 'comment_id': comment.comment_id if comment else None,
451 451 'status': {'given': status, 'was_changed': status_change},
452 452 }
453 453 return data
454 454
455 455
456 456 @jsonrpc_method()
457 457 def create_pull_request(
458 458 request, apiuser, source_repo, target_repo, source_ref, target_ref,
459 459 title, description=Optional(''), reviewers=Optional(None)):
460 460 """
461 461 Creates a new pull request.
462 462
463 463 Accepts refs in the following formats:
464 464
465 465 * branch:<branch_name>:<sha>
466 466 * branch:<branch_name>
467 467 * bookmark:<bookmark_name>:<sha> (Mercurial only)
468 468 * bookmark:<bookmark_name> (Mercurial only)
469 469
470 470 :param apiuser: This is filled automatically from the |authtoken|.
471 471 :type apiuser: AuthUser
472 472 :param source_repo: Set the source repository name.
473 473 :type source_repo: str
474 474 :param target_repo: Set the target repository name.
475 475 :type target_repo: str
476 476 :param source_ref: Set the source ref name.
477 477 :type source_ref: str
478 478 :param target_ref: Set the target ref name.
479 479 :type target_ref: str
480 480 :param title: Set the pull request title.
481 481 :type title: str
482 482 :param description: Set the pull request description.
483 483 :type description: Optional(str)
484 484 :param reviewers: Set the new pull request reviewers list.
485 Reviewer defined by review rules will be added automatically to the
486 defined list.
485 487 :type reviewers: Optional(list)
486 488 Accepts username strings or objects of the format:
487 489
488 {'username': 'nick', 'reasons': ['original author'], 'mandatory': <bool>}
490 [{'username': 'nick', 'reasons': ['original author'], 'mandatory': <bool>}]
489 491 """
490 492
491 source = get_repo_or_error(source_repo)
492 target = get_repo_or_error(target_repo)
493 source_db_repo = get_repo_or_error(source_repo)
494 target_db_repo = get_repo_or_error(target_repo)
493 495 if not has_superadmin_permission(apiuser):
494 496 _perms = ('repository.admin', 'repository.write', 'repository.read',)
495 validate_repo_permissions(apiuser, source_repo, source, _perms)
497 validate_repo_permissions(apiuser, source_repo, source_db_repo, _perms)
496 498
497 full_source_ref = resolve_ref_or_error(source_ref, source)
498 full_target_ref = resolve_ref_or_error(target_ref, target)
499 source_commit = get_commit_or_error(full_source_ref, source)
500 target_commit = get_commit_or_error(full_target_ref, target)
501 source_scm = source.scm_instance()
502 target_scm = target.scm_instance()
499 full_source_ref = resolve_ref_or_error(source_ref, source_db_repo)
500 full_target_ref = resolve_ref_or_error(target_ref, target_db_repo)
501 source_commit = get_commit_or_error(full_source_ref, source_db_repo)
502 target_commit = get_commit_or_error(full_target_ref, target_db_repo)
503 source_scm = source_db_repo.scm_instance()
504 target_scm = target_db_repo.scm_instance()
503 505
504 506 commit_ranges = target_scm.compare(
505 507 target_commit.raw_id, source_commit.raw_id, source_scm,
506 508 merge=True, pre_load=[])
507 509
508 510 ancestor = target_scm.get_common_ancestor(
509 511 target_commit.raw_id, source_commit.raw_id, source_scm)
510 512
511 513 if not commit_ranges:
512 514 raise JSONRPCError('no commits found')
513 515
514 516 if not ancestor:
515 517 raise JSONRPCError('no common ancestor found')
516 518
517 519 reviewer_objects = Optional.extract(reviewers) or []
520
518 521 if reviewer_objects:
519 522 schema = ReviewerListSchema()
520 523 try:
521 524 reviewer_objects = schema.deserialize(reviewer_objects)
522 525 except Invalid as err:
523 526 raise JSONRPCValidationError(colander_exc=err)
524 527
525 reviewers = []
526 for reviewer_object in reviewer_objects:
527 user = get_user_or_error(reviewer_object['username'])
528 reasons = reviewer_object['reasons']
529 mandatory = reviewer_object['mandatory']
530 reviewers.append((user.user_id, reasons, mandatory))
528 # validate users
529 for reviewer_object in reviewer_objects:
530 user = get_user_or_error(reviewer_object['username'])
531 reviewer_object['user_id'] = user.user_id
532
533 get_default_reviewers_data, get_validated_reviewers = \
534 PullRequestModel().get_reviewer_functions()
535
536 reviewer_rules = get_default_reviewers_data(
537 apiuser.get_instance(), source_db_repo,
538 source_commit, target_db_repo, target_commit)
539
540 # specified rules are later re-validated, thus we can assume users will
541 # eventually provide those that meet the reviewer criteria.
542 if not reviewer_objects:
543 reviewer_objects = reviewer_rules['reviewers']
544
545 try:
546 reviewers = get_validated_reviewers(
547 reviewer_objects, reviewer_rules)
548 except ValueError as e:
549 raise JSONRPCError('Reviewers Validation: {}'.format(e))
531 550
532 551 pull_request_model = PullRequestModel()
533 552 pull_request = pull_request_model.create(
534 553 created_by=apiuser.user_id,
535 554 source_repo=source_repo,
536 555 source_ref=full_source_ref,
537 556 target_repo=target_repo,
538 557 target_ref=full_target_ref,
539 558 revisions=reversed(
540 559 [commit.raw_id for commit in reversed(commit_ranges)]),
541 560 reviewers=reviewers,
542 561 title=title,
543 562 description=Optional.extract(description)
544 563 )
545 564
546 565 Session().commit()
547 566 data = {
548 567 'msg': 'Created new pull request `{}`'.format(title),
549 568 'pull_request_id': pull_request.pull_request_id,
550 569 }
551 570 return data
552 571
553 572
554 573 @jsonrpc_method()
555 574 def update_pull_request(
556 575 request, apiuser, repoid, pullrequestid, title=Optional(''),
557 576 description=Optional(''), reviewers=Optional(None),
558 577 update_commits=Optional(None)):
559 578 """
560 579 Updates a pull request.
561 580
562 581 :param apiuser: This is filled automatically from the |authtoken|.
563 582 :type apiuser: AuthUser
564 583 :param repoid: The repository name or repository ID.
565 584 :type repoid: str or int
566 585 :param pullrequestid: The pull request ID.
567 586 :type pullrequestid: int
568 587 :param title: Set the pull request title.
569 588 :type title: str
570 589 :param description: Update pull request description.
571 590 :type description: Optional(str)
572 591 :param reviewers: Update pull request reviewers list with new value.
573 592 :type reviewers: Optional(list)
574 593 Accepts username strings or objects of the format:
575 594
576 {'username': 'nick', 'reasons': ['original author'], 'mandatory': <bool>}
595 [{'username': 'nick', 'reasons': ['original author'], 'mandatory': <bool>}]
577 596
578 597 :param update_commits: Trigger update of commits for this pull request
579 598 :type: update_commits: Optional(bool)
580 599
581 600 Example output:
582 601
583 602 .. code-block:: bash
584 603
585 604 id : <id_given_in_input>
586 605 result : {
587 606 "msg": "Updated pull request `63`",
588 607 "pull_request": <pull_request_object>,
589 608 "updated_reviewers": {
590 609 "added": [
591 610 "username"
592 611 ],
593 612 "removed": []
594 613 },
595 614 "updated_commits": {
596 615 "added": [
597 616 "<sha1_hash>"
598 617 ],
599 618 "common": [
600 619 "<sha1_hash>",
601 620 "<sha1_hash>",
602 621 ],
603 622 "removed": []
604 623 }
605 624 }
606 625 error : null
607 626 """
608 627
609 628 repo = get_repo_or_error(repoid)
610 629 pull_request = get_pull_request_or_error(pullrequestid)
611 630 if not PullRequestModel().check_user_update(
612 631 pull_request, apiuser, api=True):
613 632 raise JSONRPCError(
614 633 'pull request `%s` update failed, no permission to update.' % (
615 634 pullrequestid,))
616 635 if pull_request.is_closed():
617 636 raise JSONRPCError(
618 637 'pull request `%s` update failed, pull request is closed' % (
619 638 pullrequestid,))
620 639
621 640 reviewer_objects = Optional.extract(reviewers) or []
641
622 642 if reviewer_objects:
623 643 schema = ReviewerListSchema()
624 644 try:
625 645 reviewer_objects = schema.deserialize(reviewer_objects)
626 646 except Invalid as err:
627 647 raise JSONRPCValidationError(colander_exc=err)
628 648
629 reviewers = []
630 for reviewer_object in reviewer_objects:
631 user = get_user_or_error(reviewer_object['username'])
632 reasons = reviewer_object['reasons']
633 mandatory = reviewer_object['mandatory']
634 reviewers.append((user.user_id, reasons, mandatory))
649 # validate users
650 for reviewer_object in reviewer_objects:
651 user = get_user_or_error(reviewer_object['username'])
652 reviewer_object['user_id'] = user.user_id
653
654 get_default_reviewers_data, get_validated_reviewers = \
655 PullRequestModel().get_reviewer_functions()
656
657 # re-use stored rules
658 reviewer_rules = pull_request.reviewer_data
659 try:
660 reviewers = get_validated_reviewers(
661 reviewer_objects, reviewer_rules)
662 except ValueError as e:
663 raise JSONRPCError('Reviewers Validation: {}'.format(e))
664 else:
665 reviewers = []
635 666
636 667 title = Optional.extract(title)
637 668 description = Optional.extract(description)
638 669 if title or description:
639 670 PullRequestModel().edit(
640 671 pull_request, title or pull_request.title,
641 672 description or pull_request.description)
642 673 Session().commit()
643 674
644 675 commit_changes = {"added": [], "common": [], "removed": []}
645 676 if str2bool(Optional.extract(update_commits)):
646 677 if PullRequestModel().has_valid_update_type(pull_request):
647 678 update_response = PullRequestModel().update_commits(
648 679 pull_request)
649 680 commit_changes = update_response.changes or commit_changes
650 681 Session().commit()
651 682
652 683 reviewers_changes = {"added": [], "removed": []}
653 684 if reviewers:
654 685 added_reviewers, removed_reviewers = \
655 686 PullRequestModel().update_reviewers(pull_request, reviewers)
656 687
657 688 reviewers_changes['added'] = sorted(
658 689 [get_user_or_error(n).username for n in added_reviewers])
659 690 reviewers_changes['removed'] = sorted(
660 691 [get_user_or_error(n).username for n in removed_reviewers])
661 692 Session().commit()
662 693
663 694 data = {
664 695 'msg': 'Updated pull request `{}`'.format(
665 696 pull_request.pull_request_id),
666 697 'pull_request': pull_request.get_api_data(),
667 698 'updated_commits': commit_changes,
668 699 'updated_reviewers': reviewers_changes
669 700 }
670 701
671 702 return data
672 703
673 704
674 705 @jsonrpc_method()
675 706 def close_pull_request(
676 707 request, apiuser, repoid, pullrequestid,
677 708 userid=Optional(OAttr('apiuser')), message=Optional('')):
678 709 """
679 710 Close the pull request specified by `pullrequestid`.
680 711
681 712 :param apiuser: This is filled automatically from the |authtoken|.
682 713 :type apiuser: AuthUser
683 714 :param repoid: Repository name or repository ID to which the pull
684 715 request belongs.
685 716 :type repoid: str or int
686 717 :param pullrequestid: ID of the pull request to be closed.
687 718 :type pullrequestid: int
688 719 :param userid: Close the pull request as this user.
689 720 :type userid: Optional(str or int)
690 721 :param message: Optional message to close the Pull Request with. If not
691 722 specified it will be generated automatically.
692 723 :type message: Optional(str)
693 724
694 725 Example output:
695 726
696 727 .. code-block:: bash
697 728
698 729 "id": <id_given_in_input>,
699 730 "result": {
700 731 "pull_request_id": "<int>",
701 732 "close_status": "<str:status_lbl>,
702 733 "closed": "<bool>"
703 734 },
704 735 "error": null
705 736
706 737 """
707 738 _ = request.translate
708 739
709 740 repo = get_repo_or_error(repoid)
710 741 if not isinstance(userid, Optional):
711 742 if (has_superadmin_permission(apiuser) or
712 743 HasRepoPermissionAnyApi('repository.admin')(
713 744 user=apiuser, repo_name=repo.repo_name)):
714 745 apiuser = get_user_or_error(userid)
715 746 else:
716 747 raise JSONRPCError('userid is not the same as your user')
717 748
718 749 pull_request = get_pull_request_or_error(pullrequestid)
719 750
720 751 if pull_request.is_closed():
721 752 raise JSONRPCError(
722 753 'pull request `%s` is already closed' % (pullrequestid,))
723 754
724 755 # only owner or admin or person with write permissions
725 756 allowed_to_close = PullRequestModel().check_user_update(
726 757 pull_request, apiuser, api=True)
727 758
728 759 if not allowed_to_close:
729 760 raise JSONRPCError(
730 761 'pull request `%s` close failed, no permission to close.' % (
731 762 pullrequestid,))
732 763
733 764 # message we're using to close the PR, else it's automatically generated
734 765 message = Optional.extract(message)
735 766
736 767 # finally close the PR, with proper message comment
737 768 comment, status = PullRequestModel().close_pull_request_with_comment(
738 769 pull_request, apiuser, repo, message=message)
739 770 status_lbl = ChangesetStatus.get_status_lbl(status)
740 771
741 772 Session().commit()
742 773
743 774 data = {
744 775 'pull_request_id': pull_request.pull_request_id,
745 776 'close_status': status_lbl,
746 777 'closed': True,
747 778 }
748 779 return data
@@ -1,1008 +1,1008 b''
1 1 # -*- coding: utf-8 -*-
2 2
3 3 # Copyright (C) 2012-2017 RhodeCode GmbH
4 4 #
5 5 # This program is free software: you can redistribute it and/or modify
6 6 # it under the terms of the GNU Affero General Public License, version 3
7 7 # (only), as published by the Free Software Foundation.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU Affero General Public License
15 15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 16 #
17 17 # This program is dual-licensed. If you wish to learn more about the
18 18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20 20
21 21 """
22 22 pull requests controller for rhodecode for initializing pull requests
23 23 """
24 24 import types
25 25
26 26 import peppercorn
27 27 import formencode
28 28 import logging
29 29 import collections
30 30
31 31 from webob.exc import HTTPNotFound, HTTPForbidden, HTTPBadRequest
32 32 from pylons import request, tmpl_context as c, url
33 33 from pylons.controllers.util import redirect
34 34 from pylons.i18n.translation import _
35 35 from pyramid.threadlocal import get_current_registry
36 36 from sqlalchemy.sql import func
37 37 from sqlalchemy.sql.expression import or_
38 38
39 39 from rhodecode import events
40 40 from rhodecode.lib import auth, diffs, helpers as h, codeblocks
41 41 from rhodecode.lib.ext_json import json
42 42 from rhodecode.lib.base import (
43 43 BaseRepoController, render, vcs_operation_context)
44 44 from rhodecode.lib.auth import (
45 45 LoginRequired, HasRepoPermissionAnyDecorator, NotAnonymous,
46 46 HasAcceptedRepoType, XHRRequired)
47 47 from rhodecode.lib.channelstream import channelstream_request
48 48 from rhodecode.lib.utils import jsonify
49 49 from rhodecode.lib.utils2 import (
50 50 safe_int, safe_str, str2bool, safe_unicode)
51 51 from rhodecode.lib.vcs.backends.base import (
52 52 EmptyCommit, UpdateFailureReason, EmptyRepository)
53 53 from rhodecode.lib.vcs.exceptions import (
54 54 EmptyRepositoryError, CommitDoesNotExistError, RepositoryRequirementError,
55 55 NodeDoesNotExistError)
56 56
57 57 from rhodecode.model.changeset_status import ChangesetStatusModel
58 58 from rhodecode.model.comment import CommentsModel
59 59 from rhodecode.model.db import (PullRequest, ChangesetStatus, ChangesetComment,
60 60 Repository, PullRequestVersion)
61 61 from rhodecode.model.forms import PullRequestForm
62 62 from rhodecode.model.meta import Session
63 63 from rhodecode.model.pull_request import PullRequestModel, MergeCheck
64 64
65 65 log = logging.getLogger(__name__)
66 66
67 67
68 68 class PullrequestsController(BaseRepoController):
69 69
70 70 def __before__(self):
71 71 super(PullrequestsController, self).__before__()
72 72 c.REVIEW_STATUS_APPROVED = ChangesetStatus.STATUS_APPROVED
73 73 c.REVIEW_STATUS_REJECTED = ChangesetStatus.STATUS_REJECTED
74 74
75 75 @LoginRequired()
76 76 @NotAnonymous()
77 77 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
78 78 'repository.admin')
79 79 @HasAcceptedRepoType('git', 'hg')
80 80 def index(self):
81 81 source_repo = c.rhodecode_db_repo
82 82
83 83 try:
84 84 source_repo.scm_instance().get_commit()
85 85 except EmptyRepositoryError:
86 86 h.flash(h.literal(_('There are no commits yet')),
87 87 category='warning')
88 88 redirect(h.route_path('repo_summary', repo_name=source_repo.repo_name))
89 89
90 90 commit_id = request.GET.get('commit')
91 91 branch_ref = request.GET.get('branch')
92 92 bookmark_ref = request.GET.get('bookmark')
93 93
94 94 try:
95 95 source_repo_data = PullRequestModel().generate_repo_data(
96 96 source_repo, commit_id=commit_id,
97 97 branch=branch_ref, bookmark=bookmark_ref)
98 98 except CommitDoesNotExistError as e:
99 99 log.exception(e)
100 100 h.flash(_('Commit does not exist'), 'error')
101 101 redirect(url('pullrequest_home', repo_name=source_repo.repo_name))
102 102
103 103 default_target_repo = source_repo
104 104
105 105 if source_repo.parent:
106 106 parent_vcs_obj = source_repo.parent.scm_instance()
107 107 if parent_vcs_obj and not parent_vcs_obj.is_empty():
108 108 # change default if we have a parent repo
109 109 default_target_repo = source_repo.parent
110 110
111 111 target_repo_data = PullRequestModel().generate_repo_data(
112 112 default_target_repo)
113 113
114 114 selected_source_ref = source_repo_data['refs']['selected_ref']
115 115
116 116 title_source_ref = selected_source_ref.split(':', 2)[1]
117 117 c.default_title = PullRequestModel().generate_pullrequest_title(
118 118 source=source_repo.repo_name,
119 119 source_ref=title_source_ref,
120 120 target=default_target_repo.repo_name
121 121 )
122 122
123 123 c.default_repo_data = {
124 124 'source_repo_name': source_repo.repo_name,
125 125 'source_refs_json': json.dumps(source_repo_data),
126 126 'target_repo_name': default_target_repo.repo_name,
127 127 'target_refs_json': json.dumps(target_repo_data),
128 128 }
129 129 c.default_source_ref = selected_source_ref
130 130
131 131 return render('/pullrequests/pullrequest.mako')
132 132
133 133 @LoginRequired()
134 134 @NotAnonymous()
135 135 @XHRRequired()
136 136 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
137 137 'repository.admin')
138 138 @jsonify
139 139 def get_repo_refs(self, repo_name, target_repo_name):
140 140 repo = Repository.get_by_repo_name(target_repo_name)
141 141 if not repo:
142 142 raise HTTPNotFound
143 143 return PullRequestModel().generate_repo_data(repo)
144 144
145 145 @LoginRequired()
146 146 @NotAnonymous()
147 147 @XHRRequired()
148 148 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
149 149 'repository.admin')
150 150 @jsonify
151 151 def get_repo_destinations(self, repo_name):
152 152 repo = Repository.get_by_repo_name(repo_name)
153 153 if not repo:
154 154 raise HTTPNotFound
155 155 filter_query = request.GET.get('query')
156 156
157 157 query = Repository.query() \
158 158 .order_by(func.length(Repository.repo_name)) \
159 159 .filter(or_(
160 160 Repository.repo_name == repo.repo_name,
161 161 Repository.fork_id == repo.repo_id))
162 162
163 163 if filter_query:
164 164 ilike_expression = u'%{}%'.format(safe_unicode(filter_query))
165 165 query = query.filter(
166 166 Repository.repo_name.ilike(ilike_expression))
167 167
168 168 add_parent = False
169 169 if repo.parent:
170 170 if filter_query in repo.parent.repo_name:
171 171 parent_vcs_obj = repo.parent.scm_instance()
172 172 if parent_vcs_obj and not parent_vcs_obj.is_empty():
173 173 add_parent = True
174 174
175 175 limit = 20 - 1 if add_parent else 20
176 176 all_repos = query.limit(limit).all()
177 177 if add_parent:
178 178 all_repos += [repo.parent]
179 179
180 180 repos = []
181 181 for obj in self.scm_model.get_repos(all_repos):
182 182 repos.append({
183 183 'id': obj['name'],
184 184 'text': obj['name'],
185 185 'type': 'repo',
186 186 'obj': obj['dbrepo']
187 187 })
188 188
189 189 data = {
190 190 'more': False,
191 191 'results': [{
192 192 'text': _('Repositories'),
193 193 'children': repos
194 194 }] if repos else []
195 195 }
196 196 return data
197 197
198 198 @LoginRequired()
199 199 @NotAnonymous()
200 200 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
201 201 'repository.admin')
202 202 @HasAcceptedRepoType('git', 'hg')
203 203 @auth.CSRFRequired()
204 204 def create(self, repo_name):
205 205 repo = Repository.get_by_repo_name(repo_name)
206 206 if not repo:
207 207 raise HTTPNotFound
208 208
209 209 controls = peppercorn.parse(request.POST.items())
210 210
211 211 try:
212 212 _form = PullRequestForm(repo.repo_id)().to_python(controls)
213 213 except formencode.Invalid as errors:
214 214 if errors.error_dict.get('revisions'):
215 215 msg = 'Revisions: %s' % errors.error_dict['revisions']
216 216 elif errors.error_dict.get('pullrequest_title'):
217 217 msg = _('Pull request requires a title with min. 3 chars')
218 218 else:
219 219 msg = _('Error creating pull request: {}').format(errors)
220 220 log.exception(msg)
221 221 h.flash(msg, 'error')
222 222
223 223 # would rather just go back to form ...
224 224 return redirect(url('pullrequest_home', repo_name=repo_name))
225 225
226 226 source_repo = _form['source_repo']
227 227 source_ref = _form['source_ref']
228 228 target_repo = _form['target_repo']
229 229 target_ref = _form['target_ref']
230 230 commit_ids = _form['revisions'][::-1]
231 231
232 232 # find the ancestor for this pr
233 233 source_db_repo = Repository.get_by_repo_name(_form['source_repo'])
234 234 target_db_repo = Repository.get_by_repo_name(_form['target_repo'])
235 235
236 236 source_scm = source_db_repo.scm_instance()
237 237 target_scm = target_db_repo.scm_instance()
238 238
239 239 source_commit = source_scm.get_commit(source_ref.split(':')[-1])
240 240 target_commit = target_scm.get_commit(target_ref.split(':')[-1])
241 241
242 242 ancestor = source_scm.get_common_ancestor(
243 243 source_commit.raw_id, target_commit.raw_id, target_scm)
244 244
245 245 target_ref_type, target_ref_name, __ = _form['target_ref'].split(':')
246 246 target_ref = ':'.join((target_ref_type, target_ref_name, ancestor))
247 247
248 248 pullrequest_title = _form['pullrequest_title']
249 249 title_source_ref = source_ref.split(':', 2)[1]
250 250 if not pullrequest_title:
251 251 pullrequest_title = PullRequestModel().generate_pullrequest_title(
252 252 source=source_repo,
253 253 source_ref=title_source_ref,
254 254 target=target_repo
255 255 )
256 256
257 257 description = _form['pullrequest_desc']
258 258
259 259 get_default_reviewers_data, validate_default_reviewers = \
260 260 PullRequestModel().get_reviewer_functions()
261 261
262 262 # recalculate reviewers logic, to make sure we can validate this
263 263 reviewer_rules = get_default_reviewers_data(
264 264 c.rhodecode_user.get_instance(), source_db_repo,
265 265 source_commit, target_db_repo, target_commit)
266 266
267 reviewers = validate_default_reviewers(
268 _form['review_members'], reviewer_rules)
267 given_reviewers = _form['review_members']
268 reviewers = validate_default_reviewers(given_reviewers, reviewer_rules)
269 269
270 270 try:
271 271 pull_request = PullRequestModel().create(
272 272 c.rhodecode_user.user_id, source_repo, source_ref, target_repo,
273 273 target_ref, commit_ids, reviewers, pullrequest_title,
274 274 description, reviewer_rules
275 275 )
276 276 Session().commit()
277 277 h.flash(_('Successfully opened new pull request'),
278 278 category='success')
279 279 except Exception as e:
280 280 msg = _('Error occurred during creation of this pull request.')
281 281 log.exception(msg)
282 282 h.flash(msg, category='error')
283 283 return redirect(url('pullrequest_home', repo_name=repo_name))
284 284
285 285 return redirect(url('pullrequest_show', repo_name=target_repo,
286 286 pull_request_id=pull_request.pull_request_id))
287 287
288 288 @LoginRequired()
289 289 @NotAnonymous()
290 290 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
291 291 'repository.admin')
292 292 @auth.CSRFRequired()
293 293 @jsonify
294 294 def update(self, repo_name, pull_request_id):
295 295 pull_request_id = safe_int(pull_request_id)
296 296 pull_request = PullRequest.get_or_404(pull_request_id)
297 297 # only owner or admin can update it
298 298 allowed_to_update = PullRequestModel().check_user_update(
299 299 pull_request, c.rhodecode_user)
300 300 if allowed_to_update:
301 301 controls = peppercorn.parse(request.POST.items())
302 302
303 303 if 'review_members' in controls:
304 304 self._update_reviewers(
305 305 pull_request_id, controls['review_members'],
306 306 pull_request.reviewer_data)
307 307 elif str2bool(request.POST.get('update_commits', 'false')):
308 308 self._update_commits(pull_request)
309 309 elif str2bool(request.POST.get('edit_pull_request', 'false')):
310 310 self._edit_pull_request(pull_request)
311 311 else:
312 312 raise HTTPBadRequest()
313 313 return True
314 314 raise HTTPForbidden()
315 315
316 316 def _edit_pull_request(self, pull_request):
317 317 try:
318 318 PullRequestModel().edit(
319 319 pull_request, request.POST.get('title'),
320 320 request.POST.get('description'))
321 321 except ValueError:
322 322 msg = _(u'Cannot update closed pull requests.')
323 323 h.flash(msg, category='error')
324 324 return
325 325 else:
326 326 Session().commit()
327 327
328 328 msg = _(u'Pull request title & description updated.')
329 329 h.flash(msg, category='success')
330 330 return
331 331
332 332 def _update_commits(self, pull_request):
333 333 resp = PullRequestModel().update_commits(pull_request)
334 334
335 335 if resp.executed:
336 336
337 337 if resp.target_changed and resp.source_changed:
338 338 changed = 'target and source repositories'
339 339 elif resp.target_changed and not resp.source_changed:
340 340 changed = 'target repository'
341 341 elif not resp.target_changed and resp.source_changed:
342 342 changed = 'source repository'
343 343 else:
344 344 changed = 'nothing'
345 345
346 346 msg = _(
347 347 u'Pull request updated to "{source_commit_id}" with '
348 348 u'{count_added} added, {count_removed} removed commits. '
349 349 u'Source of changes: {change_source}')
350 350 msg = msg.format(
351 351 source_commit_id=pull_request.source_ref_parts.commit_id,
352 352 count_added=len(resp.changes.added),
353 353 count_removed=len(resp.changes.removed),
354 354 change_source=changed)
355 355 h.flash(msg, category='success')
356 356
357 357 registry = get_current_registry()
358 358 rhodecode_plugins = getattr(registry, 'rhodecode_plugins', {})
359 359 channelstream_config = rhodecode_plugins.get('channelstream', {})
360 360 if channelstream_config.get('enabled'):
361 361 message = msg + (
362 362 ' - <a onclick="window.location.reload()">'
363 363 '<strong>{}</strong></a>'.format(_('Reload page')))
364 364 channel = '/repo${}$/pr/{}'.format(
365 365 pull_request.target_repo.repo_name,
366 366 pull_request.pull_request_id
367 367 )
368 368 payload = {
369 369 'type': 'message',
370 370 'user': 'system',
371 371 'exclude_users': [request.user.username],
372 372 'channel': channel,
373 373 'message': {
374 374 'message': message,
375 375 'level': 'success',
376 376 'topic': '/notifications'
377 377 }
378 378 }
379 379 channelstream_request(
380 380 channelstream_config, [payload], '/message',
381 381 raise_exc=False)
382 382 else:
383 383 msg = PullRequestModel.UPDATE_STATUS_MESSAGES[resp.reason]
384 384 warning_reasons = [
385 385 UpdateFailureReason.NO_CHANGE,
386 386 UpdateFailureReason.WRONG_REF_TYPE,
387 387 ]
388 388 category = 'warning' if resp.reason in warning_reasons else 'error'
389 389 h.flash(msg, category=category)
390 390
391 391 @auth.CSRFRequired()
392 392 @LoginRequired()
393 393 @NotAnonymous()
394 394 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
395 395 'repository.admin')
396 396 def merge(self, repo_name, pull_request_id):
397 397 """
398 398 POST /{repo_name}/pull-request/{pull_request_id}
399 399
400 400 Merge will perform a server-side merge of the specified
401 401 pull request, if the pull request is approved and mergeable.
402 402 After successful merging, the pull request is automatically
403 403 closed, with a relevant comment.
404 404 """
405 405 pull_request_id = safe_int(pull_request_id)
406 406 pull_request = PullRequest.get_or_404(pull_request_id)
407 407 user = c.rhodecode_user
408 408
409 409 check = MergeCheck.validate(pull_request, user)
410 410 merge_possible = not check.failed
411 411
412 412 for err_type, error_msg in check.errors:
413 413 h.flash(error_msg, category=err_type)
414 414
415 415 if merge_possible:
416 416 log.debug("Pre-conditions checked, trying to merge.")
417 417 extras = vcs_operation_context(
418 418 request.environ, repo_name=pull_request.target_repo.repo_name,
419 419 username=user.username, action='push',
420 420 scm=pull_request.target_repo.repo_type)
421 421 self._merge_pull_request(pull_request, user, extras)
422 422
423 423 return redirect(url(
424 424 'pullrequest_show',
425 425 repo_name=pull_request.target_repo.repo_name,
426 426 pull_request_id=pull_request.pull_request_id))
427 427
428 428 def _merge_pull_request(self, pull_request, user, extras):
429 429 merge_resp = PullRequestModel().merge(
430 430 pull_request, user, extras=extras)
431 431
432 432 if merge_resp.executed:
433 433 log.debug("The merge was successful, closing the pull request.")
434 434 PullRequestModel().close_pull_request(
435 435 pull_request.pull_request_id, user)
436 436 Session().commit()
437 437 msg = _('Pull request was successfully merged and closed.')
438 438 h.flash(msg, category='success')
439 439 else:
440 440 log.debug(
441 441 "The merge was not successful. Merge response: %s",
442 442 merge_resp)
443 443 msg = PullRequestModel().merge_status_message(
444 444 merge_resp.failure_reason)
445 445 h.flash(msg, category='error')
446 446
447 447 def _update_reviewers(self, pull_request_id, review_members, reviewer_rules):
448 448
449 449 get_default_reviewers_data, validate_default_reviewers = \
450 450 PullRequestModel().get_reviewer_functions()
451 451
452 452 try:
453 453 reviewers = validate_default_reviewers(review_members, reviewer_rules)
454 454 except ValueError as e:
455 log.error('Reviewers Validation:{}'.format(e))
455 log.error('Reviewers Validation: {}'.format(e))
456 456 h.flash(e, category='error')
457 457 return
458 458
459 459 PullRequestModel().update_reviewers(pull_request_id, reviewers)
460 460 h.flash(_('Pull request reviewers updated.'), category='success')
461 461 Session().commit()
462 462
463 463 @LoginRequired()
464 464 @NotAnonymous()
465 465 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
466 466 'repository.admin')
467 467 @auth.CSRFRequired()
468 468 @jsonify
469 469 def delete(self, repo_name, pull_request_id):
470 470 pull_request_id = safe_int(pull_request_id)
471 471 pull_request = PullRequest.get_or_404(pull_request_id)
472 472
473 473 pr_closed = pull_request.is_closed()
474 474 allowed_to_delete = PullRequestModel().check_user_delete(
475 475 pull_request, c.rhodecode_user) and not pr_closed
476 476
477 477 # only owner can delete it !
478 478 if allowed_to_delete:
479 479 PullRequestModel().delete(pull_request)
480 480 Session().commit()
481 481 h.flash(_('Successfully deleted pull request'),
482 482 category='success')
483 483 return redirect(url('my_account_pullrequests'))
484 484
485 485 h.flash(_('Your are not allowed to delete this pull request'),
486 486 category='error')
487 487 raise HTTPForbidden()
488 488
489 489 def _get_pr_version(self, pull_request_id, version=None):
490 490 pull_request_id = safe_int(pull_request_id)
491 491 at_version = None
492 492
493 493 if version and version == 'latest':
494 494 pull_request_ver = PullRequest.get(pull_request_id)
495 495 pull_request_obj = pull_request_ver
496 496 _org_pull_request_obj = pull_request_obj
497 497 at_version = 'latest'
498 498 elif version:
499 499 pull_request_ver = PullRequestVersion.get_or_404(version)
500 500 pull_request_obj = pull_request_ver
501 501 _org_pull_request_obj = pull_request_ver.pull_request
502 502 at_version = pull_request_ver.pull_request_version_id
503 503 else:
504 504 _org_pull_request_obj = pull_request_obj = PullRequest.get_or_404(
505 505 pull_request_id)
506 506
507 507 pull_request_display_obj = PullRequest.get_pr_display_object(
508 508 pull_request_obj, _org_pull_request_obj)
509 509
510 510 return _org_pull_request_obj, pull_request_obj, \
511 511 pull_request_display_obj, at_version
512 512
513 513 def _get_diffset(
514 514 self, source_repo, source_ref_id, target_ref_id, target_commit,
515 515 source_commit, diff_limit, file_limit, display_inline_comments):
516 516 vcs_diff = PullRequestModel().get_diff(
517 517 source_repo, source_ref_id, target_ref_id)
518 518
519 519 diff_processor = diffs.DiffProcessor(
520 520 vcs_diff, format='newdiff', diff_limit=diff_limit,
521 521 file_limit=file_limit, show_full_diff=c.fulldiff)
522 522
523 523 _parsed = diff_processor.prepare()
524 524
525 525 def _node_getter(commit):
526 526 def get_node(fname):
527 527 try:
528 528 return commit.get_node(fname)
529 529 except NodeDoesNotExistError:
530 530 return None
531 531
532 532 return get_node
533 533
534 534 diffset = codeblocks.DiffSet(
535 535 repo_name=c.repo_name,
536 536 source_repo_name=c.source_repo.repo_name,
537 537 source_node_getter=_node_getter(target_commit),
538 538 target_node_getter=_node_getter(source_commit),
539 539 comments=display_inline_comments
540 540 )
541 541 diffset = diffset.render_patchset(
542 542 _parsed, target_commit.raw_id, source_commit.raw_id)
543 543
544 544 return diffset
545 545
546 546 @LoginRequired()
547 547 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
548 548 'repository.admin')
549 549 def show(self, repo_name, pull_request_id):
550 550 pull_request_id = safe_int(pull_request_id)
551 551 version = request.GET.get('version')
552 552 from_version = request.GET.get('from_version') or version
553 553 merge_checks = request.GET.get('merge_checks')
554 554 c.fulldiff = str2bool(request.GET.get('fulldiff'))
555 555
556 556 (pull_request_latest,
557 557 pull_request_at_ver,
558 558 pull_request_display_obj,
559 559 at_version) = self._get_pr_version(
560 560 pull_request_id, version=version)
561 561 pr_closed = pull_request_latest.is_closed()
562 562
563 563 if pr_closed and (version or from_version):
564 564 # not allow to browse versions
565 565 return redirect(h.url('pullrequest_show', repo_name=repo_name,
566 566 pull_request_id=pull_request_id))
567 567
568 568 versions = pull_request_display_obj.versions()
569 569
570 570 c.at_version = at_version
571 571 c.at_version_num = (at_version
572 572 if at_version and at_version != 'latest'
573 573 else None)
574 574 c.at_version_pos = ChangesetComment.get_index_from_version(
575 575 c.at_version_num, versions)
576 576
577 577 (prev_pull_request_latest,
578 578 prev_pull_request_at_ver,
579 579 prev_pull_request_display_obj,
580 580 prev_at_version) = self._get_pr_version(
581 581 pull_request_id, version=from_version)
582 582
583 583 c.from_version = prev_at_version
584 584 c.from_version_num = (prev_at_version
585 585 if prev_at_version and prev_at_version != 'latest'
586 586 else None)
587 587 c.from_version_pos = ChangesetComment.get_index_from_version(
588 588 c.from_version_num, versions)
589 589
590 590 # define if we're in COMPARE mode or VIEW at version mode
591 591 compare = at_version != prev_at_version
592 592
593 593 # pull_requests repo_name we opened it against
594 594 # ie. target_repo must match
595 595 if repo_name != pull_request_at_ver.target_repo.repo_name:
596 596 raise HTTPNotFound
597 597
598 598 c.shadow_clone_url = PullRequestModel().get_shadow_clone_url(
599 599 pull_request_at_ver)
600 600
601 601 c.pull_request = pull_request_display_obj
602 602 c.pull_request_latest = pull_request_latest
603 603
604 604 if compare or (at_version and not at_version == 'latest'):
605 605 c.allowed_to_change_status = False
606 606 c.allowed_to_update = False
607 607 c.allowed_to_merge = False
608 608 c.allowed_to_delete = False
609 609 c.allowed_to_comment = False
610 610 c.allowed_to_close = False
611 611 else:
612 612 can_change_status = PullRequestModel().check_user_change_status(
613 613 pull_request_at_ver, c.rhodecode_user)
614 614 c.allowed_to_change_status = can_change_status and not pr_closed
615 615
616 616 c.allowed_to_update = PullRequestModel().check_user_update(
617 617 pull_request_latest, c.rhodecode_user) and not pr_closed
618 618 c.allowed_to_merge = PullRequestModel().check_user_merge(
619 619 pull_request_latest, c.rhodecode_user) and not pr_closed
620 620 c.allowed_to_delete = PullRequestModel().check_user_delete(
621 621 pull_request_latest, c.rhodecode_user) and not pr_closed
622 622 c.allowed_to_comment = not pr_closed
623 623 c.allowed_to_close = c.allowed_to_merge and not pr_closed
624 624
625 625 c.forbid_adding_reviewers = False
626 626 c.forbid_author_to_review = False
627 627 c.forbid_commit_author_to_review = False
628 628
629 629 if pull_request_latest.reviewer_data and \
630 630 'rules' in pull_request_latest.reviewer_data:
631 631 rules = pull_request_latest.reviewer_data['rules'] or {}
632 632 try:
633 633 c.forbid_adding_reviewers = rules.get(
634 634 'forbid_adding_reviewers')
635 635 c.forbid_author_to_review = rules.get(
636 636 'forbid_author_to_review')
637 637 c.forbid_commit_author_to_review = rules.get(
638 638 'forbid_commit_author_to_review')
639 639 except Exception:
640 640 pass
641 641
642 642 # check merge capabilities
643 643 _merge_check = MergeCheck.validate(
644 644 pull_request_latest, user=c.rhodecode_user)
645 645 c.pr_merge_errors = _merge_check.error_details
646 646 c.pr_merge_possible = not _merge_check.failed
647 647 c.pr_merge_message = _merge_check.merge_msg
648 648
649 649 c.pull_request_review_status = _merge_check.review_status
650 650 if merge_checks:
651 651 return render('/pullrequests/pullrequest_merge_checks.mako')
652 652
653 653 comments_model = CommentsModel()
654 654
655 655 # reviewers and statuses
656 656 c.pull_request_reviewers = pull_request_at_ver.reviewers_statuses()
657 657 allowed_reviewers = [x[0].user_id for x in c.pull_request_reviewers]
658 658
659 659 # GENERAL COMMENTS with versions #
660 660 q = comments_model._all_general_comments_of_pull_request(pull_request_latest)
661 661 q = q.order_by(ChangesetComment.comment_id.asc())
662 662 general_comments = q
663 663
664 664 # pick comments we want to render at current version
665 665 c.comment_versions = comments_model.aggregate_comments(
666 666 general_comments, versions, c.at_version_num)
667 667 c.comments = c.comment_versions[c.at_version_num]['until']
668 668
669 669 # INLINE COMMENTS with versions #
670 670 q = comments_model._all_inline_comments_of_pull_request(pull_request_latest)
671 671 q = q.order_by(ChangesetComment.comment_id.asc())
672 672 inline_comments = q
673 673
674 674 c.inline_versions = comments_model.aggregate_comments(
675 675 inline_comments, versions, c.at_version_num, inline=True)
676 676
677 677 # inject latest version
678 678 latest_ver = PullRequest.get_pr_display_object(
679 679 pull_request_latest, pull_request_latest)
680 680
681 681 c.versions = versions + [latest_ver]
682 682
683 683 # if we use version, then do not show later comments
684 684 # than current version
685 685 display_inline_comments = collections.defaultdict(
686 686 lambda: collections.defaultdict(list))
687 687 for co in inline_comments:
688 688 if c.at_version_num:
689 689 # pick comments that are at least UPTO given version, so we
690 690 # don't render comments for higher version
691 691 should_render = co.pull_request_version_id and \
692 692 co.pull_request_version_id <= c.at_version_num
693 693 else:
694 694 # showing all, for 'latest'
695 695 should_render = True
696 696
697 697 if should_render:
698 698 display_inline_comments[co.f_path][co.line_no].append(co)
699 699
700 700 # load diff data into template context, if we use compare mode then
701 701 # diff is calculated based on changes between versions of PR
702 702
703 703 source_repo = pull_request_at_ver.source_repo
704 704 source_ref_id = pull_request_at_ver.source_ref_parts.commit_id
705 705
706 706 target_repo = pull_request_at_ver.target_repo
707 707 target_ref_id = pull_request_at_ver.target_ref_parts.commit_id
708 708
709 709 if compare:
710 710 # in compare switch the diff base to latest commit from prev version
711 711 target_ref_id = prev_pull_request_display_obj.revisions[0]
712 712
713 713 # despite opening commits for bookmarks/branches/tags, we always
714 714 # convert this to rev to prevent changes after bookmark or branch change
715 715 c.source_ref_type = 'rev'
716 716 c.source_ref = source_ref_id
717 717
718 718 c.target_ref_type = 'rev'
719 719 c.target_ref = target_ref_id
720 720
721 721 c.source_repo = source_repo
722 722 c.target_repo = target_repo
723 723
724 724 # diff_limit is the old behavior, will cut off the whole diff
725 725 # if the limit is applied otherwise will just hide the
726 726 # big files from the front-end
727 727 diff_limit = self.cut_off_limit_diff
728 728 file_limit = self.cut_off_limit_file
729 729
730 730 c.commit_ranges = []
731 731 source_commit = EmptyCommit()
732 732 target_commit = EmptyCommit()
733 733 c.missing_requirements = False
734 734
735 735 source_scm = source_repo.scm_instance()
736 736 target_scm = target_repo.scm_instance()
737 737
738 738 # try first shadow repo, fallback to regular repo
739 739 try:
740 740 commits_source_repo = pull_request_latest.get_shadow_repo()
741 741 except Exception:
742 742 log.debug('Failed to get shadow repo', exc_info=True)
743 743 commits_source_repo = source_scm
744 744
745 745 c.commits_source_repo = commits_source_repo
746 746 commit_cache = {}
747 747 try:
748 748 pre_load = ["author", "branch", "date", "message"]
749 749 show_revs = pull_request_at_ver.revisions
750 750 for rev in show_revs:
751 751 comm = commits_source_repo.get_commit(
752 752 commit_id=rev, pre_load=pre_load)
753 753 c.commit_ranges.append(comm)
754 754 commit_cache[comm.raw_id] = comm
755 755
756 756 # Order here matters, we first need to get target, and then
757 757 # the source
758 758 target_commit = commits_source_repo.get_commit(
759 759 commit_id=safe_str(target_ref_id))
760 760
761 761 source_commit = commits_source_repo.get_commit(
762 762 commit_id=safe_str(source_ref_id))
763 763
764 764 except CommitDoesNotExistError:
765 765 log.warning(
766 766 'Failed to get commit from `{}` repo'.format(
767 767 commits_source_repo), exc_info=True)
768 768 except RepositoryRequirementError:
769 769 log.warning(
770 770 'Failed to get all required data from repo', exc_info=True)
771 771 c.missing_requirements = True
772 772
773 773 c.ancestor = None # set it to None, to hide it from PR view
774 774
775 775 try:
776 776 ancestor_id = source_scm.get_common_ancestor(
777 777 source_commit.raw_id, target_commit.raw_id, target_scm)
778 778 c.ancestor_commit = source_scm.get_commit(ancestor_id)
779 779 except Exception:
780 780 c.ancestor_commit = None
781 781
782 782 c.statuses = source_repo.statuses(
783 783 [x.raw_id for x in c.commit_ranges])
784 784
785 785 # auto collapse if we have more than limit
786 786 collapse_limit = diffs.DiffProcessor._collapse_commits_over
787 787 c.collapse_all_commits = len(c.commit_ranges) > collapse_limit
788 788 c.compare_mode = compare
789 789
790 790 c.missing_commits = False
791 791 if (c.missing_requirements or isinstance(source_commit, EmptyCommit)
792 792 or source_commit == target_commit):
793 793
794 794 c.missing_commits = True
795 795 else:
796 796
797 797 c.diffset = self._get_diffset(
798 798 commits_source_repo, source_ref_id, target_ref_id,
799 799 target_commit, source_commit,
800 800 diff_limit, file_limit, display_inline_comments)
801 801
802 802 c.limited_diff = c.diffset.limited_diff
803 803
804 804 # calculate removed files that are bound to comments
805 805 comment_deleted_files = [
806 806 fname for fname in display_inline_comments
807 807 if fname not in c.diffset.file_stats]
808 808
809 809 c.deleted_files_comments = collections.defaultdict(dict)
810 810 for fname, per_line_comments in display_inline_comments.items():
811 811 if fname in comment_deleted_files:
812 812 c.deleted_files_comments[fname]['stats'] = 0
813 813 c.deleted_files_comments[fname]['comments'] = list()
814 814 for lno, comments in per_line_comments.items():
815 815 c.deleted_files_comments[fname]['comments'].extend(
816 816 comments)
817 817
818 818 # this is a hack to properly display links, when creating PR, the
819 819 # compare view and others uses different notation, and
820 820 # compare_commits.mako renders links based on the target_repo.
821 821 # We need to swap that here to generate it properly on the html side
822 822 c.target_repo = c.source_repo
823 823
824 824 c.commit_statuses = ChangesetStatus.STATUSES
825 825
826 826 c.show_version_changes = not pr_closed
827 827 if c.show_version_changes:
828 828 cur_obj = pull_request_at_ver
829 829 prev_obj = prev_pull_request_at_ver
830 830
831 831 old_commit_ids = prev_obj.revisions
832 832 new_commit_ids = cur_obj.revisions
833 833 commit_changes = PullRequestModel()._calculate_commit_id_changes(
834 834 old_commit_ids, new_commit_ids)
835 835 c.commit_changes_summary = commit_changes
836 836
837 837 # calculate the diff for commits between versions
838 838 c.commit_changes = []
839 839 mark = lambda cs, fw: list(
840 840 h.itertools.izip_longest([], cs, fillvalue=fw))
841 841 for c_type, raw_id in mark(commit_changes.added, 'a') \
842 842 + mark(commit_changes.removed, 'r') \
843 843 + mark(commit_changes.common, 'c'):
844 844
845 845 if raw_id in commit_cache:
846 846 commit = commit_cache[raw_id]
847 847 else:
848 848 try:
849 849 commit = commits_source_repo.get_commit(raw_id)
850 850 except CommitDoesNotExistError:
851 851 # in case we fail extracting still use "dummy" commit
852 852 # for display in commit diff
853 853 commit = h.AttributeDict(
854 854 {'raw_id': raw_id,
855 855 'message': 'EMPTY or MISSING COMMIT'})
856 856 c.commit_changes.append([c_type, commit])
857 857
858 858 # current user review statuses for each version
859 859 c.review_versions = {}
860 860 if c.rhodecode_user.user_id in allowed_reviewers:
861 861 for co in general_comments:
862 862 if co.author.user_id == c.rhodecode_user.user_id:
863 863 # each comment has a status change
864 864 status = co.status_change
865 865 if status:
866 866 _ver_pr = status[0].comment.pull_request_version_id
867 867 c.review_versions[_ver_pr] = status[0]
868 868
869 869 return render('/pullrequests/pullrequest_show.mako')
870 870
871 871 @LoginRequired()
872 872 @NotAnonymous()
873 873 @HasRepoPermissionAnyDecorator(
874 874 'repository.read', 'repository.write', 'repository.admin')
875 875 @auth.CSRFRequired()
876 876 @jsonify
877 877 def comment(self, repo_name, pull_request_id):
878 878 pull_request_id = safe_int(pull_request_id)
879 879 pull_request = PullRequest.get_or_404(pull_request_id)
880 880 if pull_request.is_closed():
881 881 log.debug('comment: forbidden because pull request is closed')
882 882 raise HTTPForbidden()
883 883
884 884 status = request.POST.get('changeset_status', None)
885 885 text = request.POST.get('text')
886 886 comment_type = request.POST.get('comment_type')
887 887 resolves_comment_id = request.POST.get('resolves_comment_id', None)
888 888 close_pull_request = request.POST.get('close_pull_request')
889 889
890 890 # the logic here should work like following, if we submit close
891 891 # pr comment, use `close_pull_request_with_comment` function
892 892 # else handle regular comment logic
893 893 user = c.rhodecode_user
894 894 repo = c.rhodecode_db_repo
895 895
896 896 if close_pull_request:
897 897 # only owner or admin or person with write permissions
898 898 allowed_to_close = PullRequestModel().check_user_update(
899 899 pull_request, c.rhodecode_user)
900 900 if not allowed_to_close:
901 901 log.debug('comment: forbidden because not allowed to close '
902 902 'pull request %s', pull_request_id)
903 903 raise HTTPForbidden()
904 904 comment, status = PullRequestModel().close_pull_request_with_comment(
905 905 pull_request, user, repo, message=text)
906 906 Session().flush()
907 907 events.trigger(
908 908 events.PullRequestCommentEvent(pull_request, comment))
909 909
910 910 else:
911 911 # regular comment case, could be inline, or one with status.
912 912 # for that one we check also permissions
913 913
914 914 allowed_to_change_status = PullRequestModel().check_user_change_status(
915 915 pull_request, c.rhodecode_user)
916 916
917 917 if status and allowed_to_change_status:
918 918 message = (_('Status change %(transition_icon)s %(status)s')
919 919 % {'transition_icon': '>',
920 920 'status': ChangesetStatus.get_status_lbl(status)})
921 921 text = text or message
922 922
923 923 comment = CommentsModel().create(
924 924 text=text,
925 925 repo=c.rhodecode_db_repo.repo_id,
926 926 user=c.rhodecode_user.user_id,
927 927 pull_request=pull_request_id,
928 928 f_path=request.POST.get('f_path'),
929 929 line_no=request.POST.get('line'),
930 930 status_change=(ChangesetStatus.get_status_lbl(status)
931 931 if status and allowed_to_change_status else None),
932 932 status_change_type=(status
933 933 if status and allowed_to_change_status else None),
934 934 comment_type=comment_type,
935 935 resolves_comment_id=resolves_comment_id
936 936 )
937 937
938 938 if allowed_to_change_status:
939 939 # calculate old status before we change it
940 940 old_calculated_status = pull_request.calculated_review_status()
941 941
942 942 # get status if set !
943 943 if status:
944 944 ChangesetStatusModel().set_status(
945 945 c.rhodecode_db_repo.repo_id,
946 946 status,
947 947 c.rhodecode_user.user_id,
948 948 comment,
949 949 pull_request=pull_request_id
950 950 )
951 951
952 952 Session().flush()
953 953 events.trigger(
954 954 events.PullRequestCommentEvent(pull_request, comment))
955 955
956 956 # we now calculate the status of pull request, and based on that
957 957 # calculation we set the commits status
958 958 calculated_status = pull_request.calculated_review_status()
959 959 if old_calculated_status != calculated_status:
960 960 PullRequestModel()._trigger_pull_request_hook(
961 961 pull_request, c.rhodecode_user, 'review_status_change')
962 962
963 963 Session().commit()
964 964
965 965 if not request.is_xhr:
966 966 return redirect(h.url('pullrequest_show', repo_name=repo_name,
967 967 pull_request_id=pull_request_id))
968 968
969 969 data = {
970 970 'target_id': h.safeid(h.safe_unicode(request.POST.get('f_path'))),
971 971 }
972 972 if comment:
973 973 c.co = comment
974 974 rendered_comment = render('changeset/changeset_comment_block.mako')
975 975 data.update(comment.get_dict())
976 976 data.update({'rendered_text': rendered_comment})
977 977
978 978 return data
979 979
980 980 @LoginRequired()
981 981 @NotAnonymous()
982 982 @HasRepoPermissionAnyDecorator('repository.read', 'repository.write',
983 983 'repository.admin')
984 984 @auth.CSRFRequired()
985 985 @jsonify
986 986 def delete_comment(self, repo_name, comment_id):
987 987 return self._delete_comment(comment_id)
988 988
989 989 def _delete_comment(self, comment_id):
990 990 comment_id = safe_int(comment_id)
991 991 co = ChangesetComment.get_or_404(comment_id)
992 992 if co.pull_request.is_closed():
993 993 # don't allow deleting comments on closed pull request
994 994 raise HTTPForbidden()
995 995
996 996 is_owner = co.author.user_id == c.rhodecode_user.user_id
997 997 is_repo_admin = h.HasRepoPermissionAny('repository.admin')(c.repo_name)
998 998 if h.HasPermissionAny('hg.admin')() or is_repo_admin or is_owner:
999 999 old_calculated_status = co.pull_request.calculated_review_status()
1000 1000 CommentsModel().delete(comment=co)
1001 1001 Session().commit()
1002 1002 calculated_status = co.pull_request.calculated_review_status()
1003 1003 if old_calculated_status != calculated_status:
1004 1004 PullRequestModel()._trigger_pull_request_hook(
1005 1005 co.pull_request, c.rhodecode_user, 'review_status_change')
1006 1006 return True
1007 1007 else:
1008 1008 raise HTTPForbidden()
@@ -1,2028 +1,2035 b''
1 1 # -*- coding: utf-8 -*-
2 2
3 3 # Copyright (C) 2010-2017 RhodeCode GmbH
4 4 #
5 5 # This program is free software: you can redistribute it and/or modify
6 6 # it under the terms of the GNU Affero General Public License, version 3
7 7 # (only), as published by the Free Software Foundation.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU Affero General Public License
15 15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 16 #
17 17 # This program is dual-licensed. If you wish to learn more about the
18 18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20 20
21 21 """
22 22 Helper functions
23 23
24 24 Consists of functions to typically be used within templates, but also
25 25 available to Controllers. This module is available to both as 'h'.
26 26 """
27 27
28 28 import random
29 29 import hashlib
30 30 import StringIO
31 31 import urllib
32 32 import math
33 33 import logging
34 34 import re
35 35 import urlparse
36 36 import time
37 37 import string
38 38 import hashlib
39 39 from collections import OrderedDict
40 40
41 41 import pygments
42 42 import itertools
43 43 import fnmatch
44 44
45 45 from datetime import datetime
46 46 from functools import partial
47 47 from pygments.formatters.html import HtmlFormatter
48 48 from pygments import highlight as code_highlight
49 49 from pygments.lexers import (
50 50 get_lexer_by_name, get_lexer_for_filename, get_lexer_for_mimetype)
51 51 from pylons import url as pylons_url
52 52 from pylons.i18n.translation import _, ungettext
53 53 from pyramid.threadlocal import get_current_request
54 54
55 55 from webhelpers.html import literal, HTML, escape
56 56 from webhelpers.html.tools import *
57 57 from webhelpers.html.builder import make_tag
58 58 from webhelpers.html.tags import auto_discovery_link, checkbox, css_classes, \
59 59 end_form, file, form as wh_form, hidden, image, javascript_link, link_to, \
60 60 link_to_if, link_to_unless, ol, required_legend, select, stylesheet_link, \
61 61 submit, text, password, textarea, title, ul, xml_declaration, radio
62 62 from webhelpers.html.tools import auto_link, button_to, highlight, \
63 63 js_obfuscate, mail_to, strip_links, strip_tags, tag_re
64 64 from webhelpers.pylonslib import Flash as _Flash
65 65 from webhelpers.text import chop_at, collapse, convert_accented_entities, \
66 66 convert_misc_entities, lchop, plural, rchop, remove_formatting, \
67 67 replace_whitespace, urlify, truncate, wrap_paragraphs
68 68 from webhelpers.date import time_ago_in_words
69 69 from webhelpers.paginate import Page as _Page
70 70 from webhelpers.html.tags import _set_input_attrs, _set_id_attr, \
71 71 convert_boolean_attrs, NotGiven, _make_safe_id_component
72 72 from webhelpers2.number import format_byte_size
73 73
74 74 from rhodecode.lib.action_parser import action_parser
75 75 from rhodecode.lib.ext_json import json
76 76 from rhodecode.lib.utils import repo_name_slug, get_custom_lexer
77 77 from rhodecode.lib.utils2 import str2bool, safe_unicode, safe_str, \
78 78 get_commit_safe, datetime_to_time, time_to_datetime, time_to_utcdatetime, \
79 79 AttributeDict, safe_int, md5, md5_safe
80 80 from rhodecode.lib.markup_renderer import MarkupRenderer, relative_links
81 81 from rhodecode.lib.vcs.exceptions import CommitDoesNotExistError
82 82 from rhodecode.lib.vcs.backends.base import BaseChangeset, EmptyCommit
83 83 from rhodecode.config.conf import DATE_FORMAT, DATETIME_FORMAT
84 84 from rhodecode.model.changeset_status import ChangesetStatusModel
85 85 from rhodecode.model.db import Permission, User, Repository
86 86 from rhodecode.model.repo_group import RepoGroupModel
87 87 from rhodecode.model.settings import IssueTrackerSettingsModel
88 88
89 89 log = logging.getLogger(__name__)
90 90
91 91
92 92 DEFAULT_USER = User.DEFAULT_USER
93 93 DEFAULT_USER_EMAIL = User.DEFAULT_USER_EMAIL
94 94
95 95
96 96 def url(*args, **kw):
97 97 return pylons_url(*args, **kw)
98 98
99 99
100 100 def pylons_url_current(*args, **kw):
101 101 """
102 102 This function overrides pylons.url.current() which returns the current
103 103 path so that it will also work from a pyramid only context. This
104 104 should be removed once port to pyramid is complete.
105 105 """
106 106 if not args and not kw:
107 107 request = get_current_request()
108 108 return request.path
109 109 return pylons_url.current(*args, **kw)
110 110
111 111 url.current = pylons_url_current
112 112
113 113
114 114 def url_replace(**qargs):
115 115 """ Returns the current request url while replacing query string args """
116 116
117 117 request = get_current_request()
118 118 new_args = request.GET.mixed()
119 119 new_args.update(qargs)
120 120 return url('', **new_args)
121 121
122 122
123 123 def asset(path, ver=None, **kwargs):
124 124 """
125 125 Helper to generate a static asset file path for rhodecode assets
126 126
127 127 eg. h.asset('images/image.png', ver='3923')
128 128
129 129 :param path: path of asset
130 130 :param ver: optional version query param to append as ?ver=
131 131 """
132 132 request = get_current_request()
133 133 query = {}
134 134 query.update(kwargs)
135 135 if ver:
136 136 query = {'ver': ver}
137 137 return request.static_path(
138 138 'rhodecode:public/{}'.format(path), _query=query)
139 139
140 140
141 141 default_html_escape_table = {
142 142 ord('&'): u'&amp;',
143 143 ord('<'): u'&lt;',
144 144 ord('>'): u'&gt;',
145 145 ord('"'): u'&quot;',
146 146 ord("'"): u'&#39;',
147 147 }
148 148
149 149
150 150 def html_escape(text, html_escape_table=default_html_escape_table):
151 151 """Produce entities within text."""
152 152 return text.translate(html_escape_table)
153 153
154 154
155 155 def chop_at_smart(s, sub, inclusive=False, suffix_if_chopped=None):
156 156 """
157 157 Truncate string ``s`` at the first occurrence of ``sub``.
158 158
159 159 If ``inclusive`` is true, truncate just after ``sub`` rather than at it.
160 160 """
161 161 suffix_if_chopped = suffix_if_chopped or ''
162 162 pos = s.find(sub)
163 163 if pos == -1:
164 164 return s
165 165
166 166 if inclusive:
167 167 pos += len(sub)
168 168
169 169 chopped = s[:pos]
170 170 left = s[pos:].strip()
171 171
172 172 if left and suffix_if_chopped:
173 173 chopped += suffix_if_chopped
174 174
175 175 return chopped
176 176
177 177
178 178 def shorter(text, size=20):
179 179 postfix = '...'
180 180 if len(text) > size:
181 181 return text[:size - len(postfix)] + postfix
182 182 return text
183 183
184 184
185 185 def _reset(name, value=None, id=NotGiven, type="reset", **attrs):
186 186 """
187 187 Reset button
188 188 """
189 189 _set_input_attrs(attrs, type, name, value)
190 190 _set_id_attr(attrs, id, name)
191 191 convert_boolean_attrs(attrs, ["disabled"])
192 192 return HTML.input(**attrs)
193 193
194 194 reset = _reset
195 195 safeid = _make_safe_id_component
196 196
197 197
198 198 def branding(name, length=40):
199 199 return truncate(name, length, indicator="")
200 200
201 201
202 202 def FID(raw_id, path):
203 203 """
204 204 Creates a unique ID for filenode based on it's hash of path and commit
205 205 it's safe to use in urls
206 206
207 207 :param raw_id:
208 208 :param path:
209 209 """
210 210
211 211 return 'c-%s-%s' % (short_id(raw_id), md5_safe(path)[:12])
212 212
213 213
214 214 class _GetError(object):
215 215 """Get error from form_errors, and represent it as span wrapped error
216 216 message
217 217
218 218 :param field_name: field to fetch errors for
219 219 :param form_errors: form errors dict
220 220 """
221 221
222 222 def __call__(self, field_name, form_errors):
223 223 tmpl = """<span class="error_msg">%s</span>"""
224 224 if form_errors and field_name in form_errors:
225 225 return literal(tmpl % form_errors.get(field_name))
226 226
227 227 get_error = _GetError()
228 228
229 229
230 230 class _ToolTip(object):
231 231
232 232 def __call__(self, tooltip_title, trim_at=50):
233 233 """
234 234 Special function just to wrap our text into nice formatted
235 235 autowrapped text
236 236
237 237 :param tooltip_title:
238 238 """
239 239 tooltip_title = escape(tooltip_title)
240 240 tooltip_title = tooltip_title.replace('<', '&lt;').replace('>', '&gt;')
241 241 return tooltip_title
242 242 tooltip = _ToolTip()
243 243
244 244
245 245 def files_breadcrumbs(repo_name, commit_id, file_path):
246 246 if isinstance(file_path, str):
247 247 file_path = safe_unicode(file_path)
248 248
249 249 # TODO: johbo: Is this always a url like path, or is this operating
250 250 # system dependent?
251 251 path_segments = file_path.split('/')
252 252
253 253 repo_name_html = escape(repo_name)
254 254 if len(path_segments) == 1 and path_segments[0] == '':
255 255 url_segments = [repo_name_html]
256 256 else:
257 257 url_segments = [
258 258 link_to(
259 259 repo_name_html,
260 260 url('files_home',
261 261 repo_name=repo_name,
262 262 revision=commit_id,
263 263 f_path=''),
264 264 class_='pjax-link')]
265 265
266 266 last_cnt = len(path_segments) - 1
267 267 for cnt, segment in enumerate(path_segments):
268 268 if not segment:
269 269 continue
270 270 segment_html = escape(segment)
271 271
272 272 if cnt != last_cnt:
273 273 url_segments.append(
274 274 link_to(
275 275 segment_html,
276 276 url('files_home',
277 277 repo_name=repo_name,
278 278 revision=commit_id,
279 279 f_path='/'.join(path_segments[:cnt + 1])),
280 280 class_='pjax-link'))
281 281 else:
282 282 url_segments.append(segment_html)
283 283
284 284 return literal('/'.join(url_segments))
285 285
286 286
287 287 class CodeHtmlFormatter(HtmlFormatter):
288 288 """
289 289 My code Html Formatter for source codes
290 290 """
291 291
292 292 def wrap(self, source, outfile):
293 293 return self._wrap_div(self._wrap_pre(self._wrap_code(source)))
294 294
295 295 def _wrap_code(self, source):
296 296 for cnt, it in enumerate(source):
297 297 i, t = it
298 298 t = '<div id="L%s">%s</div>' % (cnt + 1, t)
299 299 yield i, t
300 300
301 301 def _wrap_tablelinenos(self, inner):
302 302 dummyoutfile = StringIO.StringIO()
303 303 lncount = 0
304 304 for t, line in inner:
305 305 if t:
306 306 lncount += 1
307 307 dummyoutfile.write(line)
308 308
309 309 fl = self.linenostart
310 310 mw = len(str(lncount + fl - 1))
311 311 sp = self.linenospecial
312 312 st = self.linenostep
313 313 la = self.lineanchors
314 314 aln = self.anchorlinenos
315 315 nocls = self.noclasses
316 316 if sp:
317 317 lines = []
318 318
319 319 for i in range(fl, fl + lncount):
320 320 if i % st == 0:
321 321 if i % sp == 0:
322 322 if aln:
323 323 lines.append('<a href="#%s%d" class="special">%*d</a>' %
324 324 (la, i, mw, i))
325 325 else:
326 326 lines.append('<span class="special">%*d</span>' % (mw, i))
327 327 else:
328 328 if aln:
329 329 lines.append('<a href="#%s%d">%*d</a>' % (la, i, mw, i))
330 330 else:
331 331 lines.append('%*d' % (mw, i))
332 332 else:
333 333 lines.append('')
334 334 ls = '\n'.join(lines)
335 335 else:
336 336 lines = []
337 337 for i in range(fl, fl + lncount):
338 338 if i % st == 0:
339 339 if aln:
340 340 lines.append('<a href="#%s%d">%*d</a>' % (la, i, mw, i))
341 341 else:
342 342 lines.append('%*d' % (mw, i))
343 343 else:
344 344 lines.append('')
345 345 ls = '\n'.join(lines)
346 346
347 347 # in case you wonder about the seemingly redundant <div> here: since the
348 348 # content in the other cell also is wrapped in a div, some browsers in
349 349 # some configurations seem to mess up the formatting...
350 350 if nocls:
351 351 yield 0, ('<table class="%stable">' % self.cssclass +
352 352 '<tr><td><div class="linenodiv" '
353 353 'style="background-color: #f0f0f0; padding-right: 10px">'
354 354 '<pre style="line-height: 125%">' +
355 355 ls + '</pre></div></td><td id="hlcode" class="code">')
356 356 else:
357 357 yield 0, ('<table class="%stable">' % self.cssclass +
358 358 '<tr><td class="linenos"><div class="linenodiv"><pre>' +
359 359 ls + '</pre></div></td><td id="hlcode" class="code">')
360 360 yield 0, dummyoutfile.getvalue()
361 361 yield 0, '</td></tr></table>'
362 362
363 363
364 364 class SearchContentCodeHtmlFormatter(CodeHtmlFormatter):
365 365 def __init__(self, **kw):
366 366 # only show these line numbers if set
367 367 self.only_lines = kw.pop('only_line_numbers', [])
368 368 self.query_terms = kw.pop('query_terms', [])
369 369 self.max_lines = kw.pop('max_lines', 5)
370 370 self.line_context = kw.pop('line_context', 3)
371 371 self.url = kw.pop('url', None)
372 372
373 373 super(CodeHtmlFormatter, self).__init__(**kw)
374 374
375 375 def _wrap_code(self, source):
376 376 for cnt, it in enumerate(source):
377 377 i, t = it
378 378 t = '<pre>%s</pre>' % t
379 379 yield i, t
380 380
381 381 def _wrap_tablelinenos(self, inner):
382 382 yield 0, '<table class="code-highlight %stable">' % self.cssclass
383 383
384 384 last_shown_line_number = 0
385 385 current_line_number = 1
386 386
387 387 for t, line in inner:
388 388 if not t:
389 389 yield t, line
390 390 continue
391 391
392 392 if current_line_number in self.only_lines:
393 393 if last_shown_line_number + 1 != current_line_number:
394 394 yield 0, '<tr>'
395 395 yield 0, '<td class="line">...</td>'
396 396 yield 0, '<td id="hlcode" class="code"></td>'
397 397 yield 0, '</tr>'
398 398
399 399 yield 0, '<tr>'
400 400 if self.url:
401 401 yield 0, '<td class="line"><a href="%s#L%i">%i</a></td>' % (
402 402 self.url, current_line_number, current_line_number)
403 403 else:
404 404 yield 0, '<td class="line"><a href="">%i</a></td>' % (
405 405 current_line_number)
406 406 yield 0, '<td id="hlcode" class="code">' + line + '</td>'
407 407 yield 0, '</tr>'
408 408
409 409 last_shown_line_number = current_line_number
410 410
411 411 current_line_number += 1
412 412
413 413
414 414 yield 0, '</table>'
415 415
416 416
417 417 def extract_phrases(text_query):
418 418 """
419 419 Extracts phrases from search term string making sure phrases
420 420 contained in double quotes are kept together - and discarding empty values
421 421 or fully whitespace values eg.
422 422
423 423 'some text "a phrase" more' => ['some', 'text', 'a phrase', 'more']
424 424
425 425 """
426 426
427 427 in_phrase = False
428 428 buf = ''
429 429 phrases = []
430 430 for char in text_query:
431 431 if in_phrase:
432 432 if char == '"': # end phrase
433 433 phrases.append(buf)
434 434 buf = ''
435 435 in_phrase = False
436 436 continue
437 437 else:
438 438 buf += char
439 439 continue
440 440 else:
441 441 if char == '"': # start phrase
442 442 in_phrase = True
443 443 phrases.append(buf)
444 444 buf = ''
445 445 continue
446 446 elif char == ' ':
447 447 phrases.append(buf)
448 448 buf = ''
449 449 continue
450 450 else:
451 451 buf += char
452 452
453 453 phrases.append(buf)
454 454 phrases = [phrase.strip() for phrase in phrases if phrase.strip()]
455 455 return phrases
456 456
457 457
458 458 def get_matching_offsets(text, phrases):
459 459 """
460 460 Returns a list of string offsets in `text` that the list of `terms` match
461 461
462 462 >>> get_matching_offsets('some text here', ['some', 'here'])
463 463 [(0, 4), (10, 14)]
464 464
465 465 """
466 466 offsets = []
467 467 for phrase in phrases:
468 468 for match in re.finditer(phrase, text):
469 469 offsets.append((match.start(), match.end()))
470 470
471 471 return offsets
472 472
473 473
474 474 def normalize_text_for_matching(x):
475 475 """
476 476 Replaces all non alnum characters to spaces and lower cases the string,
477 477 useful for comparing two text strings without punctuation
478 478 """
479 479 return re.sub(r'[^\w]', ' ', x.lower())
480 480
481 481
482 482 def get_matching_line_offsets(lines, terms):
483 483 """ Return a set of `lines` indices (starting from 1) matching a
484 484 text search query, along with `context` lines above/below matching lines
485 485
486 486 :param lines: list of strings representing lines
487 487 :param terms: search term string to match in lines eg. 'some text'
488 488 :param context: number of lines above/below a matching line to add to result
489 489 :param max_lines: cut off for lines of interest
490 490 eg.
491 491
492 492 text = '''
493 493 words words words
494 494 words words words
495 495 some text some
496 496 words words words
497 497 words words words
498 498 text here what
499 499 '''
500 500 get_matching_line_offsets(text, 'text', context=1)
501 501 {3: [(5, 9)], 6: [(0, 4)]]
502 502
503 503 """
504 504 matching_lines = {}
505 505 phrases = [normalize_text_for_matching(phrase)
506 506 for phrase in extract_phrases(terms)]
507 507
508 508 for line_index, line in enumerate(lines, start=1):
509 509 match_offsets = get_matching_offsets(
510 510 normalize_text_for_matching(line), phrases)
511 511 if match_offsets:
512 512 matching_lines[line_index] = match_offsets
513 513
514 514 return matching_lines
515 515
516 516
517 517 def hsv_to_rgb(h, s, v):
518 518 """ Convert hsv color values to rgb """
519 519
520 520 if s == 0.0:
521 521 return v, v, v
522 522 i = int(h * 6.0) # XXX assume int() truncates!
523 523 f = (h * 6.0) - i
524 524 p = v * (1.0 - s)
525 525 q = v * (1.0 - s * f)
526 526 t = v * (1.0 - s * (1.0 - f))
527 527 i = i % 6
528 528 if i == 0:
529 529 return v, t, p
530 530 if i == 1:
531 531 return q, v, p
532 532 if i == 2:
533 533 return p, v, t
534 534 if i == 3:
535 535 return p, q, v
536 536 if i == 4:
537 537 return t, p, v
538 538 if i == 5:
539 539 return v, p, q
540 540
541 541
542 542 def unique_color_generator(n=10000, saturation=0.10, lightness=0.95):
543 543 """
544 544 Generator for getting n of evenly distributed colors using
545 545 hsv color and golden ratio. It always return same order of colors
546 546
547 547 :param n: number of colors to generate
548 548 :param saturation: saturation of returned colors
549 549 :param lightness: lightness of returned colors
550 550 :returns: RGB tuple
551 551 """
552 552
553 553 golden_ratio = 0.618033988749895
554 554 h = 0.22717784590367374
555 555
556 556 for _ in xrange(n):
557 557 h += golden_ratio
558 558 h %= 1
559 559 HSV_tuple = [h, saturation, lightness]
560 560 RGB_tuple = hsv_to_rgb(*HSV_tuple)
561 561 yield map(lambda x: str(int(x * 256)), RGB_tuple)
562 562
563 563
564 564 def color_hasher(n=10000, saturation=0.10, lightness=0.95):
565 565 """
566 566 Returns a function which when called with an argument returns a unique
567 567 color for that argument, eg.
568 568
569 569 :param n: number of colors to generate
570 570 :param saturation: saturation of returned colors
571 571 :param lightness: lightness of returned colors
572 572 :returns: css RGB string
573 573
574 574 >>> color_hash = color_hasher()
575 575 >>> color_hash('hello')
576 576 'rgb(34, 12, 59)'
577 577 >>> color_hash('hello')
578 578 'rgb(34, 12, 59)'
579 579 >>> color_hash('other')
580 580 'rgb(90, 224, 159)'
581 581 """
582 582
583 583 color_dict = {}
584 584 cgenerator = unique_color_generator(
585 585 saturation=saturation, lightness=lightness)
586 586
587 587 def get_color_string(thing):
588 588 if thing in color_dict:
589 589 col = color_dict[thing]
590 590 else:
591 591 col = color_dict[thing] = cgenerator.next()
592 592 return "rgb(%s)" % (', '.join(col))
593 593
594 594 return get_color_string
595 595
596 596
597 597 def get_lexer_safe(mimetype=None, filepath=None):
598 598 """
599 599 Tries to return a relevant pygments lexer using mimetype/filepath name,
600 600 defaulting to plain text if none could be found
601 601 """
602 602 lexer = None
603 603 try:
604 604 if mimetype:
605 605 lexer = get_lexer_for_mimetype(mimetype)
606 606 if not lexer:
607 607 lexer = get_lexer_for_filename(filepath)
608 608 except pygments.util.ClassNotFound:
609 609 pass
610 610
611 611 if not lexer:
612 612 lexer = get_lexer_by_name('text')
613 613
614 614 return lexer
615 615
616 616
617 617 def get_lexer_for_filenode(filenode):
618 618 lexer = get_custom_lexer(filenode.extension) or filenode.lexer
619 619 return lexer
620 620
621 621
622 622 def pygmentize(filenode, **kwargs):
623 623 """
624 624 pygmentize function using pygments
625 625
626 626 :param filenode:
627 627 """
628 628 lexer = get_lexer_for_filenode(filenode)
629 629 return literal(code_highlight(filenode.content, lexer,
630 630 CodeHtmlFormatter(**kwargs)))
631 631
632 632
633 633 def is_following_repo(repo_name, user_id):
634 634 from rhodecode.model.scm import ScmModel
635 635 return ScmModel().is_following_repo(repo_name, user_id)
636 636
637 637
638 638 class _Message(object):
639 639 """A message returned by ``Flash.pop_messages()``.
640 640
641 641 Converting the message to a string returns the message text. Instances
642 642 also have the following attributes:
643 643
644 644 * ``message``: the message text.
645 645 * ``category``: the category specified when the message was created.
646 646 """
647 647
648 648 def __init__(self, category, message):
649 649 self.category = category
650 650 self.message = message
651 651
652 652 def __str__(self):
653 653 return self.message
654 654
655 655 __unicode__ = __str__
656 656
657 657 def __html__(self):
658 658 return escape(safe_unicode(self.message))
659 659
660 660
661 661 class Flash(_Flash):
662 662
663 663 def pop_messages(self):
664 664 """Return all accumulated messages and delete them from the session.
665 665
666 666 The return value is a list of ``Message`` objects.
667 667 """
668 668 from pylons import session
669 669
670 670 messages = []
671 671
672 672 # Pop the 'old' pylons flash messages. They are tuples of the form
673 673 # (category, message)
674 674 for cat, msg in session.pop(self.session_key, []):
675 675 messages.append(_Message(cat, msg))
676 676
677 677 # Pop the 'new' pyramid flash messages for each category as list
678 678 # of strings.
679 679 for cat in self.categories:
680 680 for msg in session.pop_flash(queue=cat):
681 681 messages.append(_Message(cat, msg))
682 682 # Map messages from the default queue to the 'notice' category.
683 683 for msg in session.pop_flash():
684 684 messages.append(_Message('notice', msg))
685 685
686 686 session.save()
687 687 return messages
688 688
689 689 def json_alerts(self):
690 690 payloads = []
691 691 messages = flash.pop_messages()
692 692 if messages:
693 693 for message in messages:
694 694 subdata = {}
695 695 if hasattr(message.message, 'rsplit'):
696 696 flash_data = message.message.rsplit('|DELIM|', 1)
697 697 org_message = flash_data[0]
698 698 if len(flash_data) > 1:
699 699 subdata = json.loads(flash_data[1])
700 700 else:
701 701 org_message = message.message
702 702 payloads.append({
703 703 'message': {
704 704 'message': u'{}'.format(org_message),
705 705 'level': message.category,
706 706 'force': True,
707 707 'subdata': subdata
708 708 }
709 709 })
710 710 return json.dumps(payloads)
711 711
712 712 flash = Flash()
713 713
714 714 #==============================================================================
715 715 # SCM FILTERS available via h.
716 716 #==============================================================================
717 717 from rhodecode.lib.vcs.utils import author_name, author_email
718 718 from rhodecode.lib.utils2 import credentials_filter, age as _age
719 719 from rhodecode.model.db import User, ChangesetStatus
720 720
721 721 age = _age
722 722 capitalize = lambda x: x.capitalize()
723 723 email = author_email
724 724 short_id = lambda x: x[:12]
725 725 hide_credentials = lambda x: ''.join(credentials_filter(x))
726 726
727 727
728 728 def age_component(datetime_iso, value=None, time_is_local=False):
729 729 title = value or format_date(datetime_iso)
730 730 tzinfo = '+00:00'
731 731
732 732 # detect if we have a timezone info, otherwise, add it
733 733 if isinstance(datetime_iso, datetime) and not datetime_iso.tzinfo:
734 734 if time_is_local:
735 735 tzinfo = time.strftime("+%H:%M",
736 736 time.gmtime(
737 737 (datetime.now() - datetime.utcnow()).seconds + 1
738 738 )
739 739 )
740 740
741 741 return literal(
742 742 '<time class="timeago tooltip" '
743 743 'title="{1}{2}" datetime="{0}{2}">{1}</time>'.format(
744 744 datetime_iso, title, tzinfo))
745 745
746 746
747 747 def _shorten_commit_id(commit_id):
748 748 from rhodecode import CONFIG
749 749 def_len = safe_int(CONFIG.get('rhodecode_show_sha_length', 12))
750 750 return commit_id[:def_len]
751 751
752 752
753 753 def show_id(commit):
754 754 """
755 755 Configurable function that shows ID
756 756 by default it's r123:fffeeefffeee
757 757
758 758 :param commit: commit instance
759 759 """
760 760 from rhodecode import CONFIG
761 761 show_idx = str2bool(CONFIG.get('rhodecode_show_revision_number', True))
762 762
763 763 raw_id = _shorten_commit_id(commit.raw_id)
764 764 if show_idx:
765 765 return 'r%s:%s' % (commit.idx, raw_id)
766 766 else:
767 767 return '%s' % (raw_id, )
768 768
769 769
770 770 def format_date(date):
771 771 """
772 772 use a standardized formatting for dates used in RhodeCode
773 773
774 774 :param date: date/datetime object
775 775 :return: formatted date
776 776 """
777 777
778 778 if date:
779 779 _fmt = "%a, %d %b %Y %H:%M:%S"
780 780 return safe_unicode(date.strftime(_fmt))
781 781
782 782 return u""
783 783
784 784
785 785 class _RepoChecker(object):
786 786
787 787 def __init__(self, backend_alias):
788 788 self._backend_alias = backend_alias
789 789
790 790 def __call__(self, repository):
791 791 if hasattr(repository, 'alias'):
792 792 _type = repository.alias
793 793 elif hasattr(repository, 'repo_type'):
794 794 _type = repository.repo_type
795 795 else:
796 796 _type = repository
797 797 return _type == self._backend_alias
798 798
799 799 is_git = _RepoChecker('git')
800 800 is_hg = _RepoChecker('hg')
801 801 is_svn = _RepoChecker('svn')
802 802
803 803
804 804 def get_repo_type_by_name(repo_name):
805 805 repo = Repository.get_by_repo_name(repo_name)
806 806 return repo.repo_type
807 807
808 808
809 809 def is_svn_without_proxy(repository):
810 810 if is_svn(repository):
811 811 from rhodecode.model.settings import VcsSettingsModel
812 812 conf = VcsSettingsModel().get_ui_settings_as_config_obj()
813 813 return not str2bool(conf.get('vcs_svn_proxy', 'http_requests_enabled'))
814 814 return False
815 815
816 816
817 817 def discover_user(author):
818 818 """
819 819 Tries to discover RhodeCode User based on the autho string. Author string
820 820 is typically `FirstName LastName <email@address.com>`
821 821 """
822 822
823 823 # if author is already an instance use it for extraction
824 824 if isinstance(author, User):
825 825 return author
826 826
827 827 # Valid email in the attribute passed, see if they're in the system
828 828 _email = author_email(author)
829 829 if _email != '':
830 830 user = User.get_by_email(_email, case_insensitive=True, cache=True)
831 831 if user is not None:
832 832 return user
833 833
834 834 # Maybe it's a username, we try to extract it and fetch by username ?
835 835 _author = author_name(author)
836 836 user = User.get_by_username(_author, case_insensitive=True, cache=True)
837 837 if user is not None:
838 838 return user
839 839
840 840 return None
841 841
842 842
843 843 def email_or_none(author):
844 844 # extract email from the commit string
845 845 _email = author_email(author)
846 846
847 847 # If we have an email, use it, otherwise
848 848 # see if it contains a username we can get an email from
849 849 if _email != '':
850 850 return _email
851 851 else:
852 852 user = User.get_by_username(
853 853 author_name(author), case_insensitive=True, cache=True)
854 854
855 855 if user is not None:
856 856 return user.email
857 857
858 858 # No valid email, not a valid user in the system, none!
859 859 return None
860 860
861 861
862 862 def link_to_user(author, length=0, **kwargs):
863 863 user = discover_user(author)
864 864 # user can be None, but if we have it already it means we can re-use it
865 865 # in the person() function, so we save 1 intensive-query
866 866 if user:
867 867 author = user
868 868
869 869 display_person = person(author, 'username_or_name_or_email')
870 870 if length:
871 871 display_person = shorter(display_person, length)
872 872
873 873 if user:
874 874 return link_to(
875 875 escape(display_person),
876 876 route_path('user_profile', username=user.username),
877 877 **kwargs)
878 878 else:
879 879 return escape(display_person)
880 880
881 881
882 882 def person(author, show_attr="username_and_name"):
883 883 user = discover_user(author)
884 884 if user:
885 885 return getattr(user, show_attr)
886 886 else:
887 887 _author = author_name(author)
888 888 _email = email(author)
889 889 return _author or _email
890 890
891 891
892 892 def author_string(email):
893 893 if email:
894 894 user = User.get_by_email(email, case_insensitive=True, cache=True)
895 895 if user:
896 896 if user.firstname or user.lastname:
897 897 return '%s %s &lt;%s&gt;' % (
898 898 escape(user.firstname), escape(user.lastname), email)
899 899 else:
900 900 return email
901 901 else:
902 902 return email
903 903 else:
904 904 return None
905 905
906 906
907 907 def person_by_id(id_, show_attr="username_and_name"):
908 908 # attr to return from fetched user
909 909 person_getter = lambda usr: getattr(usr, show_attr)
910 910
911 911 #maybe it's an ID ?
912 912 if str(id_).isdigit() or isinstance(id_, int):
913 913 id_ = int(id_)
914 914 user = User.get(id_)
915 915 if user is not None:
916 916 return person_getter(user)
917 917 return id_
918 918
919 919
920 920 def gravatar_with_user(author, show_disabled=False):
921 921 from rhodecode.lib.utils import PartialRenderer
922 922 _render = PartialRenderer('base/base.mako')
923 923 return _render('gravatar_with_user', author, show_disabled=show_disabled)
924 924
925 925
926 926 def desc_stylize(value):
927 927 """
928 928 converts tags from value into html equivalent
929 929
930 930 :param value:
931 931 """
932 932 if not value:
933 933 return ''
934 934
935 935 value = re.sub(r'\[see\ \=\>\ *([a-zA-Z0-9\/\=\?\&\ \:\/\.\-]*)\]',
936 936 '<div class="metatag" tag="see">see =&gt; \\1 </div>', value)
937 937 value = re.sub(r'\[license\ \=\>\ *([a-zA-Z0-9\/\=\?\&\ \:\/\.\-]*)\]',
938 938 '<div class="metatag" tag="license"><a href="http:\/\/www.opensource.org/licenses/\\1">\\1</a></div>', value)
939 939 value = re.sub(r'\[(requires|recommends|conflicts|base)\ \=\>\ *([a-zA-Z0-9\-\/]*)\]',
940 940 '<div class="metatag" tag="\\1">\\1 =&gt; <a href="/\\2">\\2</a></div>', value)
941 941 value = re.sub(r'\[(lang|language)\ \=\>\ *([a-zA-Z\-\/\#\+]*)\]',
942 942 '<div class="metatag" tag="lang">\\2</div>', value)
943 943 value = re.sub(r'\[([a-z]+)\]',
944 944 '<div class="metatag" tag="\\1">\\1</div>', value)
945 945
946 946 return value
947 947
948 948
949 949 def escaped_stylize(value):
950 950 """
951 951 converts tags from value into html equivalent, but escaping its value first
952 952 """
953 953 if not value:
954 954 return ''
955 955
956 956 # Using default webhelper escape method, but has to force it as a
957 957 # plain unicode instead of a markup tag to be used in regex expressions
958 958 value = unicode(escape(safe_unicode(value)))
959 959
960 960 value = re.sub(r'\[see\ \=\&gt;\ *([a-zA-Z0-9\/\=\?\&amp;\ \:\/\.\-]*)\]',
961 961 '<div class="metatag" tag="see">see =&gt; \\1 </div>', value)
962 962 value = re.sub(r'\[license\ \=\&gt;\ *([a-zA-Z0-9\/\=\?\&amp;\ \:\/\.\-]*)\]',
963 963 '<div class="metatag" tag="license"><a href="http:\/\/www.opensource.org/licenses/\\1">\\1</a></div>', value)
964 964 value = re.sub(r'\[(requires|recommends|conflicts|base)\ \=\&gt;\ *([a-zA-Z0-9\-\/]*)\]',
965 965 '<div class="metatag" tag="\\1">\\1 =&gt; <a href="/\\2">\\2</a></div>', value)
966 966 value = re.sub(r'\[(lang|language)\ \=\&gt;\ *([a-zA-Z\-\/\#\+]*)\]',
967 967 '<div class="metatag" tag="lang">\\2</div>', value)
968 968 value = re.sub(r'\[([a-z]+)\]',
969 969 '<div class="metatag" tag="\\1">\\1</div>', value)
970 970
971 971 return value
972 972
973 973
974 974 def bool2icon(value):
975 975 """
976 976 Returns boolean value of a given value, represented as html element with
977 977 classes that will represent icons
978 978
979 979 :param value: given value to convert to html node
980 980 """
981 981
982 982 if value: # does bool conversion
983 983 return HTML.tag('i', class_="icon-true")
984 984 else: # not true as bool
985 985 return HTML.tag('i', class_="icon-false")
986 986
987 987
988 988 #==============================================================================
989 989 # PERMS
990 990 #==============================================================================
991 991 from rhodecode.lib.auth import HasPermissionAny, HasPermissionAll, \
992 992 HasRepoPermissionAny, HasRepoPermissionAll, HasRepoGroupPermissionAll, \
993 993 HasRepoGroupPermissionAny, HasRepoPermissionAnyApi, get_csrf_token, \
994 994 csrf_token_key
995 995
996 996
997 997 #==============================================================================
998 998 # GRAVATAR URL
999 999 #==============================================================================
1000 1000 class InitialsGravatar(object):
1001 1001 def __init__(self, email_address, first_name, last_name, size=30,
1002 1002 background=None, text_color='#fff'):
1003 1003 self.size = size
1004 1004 self.first_name = first_name
1005 1005 self.last_name = last_name
1006 1006 self.email_address = email_address
1007 1007 self.background = background or self.str2color(email_address)
1008 1008 self.text_color = text_color
1009 1009
1010 1010 def get_color_bank(self):
1011 1011 """
1012 1012 returns a predefined list of colors that gravatars can use.
1013 1013 Those are randomized distinct colors that guarantee readability and
1014 1014 uniqueness.
1015 1015
1016 1016 generated with: http://phrogz.net/css/distinct-colors.html
1017 1017 """
1018 1018 return [
1019 1019 '#bf3030', '#a67f53', '#00ff00', '#5989b3', '#392040', '#d90000',
1020 1020 '#402910', '#204020', '#79baf2', '#a700b3', '#bf6060', '#7f5320',
1021 1021 '#008000', '#003059', '#ee00ff', '#ff0000', '#8c4b00', '#007300',
1022 1022 '#005fb3', '#de73e6', '#ff4040', '#ffaa00', '#3df255', '#203140',
1023 1023 '#47004d', '#591616', '#664400', '#59b365', '#0d2133', '#83008c',
1024 1024 '#592d2d', '#bf9f60', '#73e682', '#1d3f73', '#73006b', '#402020',
1025 1025 '#b2862d', '#397341', '#597db3', '#e600d6', '#a60000', '#736039',
1026 1026 '#00b318', '#79aaf2', '#330d30', '#ff8080', '#403010', '#16591f',
1027 1027 '#002459', '#8c4688', '#e50000', '#ffbf40', '#00732e', '#102340',
1028 1028 '#bf60ac', '#8c4646', '#cc8800', '#00a642', '#1d3473', '#b32d98',
1029 1029 '#660e00', '#ffd580', '#80ffb2', '#7391e6', '#733967', '#d97b6c',
1030 1030 '#8c5e00', '#59b389', '#3967e6', '#590047', '#73281d', '#665200',
1031 1031 '#00e67a', '#2d50b3', '#8c2377', '#734139', '#b2982d', '#16593a',
1032 1032 '#001859', '#ff00aa', '#a65e53', '#ffcc00', '#0d3321', '#2d3959',
1033 1033 '#731d56', '#401610', '#4c3d00', '#468c6c', '#002ca6', '#d936a3',
1034 1034 '#d94c36', '#403920', '#36d9a3', '#0d1733', '#592d4a', '#993626',
1035 1035 '#cca300', '#00734d', '#46598c', '#8c005e', '#7f1100', '#8c7000',
1036 1036 '#00a66f', '#7382e6', '#b32d74', '#d9896c', '#ffe680', '#1d7362',
1037 1037 '#364cd9', '#73003d', '#d93a00', '#998a4d', '#59b3a1', '#5965b3',
1038 1038 '#e5007a', '#73341d', '#665f00', '#00b38f', '#0018b3', '#59163a',
1039 1039 '#b2502d', '#bfb960', '#00ffcc', '#23318c', '#a6537f', '#734939',
1040 1040 '#b2a700', '#104036', '#3d3df2', '#402031', '#e56739', '#736f39',
1041 1041 '#79f2ea', '#000059', '#401029', '#4c1400', '#ffee00', '#005953',
1042 1042 '#101040', '#990052', '#402820', '#403d10', '#00ffee', '#0000d9',
1043 1043 '#ff80c4', '#a66953', '#eeff00', '#00ccbe', '#8080ff', '#e673a1',
1044 1044 '#a62c00', '#474d00', '#1a3331', '#46468c', '#733950', '#662900',
1045 1045 '#858c23', '#238c85', '#0f0073', '#b20047', '#d9986c', '#becc00',
1046 1046 '#396f73', '#281d73', '#ff0066', '#ff6600', '#dee673', '#59adb3',
1047 1047 '#6559b3', '#590024', '#b2622d', '#98b32d', '#36ced9', '#332d59',
1048 1048 '#40001a', '#733f1d', '#526600', '#005359', '#242040', '#bf6079',
1049 1049 '#735039', '#cef23d', '#007780', '#5630bf', '#66001b', '#b24700',
1050 1050 '#acbf60', '#1d6273', '#25008c', '#731d34', '#a67453', '#50592d',
1051 1051 '#00ccff', '#6600ff', '#ff0044', '#4c1f00', '#8a994d', '#79daf2',
1052 1052 '#a173e6', '#d93662', '#402310', '#aaff00', '#2d98b3', '#8c40ff',
1053 1053 '#592d39', '#ff8c40', '#354020', '#103640', '#1a0040', '#331a20',
1054 1054 '#331400', '#334d00', '#1d5673', '#583973', '#7f0022', '#4c3626',
1055 1055 '#88cc00', '#36a3d9', '#3d0073', '#d9364c', '#33241a', '#698c23',
1056 1056 '#5995b3', '#300059', '#e57382', '#7f3300', '#366600', '#00aaff',
1057 1057 '#3a1659', '#733941', '#663600', '#74b32d', '#003c59', '#7f53a6',
1058 1058 '#73000f', '#ff8800', '#baf279', '#79caf2', '#291040', '#a6293a',
1059 1059 '#b2742d', '#587339', '#0077b3', '#632699', '#400009', '#d9a66c',
1060 1060 '#294010', '#2d4a59', '#aa00ff', '#4c131b', '#b25f00', '#5ce600',
1061 1061 '#267399', '#a336d9', '#990014', '#664e33', '#86bf60', '#0088ff',
1062 1062 '#7700b3', '#593a16', '#073300', '#1d4b73', '#ac60bf', '#e59539',
1063 1063 '#4f8c46', '#368dd9', '#5c0073'
1064 1064 ]
1065 1065
1066 1066 def rgb_to_hex_color(self, rgb_tuple):
1067 1067 """
1068 1068 Converts an rgb_tuple passed to an hex color.
1069 1069
1070 1070 :param rgb_tuple: tuple with 3 ints represents rgb color space
1071 1071 """
1072 1072 return '#' + ("".join(map(chr, rgb_tuple)).encode('hex'))
1073 1073
1074 1074 def email_to_int_list(self, email_str):
1075 1075 """
1076 1076 Get every byte of the hex digest value of email and turn it to integer.
1077 1077 It's going to be always between 0-255
1078 1078 """
1079 1079 digest = md5_safe(email_str.lower())
1080 1080 return [int(digest[i * 2:i * 2 + 2], 16) for i in range(16)]
1081 1081
1082 1082 def pick_color_bank_index(self, email_str, color_bank):
1083 1083 return self.email_to_int_list(email_str)[0] % len(color_bank)
1084 1084
1085 1085 def str2color(self, email_str):
1086 1086 """
1087 1087 Tries to map in a stable algorithm an email to color
1088 1088
1089 1089 :param email_str:
1090 1090 """
1091 1091 color_bank = self.get_color_bank()
1092 1092 # pick position (module it's length so we always find it in the
1093 1093 # bank even if it's smaller than 256 values
1094 1094 pos = self.pick_color_bank_index(email_str, color_bank)
1095 1095 return color_bank[pos]
1096 1096
1097 1097 def normalize_email(self, email_address):
1098 1098 import unicodedata
1099 1099 # default host used to fill in the fake/missing email
1100 1100 default_host = u'localhost'
1101 1101
1102 1102 if not email_address:
1103 1103 email_address = u'%s@%s' % (User.DEFAULT_USER, default_host)
1104 1104
1105 1105 email_address = safe_unicode(email_address)
1106 1106
1107 1107 if u'@' not in email_address:
1108 1108 email_address = u'%s@%s' % (email_address, default_host)
1109 1109
1110 1110 if email_address.endswith(u'@'):
1111 1111 email_address = u'%s%s' % (email_address, default_host)
1112 1112
1113 1113 email_address = unicodedata.normalize('NFKD', email_address)\
1114 1114 .encode('ascii', 'ignore')
1115 1115 return email_address
1116 1116
1117 1117 def get_initials(self):
1118 1118 """
1119 1119 Returns 2 letter initials calculated based on the input.
1120 1120 The algorithm picks first given email address, and takes first letter
1121 1121 of part before @, and then the first letter of server name. In case
1122 1122 the part before @ is in a format of `somestring.somestring2` it replaces
1123 1123 the server letter with first letter of somestring2
1124 1124
1125 1125 In case function was initialized with both first and lastname, this
1126 1126 overrides the extraction from email by first letter of the first and
1127 1127 last name. We add special logic to that functionality, In case Full name
1128 1128 is compound, like Guido Von Rossum, we use last part of the last name
1129 1129 (Von Rossum) picking `R`.
1130 1130
1131 1131 Function also normalizes the non-ascii characters to they ascii
1132 1132 representation, eg Ą => A
1133 1133 """
1134 1134 import unicodedata
1135 1135 # replace non-ascii to ascii
1136 1136 first_name = unicodedata.normalize(
1137 1137 'NFKD', safe_unicode(self.first_name)).encode('ascii', 'ignore')
1138 1138 last_name = unicodedata.normalize(
1139 1139 'NFKD', safe_unicode(self.last_name)).encode('ascii', 'ignore')
1140 1140
1141 1141 # do NFKD encoding, and also make sure email has proper format
1142 1142 email_address = self.normalize_email(self.email_address)
1143 1143
1144 1144 # first push the email initials
1145 1145 prefix, server = email_address.split('@', 1)
1146 1146
1147 1147 # check if prefix is maybe a 'firstname.lastname' syntax
1148 1148 _dot_split = prefix.rsplit('.', 1)
1149 1149 if len(_dot_split) == 2:
1150 1150 initials = [_dot_split[0][0], _dot_split[1][0]]
1151 1151 else:
1152 1152 initials = [prefix[0], server[0]]
1153 1153
1154 1154 # then try to replace either firtname or lastname
1155 1155 fn_letter = (first_name or " ")[0].strip()
1156 1156 ln_letter = (last_name.split(' ', 1)[-1] or " ")[0].strip()
1157 1157
1158 1158 if fn_letter:
1159 1159 initials[0] = fn_letter
1160 1160
1161 1161 if ln_letter:
1162 1162 initials[1] = ln_letter
1163 1163
1164 1164 return ''.join(initials).upper()
1165 1165
1166 1166 def get_img_data_by_type(self, font_family, img_type):
1167 1167 default_user = """
1168 1168 <svg xmlns="http://www.w3.org/2000/svg"
1169 1169 version="1.1" x="0px" y="0px" width="{size}" height="{size}"
1170 1170 viewBox="-15 -10 439.165 429.164"
1171 1171
1172 1172 xml:space="preserve"
1173 1173 style="background:{background};" >
1174 1174
1175 1175 <path d="M204.583,216.671c50.664,0,91.74-48.075,
1176 1176 91.74-107.378c0-82.237-41.074-107.377-91.74-107.377
1177 1177 c-50.668,0-91.74,25.14-91.74,107.377C112.844,
1178 1178 168.596,153.916,216.671,
1179 1179 204.583,216.671z" fill="{text_color}"/>
1180 1180 <path d="M407.164,374.717L360.88,
1181 1181 270.454c-2.117-4.771-5.836-8.728-10.465-11.138l-71.83-37.392
1182 1182 c-1.584-0.823-3.502-0.663-4.926,0.415c-20.316,
1183 1183 15.366-44.203,23.488-69.076,23.488c-24.877,
1184 1184 0-48.762-8.122-69.078-23.488
1185 1185 c-1.428-1.078-3.346-1.238-4.93-0.415L58.75,
1186 1186 259.316c-4.631,2.41-8.346,6.365-10.465,11.138L2.001,374.717
1187 1187 c-3.191,7.188-2.537,15.412,1.75,22.005c4.285,
1188 1188 6.592,11.537,10.526,19.4,10.526h362.861c7.863,0,15.117-3.936,
1189 1189 19.402-10.527 C409.699,390.129,
1190 1190 410.355,381.902,407.164,374.717z" fill="{text_color}"/>
1191 1191 </svg>""".format(
1192 1192 size=self.size,
1193 1193 background='#979797', # @grey4
1194 1194 text_color=self.text_color,
1195 1195 font_family=font_family)
1196 1196
1197 1197 return {
1198 1198 "default_user": default_user
1199 1199 }[img_type]
1200 1200
1201 1201 def get_img_data(self, svg_type=None):
1202 1202 """
1203 1203 generates the svg metadata for image
1204 1204 """
1205 1205
1206 1206 font_family = ','.join([
1207 1207 'proximanovaregular',
1208 1208 'Proxima Nova Regular',
1209 1209 'Proxima Nova',
1210 1210 'Arial',
1211 1211 'Lucida Grande',
1212 1212 'sans-serif'
1213 1213 ])
1214 1214 if svg_type:
1215 1215 return self.get_img_data_by_type(font_family, svg_type)
1216 1216
1217 1217 initials = self.get_initials()
1218 1218 img_data = """
1219 1219 <svg xmlns="http://www.w3.org/2000/svg" pointer-events="none"
1220 1220 width="{size}" height="{size}"
1221 1221 style="width: 100%; height: 100%; background-color: {background}"
1222 1222 viewBox="0 0 {size} {size}">
1223 1223 <text text-anchor="middle" y="50%" x="50%" dy="0.35em"
1224 1224 pointer-events="auto" fill="{text_color}"
1225 1225 font-family="{font_family}"
1226 1226 style="font-weight: 400; font-size: {f_size}px;">{text}
1227 1227 </text>
1228 1228 </svg>""".format(
1229 1229 size=self.size,
1230 1230 f_size=self.size/1.85, # scale the text inside the box nicely
1231 1231 background=self.background,
1232 1232 text_color=self.text_color,
1233 1233 text=initials.upper(),
1234 1234 font_family=font_family)
1235 1235
1236 1236 return img_data
1237 1237
1238 1238 def generate_svg(self, svg_type=None):
1239 1239 img_data = self.get_img_data(svg_type)
1240 1240 return "data:image/svg+xml;base64,%s" % img_data.encode('base64')
1241 1241
1242 1242
1243 1243 def initials_gravatar(email_address, first_name, last_name, size=30):
1244 1244 svg_type = None
1245 1245 if email_address == User.DEFAULT_USER_EMAIL:
1246 1246 svg_type = 'default_user'
1247 1247 klass = InitialsGravatar(email_address, first_name, last_name, size)
1248 1248 return klass.generate_svg(svg_type=svg_type)
1249 1249
1250 1250
1251 def gravatar_url(email_address, size=30):
1252 # doh, we need to re-import those to mock it later
1253 from pylons import tmpl_context as c
1251 def gravatar_url(email_address, size=30, request=None):
1252 request = get_current_request()
1253 if request and hasattr(request, 'call_context'):
1254 _use_gravatar = request.call_context.visual.use_gravatar
1255 _gravatar_url = request.call_context.visual.gravatar_url
1256 else:
1257 # doh, we need to re-import those to mock it later
1258 from pylons import tmpl_context as c
1254 1259
1255 _use_gravatar = c.visual.use_gravatar
1256 _gravatar_url = c.visual.gravatar_url or User.DEFAULT_GRAVATAR_URL
1260 _use_gravatar = c.visual.use_gravatar
1261 _gravatar_url = c.visual.gravatar_url
1262
1263 _gravatar_url = _gravatar_url or User.DEFAULT_GRAVATAR_URL
1257 1264
1258 1265 email_address = email_address or User.DEFAULT_USER_EMAIL
1259 1266 if isinstance(email_address, unicode):
1260 1267 # hashlib crashes on unicode items
1261 1268 email_address = safe_str(email_address)
1262 1269
1263 1270 # empty email or default user
1264 1271 if not email_address or email_address == User.DEFAULT_USER_EMAIL:
1265 1272 return initials_gravatar(User.DEFAULT_USER_EMAIL, '', '', size=size)
1266 1273
1267 1274 if _use_gravatar:
1268 1275 # TODO: Disuse pyramid thread locals. Think about another solution to
1269 1276 # get the host and schema here.
1270 1277 request = get_current_request()
1271 1278 tmpl = safe_str(_gravatar_url)
1272 1279 tmpl = tmpl.replace('{email}', email_address)\
1273 1280 .replace('{md5email}', md5_safe(email_address.lower())) \
1274 1281 .replace('{netloc}', request.host)\
1275 1282 .replace('{scheme}', request.scheme)\
1276 1283 .replace('{size}', safe_str(size))
1277 1284 return tmpl
1278 1285 else:
1279 1286 return initials_gravatar(email_address, '', '', size=size)
1280 1287
1281 1288
1282 1289 class Page(_Page):
1283 1290 """
1284 1291 Custom pager to match rendering style with paginator
1285 1292 """
1286 1293
1287 1294 def _get_pos(self, cur_page, max_page, items):
1288 1295 edge = (items / 2) + 1
1289 1296 if (cur_page <= edge):
1290 1297 radius = max(items / 2, items - cur_page)
1291 1298 elif (max_page - cur_page) < edge:
1292 1299 radius = (items - 1) - (max_page - cur_page)
1293 1300 else:
1294 1301 radius = items / 2
1295 1302
1296 1303 left = max(1, (cur_page - (radius)))
1297 1304 right = min(max_page, cur_page + (radius))
1298 1305 return left, cur_page, right
1299 1306
1300 1307 def _range(self, regexp_match):
1301 1308 """
1302 1309 Return range of linked pages (e.g. '1 2 [3] 4 5 6 7 8').
1303 1310
1304 1311 Arguments:
1305 1312
1306 1313 regexp_match
1307 1314 A "re" (regular expressions) match object containing the
1308 1315 radius of linked pages around the current page in
1309 1316 regexp_match.group(1) as a string
1310 1317
1311 1318 This function is supposed to be called as a callable in
1312 1319 re.sub.
1313 1320
1314 1321 """
1315 1322 radius = int(regexp_match.group(1))
1316 1323
1317 1324 # Compute the first and last page number within the radius
1318 1325 # e.g. '1 .. 5 6 [7] 8 9 .. 12'
1319 1326 # -> leftmost_page = 5
1320 1327 # -> rightmost_page = 9
1321 1328 leftmost_page, _cur, rightmost_page = self._get_pos(self.page,
1322 1329 self.last_page,
1323 1330 (radius * 2) + 1)
1324 1331 nav_items = []
1325 1332
1326 1333 # Create a link to the first page (unless we are on the first page
1327 1334 # or there would be no need to insert '..' spacers)
1328 1335 if self.page != self.first_page and self.first_page < leftmost_page:
1329 1336 nav_items.append(self._pagerlink(self.first_page, self.first_page))
1330 1337
1331 1338 # Insert dots if there are pages between the first page
1332 1339 # and the currently displayed page range
1333 1340 if leftmost_page - self.first_page > 1:
1334 1341 # Wrap in a SPAN tag if nolink_attr is set
1335 1342 text = '..'
1336 1343 if self.dotdot_attr:
1337 1344 text = HTML.span(c=text, **self.dotdot_attr)
1338 1345 nav_items.append(text)
1339 1346
1340 1347 for thispage in xrange(leftmost_page, rightmost_page + 1):
1341 1348 # Hilight the current page number and do not use a link
1342 1349 if thispage == self.page:
1343 1350 text = '%s' % (thispage,)
1344 1351 # Wrap in a SPAN tag if nolink_attr is set
1345 1352 if self.curpage_attr:
1346 1353 text = HTML.span(c=text, **self.curpage_attr)
1347 1354 nav_items.append(text)
1348 1355 # Otherwise create just a link to that page
1349 1356 else:
1350 1357 text = '%s' % (thispage,)
1351 1358 nav_items.append(self._pagerlink(thispage, text))
1352 1359
1353 1360 # Insert dots if there are pages between the displayed
1354 1361 # page numbers and the end of the page range
1355 1362 if self.last_page - rightmost_page > 1:
1356 1363 text = '..'
1357 1364 # Wrap in a SPAN tag if nolink_attr is set
1358 1365 if self.dotdot_attr:
1359 1366 text = HTML.span(c=text, **self.dotdot_attr)
1360 1367 nav_items.append(text)
1361 1368
1362 1369 # Create a link to the very last page (unless we are on the last
1363 1370 # page or there would be no need to insert '..' spacers)
1364 1371 if self.page != self.last_page and rightmost_page < self.last_page:
1365 1372 nav_items.append(self._pagerlink(self.last_page, self.last_page))
1366 1373
1367 1374 ## prerender links
1368 1375 #_page_link = url.current()
1369 1376 #nav_items.append(literal('<link rel="prerender" href="%s?page=%s">' % (_page_link, str(int(self.page)+1))))
1370 1377 #nav_items.append(literal('<link rel="prefetch" href="%s?page=%s">' % (_page_link, str(int(self.page)+1))))
1371 1378 return self.separator.join(nav_items)
1372 1379
1373 1380 def pager(self, format='~2~', page_param='page', partial_param='partial',
1374 1381 show_if_single_page=False, separator=' ', onclick=None,
1375 1382 symbol_first='<<', symbol_last='>>',
1376 1383 symbol_previous='<', symbol_next='>',
1377 1384 link_attr={'class': 'pager_link', 'rel': 'prerender'},
1378 1385 curpage_attr={'class': 'pager_curpage'},
1379 1386 dotdot_attr={'class': 'pager_dotdot'}, **kwargs):
1380 1387
1381 1388 self.curpage_attr = curpage_attr
1382 1389 self.separator = separator
1383 1390 self.pager_kwargs = kwargs
1384 1391 self.page_param = page_param
1385 1392 self.partial_param = partial_param
1386 1393 self.onclick = onclick
1387 1394 self.link_attr = link_attr
1388 1395 self.dotdot_attr = dotdot_attr
1389 1396
1390 1397 # Don't show navigator if there is no more than one page
1391 1398 if self.page_count == 0 or (self.page_count == 1 and not show_if_single_page):
1392 1399 return ''
1393 1400
1394 1401 from string import Template
1395 1402 # Replace ~...~ in token format by range of pages
1396 1403 result = re.sub(r'~(\d+)~', self._range, format)
1397 1404
1398 1405 # Interpolate '%' variables
1399 1406 result = Template(result).safe_substitute({
1400 1407 'first_page': self.first_page,
1401 1408 'last_page': self.last_page,
1402 1409 'page': self.page,
1403 1410 'page_count': self.page_count,
1404 1411 'items_per_page': self.items_per_page,
1405 1412 'first_item': self.first_item,
1406 1413 'last_item': self.last_item,
1407 1414 'item_count': self.item_count,
1408 1415 'link_first': self.page > self.first_page and \
1409 1416 self._pagerlink(self.first_page, symbol_first) or '',
1410 1417 'link_last': self.page < self.last_page and \
1411 1418 self._pagerlink(self.last_page, symbol_last) or '',
1412 1419 'link_previous': self.previous_page and \
1413 1420 self._pagerlink(self.previous_page, symbol_previous) \
1414 1421 or HTML.span(symbol_previous, class_="pg-previous disabled"),
1415 1422 'link_next': self.next_page and \
1416 1423 self._pagerlink(self.next_page, symbol_next) \
1417 1424 or HTML.span(symbol_next, class_="pg-next disabled")
1418 1425 })
1419 1426
1420 1427 return literal(result)
1421 1428
1422 1429
1423 1430 #==============================================================================
1424 1431 # REPO PAGER, PAGER FOR REPOSITORY
1425 1432 #==============================================================================
1426 1433 class RepoPage(Page):
1427 1434
1428 1435 def __init__(self, collection, page=1, items_per_page=20,
1429 1436 item_count=None, url=None, **kwargs):
1430 1437
1431 1438 """Create a "RepoPage" instance. special pager for paging
1432 1439 repository
1433 1440 """
1434 1441 self._url_generator = url
1435 1442
1436 1443 # Safe the kwargs class-wide so they can be used in the pager() method
1437 1444 self.kwargs = kwargs
1438 1445
1439 1446 # Save a reference to the collection
1440 1447 self.original_collection = collection
1441 1448
1442 1449 self.collection = collection
1443 1450
1444 1451 # The self.page is the number of the current page.
1445 1452 # The first page has the number 1!
1446 1453 try:
1447 1454 self.page = int(page) # make it int() if we get it as a string
1448 1455 except (ValueError, TypeError):
1449 1456 self.page = 1
1450 1457
1451 1458 self.items_per_page = items_per_page
1452 1459
1453 1460 # Unless the user tells us how many items the collections has
1454 1461 # we calculate that ourselves.
1455 1462 if item_count is not None:
1456 1463 self.item_count = item_count
1457 1464 else:
1458 1465 self.item_count = len(self.collection)
1459 1466
1460 1467 # Compute the number of the first and last available page
1461 1468 if self.item_count > 0:
1462 1469 self.first_page = 1
1463 1470 self.page_count = int(math.ceil(float(self.item_count) /
1464 1471 self.items_per_page))
1465 1472 self.last_page = self.first_page + self.page_count - 1
1466 1473
1467 1474 # Make sure that the requested page number is the range of
1468 1475 # valid pages
1469 1476 if self.page > self.last_page:
1470 1477 self.page = self.last_page
1471 1478 elif self.page < self.first_page:
1472 1479 self.page = self.first_page
1473 1480
1474 1481 # Note: the number of items on this page can be less than
1475 1482 # items_per_page if the last page is not full
1476 1483 self.first_item = max(0, (self.item_count) - (self.page *
1477 1484 items_per_page))
1478 1485 self.last_item = ((self.item_count - 1) - items_per_page *
1479 1486 (self.page - 1))
1480 1487
1481 1488 self.items = list(self.collection[self.first_item:self.last_item + 1])
1482 1489
1483 1490 # Links to previous and next page
1484 1491 if self.page > self.first_page:
1485 1492 self.previous_page = self.page - 1
1486 1493 else:
1487 1494 self.previous_page = None
1488 1495
1489 1496 if self.page < self.last_page:
1490 1497 self.next_page = self.page + 1
1491 1498 else:
1492 1499 self.next_page = None
1493 1500
1494 1501 # No items available
1495 1502 else:
1496 1503 self.first_page = None
1497 1504 self.page_count = 0
1498 1505 self.last_page = None
1499 1506 self.first_item = None
1500 1507 self.last_item = None
1501 1508 self.previous_page = None
1502 1509 self.next_page = None
1503 1510 self.items = []
1504 1511
1505 1512 # This is a subclass of the 'list' type. Initialise the list now.
1506 1513 list.__init__(self, reversed(self.items))
1507 1514
1508 1515
1509 1516 def changed_tooltip(nodes):
1510 1517 """
1511 1518 Generates a html string for changed nodes in commit page.
1512 1519 It limits the output to 30 entries
1513 1520
1514 1521 :param nodes: LazyNodesGenerator
1515 1522 """
1516 1523 if nodes:
1517 1524 pref = ': <br/> '
1518 1525 suf = ''
1519 1526 if len(nodes) > 30:
1520 1527 suf = '<br/>' + _(' and %s more') % (len(nodes) - 30)
1521 1528 return literal(pref + '<br/> '.join([safe_unicode(x.path)
1522 1529 for x in nodes[:30]]) + suf)
1523 1530 else:
1524 1531 return ': ' + _('No Files')
1525 1532
1526 1533
1527 1534 def breadcrumb_repo_link(repo):
1528 1535 """
1529 1536 Makes a breadcrumbs path link to repo
1530 1537
1531 1538 ex::
1532 1539 group >> subgroup >> repo
1533 1540
1534 1541 :param repo: a Repository instance
1535 1542 """
1536 1543
1537 1544 path = [
1538 1545 link_to(group.name, route_path('repo_group_home', repo_group_name=group.group_name))
1539 1546 for group in repo.groups_with_parents
1540 1547 ] + [
1541 1548 link_to(repo.just_name, route_path('repo_summary', repo_name=repo.repo_name))
1542 1549 ]
1543 1550
1544 1551 return literal(' &raquo; '.join(path))
1545 1552
1546 1553
1547 1554 def format_byte_size_binary(file_size):
1548 1555 """
1549 1556 Formats file/folder sizes to standard.
1550 1557 """
1551 1558 formatted_size = format_byte_size(file_size, binary=True)
1552 1559 return formatted_size
1553 1560
1554 1561
1555 1562 def urlify_text(text_, safe=True):
1556 1563 """
1557 1564 Extrac urls from text and make html links out of them
1558 1565
1559 1566 :param text_:
1560 1567 """
1561 1568
1562 1569 url_pat = re.compile(r'''(http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@#.&+]'''
1563 1570 '''|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+)''')
1564 1571
1565 1572 def url_func(match_obj):
1566 1573 url_full = match_obj.groups()[0]
1567 1574 return '<a href="%(url)s">%(url)s</a>' % ({'url': url_full})
1568 1575 _newtext = url_pat.sub(url_func, text_)
1569 1576 if safe:
1570 1577 return literal(_newtext)
1571 1578 return _newtext
1572 1579
1573 1580
1574 1581 def urlify_commits(text_, repository):
1575 1582 """
1576 1583 Extract commit ids from text and make link from them
1577 1584
1578 1585 :param text_:
1579 1586 :param repository: repo name to build the URL with
1580 1587 """
1581 1588 from pylons import url # doh, we need to re-import url to mock it later
1582 1589 URL_PAT = re.compile(r'(^|\s)([0-9a-fA-F]{12,40})($|\s)')
1583 1590
1584 1591 def url_func(match_obj):
1585 1592 commit_id = match_obj.groups()[1]
1586 1593 pref = match_obj.groups()[0]
1587 1594 suf = match_obj.groups()[2]
1588 1595
1589 1596 tmpl = (
1590 1597 '%(pref)s<a class="%(cls)s" href="%(url)s">'
1591 1598 '%(commit_id)s</a>%(suf)s'
1592 1599 )
1593 1600 return tmpl % {
1594 1601 'pref': pref,
1595 1602 'cls': 'revision-link',
1596 1603 'url': url('changeset_home', repo_name=repository,
1597 1604 revision=commit_id, qualified=True),
1598 1605 'commit_id': commit_id,
1599 1606 'suf': suf
1600 1607 }
1601 1608
1602 1609 newtext = URL_PAT.sub(url_func, text_)
1603 1610
1604 1611 return newtext
1605 1612
1606 1613
1607 1614 def _process_url_func(match_obj, repo_name, uid, entry,
1608 1615 return_raw_data=False, link_format='html'):
1609 1616 pref = ''
1610 1617 if match_obj.group().startswith(' '):
1611 1618 pref = ' '
1612 1619
1613 1620 issue_id = ''.join(match_obj.groups())
1614 1621
1615 1622 if link_format == 'html':
1616 1623 tmpl = (
1617 1624 '%(pref)s<a class="%(cls)s" href="%(url)s">'
1618 1625 '%(issue-prefix)s%(id-repr)s'
1619 1626 '</a>')
1620 1627 elif link_format == 'rst':
1621 1628 tmpl = '`%(issue-prefix)s%(id-repr)s <%(url)s>`_'
1622 1629 elif link_format == 'markdown':
1623 1630 tmpl = '[%(issue-prefix)s%(id-repr)s](%(url)s)'
1624 1631 else:
1625 1632 raise ValueError('Bad link_format:{}'.format(link_format))
1626 1633
1627 1634 (repo_name_cleaned,
1628 1635 parent_group_name) = RepoGroupModel().\
1629 1636 _get_group_name_and_parent(repo_name)
1630 1637
1631 1638 # variables replacement
1632 1639 named_vars = {
1633 1640 'id': issue_id,
1634 1641 'repo': repo_name,
1635 1642 'repo_name': repo_name_cleaned,
1636 1643 'group_name': parent_group_name
1637 1644 }
1638 1645 # named regex variables
1639 1646 named_vars.update(match_obj.groupdict())
1640 1647 _url = string.Template(entry['url']).safe_substitute(**named_vars)
1641 1648
1642 1649 data = {
1643 1650 'pref': pref,
1644 1651 'cls': 'issue-tracker-link',
1645 1652 'url': _url,
1646 1653 'id-repr': issue_id,
1647 1654 'issue-prefix': entry['pref'],
1648 1655 'serv': entry['url'],
1649 1656 }
1650 1657 if return_raw_data:
1651 1658 return {
1652 1659 'id': issue_id,
1653 1660 'url': _url
1654 1661 }
1655 1662 return tmpl % data
1656 1663
1657 1664
1658 1665 def process_patterns(text_string, repo_name, link_format='html'):
1659 1666 allowed_formats = ['html', 'rst', 'markdown']
1660 1667 if link_format not in allowed_formats:
1661 1668 raise ValueError('Link format can be only one of:{} got {}'.format(
1662 1669 allowed_formats, link_format))
1663 1670
1664 1671 repo = None
1665 1672 if repo_name:
1666 1673 # Retrieving repo_name to avoid invalid repo_name to explode on
1667 1674 # IssueTrackerSettingsModel but still passing invalid name further down
1668 1675 repo = Repository.get_by_repo_name(repo_name, cache=True)
1669 1676
1670 1677 settings_model = IssueTrackerSettingsModel(repo=repo)
1671 1678 active_entries = settings_model.get_settings(cache=True)
1672 1679
1673 1680 issues_data = []
1674 1681 newtext = text_string
1675 1682
1676 1683 for uid, entry in active_entries.items():
1677 1684 log.debug('found issue tracker entry with uid %s' % (uid,))
1678 1685
1679 1686 if not (entry['pat'] and entry['url']):
1680 1687 log.debug('skipping due to missing data')
1681 1688 continue
1682 1689
1683 1690 log.debug('issue tracker entry: uid: `%s` PAT:%s URL:%s PREFIX:%s'
1684 1691 % (uid, entry['pat'], entry['url'], entry['pref']))
1685 1692
1686 1693 try:
1687 1694 pattern = re.compile(r'%s' % entry['pat'])
1688 1695 except re.error:
1689 1696 log.exception(
1690 1697 'issue tracker pattern: `%s` failed to compile',
1691 1698 entry['pat'])
1692 1699 continue
1693 1700
1694 1701 data_func = partial(
1695 1702 _process_url_func, repo_name=repo_name, entry=entry, uid=uid,
1696 1703 return_raw_data=True)
1697 1704
1698 1705 for match_obj in pattern.finditer(text_string):
1699 1706 issues_data.append(data_func(match_obj))
1700 1707
1701 1708 url_func = partial(
1702 1709 _process_url_func, repo_name=repo_name, entry=entry, uid=uid,
1703 1710 link_format=link_format)
1704 1711
1705 1712 newtext = pattern.sub(url_func, newtext)
1706 1713 log.debug('processed prefix:uid `%s`' % (uid,))
1707 1714
1708 1715 return newtext, issues_data
1709 1716
1710 1717
1711 1718 def urlify_commit_message(commit_text, repository=None):
1712 1719 """
1713 1720 Parses given text message and makes proper links.
1714 1721 issues are linked to given issue-server, and rest is a commit link
1715 1722
1716 1723 :param commit_text:
1717 1724 :param repository:
1718 1725 """
1719 1726 from pylons import url # doh, we need to re-import url to mock it later
1720 1727
1721 1728 def escaper(string):
1722 1729 return string.replace('<', '&lt;').replace('>', '&gt;')
1723 1730
1724 1731 newtext = escaper(commit_text)
1725 1732
1726 1733 # extract http/https links and make them real urls
1727 1734 newtext = urlify_text(newtext, safe=False)
1728 1735
1729 1736 # urlify commits - extract commit ids and make link out of them, if we have
1730 1737 # the scope of repository present.
1731 1738 if repository:
1732 1739 newtext = urlify_commits(newtext, repository)
1733 1740
1734 1741 # process issue tracker patterns
1735 1742 newtext, issues = process_patterns(newtext, repository or '')
1736 1743
1737 1744 return literal(newtext)
1738 1745
1739 1746
1740 1747 def render_binary(repo_name, file_obj):
1741 1748 """
1742 1749 Choose how to render a binary file
1743 1750 """
1744 1751 filename = file_obj.name
1745 1752
1746 1753 # images
1747 1754 for ext in ['*.png', '*.jpg', '*.ico', '*.gif']:
1748 1755 if fnmatch.fnmatch(filename, pat=ext):
1749 1756 alt = filename
1750 1757 src = url('files_raw_home', repo_name=repo_name,
1751 1758 revision=file_obj.commit.raw_id, f_path=file_obj.path)
1752 1759 return literal('<img class="rendered-binary" alt="{}" src="{}">'.format(alt, src))
1753 1760
1754 1761
1755 1762 def renderer_from_filename(filename, exclude=None):
1756 1763 """
1757 1764 choose a renderer based on filename, this works only for text based files
1758 1765 """
1759 1766
1760 1767 # ipython
1761 1768 for ext in ['*.ipynb']:
1762 1769 if fnmatch.fnmatch(filename, pat=ext):
1763 1770 return 'jupyter'
1764 1771
1765 1772 is_markup = MarkupRenderer.renderer_from_filename(filename, exclude=exclude)
1766 1773 if is_markup:
1767 1774 return is_markup
1768 1775 return None
1769 1776
1770 1777
1771 1778 def render(source, renderer='rst', mentions=False, relative_url=None,
1772 1779 repo_name=None):
1773 1780
1774 1781 def maybe_convert_relative_links(html_source):
1775 1782 if relative_url:
1776 1783 return relative_links(html_source, relative_url)
1777 1784 return html_source
1778 1785
1779 1786 if renderer == 'rst':
1780 1787 if repo_name:
1781 1788 # process patterns on comments if we pass in repo name
1782 1789 source, issues = process_patterns(
1783 1790 source, repo_name, link_format='rst')
1784 1791
1785 1792 return literal(
1786 1793 '<div class="rst-block">%s</div>' %
1787 1794 maybe_convert_relative_links(
1788 1795 MarkupRenderer.rst(source, mentions=mentions)))
1789 1796 elif renderer == 'markdown':
1790 1797 if repo_name:
1791 1798 # process patterns on comments if we pass in repo name
1792 1799 source, issues = process_patterns(
1793 1800 source, repo_name, link_format='markdown')
1794 1801
1795 1802 return literal(
1796 1803 '<div class="markdown-block">%s</div>' %
1797 1804 maybe_convert_relative_links(
1798 1805 MarkupRenderer.markdown(source, flavored=True,
1799 1806 mentions=mentions)))
1800 1807 elif renderer == 'jupyter':
1801 1808 return literal(
1802 1809 '<div class="ipynb">%s</div>' %
1803 1810 maybe_convert_relative_links(
1804 1811 MarkupRenderer.jupyter(source)))
1805 1812
1806 1813 # None means just show the file-source
1807 1814 return None
1808 1815
1809 1816
1810 1817 def commit_status(repo, commit_id):
1811 1818 return ChangesetStatusModel().get_status(repo, commit_id)
1812 1819
1813 1820
1814 1821 def commit_status_lbl(commit_status):
1815 1822 return dict(ChangesetStatus.STATUSES).get(commit_status)
1816 1823
1817 1824
1818 1825 def commit_time(repo_name, commit_id):
1819 1826 repo = Repository.get_by_repo_name(repo_name)
1820 1827 commit = repo.get_commit(commit_id=commit_id)
1821 1828 return commit.date
1822 1829
1823 1830
1824 1831 def get_permission_name(key):
1825 1832 return dict(Permission.PERMS).get(key)
1826 1833
1827 1834
1828 1835 def journal_filter_help():
1829 1836 return _(
1830 1837 'Example filter terms:\n' +
1831 1838 ' repository:vcs\n' +
1832 1839 ' username:marcin\n' +
1833 1840 ' action:*push*\n' +
1834 1841 ' ip:127.0.0.1\n' +
1835 1842 ' date:20120101\n' +
1836 1843 ' date:[20120101100000 TO 20120102]\n' +
1837 1844 '\n' +
1838 1845 'Generate wildcards using \'*\' character:\n' +
1839 1846 ' "repository:vcs*" - search everything starting with \'vcs\'\n' +
1840 1847 ' "repository:*vcs*" - search for repository containing \'vcs\'\n' +
1841 1848 '\n' +
1842 1849 'Optional AND / OR operators in queries\n' +
1843 1850 ' "repository:vcs OR repository:test"\n' +
1844 1851 ' "username:test AND repository:test*"\n'
1845 1852 )
1846 1853
1847 1854
1848 1855 def search_filter_help(searcher):
1849 1856
1850 1857 terms = ''
1851 1858 return _(
1852 1859 'Example filter terms for `{searcher}` search:\n' +
1853 1860 '{terms}\n' +
1854 1861 'Generate wildcards using \'*\' character:\n' +
1855 1862 ' "repo_name:vcs*" - search everything starting with \'vcs\'\n' +
1856 1863 ' "repo_name:*vcs*" - search for repository containing \'vcs\'\n' +
1857 1864 '\n' +
1858 1865 'Optional AND / OR operators in queries\n' +
1859 1866 ' "repo_name:vcs OR repo_name:test"\n' +
1860 1867 ' "owner:test AND repo_name:test*"\n' +
1861 1868 'More: {search_doc}'
1862 1869 ).format(searcher=searcher.name,
1863 1870 terms=terms, search_doc=searcher.query_lang_doc)
1864 1871
1865 1872
1866 1873 def not_mapped_error(repo_name):
1867 1874 flash(_('%s repository is not mapped to db perhaps'
1868 1875 ' it was created or renamed from the filesystem'
1869 1876 ' please run the application again'
1870 1877 ' in order to rescan repositories') % repo_name, category='error')
1871 1878
1872 1879
1873 1880 def ip_range(ip_addr):
1874 1881 from rhodecode.model.db import UserIpMap
1875 1882 s, e = UserIpMap._get_ip_range(ip_addr)
1876 1883 return '%s - %s' % (s, e)
1877 1884
1878 1885
1879 1886 def form(url, method='post', needs_csrf_token=True, **attrs):
1880 1887 """Wrapper around webhelpers.tags.form to prevent CSRF attacks."""
1881 1888 if method.lower() != 'get' and needs_csrf_token:
1882 1889 raise Exception(
1883 1890 'Forms to POST/PUT/DELETE endpoints should have (in general) a ' +
1884 1891 'CSRF token. If the endpoint does not require such token you can ' +
1885 1892 'explicitly set the parameter needs_csrf_token to false.')
1886 1893
1887 1894 return wh_form(url, method=method, **attrs)
1888 1895
1889 1896
1890 1897 def secure_form(url, method="POST", multipart=False, **attrs):
1891 1898 """Start a form tag that points the action to an url. This
1892 1899 form tag will also include the hidden field containing
1893 1900 the auth token.
1894 1901
1895 1902 The url options should be given either as a string, or as a
1896 1903 ``url()`` function. The method for the form defaults to POST.
1897 1904
1898 1905 Options:
1899 1906
1900 1907 ``multipart``
1901 1908 If set to True, the enctype is set to "multipart/form-data".
1902 1909 ``method``
1903 1910 The method to use when submitting the form, usually either
1904 1911 "GET" or "POST". If "PUT", "DELETE", or another verb is used, a
1905 1912 hidden input with name _method is added to simulate the verb
1906 1913 over POST.
1907 1914
1908 1915 """
1909 1916 from webhelpers.pylonslib.secure_form import insecure_form
1910 1917 form = insecure_form(url, method, multipart, **attrs)
1911 1918 token = csrf_input()
1912 1919 return literal("%s\n%s" % (form, token))
1913 1920
1914 1921 def csrf_input():
1915 1922 return literal(
1916 1923 '<input type="hidden" id="{}" name="{}" value="{}">'.format(
1917 1924 csrf_token_key, csrf_token_key, get_csrf_token()))
1918 1925
1919 1926 def dropdownmenu(name, selected, options, enable_filter=False, **attrs):
1920 1927 select_html = select(name, selected, options, **attrs)
1921 1928 select2 = """
1922 1929 <script>
1923 1930 $(document).ready(function() {
1924 1931 $('#%s').select2({
1925 1932 containerCssClass: 'drop-menu',
1926 1933 dropdownCssClass: 'drop-menu-dropdown',
1927 1934 dropdownAutoWidth: true%s
1928 1935 });
1929 1936 });
1930 1937 </script>
1931 1938 """
1932 1939 filter_option = """,
1933 1940 minimumResultsForSearch: -1
1934 1941 """
1935 1942 input_id = attrs.get('id') or name
1936 1943 filter_enabled = "" if enable_filter else filter_option
1937 1944 select_script = literal(select2 % (input_id, filter_enabled))
1938 1945
1939 1946 return literal(select_html+select_script)
1940 1947
1941 1948
1942 1949 def get_visual_attr(tmpl_context_var, attr_name):
1943 1950 """
1944 1951 A safe way to get a variable from visual variable of template context
1945 1952
1946 1953 :param tmpl_context_var: instance of tmpl_context, usually present as `c`
1947 1954 :param attr_name: name of the attribute we fetch from the c.visual
1948 1955 """
1949 1956 visual = getattr(tmpl_context_var, 'visual', None)
1950 1957 if not visual:
1951 1958 return
1952 1959 else:
1953 1960 return getattr(visual, attr_name, None)
1954 1961
1955 1962
1956 1963 def get_last_path_part(file_node):
1957 1964 if not file_node.path:
1958 1965 return u''
1959 1966
1960 1967 path = safe_unicode(file_node.path.split('/')[-1])
1961 1968 return u'../' + path
1962 1969
1963 1970
1964 1971 def route_url(*args, **kwargs):
1965 1972 """
1966 1973 Wrapper around pyramids `route_url` (fully qualified url) function.
1967 1974 It is used to generate URLs from within pylons views or templates.
1968 1975 This will be removed when pyramid migration if finished.
1969 1976 """
1970 1977 req = get_current_request()
1971 1978 return req.route_url(*args, **kwargs)
1972 1979
1973 1980
1974 1981 def route_path(*args, **kwargs):
1975 1982 """
1976 1983 Wrapper around pyramids `route_path` function. It is used to generate
1977 1984 URLs from within pylons views or templates. This will be removed when
1978 1985 pyramid migration if finished.
1979 1986 """
1980 1987 req = get_current_request()
1981 1988 return req.route_path(*args, **kwargs)
1982 1989
1983 1990
1984 1991 def route_path_or_none(*args, **kwargs):
1985 1992 try:
1986 1993 return route_path(*args, **kwargs)
1987 1994 except KeyError:
1988 1995 return None
1989 1996
1990 1997
1991 1998 def static_url(*args, **kwds):
1992 1999 """
1993 2000 Wrapper around pyramids `route_path` function. It is used to generate
1994 2001 URLs from within pylons views or templates. This will be removed when
1995 2002 pyramid migration if finished.
1996 2003 """
1997 2004 req = get_current_request()
1998 2005 return req.static_url(*args, **kwds)
1999 2006
2000 2007
2001 2008 def resource_path(*args, **kwds):
2002 2009 """
2003 2010 Wrapper around pyramids `route_path` function. It is used to generate
2004 2011 URLs from within pylons views or templates. This will be removed when
2005 2012 pyramid migration if finished.
2006 2013 """
2007 2014 req = get_current_request()
2008 2015 return req.resource_path(*args, **kwds)
2009 2016
2010 2017
2011 2018 def api_call_example(method, args):
2012 2019 """
2013 2020 Generates an API call example via CURL
2014 2021 """
2015 2022 args_json = json.dumps(OrderedDict([
2016 2023 ('id', 1),
2017 2024 ('auth_token', 'SECRET'),
2018 2025 ('method', method),
2019 2026 ('args', args)
2020 2027 ]))
2021 2028 return literal(
2022 2029 "curl {api_url} -X POST -H 'content-type:text/plain' --data-binary '{data}'"
2023 2030 "<br/><br/>SECRET can be found in <a href=\"{token_url}\">auth-tokens</a> page, "
2024 2031 "and needs to be of `api calls` role."
2025 2032 .format(
2026 2033 api_url=route_url('apiv2'),
2027 2034 token_url=route_url('my_account_auth_tokens'),
2028 2035 data=args_json))
@@ -1,34 +1,34 b''
1 1 # -*- coding: utf-8 -*-
2 2
3 3 # Copyright (C) 2016-2017 RhodeCode GmbH
4 4 #
5 5 # This program is free software: you can redistribute it and/or modify
6 6 # it under the terms of the GNU Affero General Public License, version 3
7 7 # (only), as published by the Free Software Foundation.
8 8 #
9 9 # This program is distributed in the hope that it will be useful,
10 10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 12 # GNU General Public License for more details.
13 13 #
14 14 # You should have received a copy of the GNU Affero General Public License
15 15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 16 #
17 17 # This program is dual-licensed. If you wish to learn more about the
18 18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20 20
21 21 import colander
22 22 from rhodecode.model.validation_schema import validators, preparers, types
23 23
24 24
25 25 class ReviewerSchema(colander.MappingSchema):
26 26 username = colander.SchemaNode(types.StrOrIntType())
27 reasons = colander.SchemaNode(colander.List(), missing=[])
27 reasons = colander.SchemaNode(colander.List(), missing=['no reason specified'])
28 28 mandatory = colander.SchemaNode(colander.Boolean(), missing=False)
29 29
30 30
31 31 class ReviewerListSchema(colander.SequenceSchema):
32 32 reviewers = ReviewerSchema()
33 33
34 34
General Comments 0
You need to be logged in to leave comments. Login now