##// END OF EJS Templates
python3: fixed usage of .next() and .func_name
super-admin -
r4936:0d6cd344 default
parent child Browse files
Show More

The requested changes are too big and content was truncated. Show full diff

@@ -1,479 +1,479 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2016-2020 RhodeCode GmbH
3 # Copyright (C) 2016-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 import re
21 import re
22 import logging
22 import logging
23 import formencode
23 import formencode
24 import formencode.htmlfill
24 import formencode.htmlfill
25 import datetime
25 import datetime
26 from pyramid.interfaces import IRoutesMapper
26 from pyramid.interfaces import IRoutesMapper
27
27
28 from pyramid.httpexceptions import HTTPFound
28 from pyramid.httpexceptions import HTTPFound
29 from pyramid.renderers import render
29 from pyramid.renderers import render
30 from pyramid.response import Response
30 from pyramid.response import Response
31
31
32 from rhodecode.apps._base import BaseAppView, DataGridAppView
32 from rhodecode.apps._base import BaseAppView, DataGridAppView
33 from rhodecode.apps.ssh_support import SshKeyFileChangeEvent
33 from rhodecode.apps.ssh_support import SshKeyFileChangeEvent
34 from rhodecode import events
34 from rhodecode import events
35
35
36 from rhodecode.lib import helpers as h
36 from rhodecode.lib import helpers as h
37 from rhodecode.lib.auth import (
37 from rhodecode.lib.auth import (
38 LoginRequired, HasPermissionAllDecorator, CSRFRequired)
38 LoginRequired, HasPermissionAllDecorator, CSRFRequired)
39 from rhodecode.lib.utils2 import aslist, safe_unicode
39 from rhodecode.lib.utils2 import aslist, safe_unicode
40 from rhodecode.model.db import (
40 from rhodecode.model.db import (
41 or_, coalesce, User, UserIpMap, UserSshKeys)
41 or_, coalesce, User, UserIpMap, UserSshKeys)
42 from rhodecode.model.forms import (
42 from rhodecode.model.forms import (
43 ApplicationPermissionsForm, ObjectPermissionsForm, UserPermissionsForm)
43 ApplicationPermissionsForm, ObjectPermissionsForm, UserPermissionsForm)
44 from rhodecode.model.meta import Session
44 from rhodecode.model.meta import Session
45 from rhodecode.model.permission import PermissionModel
45 from rhodecode.model.permission import PermissionModel
46 from rhodecode.model.settings import SettingsModel
46 from rhodecode.model.settings import SettingsModel
47
47
48
48
49 log = logging.getLogger(__name__)
49 log = logging.getLogger(__name__)
50
50
51
51
52 class AdminPermissionsView(BaseAppView, DataGridAppView):
52 class AdminPermissionsView(BaseAppView, DataGridAppView):
53 def load_default_context(self):
53 def load_default_context(self):
54 c = self._get_local_tmpl_context()
54 c = self._get_local_tmpl_context()
55 PermissionModel().set_global_permission_choices(
55 PermissionModel().set_global_permission_choices(
56 c, gettext_translator=self.request.translate)
56 c, gettext_translator=self.request.translate)
57 return c
57 return c
58
58
59 @LoginRequired()
59 @LoginRequired()
60 @HasPermissionAllDecorator('hg.admin')
60 @HasPermissionAllDecorator('hg.admin')
61 def permissions_application(self):
61 def permissions_application(self):
62 c = self.load_default_context()
62 c = self.load_default_context()
63 c.active = 'application'
63 c.active = 'application'
64
64
65 c.user = User.get_default_user(refresh=True)
65 c.user = User.get_default_user(refresh=True)
66
66
67 app_settings = c.rc_config
67 app_settings = c.rc_config
68
68
69 defaults = {
69 defaults = {
70 'anonymous': c.user.active,
70 'anonymous': c.user.active,
71 'default_register_message': app_settings.get(
71 'default_register_message': app_settings.get(
72 'rhodecode_register_message')
72 'rhodecode_register_message')
73 }
73 }
74 defaults.update(c.user.get_default_perms())
74 defaults.update(c.user.get_default_perms())
75
75
76 data = render('rhodecode:templates/admin/permissions/permissions.mako',
76 data = render('rhodecode:templates/admin/permissions/permissions.mako',
77 self._get_template_context(c), self.request)
77 self._get_template_context(c), self.request)
78 html = formencode.htmlfill.render(
78 html = formencode.htmlfill.render(
79 data,
79 data,
80 defaults=defaults,
80 defaults=defaults,
81 encoding="UTF-8",
81 encoding="UTF-8",
82 force_defaults=False
82 force_defaults=False
83 )
83 )
84 return Response(html)
84 return Response(html)
85
85
86 @LoginRequired()
86 @LoginRequired()
87 @HasPermissionAllDecorator('hg.admin')
87 @HasPermissionAllDecorator('hg.admin')
88 @CSRFRequired()
88 @CSRFRequired()
89 def permissions_application_update(self):
89 def permissions_application_update(self):
90 _ = self.request.translate
90 _ = self.request.translate
91 c = self.load_default_context()
91 c = self.load_default_context()
92 c.active = 'application'
92 c.active = 'application'
93
93
94 _form = ApplicationPermissionsForm(
94 _form = ApplicationPermissionsForm(
95 self.request.translate,
95 self.request.translate,
96 [x[0] for x in c.register_choices],
96 [x[0] for x in c.register_choices],
97 [x[0] for x in c.password_reset_choices],
97 [x[0] for x in c.password_reset_choices],
98 [x[0] for x in c.extern_activate_choices])()
98 [x[0] for x in c.extern_activate_choices])()
99
99
100 try:
100 try:
101 form_result = _form.to_python(dict(self.request.POST))
101 form_result = _form.to_python(dict(self.request.POST))
102 form_result.update({'perm_user_name': User.DEFAULT_USER})
102 form_result.update({'perm_user_name': User.DEFAULT_USER})
103 PermissionModel().update_application_permissions(form_result)
103 PermissionModel().update_application_permissions(form_result)
104
104
105 settings = [
105 settings = [
106 ('register_message', 'default_register_message'),
106 ('register_message', 'default_register_message'),
107 ]
107 ]
108 for setting, form_key in settings:
108 for setting, form_key in settings:
109 sett = SettingsModel().create_or_update_setting(
109 sett = SettingsModel().create_or_update_setting(
110 setting, form_result[form_key])
110 setting, form_result[form_key])
111 Session().add(sett)
111 Session().add(sett)
112
112
113 Session().commit()
113 Session().commit()
114 h.flash(_('Application permissions updated successfully'),
114 h.flash(_('Application permissions updated successfully'),
115 category='success')
115 category='success')
116
116
117 except formencode.Invalid as errors:
117 except formencode.Invalid as errors:
118 defaults = errors.value
118 defaults = errors.value
119
119
120 data = render(
120 data = render(
121 'rhodecode:templates/admin/permissions/permissions.mako',
121 'rhodecode:templates/admin/permissions/permissions.mako',
122 self._get_template_context(c), self.request)
122 self._get_template_context(c), self.request)
123 html = formencode.htmlfill.render(
123 html = formencode.htmlfill.render(
124 data,
124 data,
125 defaults=defaults,
125 defaults=defaults,
126 errors=errors.error_dict or {},
126 errors=errors.error_dict or {},
127 prefix_error=False,
127 prefix_error=False,
128 encoding="UTF-8",
128 encoding="UTF-8",
129 force_defaults=False
129 force_defaults=False
130 )
130 )
131 return Response(html)
131 return Response(html)
132
132
133 except Exception:
133 except Exception:
134 log.exception("Exception during update of permissions")
134 log.exception("Exception during update of permissions")
135 h.flash(_('Error occurred during update of permissions'),
135 h.flash(_('Error occurred during update of permissions'),
136 category='error')
136 category='error')
137
137
138 affected_user_ids = [User.get_default_user_id()]
138 affected_user_ids = [User.get_default_user_id()]
139 PermissionModel().trigger_permission_flush(affected_user_ids)
139 PermissionModel().trigger_permission_flush(affected_user_ids)
140
140
141 raise HTTPFound(h.route_path('admin_permissions_application'))
141 raise HTTPFound(h.route_path('admin_permissions_application'))
142
142
143 @LoginRequired()
143 @LoginRequired()
144 @HasPermissionAllDecorator('hg.admin')
144 @HasPermissionAllDecorator('hg.admin')
145 def permissions_objects(self):
145 def permissions_objects(self):
146 c = self.load_default_context()
146 c = self.load_default_context()
147 c.active = 'objects'
147 c.active = 'objects'
148
148
149 c.user = User.get_default_user(refresh=True)
149 c.user = User.get_default_user(refresh=True)
150 defaults = {}
150 defaults = {}
151 defaults.update(c.user.get_default_perms())
151 defaults.update(c.user.get_default_perms())
152
152
153 data = render(
153 data = render(
154 'rhodecode:templates/admin/permissions/permissions.mako',
154 'rhodecode:templates/admin/permissions/permissions.mako',
155 self._get_template_context(c), self.request)
155 self._get_template_context(c), self.request)
156 html = formencode.htmlfill.render(
156 html = formencode.htmlfill.render(
157 data,
157 data,
158 defaults=defaults,
158 defaults=defaults,
159 encoding="UTF-8",
159 encoding="UTF-8",
160 force_defaults=False
160 force_defaults=False
161 )
161 )
162 return Response(html)
162 return Response(html)
163
163
164 @LoginRequired()
164 @LoginRequired()
165 @HasPermissionAllDecorator('hg.admin')
165 @HasPermissionAllDecorator('hg.admin')
166 @CSRFRequired()
166 @CSRFRequired()
167 def permissions_objects_update(self):
167 def permissions_objects_update(self):
168 _ = self.request.translate
168 _ = self.request.translate
169 c = self.load_default_context()
169 c = self.load_default_context()
170 c.active = 'objects'
170 c.active = 'objects'
171
171
172 _form = ObjectPermissionsForm(
172 _form = ObjectPermissionsForm(
173 self.request.translate,
173 self.request.translate,
174 [x[0] for x in c.repo_perms_choices],
174 [x[0] for x in c.repo_perms_choices],
175 [x[0] for x in c.group_perms_choices],
175 [x[0] for x in c.group_perms_choices],
176 [x[0] for x in c.user_group_perms_choices],
176 [x[0] for x in c.user_group_perms_choices],
177 )()
177 )()
178
178
179 try:
179 try:
180 form_result = _form.to_python(dict(self.request.POST))
180 form_result = _form.to_python(dict(self.request.POST))
181 form_result.update({'perm_user_name': User.DEFAULT_USER})
181 form_result.update({'perm_user_name': User.DEFAULT_USER})
182 PermissionModel().update_object_permissions(form_result)
182 PermissionModel().update_object_permissions(form_result)
183
183
184 Session().commit()
184 Session().commit()
185 h.flash(_('Object permissions updated successfully'),
185 h.flash(_('Object permissions updated successfully'),
186 category='success')
186 category='success')
187
187
188 except formencode.Invalid as errors:
188 except formencode.Invalid as errors:
189 defaults = errors.value
189 defaults = errors.value
190
190
191 data = render(
191 data = render(
192 'rhodecode:templates/admin/permissions/permissions.mako',
192 'rhodecode:templates/admin/permissions/permissions.mako',
193 self._get_template_context(c), self.request)
193 self._get_template_context(c), self.request)
194 html = formencode.htmlfill.render(
194 html = formencode.htmlfill.render(
195 data,
195 data,
196 defaults=defaults,
196 defaults=defaults,
197 errors=errors.error_dict or {},
197 errors=errors.error_dict or {},
198 prefix_error=False,
198 prefix_error=False,
199 encoding="UTF-8",
199 encoding="UTF-8",
200 force_defaults=False
200 force_defaults=False
201 )
201 )
202 return Response(html)
202 return Response(html)
203 except Exception:
203 except Exception:
204 log.exception("Exception during update of permissions")
204 log.exception("Exception during update of permissions")
205 h.flash(_('Error occurred during update of permissions'),
205 h.flash(_('Error occurred during update of permissions'),
206 category='error')
206 category='error')
207
207
208 affected_user_ids = [User.get_default_user_id()]
208 affected_user_ids = [User.get_default_user_id()]
209 PermissionModel().trigger_permission_flush(affected_user_ids)
209 PermissionModel().trigger_permission_flush(affected_user_ids)
210
210
211 raise HTTPFound(h.route_path('admin_permissions_object'))
211 raise HTTPFound(h.route_path('admin_permissions_object'))
212
212
213 @LoginRequired()
213 @LoginRequired()
214 @HasPermissionAllDecorator('hg.admin')
214 @HasPermissionAllDecorator('hg.admin')
215 def permissions_branch(self):
215 def permissions_branch(self):
216 c = self.load_default_context()
216 c = self.load_default_context()
217 c.active = 'branch'
217 c.active = 'branch'
218
218
219 c.user = User.get_default_user(refresh=True)
219 c.user = User.get_default_user(refresh=True)
220 defaults = {}
220 defaults = {}
221 defaults.update(c.user.get_default_perms())
221 defaults.update(c.user.get_default_perms())
222
222
223 data = render(
223 data = render(
224 'rhodecode:templates/admin/permissions/permissions.mako',
224 'rhodecode:templates/admin/permissions/permissions.mako',
225 self._get_template_context(c), self.request)
225 self._get_template_context(c), self.request)
226 html = formencode.htmlfill.render(
226 html = formencode.htmlfill.render(
227 data,
227 data,
228 defaults=defaults,
228 defaults=defaults,
229 encoding="UTF-8",
229 encoding="UTF-8",
230 force_defaults=False
230 force_defaults=False
231 )
231 )
232 return Response(html)
232 return Response(html)
233
233
234 @LoginRequired()
234 @LoginRequired()
235 @HasPermissionAllDecorator('hg.admin')
235 @HasPermissionAllDecorator('hg.admin')
236 def permissions_global(self):
236 def permissions_global(self):
237 c = self.load_default_context()
237 c = self.load_default_context()
238 c.active = 'global'
238 c.active = 'global'
239
239
240 c.user = User.get_default_user(refresh=True)
240 c.user = User.get_default_user(refresh=True)
241 defaults = {}
241 defaults = {}
242 defaults.update(c.user.get_default_perms())
242 defaults.update(c.user.get_default_perms())
243
243
244 data = render(
244 data = render(
245 'rhodecode:templates/admin/permissions/permissions.mako',
245 'rhodecode:templates/admin/permissions/permissions.mako',
246 self._get_template_context(c), self.request)
246 self._get_template_context(c), self.request)
247 html = formencode.htmlfill.render(
247 html = formencode.htmlfill.render(
248 data,
248 data,
249 defaults=defaults,
249 defaults=defaults,
250 encoding="UTF-8",
250 encoding="UTF-8",
251 force_defaults=False
251 force_defaults=False
252 )
252 )
253 return Response(html)
253 return Response(html)
254
254
255 @LoginRequired()
255 @LoginRequired()
256 @HasPermissionAllDecorator('hg.admin')
256 @HasPermissionAllDecorator('hg.admin')
257 @CSRFRequired()
257 @CSRFRequired()
258 def permissions_global_update(self):
258 def permissions_global_update(self):
259 _ = self.request.translate
259 _ = self.request.translate
260 c = self.load_default_context()
260 c = self.load_default_context()
261 c.active = 'global'
261 c.active = 'global'
262
262
263 _form = UserPermissionsForm(
263 _form = UserPermissionsForm(
264 self.request.translate,
264 self.request.translate,
265 [x[0] for x in c.repo_create_choices],
265 [x[0] for x in c.repo_create_choices],
266 [x[0] for x in c.repo_create_on_write_choices],
266 [x[0] for x in c.repo_create_on_write_choices],
267 [x[0] for x in c.repo_group_create_choices],
267 [x[0] for x in c.repo_group_create_choices],
268 [x[0] for x in c.user_group_create_choices],
268 [x[0] for x in c.user_group_create_choices],
269 [x[0] for x in c.fork_choices],
269 [x[0] for x in c.fork_choices],
270 [x[0] for x in c.inherit_default_permission_choices])()
270 [x[0] for x in c.inherit_default_permission_choices])()
271
271
272 try:
272 try:
273 form_result = _form.to_python(dict(self.request.POST))
273 form_result = _form.to_python(dict(self.request.POST))
274 form_result.update({'perm_user_name': User.DEFAULT_USER})
274 form_result.update({'perm_user_name': User.DEFAULT_USER})
275 PermissionModel().update_user_permissions(form_result)
275 PermissionModel().update_user_permissions(form_result)
276
276
277 Session().commit()
277 Session().commit()
278 h.flash(_('Global permissions updated successfully'),
278 h.flash(_('Global permissions updated successfully'),
279 category='success')
279 category='success')
280
280
281 except formencode.Invalid as errors:
281 except formencode.Invalid as errors:
282 defaults = errors.value
282 defaults = errors.value
283
283
284 data = render(
284 data = render(
285 'rhodecode:templates/admin/permissions/permissions.mako',
285 'rhodecode:templates/admin/permissions/permissions.mako',
286 self._get_template_context(c), self.request)
286 self._get_template_context(c), self.request)
287 html = formencode.htmlfill.render(
287 html = formencode.htmlfill.render(
288 data,
288 data,
289 defaults=defaults,
289 defaults=defaults,
290 errors=errors.error_dict or {},
290 errors=errors.error_dict or {},
291 prefix_error=False,
291 prefix_error=False,
292 encoding="UTF-8",
292 encoding="UTF-8",
293 force_defaults=False
293 force_defaults=False
294 )
294 )
295 return Response(html)
295 return Response(html)
296 except Exception:
296 except Exception:
297 log.exception("Exception during update of permissions")
297 log.exception("Exception during update of permissions")
298 h.flash(_('Error occurred during update of permissions'),
298 h.flash(_('Error occurred during update of permissions'),
299 category='error')
299 category='error')
300
300
301 affected_user_ids = [User.get_default_user_id()]
301 affected_user_ids = [User.get_default_user_id()]
302 PermissionModel().trigger_permission_flush(affected_user_ids)
302 PermissionModel().trigger_permission_flush(affected_user_ids)
303
303
304 raise HTTPFound(h.route_path('admin_permissions_global'))
304 raise HTTPFound(h.route_path('admin_permissions_global'))
305
305
306 @LoginRequired()
306 @LoginRequired()
307 @HasPermissionAllDecorator('hg.admin')
307 @HasPermissionAllDecorator('hg.admin')
308 def permissions_ips(self):
308 def permissions_ips(self):
309 c = self.load_default_context()
309 c = self.load_default_context()
310 c.active = 'ips'
310 c.active = 'ips'
311
311
312 c.user = User.get_default_user(refresh=True)
312 c.user = User.get_default_user(refresh=True)
313 c.user_ip_map = (
313 c.user_ip_map = (
314 UserIpMap.query().filter(UserIpMap.user == c.user).all())
314 UserIpMap.query().filter(UserIpMap.user == c.user).all())
315
315
316 return self._get_template_context(c)
316 return self._get_template_context(c)
317
317
318 @LoginRequired()
318 @LoginRequired()
319 @HasPermissionAllDecorator('hg.admin')
319 @HasPermissionAllDecorator('hg.admin')
320 def permissions_overview(self):
320 def permissions_overview(self):
321 c = self.load_default_context()
321 c = self.load_default_context()
322 c.active = 'perms'
322 c.active = 'perms'
323
323
324 c.user = User.get_default_user(refresh=True)
324 c.user = User.get_default_user(refresh=True)
325 c.perm_user = c.user.AuthUser()
325 c.perm_user = c.user.AuthUser()
326 return self._get_template_context(c)
326 return self._get_template_context(c)
327
327
328 @LoginRequired()
328 @LoginRequired()
329 @HasPermissionAllDecorator('hg.admin')
329 @HasPermissionAllDecorator('hg.admin')
330 def auth_token_access(self):
330 def auth_token_access(self):
331 from rhodecode import CONFIG
331 from rhodecode import CONFIG
332
332
333 c = self.load_default_context()
333 c = self.load_default_context()
334 c.active = 'auth_token_access'
334 c.active = 'auth_token_access'
335
335
336 c.user = User.get_default_user(refresh=True)
336 c.user = User.get_default_user(refresh=True)
337 c.perm_user = c.user.AuthUser()
337 c.perm_user = c.user.AuthUser()
338
338
339 mapper = self.request.registry.queryUtility(IRoutesMapper)
339 mapper = self.request.registry.queryUtility(IRoutesMapper)
340 c.view_data = []
340 c.view_data = []
341
341
342 _argument_prog = re.compile('\{(.*?)\}|:\((.*)\)')
342 _argument_prog = re.compile('\{(.*?)\}|:\((.*)\)')
343 introspector = self.request.registry.introspector
343 introspector = self.request.registry.introspector
344
344
345 view_intr = {}
345 view_intr = {}
346 for view_data in introspector.get_category('views'):
346 for view_data in introspector.get_category('views'):
347 intr = view_data['introspectable']
347 intr = view_data['introspectable']
348
348
349 if 'route_name' in intr and intr['attr']:
349 if 'route_name' in intr and intr['attr']:
350 view_intr[intr['route_name']] = '{}:{}'.format(
350 view_intr[intr['route_name']] = '{}:{}'.format(
351 str(intr['derived_callable'].func_name), intr['attr']
351 str(intr['derived_callable'].__name__), intr['attr']
352 )
352 )
353
353
354 c.whitelist_key = 'api_access_controllers_whitelist'
354 c.whitelist_key = 'api_access_controllers_whitelist'
355 c.whitelist_file = CONFIG.get('__file__')
355 c.whitelist_file = CONFIG.get('__file__')
356 whitelist_views = aslist(
356 whitelist_views = aslist(
357 CONFIG.get(c.whitelist_key), sep=',')
357 CONFIG.get(c.whitelist_key), sep=',')
358
358
359 for route_info in mapper.get_routes():
359 for route_info in mapper.get_routes():
360 if not route_info.name.startswith('__'):
360 if not route_info.name.startswith('__'):
361 routepath = route_info.pattern
361 routepath = route_info.pattern
362
362
363 def replace(matchobj):
363 def replace(matchobj):
364 if matchobj.group(1):
364 if matchobj.group(1):
365 return "{%s}" % matchobj.group(1).split(':')[0]
365 return "{%s}" % matchobj.group(1).split(':')[0]
366 else:
366 else:
367 return "{%s}" % matchobj.group(2)
367 return "{%s}" % matchobj.group(2)
368
368
369 routepath = _argument_prog.sub(replace, routepath)
369 routepath = _argument_prog.sub(replace, routepath)
370
370
371 if not routepath.startswith('/'):
371 if not routepath.startswith('/'):
372 routepath = '/' + routepath
372 routepath = '/' + routepath
373
373
374 view_fqn = view_intr.get(route_info.name, 'NOT AVAILABLE')
374 view_fqn = view_intr.get(route_info.name, 'NOT AVAILABLE')
375 active = view_fqn in whitelist_views
375 active = view_fqn in whitelist_views
376 c.view_data.append((route_info.name, view_fqn, routepath, active))
376 c.view_data.append((route_info.name, view_fqn, routepath, active))
377
377
378 c.whitelist_views = whitelist_views
378 c.whitelist_views = whitelist_views
379 return self._get_template_context(c)
379 return self._get_template_context(c)
380
380
381 def ssh_enabled(self):
381 def ssh_enabled(self):
382 return self.request.registry.settings.get(
382 return self.request.registry.settings.get(
383 'ssh.generate_authorized_keyfile')
383 'ssh.generate_authorized_keyfile')
384
384
385 @LoginRequired()
385 @LoginRequired()
386 @HasPermissionAllDecorator('hg.admin')
386 @HasPermissionAllDecorator('hg.admin')
387 def ssh_keys(self):
387 def ssh_keys(self):
388 c = self.load_default_context()
388 c = self.load_default_context()
389 c.active = 'ssh_keys'
389 c.active = 'ssh_keys'
390 c.ssh_enabled = self.ssh_enabled()
390 c.ssh_enabled = self.ssh_enabled()
391 return self._get_template_context(c)
391 return self._get_template_context(c)
392
392
393 @LoginRequired()
393 @LoginRequired()
394 @HasPermissionAllDecorator('hg.admin')
394 @HasPermissionAllDecorator('hg.admin')
395 def ssh_keys_data(self):
395 def ssh_keys_data(self):
396 _ = self.request.translate
396 _ = self.request.translate
397 self.load_default_context()
397 self.load_default_context()
398 column_map = {
398 column_map = {
399 'fingerprint': 'ssh_key_fingerprint',
399 'fingerprint': 'ssh_key_fingerprint',
400 'username': User.username
400 'username': User.username
401 }
401 }
402 draw, start, limit = self._extract_chunk(self.request)
402 draw, start, limit = self._extract_chunk(self.request)
403 search_q, order_by, order_dir = self._extract_ordering(
403 search_q, order_by, order_dir = self._extract_ordering(
404 self.request, column_map=column_map)
404 self.request, column_map=column_map)
405
405
406 ssh_keys_data_total_count = UserSshKeys.query()\
406 ssh_keys_data_total_count = UserSshKeys.query()\
407 .count()
407 .count()
408
408
409 # json generate
409 # json generate
410 base_q = UserSshKeys.query().join(UserSshKeys.user)
410 base_q = UserSshKeys.query().join(UserSshKeys.user)
411
411
412 if search_q:
412 if search_q:
413 like_expression = u'%{}%'.format(safe_unicode(search_q))
413 like_expression = u'%{}%'.format(safe_unicode(search_q))
414 base_q = base_q.filter(or_(
414 base_q = base_q.filter(or_(
415 User.username.ilike(like_expression),
415 User.username.ilike(like_expression),
416 UserSshKeys.ssh_key_fingerprint.ilike(like_expression),
416 UserSshKeys.ssh_key_fingerprint.ilike(like_expression),
417 ))
417 ))
418
418
419 users_data_total_filtered_count = base_q.count()
419 users_data_total_filtered_count = base_q.count()
420
420
421 sort_col = self._get_order_col(order_by, UserSshKeys)
421 sort_col = self._get_order_col(order_by, UserSshKeys)
422 if sort_col:
422 if sort_col:
423 if order_dir == 'asc':
423 if order_dir == 'asc':
424 # handle null values properly to order by NULL last
424 # handle null values properly to order by NULL last
425 if order_by in ['created_on']:
425 if order_by in ['created_on']:
426 sort_col = coalesce(sort_col, datetime.date.max)
426 sort_col = coalesce(sort_col, datetime.date.max)
427 sort_col = sort_col.asc()
427 sort_col = sort_col.asc()
428 else:
428 else:
429 # handle null values properly to order by NULL last
429 # handle null values properly to order by NULL last
430 if order_by in ['created_on']:
430 if order_by in ['created_on']:
431 sort_col = coalesce(sort_col, datetime.date.min)
431 sort_col = coalesce(sort_col, datetime.date.min)
432 sort_col = sort_col.desc()
432 sort_col = sort_col.desc()
433
433
434 base_q = base_q.order_by(sort_col)
434 base_q = base_q.order_by(sort_col)
435 base_q = base_q.offset(start).limit(limit)
435 base_q = base_q.offset(start).limit(limit)
436
436
437 ssh_keys = base_q.all()
437 ssh_keys = base_q.all()
438
438
439 ssh_keys_data = []
439 ssh_keys_data = []
440 for ssh_key in ssh_keys:
440 for ssh_key in ssh_keys:
441 ssh_keys_data.append({
441 ssh_keys_data.append({
442 "username": h.gravatar_with_user(self.request, ssh_key.user.username),
442 "username": h.gravatar_with_user(self.request, ssh_key.user.username),
443 "fingerprint": ssh_key.ssh_key_fingerprint,
443 "fingerprint": ssh_key.ssh_key_fingerprint,
444 "description": ssh_key.description,
444 "description": ssh_key.description,
445 "created_on": h.format_date(ssh_key.created_on),
445 "created_on": h.format_date(ssh_key.created_on),
446 "accessed_on": h.format_date(ssh_key.accessed_on),
446 "accessed_on": h.format_date(ssh_key.accessed_on),
447 "action": h.link_to(
447 "action": h.link_to(
448 _('Edit'), h.route_path('edit_user_ssh_keys',
448 _('Edit'), h.route_path('edit_user_ssh_keys',
449 user_id=ssh_key.user.user_id))
449 user_id=ssh_key.user.user_id))
450 })
450 })
451
451
452 data = ({
452 data = ({
453 'draw': draw,
453 'draw': draw,
454 'data': ssh_keys_data,
454 'data': ssh_keys_data,
455 'recordsTotal': ssh_keys_data_total_count,
455 'recordsTotal': ssh_keys_data_total_count,
456 'recordsFiltered': users_data_total_filtered_count,
456 'recordsFiltered': users_data_total_filtered_count,
457 })
457 })
458
458
459 return data
459 return data
460
460
461 @LoginRequired()
461 @LoginRequired()
462 @HasPermissionAllDecorator('hg.admin')
462 @HasPermissionAllDecorator('hg.admin')
463 @CSRFRequired()
463 @CSRFRequired()
464 def ssh_keys_update(self):
464 def ssh_keys_update(self):
465 _ = self.request.translate
465 _ = self.request.translate
466 self.load_default_context()
466 self.load_default_context()
467
467
468 ssh_enabled = self.ssh_enabled()
468 ssh_enabled = self.ssh_enabled()
469 key_file = self.request.registry.settings.get(
469 key_file = self.request.registry.settings.get(
470 'ssh.authorized_keys_file_path')
470 'ssh.authorized_keys_file_path')
471 if ssh_enabled:
471 if ssh_enabled:
472 events.trigger(SshKeyFileChangeEvent(), self.request.registry)
472 events.trigger(SshKeyFileChangeEvent(), self.request.registry)
473 h.flash(_('Updated SSH keys file: {}').format(key_file),
473 h.flash(_('Updated SSH keys file: {}').format(key_file),
474 category='success')
474 category='success')
475 else:
475 else:
476 h.flash(_('SSH key support is disabled in .ini file'),
476 h.flash(_('SSH key support is disabled in .ini file'),
477 category='warning')
477 category='warning')
478
478
479 raise HTTPFound(h.route_path('admin_permissions_ssh_keys'))
479 raise HTTPFound(h.route_path('admin_permissions_ssh_keys'))
@@ -1,121 +1,121 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2012-2020 RhodeCode GmbH
3 # Copyright (C) 2012-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 import time
21 import time
22 import logging
22 import logging
23
23
24 from pyramid.exceptions import ConfigurationError
24 from pyramid.exceptions import ConfigurationError
25 from zope.interface import implementer
25 from zope.interface import implementer
26
26
27 from rhodecode.authentication.interface import IAuthnPluginRegistry
27 from rhodecode.authentication.interface import IAuthnPluginRegistry
28 from rhodecode.model.settings import SettingsModel
28 from rhodecode.model.settings import SettingsModel
29 from rhodecode.lib.utils2 import safe_str
29 from rhodecode.lib.utils2 import safe_str
30 from rhodecode.lib.statsd_client import StatsdClient
30 from rhodecode.lib.statsd_client import StatsdClient
31 from rhodecode.lib import rc_cache
31 from rhodecode.lib import rc_cache
32
32
33 log = logging.getLogger(__name__)
33 log = logging.getLogger(__name__)
34
34
35
35
36 @implementer(IAuthnPluginRegistry)
36 @implementer(IAuthnPluginRegistry)
37 class AuthenticationPluginRegistry(object):
37 class AuthenticationPluginRegistry(object):
38
38
39 # INI settings key to set a fallback authentication plugin.
39 # INI settings key to set a fallback authentication plugin.
40 fallback_plugin_key = 'rhodecode.auth_plugin_fallback'
40 fallback_plugin_key = 'rhodecode.auth_plugin_fallback'
41
41
42 def __init__(self, settings):
42 def __init__(self, settings):
43 self._plugins = {}
43 self._plugins = {}
44 self._fallback_plugin = settings.get(self.fallback_plugin_key, None)
44 self._fallback_plugin = settings.get(self.fallback_plugin_key, None)
45
45
46 def add_authn_plugin(self, config, plugin):
46 def add_authn_plugin(self, config, plugin):
47 plugin_id = plugin.get_id()
47 plugin_id = plugin.get_id()
48 if plugin_id in self._plugins.keys():
48 if plugin_id in self._plugins.keys():
49 raise ConfigurationError(
49 raise ConfigurationError(
50 'Cannot register authentication plugin twice: "%s"', plugin_id)
50 'Cannot register authentication plugin twice: "%s"', plugin_id)
51 else:
51 else:
52 log.debug('Register authentication plugin: "%s"', plugin_id)
52 log.debug('Register authentication plugin: "%s"', plugin_id)
53 self._plugins[plugin_id] = plugin
53 self._plugins[plugin_id] = plugin
54
54
55 def get_plugins(self):
55 def get_plugins(self):
56 def sort_key(plugin):
56 def sort_key(plugin):
57 return str.lower(safe_str(plugin.get_display_name()))
57 return str.lower(safe_str(plugin.get_display_name()))
58
58
59 return sorted(self._plugins.values(), key=sort_key)
59 return sorted(self._plugins.values(), key=sort_key)
60
60
61 def get_plugin(self, plugin_id):
61 def get_plugin(self, plugin_id):
62 return self._plugins.get(plugin_id, None)
62 return self._plugins.get(plugin_id, None)
63
63
64 def get_plugin_by_uid(self, plugin_uid):
64 def get_plugin_by_uid(self, plugin_uid):
65 for plugin in self._plugins.values():
65 for plugin in self._plugins.values():
66 if plugin.uid == plugin_uid:
66 if plugin.uid == plugin_uid:
67 return plugin
67 return plugin
68
68
69 def get_plugins_for_authentication(self, cache=True):
69 def get_plugins_for_authentication(self, cache=True):
70 """
70 """
71 Returns a list of plugins which should be consulted when authenticating
71 Returns a list of plugins which should be consulted when authenticating
72 a user. It only returns plugins which are enabled and active.
72 a user. It only returns plugins which are enabled and active.
73 Additionally it includes the fallback plugin from the INI file, if
73 Additionally it includes the fallback plugin from the INI file, if
74 `rhodecode.auth_plugin_fallback` is set to a plugin ID.
74 `rhodecode.auth_plugin_fallback` is set to a plugin ID.
75 """
75 """
76
76
77 cache_namespace_uid = 'cache_auth_plugins'
77 cache_namespace_uid = 'cache_auth_plugins'
78 region = rc_cache.get_or_create_region('cache_general', cache_namespace_uid)
78 region = rc_cache.get_or_create_region('cache_general', cache_namespace_uid)
79
79
80 @region.conditional_cache_on_arguments(condition=cache)
80 @region.conditional_cache_on_arguments(condition=cache)
81 def _get_auth_plugins(name, key, fallback_plugin):
81 def _get_auth_plugins(name, key, fallback_plugin):
82 plugins = []
82 plugins = []
83
83
84 # Add all enabled and active plugins to the list. We iterate over the
84 # Add all enabled and active plugins to the list. We iterate over the
85 # auth_plugins setting from DB because it also represents the ordering.
85 # auth_plugins setting from DB because it also represents the ordering.
86 enabled_plugins = SettingsModel().get_auth_plugins()
86 enabled_plugins = SettingsModel().get_auth_plugins()
87 raw_settings = SettingsModel().get_all_settings(cache=False)
87 raw_settings = SettingsModel().get_all_settings(cache=False)
88 for plugin_id in enabled_plugins:
88 for plugin_id in enabled_plugins:
89 plugin = self.get_plugin(plugin_id)
89 plugin = self.get_plugin(plugin_id)
90 if plugin is not None and plugin.is_active(
90 if plugin is not None and plugin.is_active(
91 plugin_cached_settings=raw_settings):
91 plugin_cached_settings=raw_settings):
92
92
93 # inject settings into plugin, we can re-use the DB fetched settings here
93 # inject settings into plugin, we can re-use the DB fetched settings here
94 plugin._settings = plugin._propagate_settings(raw_settings)
94 plugin._settings = plugin._propagate_settings(raw_settings)
95 plugins.append(plugin)
95 plugins.append(plugin)
96
96
97 # Add the fallback plugin from ini file.
97 # Add the fallback plugin from ini file.
98 if fallback_plugin:
98 if fallback_plugin:
99 log.warn(
99 log.warn(
100 'Using fallback authentication plugin from INI file: "%s"',
100 'Using fallback authentication plugin from INI file: "%s"',
101 fallback_plugin)
101 fallback_plugin)
102 plugin = self.get_plugin(fallback_plugin)
102 plugin = self.get_plugin(fallback_plugin)
103 if plugin is not None and plugin not in plugins:
103 if plugin is not None and plugin not in plugins:
104 plugin._settings = plugin._propagate_settings(raw_settings)
104 plugin._settings = plugin._propagate_settings(raw_settings)
105 plugins.append(plugin)
105 plugins.append(plugin)
106 return plugins
106 return plugins
107
107
108 start = time.time()
108 start = time.time()
109 plugins = _get_auth_plugins('rhodecode_auth_plugins', 'v1', self._fallback_plugin)
109 plugins = _get_auth_plugins('rhodecode_auth_plugins', 'v1', self._fallback_plugin)
110
110
111 compute_time = time.time() - start
111 compute_time = time.time() - start
112 log.debug('cached method:%s took %.4fs', _get_auth_plugins.func_name, compute_time)
112 log.debug('cached method:%s took %.4fs', _get_auth_plugins.__name__, compute_time)
113
113
114 statsd = StatsdClient.statsd
114 statsd = StatsdClient.statsd
115 if statsd:
115 if statsd:
116 elapsed_time_ms = round(1000.0 * compute_time) # use ms only
116 elapsed_time_ms = round(1000.0 * compute_time) # use ms only
117 statsd.timing("rhodecode_auth_plugins_timing.histogram", elapsed_time_ms,
117 statsd.timing("rhodecode_auth_plugins_timing.histogram", elapsed_time_ms,
118 use_decimals=False)
118 use_decimals=False)
119
119
120 return plugins
120 return plugins
121
121
@@ -1,199 +1,199 b''
1 # Copyright (C) 2016-2020 RhodeCode GmbH
1 # Copyright (C) 2016-2020 RhodeCode GmbH
2 #
2 #
3 # This program is free software: you can redistribute it and/or modify
3 # This program is free software: you can redistribute it and/or modify
4 # it under the terms of the GNU Affero General Public License, version 3
4 # it under the terms of the GNU Affero General Public License, version 3
5 # (only), as published by the Free Software Foundation.
5 # (only), as published by the Free Software Foundation.
6 #
6 #
7 # This program is distributed in the hope that it will be useful,
7 # This program is distributed in the hope that it will be useful,
8 # but WITHOUT ANY WARRANTY; without even the implied warranty of
8 # but WITHOUT ANY WARRANTY; without even the implied warranty of
9 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
9 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
10 # GNU General Public License for more details.
10 # GNU General Public License for more details.
11 #
11 #
12 # You should have received a copy of the GNU Affero General Public License
12 # You should have received a copy of the GNU Affero General Public License
13 # along with this program. If not, see <http://www.gnu.org/licenses/>.
13 # along with this program. If not, see <http://www.gnu.org/licenses/>.
14 #
14 #
15 # This program is dual-licensed. If you wish to learn more about the
15 # This program is dual-licensed. If you wish to learn more about the
16 # RhodeCode Enterprise Edition, including its added features, Support services,
16 # RhodeCode Enterprise Edition, including its added features, Support services,
17 # and proprietary license terms, please see https://rhodecode.com/licenses/
17 # and proprietary license terms, please see https://rhodecode.com/licenses/
18
18
19 import logging
19 import logging
20 import os
20 import os
21 import string
21 import string
22 import functools
22 import functools
23 import collections
23 import collections
24 import urllib.request, urllib.parse, urllib.error
24 import urllib.request, urllib.parse, urllib.error
25
25
26 log = logging.getLogger('rhodecode.' + __name__)
26 log = logging.getLogger('rhodecode.' + __name__)
27
27
28
28
29 class HookResponse(object):
29 class HookResponse(object):
30 def __init__(self, status, output):
30 def __init__(self, status, output):
31 self.status = status
31 self.status = status
32 self.output = output
32 self.output = output
33
33
34 def __add__(self, other):
34 def __add__(self, other):
35 other_status = getattr(other, 'status', 0)
35 other_status = getattr(other, 'status', 0)
36 new_status = max(self.status, other_status)
36 new_status = max(self.status, other_status)
37 other_output = getattr(other, 'output', '')
37 other_output = getattr(other, 'output', '')
38 new_output = self.output + other_output
38 new_output = self.output + other_output
39
39
40 return HookResponse(new_status, new_output)
40 return HookResponse(new_status, new_output)
41
41
42 def __bool__(self):
42 def __bool__(self):
43 return self.status == 0
43 return self.status == 0
44
44
45
45
46 class DotDict(dict):
46 class DotDict(dict):
47
47
48 def __contains__(self, k):
48 def __contains__(self, k):
49 try:
49 try:
50 return dict.__contains__(self, k) or hasattr(self, k)
50 return dict.__contains__(self, k) or hasattr(self, k)
51 except:
51 except:
52 return False
52 return False
53
53
54 # only called if k not found in normal places
54 # only called if k not found in normal places
55 def __getattr__(self, k):
55 def __getattr__(self, k):
56 try:
56 try:
57 return object.__getattribute__(self, k)
57 return object.__getattribute__(self, k)
58 except AttributeError:
58 except AttributeError:
59 try:
59 try:
60 return self[k]
60 return self[k]
61 except KeyError:
61 except KeyError:
62 raise AttributeError(k)
62 raise AttributeError(k)
63
63
64 def __setattr__(self, k, v):
64 def __setattr__(self, k, v):
65 try:
65 try:
66 object.__getattribute__(self, k)
66 object.__getattribute__(self, k)
67 except AttributeError:
67 except AttributeError:
68 try:
68 try:
69 self[k] = v
69 self[k] = v
70 except:
70 except:
71 raise AttributeError(k)
71 raise AttributeError(k)
72 else:
72 else:
73 object.__setattr__(self, k, v)
73 object.__setattr__(self, k, v)
74
74
75 def __delattr__(self, k):
75 def __delattr__(self, k):
76 try:
76 try:
77 object.__getattribute__(self, k)
77 object.__getattribute__(self, k)
78 except AttributeError:
78 except AttributeError:
79 try:
79 try:
80 del self[k]
80 del self[k]
81 except KeyError:
81 except KeyError:
82 raise AttributeError(k)
82 raise AttributeError(k)
83 else:
83 else:
84 object.__delattr__(self, k)
84 object.__delattr__(self, k)
85
85
86 def toDict(self):
86 def toDict(self):
87 return unserialize(self)
87 return unserialize(self)
88
88
89 def __repr__(self):
89 def __repr__(self):
90 keys = list(self.keys())
90 keys = list(self.keys())
91 keys.sort()
91 keys.sort()
92 args = ', '.join(['%s=%r' % (key, self[key]) for key in keys])
92 args = ', '.join(['%s=%r' % (key, self[key]) for key in keys])
93 return '%s(%s)' % (self.__class__.__name__, args)
93 return '%s(%s)' % (self.__class__.__name__, args)
94
94
95 @staticmethod
95 @staticmethod
96 def fromDict(d):
96 def fromDict(d):
97 return serialize(d)
97 return serialize(d)
98
98
99
99
100 def serialize(x):
100 def serialize(x):
101 if isinstance(x, dict):
101 if isinstance(x, dict):
102 return DotDict((k, serialize(v)) for k, v in x.items())
102 return DotDict((k, serialize(v)) for k, v in x.items())
103 elif isinstance(x, (list, tuple)):
103 elif isinstance(x, (list, tuple)):
104 return type(x)(serialize(v) for v in x)
104 return type(x)(serialize(v) for v in x)
105 else:
105 else:
106 return x
106 return x
107
107
108
108
109 def unserialize(x):
109 def unserialize(x):
110 if isinstance(x, dict):
110 if isinstance(x, dict):
111 return dict((k, unserialize(v)) for k, v in x.items())
111 return dict((k, unserialize(v)) for k, v in x.items())
112 elif isinstance(x, (list, tuple)):
112 elif isinstance(x, (list, tuple)):
113 return type(x)(unserialize(v) for v in x)
113 return type(x)(unserialize(v) for v in x)
114 else:
114 else:
115 return x
115 return x
116
116
117
117
118 def _verify_kwargs(func_name, expected_parameters, kwargs):
118 def _verify_kwargs(func_name, expected_parameters, kwargs):
119 """
119 """
120 Verify that exactly `expected_parameters` are passed in as `kwargs`.
120 Verify that exactly `expected_parameters` are passed in as `kwargs`.
121 """
121 """
122 expected_parameters = set(expected_parameters)
122 expected_parameters = set(expected_parameters)
123 kwargs_keys = set(kwargs.keys())
123 kwargs_keys = set(kwargs.keys())
124 if kwargs_keys != expected_parameters:
124 if kwargs_keys != expected_parameters:
125 missing_kwargs = expected_parameters - kwargs_keys
125 missing_kwargs = expected_parameters - kwargs_keys
126 unexpected_kwargs = kwargs_keys - expected_parameters
126 unexpected_kwargs = kwargs_keys - expected_parameters
127 raise AssertionError(
127 raise AssertionError(
128 "func:%s: missing parameters: %r, unexpected parameters: %s" %
128 "func:%s: missing parameters: %r, unexpected parameters: %s" %
129 (func_name, missing_kwargs, unexpected_kwargs))
129 (func_name, missing_kwargs, unexpected_kwargs))
130
130
131
131
132 def has_kwargs(required_args):
132 def has_kwargs(required_args):
133 """
133 """
134 decorator to verify extension calls arguments.
134 decorator to verify extension calls arguments.
135
135
136 :param required_args:
136 :param required_args:
137 """
137 """
138 def wrap(func):
138 def wrap(func):
139 def wrapper(*args, **kwargs):
139 def wrapper(*args, **kwargs):
140 _verify_kwargs(func.func_name, required_args.keys(), kwargs)
140 _verify_kwargs(func.__name__, required_args.keys(), kwargs)
141 # in case there's `calls` defined on module we store the data
141 # in case there's `calls` defined on module we store the data
142 maybe_log_call(func.func_name, args, kwargs)
142 maybe_log_call(func.__name__, args, kwargs)
143 log.debug('Calling rcextensions function %s', func.func_name)
143 log.debug('Calling rcextensions function %s', func.__name__)
144 return func(*args, **kwargs)
144 return func(*args, **kwargs)
145 return wrapper
145 return wrapper
146 return wrap
146 return wrap
147
147
148
148
149 def maybe_log_call(name, args, kwargs):
149 def maybe_log_call(name, args, kwargs):
150 from rhodecode.config import rcextensions
150 from rhodecode.config import rcextensions
151 if hasattr(rcextensions, 'calls'):
151 if hasattr(rcextensions, 'calls'):
152 calls = rcextensions.calls
152 calls = rcextensions.calls
153 calls[name].append((args, kwargs))
153 calls[name].append((args, kwargs))
154
154
155
155
156 def str2bool(_str):
156 def str2bool(_str):
157 """
157 """
158 returns True/False value from given string, it tries to translate the
158 returns True/False value from given string, it tries to translate the
159 string into boolean
159 string into boolean
160
160
161 :param _str: string value to translate into boolean
161 :param _str: string value to translate into boolean
162 :rtype: boolean
162 :rtype: boolean
163 :returns: boolean from given string
163 :returns: boolean from given string
164 """
164 """
165 if _str is None:
165 if _str is None:
166 return False
166 return False
167 if _str in (True, False):
167 if _str in (True, False):
168 return _str
168 return _str
169 _str = str(_str).strip().lower()
169 _str = str(_str).strip().lower()
170 return _str in ('t', 'true', 'y', 'yes', 'on', '1')
170 return _str in ('t', 'true', 'y', 'yes', 'on', '1')
171
171
172
172
173 def aslist(obj, sep=None, strip=True):
173 def aslist(obj, sep=None, strip=True):
174 """
174 """
175 Returns given string separated by sep as list
175 Returns given string separated by sep as list
176
176
177 :param obj:
177 :param obj:
178 :param sep:
178 :param sep:
179 :param strip:
179 :param strip:
180 """
180 """
181 if isinstance(obj, (str,)):
181 if isinstance(obj, (str,)):
182 lst = obj.split(sep)
182 lst = obj.split(sep)
183 if strip:
183 if strip:
184 lst = [v.strip() for v in lst]
184 lst = [v.strip() for v in lst]
185 return lst
185 return lst
186 elif isinstance(obj, (list, tuple)):
186 elif isinstance(obj, (list, tuple)):
187 return obj
187 return obj
188 elif obj is None:
188 elif obj is None:
189 return []
189 return []
190 else:
190 else:
191 return [obj]
191 return [obj]
192
192
193
193
194 class UrlTemplate(string.Template):
194 class UrlTemplate(string.Template):
195
195
196 def safe_substitute(self, **kws):
196 def safe_substitute(self, **kws):
197 # url encode the kw for usage in url
197 # url encode the kw for usage in url
198 kws = {k: urllib.parse.quote(str(v)) for k, v in kws.items()}
198 kws = {k: urllib.parse.quote(str(v)) for k, v in kws.items()}
199 return super(UrlTemplate, self).safe_substitute(**kws)
199 return super(UrlTemplate, self).safe_substitute(**kws)
@@ -1,839 +1,839 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2 """Utilities for writing code that runs on Python 2 and 3"""
2 """Utilities for writing code that runs on Python 2 and 3"""
3
3
4 # Copyright (c) 2010-2015 Benjamin Peterson
4 # Copyright (c) 2010-2015 Benjamin Peterson
5 #
5 #
6 # Permission is hereby granted, free of charge, to any person obtaining a copy
6 # Permission is hereby granted, free of charge, to any person obtaining a copy
7 # of this software and associated documentation files (the "Software"), to deal
7 # of this software and associated documentation files (the "Software"), to deal
8 # in the Software without restriction, including without limitation the rights
8 # in the Software without restriction, including without limitation the rights
9 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9 # to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
10 # copies of the Software, and to permit persons to whom the Software is
10 # copies of the Software, and to permit persons to whom the Software is
11 # furnished to do so, subject to the following conditions:
11 # furnished to do so, subject to the following conditions:
12 #
12 #
13 # The above copyright notice and this permission notice shall be included in all
13 # The above copyright notice and this permission notice shall be included in all
14 # copies or substantial portions of the Software.
14 # copies or substantial portions of the Software.
15 #
15 #
16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 # IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 # AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22 # SOFTWARE.
22 # SOFTWARE.
23
23
24
24
25
25
26 import functools
26 import functools
27 import itertools
27 import itertools
28 import operator
28 import operator
29 import sys
29 import sys
30 import types
30 import types
31
31
32 __author__ = "Benjamin Peterson <benjamin@python.org>"
32 __author__ = "Benjamin Peterson <benjamin@python.org>"
33 __version__ = "1.9.0"
33 __version__ = "1.9.0"
34
34
35
35
36 # Useful for very coarse version differentiation.
36 # Useful for very coarse version differentiation.
37 PY2 = sys.version_info[0] == 2
37 PY2 = sys.version_info[0] == 2
38 PY3 = sys.version_info[0] == 3
38 PY3 = sys.version_info[0] == 3
39
39
40 if PY3:
40 if PY3:
41 string_types = str,
41 string_types = str,
42 integer_types = int,
42 integer_types = int,
43 class_types = type,
43 class_types = type,
44 text_type = str
44 text_type = str
45 binary_type = bytes
45 binary_type = bytes
46
46
47 MAXSIZE = sys.maxsize
47 MAXSIZE = sys.maxsize
48 else:
48 else:
49 string_types = str,
49 string_types = str,
50 integer_types = int
50 integer_types = int
51 class_types = (type, types.ClassType)
51 class_types = (type, types.ClassType)
52 text_type = unicode
52 text_type = unicode
53 binary_type = str
53 binary_type = str
54
54
55 if sys.platform.startswith("java"):
55 if sys.platform.startswith("java"):
56 # Jython always uses 32 bits.
56 # Jython always uses 32 bits.
57 MAXSIZE = int((1 << 31) - 1)
57 MAXSIZE = int((1 << 31) - 1)
58 else:
58 else:
59 # It's possible to have sizeof(long) != sizeof(Py_ssize_t).
59 # It's possible to have sizeof(long) != sizeof(Py_ssize_t).
60 class X(object):
60 class X(object):
61 def __len__(self):
61 def __len__(self):
62 return 1 << 31
62 return 1 << 31
63 try:
63 try:
64 len(X())
64 len(X())
65 except OverflowError:
65 except OverflowError:
66 # 32-bit
66 # 32-bit
67 MAXSIZE = int((1 << 31) - 1)
67 MAXSIZE = int((1 << 31) - 1)
68 else:
68 else:
69 # 64-bit
69 # 64-bit
70 MAXSIZE = int((1 << 63) - 1)
70 MAXSIZE = int((1 << 63) - 1)
71 del X
71 del X
72
72
73
73
74 def _add_doc(func, doc):
74 def _add_doc(func, doc):
75 """Add documentation to a function."""
75 """Add documentation to a function."""
76 func.__doc__ = doc
76 func.__doc__ = doc
77
77
78
78
79 def _import_module(name):
79 def _import_module(name):
80 """Import module, returning the module after the last dot."""
80 """Import module, returning the module after the last dot."""
81 __import__(name)
81 __import__(name)
82 return sys.modules[name]
82 return sys.modules[name]
83
83
84
84
85 class _LazyDescr(object):
85 class _LazyDescr(object):
86
86
87 def __init__(self, name):
87 def __init__(self, name):
88 self.name = name
88 self.name = name
89
89
90 def __get__(self, obj, tp):
90 def __get__(self, obj, tp):
91 result = self._resolve()
91 result = self._resolve()
92 setattr(obj, self.name, result) # Invokes __set__.
92 setattr(obj, self.name, result) # Invokes __set__.
93 try:
93 try:
94 # This is a bit ugly, but it avoids running this again by
94 # This is a bit ugly, but it avoids running this again by
95 # removing this descriptor.
95 # removing this descriptor.
96 delattr(obj.__class__, self.name)
96 delattr(obj.__class__, self.name)
97 except AttributeError:
97 except AttributeError:
98 pass
98 pass
99 return result
99 return result
100
100
101
101
102 class MovedModule(_LazyDescr):
102 class MovedModule(_LazyDescr):
103
103
104 def __init__(self, name, old, new=None):
104 def __init__(self, name, old, new=None):
105 super(MovedModule, self).__init__(name)
105 super(MovedModule, self).__init__(name)
106 if PY3:
106 if PY3:
107 if new is None:
107 if new is None:
108 new = name
108 new = name
109 self.mod = new
109 self.mod = new
110 else:
110 else:
111 self.mod = old
111 self.mod = old
112
112
113 def _resolve(self):
113 def _resolve(self):
114 return _import_module(self.mod)
114 return _import_module(self.mod)
115
115
116 def __getattr__(self, attr):
116 def __getattr__(self, attr):
117 _module = self._resolve()
117 _module = self._resolve()
118 value = getattr(_module, attr)
118 value = getattr(_module, attr)
119 setattr(self, attr, value)
119 setattr(self, attr, value)
120 return value
120 return value
121
121
122
122
123 class _LazyModule(types.ModuleType):
123 class _LazyModule(types.ModuleType):
124
124
125 def __init__(self, name):
125 def __init__(self, name):
126 super(_LazyModule, self).__init__(name)
126 super(_LazyModule, self).__init__(name)
127 self.__doc__ = self.__class__.__doc__
127 self.__doc__ = self.__class__.__doc__
128
128
129 def __dir__(self):
129 def __dir__(self):
130 attrs = ["__doc__", "__name__"]
130 attrs = ["__doc__", "__name__"]
131 attrs += [attr.name for attr in self._moved_attributes]
131 attrs += [attr.name for attr in self._moved_attributes]
132 return attrs
132 return attrs
133
133
134 # Subclasses should override this
134 # Subclasses should override this
135 _moved_attributes = []
135 _moved_attributes = []
136
136
137
137
138 class MovedAttribute(_LazyDescr):
138 class MovedAttribute(_LazyDescr):
139
139
140 def __init__(self, name, old_mod, new_mod, old_attr=None, new_attr=None):
140 def __init__(self, name, old_mod, new_mod, old_attr=None, new_attr=None):
141 super(MovedAttribute, self).__init__(name)
141 super(MovedAttribute, self).__init__(name)
142 if PY3:
142 if PY3:
143 if new_mod is None:
143 if new_mod is None:
144 new_mod = name
144 new_mod = name
145 self.mod = new_mod
145 self.mod = new_mod
146 if new_attr is None:
146 if new_attr is None:
147 if old_attr is None:
147 if old_attr is None:
148 new_attr = name
148 new_attr = name
149 else:
149 else:
150 new_attr = old_attr
150 new_attr = old_attr
151 self.attr = new_attr
151 self.attr = new_attr
152 else:
152 else:
153 self.mod = old_mod
153 self.mod = old_mod
154 if old_attr is None:
154 if old_attr is None:
155 old_attr = name
155 old_attr = name
156 self.attr = old_attr
156 self.attr = old_attr
157
157
158 def _resolve(self):
158 def _resolve(self):
159 module = _import_module(self.mod)
159 module = _import_module(self.mod)
160 return getattr(module, self.attr)
160 return getattr(module, self.attr)
161
161
162
162
163 class _SixMetaPathImporter(object):
163 class _SixMetaPathImporter(object):
164 """
164 """
165 A meta path importer to import six.moves and its submodules.
165 A meta path importer to import six.moves and its submodules.
166
166
167 This class implements a PEP302 finder and loader. It should be compatible
167 This class implements a PEP302 finder and loader. It should be compatible
168 with Python 2.5 and all existing versions of Python3
168 with Python 2.5 and all existing versions of Python3
169 """
169 """
170 def __init__(self, six_module_name):
170 def __init__(self, six_module_name):
171 self.name = six_module_name
171 self.name = six_module_name
172 self.known_modules = {}
172 self.known_modules = {}
173
173
174 def _add_module(self, mod, *fullnames):
174 def _add_module(self, mod, *fullnames):
175 for fullname in fullnames:
175 for fullname in fullnames:
176 self.known_modules[self.name + "." + fullname] = mod
176 self.known_modules[self.name + "." + fullname] = mod
177
177
178 def _get_module(self, fullname):
178 def _get_module(self, fullname):
179 return self.known_modules[self.name + "." + fullname]
179 return self.known_modules[self.name + "." + fullname]
180
180
181 def find_module(self, fullname, path=None):
181 def find_module(self, fullname, path=None):
182 if fullname in self.known_modules:
182 if fullname in self.known_modules:
183 return self
183 return self
184 return None
184 return None
185
185
186 def __get_module(self, fullname):
186 def __get_module(self, fullname):
187 try:
187 try:
188 return self.known_modules[fullname]
188 return self.known_modules[fullname]
189 except KeyError:
189 except KeyError:
190 raise ImportError("This loader does not know module " + fullname)
190 raise ImportError("This loader does not know module " + fullname)
191
191
192 def load_module(self, fullname):
192 def load_module(self, fullname):
193 try:
193 try:
194 # in case of a reload
194 # in case of a reload
195 return sys.modules[fullname]
195 return sys.modules[fullname]
196 except KeyError:
196 except KeyError:
197 pass
197 pass
198 mod = self.__get_module(fullname)
198 mod = self.__get_module(fullname)
199 if isinstance(mod, MovedModule):
199 if isinstance(mod, MovedModule):
200 mod = mod._resolve()
200 mod = mod._resolve()
201 else:
201 else:
202 mod.__loader__ = self
202 mod.__loader__ = self
203 sys.modules[fullname] = mod
203 sys.modules[fullname] = mod
204 return mod
204 return mod
205
205
206 def is_package(self, fullname):
206 def is_package(self, fullname):
207 """
207 """
208 Return true, if the named module is a package.
208 Return true, if the named module is a package.
209
209
210 We need this method to get correct spec objects with
210 We need this method to get correct spec objects with
211 Python 3.4 (see PEP451)
211 Python 3.4 (see PEP451)
212 """
212 """
213 return hasattr(self.__get_module(fullname), "__path__")
213 return hasattr(self.__get_module(fullname), "__path__")
214
214
215 def get_code(self, fullname):
215 def get_code(self, fullname):
216 """Return None
216 """Return None
217
217
218 Required, if is_package is implemented"""
218 Required, if is_package is implemented"""
219 self.__get_module(fullname) # eventually raises ImportError
219 self.__get_module(fullname) # eventually raises ImportError
220 return None
220 return None
221 get_source = get_code # same as get_code
221 get_source = get_code # same as get_code
222
222
223 _importer = _SixMetaPathImporter(__name__)
223 _importer = _SixMetaPathImporter(__name__)
224
224
225
225
226 class _MovedItems(_LazyModule):
226 class _MovedItems(_LazyModule):
227 """Lazy loading of moved objects"""
227 """Lazy loading of moved objects"""
228 __path__ = [] # mark as package
228 __path__ = [] # mark as package
229
229
230
230
231 _moved_attributes = [
231 _moved_attributes = [
232 MovedAttribute("cStringIO", "cStringIO", "io", "StringIO"),
232 MovedAttribute("cStringIO", "cStringIO", "io", "StringIO"),
233 MovedAttribute("filter", "itertools", "builtins", "ifilter", "filter"),
233 MovedAttribute("filter", "itertools", "builtins", "ifilter", "filter"),
234 MovedAttribute("filterfalse", "itertools", "itertools", "ifilterfalse", "filterfalse"),
234 MovedAttribute("filterfalse", "itertools", "itertools", "ifilterfalse", "filterfalse"),
235 MovedAttribute("input", "__builtin__", "builtins", "raw_input", "input"),
235 MovedAttribute("input", "__builtin__", "builtins", "raw_input", "input"),
236 MovedAttribute("intern", "__builtin__", "sys"),
236 MovedAttribute("intern", "__builtin__", "sys"),
237 MovedAttribute("map", "itertools", "builtins", "imap", "map"),
237 MovedAttribute("map", "itertools", "builtins", "imap", "map"),
238 MovedAttribute("range", "__builtin__", "builtins", "xrange", "range"),
238 MovedAttribute("range", "__builtin__", "builtins", "xrange", "range"),
239 MovedAttribute("reload_module", "__builtin__", "imp", "reload"),
239 MovedAttribute("reload_module", "__builtin__", "imp", "reload"),
240 MovedAttribute("reduce", "__builtin__", "functools"),
240 MovedAttribute("reduce", "__builtin__", "functools"),
241 MovedAttribute("shlex_quote", "pipes", "shlex", "quote"),
241 MovedAttribute("shlex_quote", "pipes", "shlex", "quote"),
242 MovedAttribute("StringIO", "StringIO", "io"),
242 MovedAttribute("StringIO", "StringIO", "io"),
243 MovedAttribute("UserDict", "UserDict", "collections"),
243 MovedAttribute("UserDict", "UserDict", "collections"),
244 MovedAttribute("UserList", "UserList", "collections"),
244 MovedAttribute("UserList", "UserList", "collections"),
245 MovedAttribute("UserString", "UserString", "collections"),
245 MovedAttribute("UserString", "UserString", "collections"),
246 MovedAttribute("xrange", "__builtin__", "builtins", "xrange", "range"),
246 MovedAttribute("xrange", "__builtin__", "builtins", "xrange", "range"),
247 MovedAttribute("zip", "itertools", "builtins", "izip", "zip"),
247 MovedAttribute("zip", "itertools", "builtins", "izip", "zip"),
248 MovedAttribute("zip_longest", "itertools", "itertools", "izip_longest", "zip_longest"),
248 MovedAttribute("zip_longest", "itertools", "itertools", "izip_longest", "zip_longest"),
249
249
250 MovedModule("builtins", "__builtin__"),
250 MovedModule("builtins", "__builtin__"),
251 MovedModule("configparser", "ConfigParser"),
251 MovedModule("configparser", "ConfigParser"),
252 MovedModule("copyreg", "copy_reg"),
252 MovedModule("copyreg", "copy_reg"),
253 MovedModule("dbm_gnu", "gdbm", "dbm.gnu"),
253 MovedModule("dbm_gnu", "gdbm", "dbm.gnu"),
254 MovedModule("_dummy_thread", "dummy_thread", "_dummy_thread"),
254 MovedModule("_dummy_thread", "dummy_thread", "_dummy_thread"),
255 MovedModule("http_cookiejar", "cookielib", "http.cookiejar"),
255 MovedModule("http_cookiejar", "cookielib", "http.cookiejar"),
256 MovedModule("http_cookies", "Cookie", "http.cookies"),
256 MovedModule("http_cookies", "Cookie", "http.cookies"),
257 MovedModule("html_entities", "htmlentitydefs", "html.entities"),
257 MovedModule("html_entities", "htmlentitydefs", "html.entities"),
258 MovedModule("html_parser", "HTMLParser", "html.parser"),
258 MovedModule("html_parser", "HTMLParser", "html.parser"),
259 MovedModule("http_client", "httplib", "http.client"),
259 MovedModule("http_client", "httplib", "http.client"),
260 MovedModule("email_mime_multipart", "email.MIMEMultipart", "email.mime.multipart"),
260 MovedModule("email_mime_multipart", "email.MIMEMultipart", "email.mime.multipart"),
261 MovedModule("email_mime_nonmultipart", "email.MIMENonMultipart", "email.mime.nonmultipart"),
261 MovedModule("email_mime_nonmultipart", "email.MIMENonMultipart", "email.mime.nonmultipart"),
262 MovedModule("email_mime_text", "email.MIMEText", "email.mime.text"),
262 MovedModule("email_mime_text", "email.MIMEText", "email.mime.text"),
263 MovedModule("email_mime_base", "email.MIMEBase", "email.mime.base"),
263 MovedModule("email_mime_base", "email.MIMEBase", "email.mime.base"),
264 MovedModule("BaseHTTPServer", "BaseHTTPServer", "http.server"),
264 MovedModule("BaseHTTPServer", "BaseHTTPServer", "http.server"),
265 MovedModule("CGIHTTPServer", "CGIHTTPServer", "http.server"),
265 MovedModule("CGIHTTPServer", "CGIHTTPServer", "http.server"),
266 MovedModule("SimpleHTTPServer", "SimpleHTTPServer", "http.server"),
266 MovedModule("SimpleHTTPServer", "SimpleHTTPServer", "http.server"),
267 MovedModule("cPickle", "cPickle", "pickle"),
267 MovedModule("cPickle", "cPickle", "pickle"),
268 MovedModule("queue", "Queue"),
268 MovedModule("queue", "Queue"),
269 MovedModule("reprlib", "repr"),
269 MovedModule("reprlib", "repr"),
270 MovedModule("socketserver", "SocketServer"),
270 MovedModule("socketserver", "SocketServer"),
271 MovedModule("_thread", "thread", "_thread"),
271 MovedModule("_thread", "thread", "_thread"),
272 MovedModule("tkinter", "Tkinter"),
272 MovedModule("tkinter", "Tkinter"),
273 MovedModule("tkinter_dialog", "Dialog", "tkinter.dialog"),
273 MovedModule("tkinter_dialog", "Dialog", "tkinter.dialog"),
274 MovedModule("tkinter_filedialog", "FileDialog", "tkinter.filedialog"),
274 MovedModule("tkinter_filedialog", "FileDialog", "tkinter.filedialog"),
275 MovedModule("tkinter_scrolledtext", "ScrolledText", "tkinter.scrolledtext"),
275 MovedModule("tkinter_scrolledtext", "ScrolledText", "tkinter.scrolledtext"),
276 MovedModule("tkinter_simpledialog", "SimpleDialog", "tkinter.simpledialog"),
276 MovedModule("tkinter_simpledialog", "SimpleDialog", "tkinter.simpledialog"),
277 MovedModule("tkinter_tix", "Tix", "tkinter.tix"),
277 MovedModule("tkinter_tix", "Tix", "tkinter.tix"),
278 MovedModule("tkinter_ttk", "ttk", "tkinter.ttk"),
278 MovedModule("tkinter_ttk", "ttk", "tkinter.ttk"),
279 MovedModule("tkinter_constants", "Tkconstants", "tkinter.constants"),
279 MovedModule("tkinter_constants", "Tkconstants", "tkinter.constants"),
280 MovedModule("tkinter_dnd", "Tkdnd", "tkinter.dnd"),
280 MovedModule("tkinter_dnd", "Tkdnd", "tkinter.dnd"),
281 MovedModule("tkinter_colorchooser", "tkColorChooser",
281 MovedModule("tkinter_colorchooser", "tkColorChooser",
282 "tkinter.colorchooser"),
282 "tkinter.colorchooser"),
283 MovedModule("tkinter_commondialog", "tkCommonDialog",
283 MovedModule("tkinter_commondialog", "tkCommonDialog",
284 "tkinter.commondialog"),
284 "tkinter.commondialog"),
285 MovedModule("tkinter_tkfiledialog", "tkFileDialog", "tkinter.filedialog"),
285 MovedModule("tkinter_tkfiledialog", "tkFileDialog", "tkinter.filedialog"),
286 MovedModule("tkinter_font", "tkFont", "tkinter.font"),
286 MovedModule("tkinter_font", "tkFont", "tkinter.font"),
287 MovedModule("tkinter_messagebox", "tkMessageBox", "tkinter.messagebox"),
287 MovedModule("tkinter_messagebox", "tkMessageBox", "tkinter.messagebox"),
288 MovedModule("tkinter_tksimpledialog", "tkSimpleDialog",
288 MovedModule("tkinter_tksimpledialog", "tkSimpleDialog",
289 "tkinter.simpledialog"),
289 "tkinter.simpledialog"),
290 MovedModule("urllib_parse", __name__ + ".moves.urllib_parse", "urllib.parse"),
290 MovedModule("urllib_parse", __name__ + ".moves.urllib_parse", "urllib.parse"),
291 MovedModule("urllib_error", __name__ + ".moves.urllib_error", "urllib.error"),
291 MovedModule("urllib_error", __name__ + ".moves.urllib_error", "urllib.error"),
292 MovedModule("urllib", __name__ + ".moves.urllib", __name__ + ".moves.urllib"),
292 MovedModule("urllib", __name__ + ".moves.urllib", __name__ + ".moves.urllib"),
293 MovedModule("urllib_robotparser", "robotparser", "urllib.robotparser"),
293 MovedModule("urllib_robotparser", "robotparser", "urllib.robotparser"),
294 MovedModule("xmlrpc_client", "xmlrpclib", "xmlrpc.client"),
294 MovedModule("xmlrpc_client", "xmlrpclib", "xmlrpc.client"),
295 MovedModule("xmlrpc_server", "SimpleXMLRPCServer", "xmlrpc.server"),
295 MovedModule("xmlrpc_server", "SimpleXMLRPCServer", "xmlrpc.server"),
296 MovedModule("winreg", "_winreg"),
296 MovedModule("winreg", "_winreg"),
297 ]
297 ]
298 for attr in _moved_attributes:
298 for attr in _moved_attributes:
299 setattr(_MovedItems, attr.name, attr)
299 setattr(_MovedItems, attr.name, attr)
300 if isinstance(attr, MovedModule):
300 if isinstance(attr, MovedModule):
301 _importer._add_module(attr, "moves." + attr.name)
301 _importer._add_module(attr, "moves." + attr.name)
302 del attr
302 del attr
303
303
304 _MovedItems._moved_attributes = _moved_attributes
304 _MovedItems._moved_attributes = _moved_attributes
305
305
306 moves = _MovedItems(__name__ + ".moves")
306 moves = _MovedItems(__name__ + ".moves")
307 _importer._add_module(moves, "moves")
307 _importer._add_module(moves, "moves")
308
308
309
309
310 class Module_six_moves_urllib_parse(_LazyModule):
310 class Module_six_moves_urllib_parse(_LazyModule):
311 """Lazy loading of moved objects in six.moves.urllib_parse"""
311 """Lazy loading of moved objects in six.moves.urllib_parse"""
312
312
313
313
314 _urllib_parse_moved_attributes = [
314 _urllib_parse_moved_attributes = [
315 MovedAttribute("ParseResult", "urlparse", "urllib.parse"),
315 MovedAttribute("ParseResult", "urlparse", "urllib.parse"),
316 MovedAttribute("SplitResult", "urlparse", "urllib.parse"),
316 MovedAttribute("SplitResult", "urlparse", "urllib.parse"),
317 MovedAttribute("parse_qs", "urlparse", "urllib.parse"),
317 MovedAttribute("parse_qs", "urlparse", "urllib.parse"),
318 MovedAttribute("parse_qsl", "urlparse", "urllib.parse"),
318 MovedAttribute("parse_qsl", "urlparse", "urllib.parse"),
319 MovedAttribute("urldefrag", "urlparse", "urllib.parse"),
319 MovedAttribute("urldefrag", "urlparse", "urllib.parse"),
320 MovedAttribute("urljoin", "urlparse", "urllib.parse"),
320 MovedAttribute("urljoin", "urlparse", "urllib.parse"),
321 MovedAttribute("urlparse", "urlparse", "urllib.parse"),
321 MovedAttribute("urlparse", "urlparse", "urllib.parse"),
322 MovedAttribute("urlsplit", "urlparse", "urllib.parse"),
322 MovedAttribute("urlsplit", "urlparse", "urllib.parse"),
323 MovedAttribute("urlunparse", "urlparse", "urllib.parse"),
323 MovedAttribute("urlunparse", "urlparse", "urllib.parse"),
324 MovedAttribute("urlunsplit", "urlparse", "urllib.parse"),
324 MovedAttribute("urlunsplit", "urlparse", "urllib.parse"),
325 MovedAttribute("quote", "urllib", "urllib.parse"),
325 MovedAttribute("quote", "urllib", "urllib.parse"),
326 MovedAttribute("quote_plus", "urllib", "urllib.parse"),
326 MovedAttribute("quote_plus", "urllib", "urllib.parse"),
327 MovedAttribute("unquote", "urllib", "urllib.parse"),
327 MovedAttribute("unquote", "urllib", "urllib.parse"),
328 MovedAttribute("unquote_plus", "urllib", "urllib.parse"),
328 MovedAttribute("unquote_plus", "urllib", "urllib.parse"),
329 MovedAttribute("urlencode", "urllib", "urllib.parse"),
329 MovedAttribute("urlencode", "urllib", "urllib.parse"),
330 MovedAttribute("splitquery", "urllib", "urllib.parse"),
330 MovedAttribute("splitquery", "urllib", "urllib.parse"),
331 MovedAttribute("splittag", "urllib", "urllib.parse"),
331 MovedAttribute("splittag", "urllib", "urllib.parse"),
332 MovedAttribute("splituser", "urllib", "urllib.parse"),
332 MovedAttribute("splituser", "urllib", "urllib.parse"),
333 MovedAttribute("uses_fragment", "urlparse", "urllib.parse"),
333 MovedAttribute("uses_fragment", "urlparse", "urllib.parse"),
334 MovedAttribute("uses_netloc", "urlparse", "urllib.parse"),
334 MovedAttribute("uses_netloc", "urlparse", "urllib.parse"),
335 MovedAttribute("uses_params", "urlparse", "urllib.parse"),
335 MovedAttribute("uses_params", "urlparse", "urllib.parse"),
336 MovedAttribute("uses_query", "urlparse", "urllib.parse"),
336 MovedAttribute("uses_query", "urlparse", "urllib.parse"),
337 MovedAttribute("uses_relative", "urlparse", "urllib.parse"),
337 MovedAttribute("uses_relative", "urlparse", "urllib.parse"),
338 ]
338 ]
339 for attr in _urllib_parse_moved_attributes:
339 for attr in _urllib_parse_moved_attributes:
340 setattr(Module_six_moves_urllib_parse, attr.name, attr)
340 setattr(Module_six_moves_urllib_parse, attr.name, attr)
341 del attr
341 del attr
342
342
343 Module_six_moves_urllib_parse._moved_attributes = _urllib_parse_moved_attributes
343 Module_six_moves_urllib_parse._moved_attributes = _urllib_parse_moved_attributes
344
344
345 _importer._add_module(Module_six_moves_urllib_parse(__name__ + ".moves.urllib_parse"),
345 _importer._add_module(Module_six_moves_urllib_parse(__name__ + ".moves.urllib_parse"),
346 "moves.urllib_parse", "moves.urllib.parse")
346 "moves.urllib_parse", "moves.urllib.parse")
347
347
348
348
349 class Module_six_moves_urllib_error(_LazyModule):
349 class Module_six_moves_urllib_error(_LazyModule):
350 """Lazy loading of moved objects in six.moves.urllib_error"""
350 """Lazy loading of moved objects in six.moves.urllib_error"""
351
351
352
352
353 _urllib_error_moved_attributes = [
353 _urllib_error_moved_attributes = [
354 MovedAttribute("URLError", "urllib2", "urllib.error"),
354 MovedAttribute("URLError", "urllib2", "urllib.error"),
355 MovedAttribute("HTTPError", "urllib2", "urllib.error"),
355 MovedAttribute("HTTPError", "urllib2", "urllib.error"),
356 MovedAttribute("ContentTooShortError", "urllib", "urllib.error"),
356 MovedAttribute("ContentTooShortError", "urllib", "urllib.error"),
357 ]
357 ]
358 for attr in _urllib_error_moved_attributes:
358 for attr in _urllib_error_moved_attributes:
359 setattr(Module_six_moves_urllib_error, attr.name, attr)
359 setattr(Module_six_moves_urllib_error, attr.name, attr)
360 del attr
360 del attr
361
361
362 Module_six_moves_urllib_error._moved_attributes = _urllib_error_moved_attributes
362 Module_six_moves_urllib_error._moved_attributes = _urllib_error_moved_attributes
363
363
364 _importer._add_module(Module_six_moves_urllib_error(__name__ + ".moves.urllib.error"),
364 _importer._add_module(Module_six_moves_urllib_error(__name__ + ".moves.urllib.error"),
365 "moves.urllib_error", "moves.urllib.error")
365 "moves.urllib_error", "moves.urllib.error")
366
366
367
367
368 class Module_six_moves_urllib_request(_LazyModule):
368 class Module_six_moves_urllib_request(_LazyModule):
369 """Lazy loading of moved objects in six.moves.urllib_request"""
369 """Lazy loading of moved objects in six.moves.urllib_request"""
370
370
371
371
372 _urllib_request_moved_attributes = [
372 _urllib_request_moved_attributes = [
373 MovedAttribute("urlopen", "urllib2", "urllib.request"),
373 MovedAttribute("urlopen", "urllib2", "urllib.request"),
374 MovedAttribute("install_opener", "urllib2", "urllib.request"),
374 MovedAttribute("install_opener", "urllib2", "urllib.request"),
375 MovedAttribute("build_opener", "urllib2", "urllib.request"),
375 MovedAttribute("build_opener", "urllib2", "urllib.request"),
376 MovedAttribute("pathname2url", "urllib", "urllib.request"),
376 MovedAttribute("pathname2url", "urllib", "urllib.request"),
377 MovedAttribute("url2pathname", "urllib", "urllib.request"),
377 MovedAttribute("url2pathname", "urllib", "urllib.request"),
378 MovedAttribute("getproxies", "urllib", "urllib.request"),
378 MovedAttribute("getproxies", "urllib", "urllib.request"),
379 MovedAttribute("Request", "urllib2", "urllib.request"),
379 MovedAttribute("Request", "urllib2", "urllib.request"),
380 MovedAttribute("OpenerDirector", "urllib2", "urllib.request"),
380 MovedAttribute("OpenerDirector", "urllib2", "urllib.request"),
381 MovedAttribute("HTTPDefaultErrorHandler", "urllib2", "urllib.request"),
381 MovedAttribute("HTTPDefaultErrorHandler", "urllib2", "urllib.request"),
382 MovedAttribute("HTTPRedirectHandler", "urllib2", "urllib.request"),
382 MovedAttribute("HTTPRedirectHandler", "urllib2", "urllib.request"),
383 MovedAttribute("HTTPCookieProcessor", "urllib2", "urllib.request"),
383 MovedAttribute("HTTPCookieProcessor", "urllib2", "urllib.request"),
384 MovedAttribute("ProxyHandler", "urllib2", "urllib.request"),
384 MovedAttribute("ProxyHandler", "urllib2", "urllib.request"),
385 MovedAttribute("BaseHandler", "urllib2", "urllib.request"),
385 MovedAttribute("BaseHandler", "urllib2", "urllib.request"),
386 MovedAttribute("HTTPPasswordMgr", "urllib2", "urllib.request"),
386 MovedAttribute("HTTPPasswordMgr", "urllib2", "urllib.request"),
387 MovedAttribute("HTTPPasswordMgrWithDefaultRealm", "urllib2", "urllib.request"),
387 MovedAttribute("HTTPPasswordMgrWithDefaultRealm", "urllib2", "urllib.request"),
388 MovedAttribute("AbstractBasicAuthHandler", "urllib2", "urllib.request"),
388 MovedAttribute("AbstractBasicAuthHandler", "urllib2", "urllib.request"),
389 MovedAttribute("HTTPBasicAuthHandler", "urllib2", "urllib.request"),
389 MovedAttribute("HTTPBasicAuthHandler", "urllib2", "urllib.request"),
390 MovedAttribute("ProxyBasicAuthHandler", "urllib2", "urllib.request"),
390 MovedAttribute("ProxyBasicAuthHandler", "urllib2", "urllib.request"),
391 MovedAttribute("AbstractDigestAuthHandler", "urllib2", "urllib.request"),
391 MovedAttribute("AbstractDigestAuthHandler", "urllib2", "urllib.request"),
392 MovedAttribute("HTTPDigestAuthHandler", "urllib2", "urllib.request"),
392 MovedAttribute("HTTPDigestAuthHandler", "urllib2", "urllib.request"),
393 MovedAttribute("ProxyDigestAuthHandler", "urllib2", "urllib.request"),
393 MovedAttribute("ProxyDigestAuthHandler", "urllib2", "urllib.request"),
394 MovedAttribute("HTTPHandler", "urllib2", "urllib.request"),
394 MovedAttribute("HTTPHandler", "urllib2", "urllib.request"),
395 MovedAttribute("HTTPSHandler", "urllib2", "urllib.request"),
395 MovedAttribute("HTTPSHandler", "urllib2", "urllib.request"),
396 MovedAttribute("FileHandler", "urllib2", "urllib.request"),
396 MovedAttribute("FileHandler", "urllib2", "urllib.request"),
397 MovedAttribute("FTPHandler", "urllib2", "urllib.request"),
397 MovedAttribute("FTPHandler", "urllib2", "urllib.request"),
398 MovedAttribute("CacheFTPHandler", "urllib2", "urllib.request"),
398 MovedAttribute("CacheFTPHandler", "urllib2", "urllib.request"),
399 MovedAttribute("UnknownHandler", "urllib2", "urllib.request"),
399 MovedAttribute("UnknownHandler", "urllib2", "urllib.request"),
400 MovedAttribute("HTTPErrorProcessor", "urllib2", "urllib.request"),
400 MovedAttribute("HTTPErrorProcessor", "urllib2", "urllib.request"),
401 MovedAttribute("urlretrieve", "urllib", "urllib.request"),
401 MovedAttribute("urlretrieve", "urllib", "urllib.request"),
402 MovedAttribute("urlcleanup", "urllib", "urllib.request"),
402 MovedAttribute("urlcleanup", "urllib", "urllib.request"),
403 MovedAttribute("URLopener", "urllib", "urllib.request"),
403 MovedAttribute("URLopener", "urllib", "urllib.request"),
404 MovedAttribute("FancyURLopener", "urllib", "urllib.request"),
404 MovedAttribute("FancyURLopener", "urllib", "urllib.request"),
405 MovedAttribute("proxy_bypass", "urllib", "urllib.request"),
405 MovedAttribute("proxy_bypass", "urllib", "urllib.request"),
406 ]
406 ]
407 for attr in _urllib_request_moved_attributes:
407 for attr in _urllib_request_moved_attributes:
408 setattr(Module_six_moves_urllib_request, attr.name, attr)
408 setattr(Module_six_moves_urllib_request, attr.name, attr)
409 del attr
409 del attr
410
410
411 Module_six_moves_urllib_request._moved_attributes = _urllib_request_moved_attributes
411 Module_six_moves_urllib_request._moved_attributes = _urllib_request_moved_attributes
412
412
413 _importer._add_module(Module_six_moves_urllib_request(__name__ + ".moves.urllib.request"),
413 _importer._add_module(Module_six_moves_urllib_request(__name__ + ".moves.urllib.request"),
414 "moves.urllib_request", "moves.urllib.request")
414 "moves.urllib_request", "moves.urllib.request")
415
415
416
416
417 class Module_six_moves_urllib_response(_LazyModule):
417 class Module_six_moves_urllib_response(_LazyModule):
418 """Lazy loading of moved objects in six.moves.urllib_response"""
418 """Lazy loading of moved objects in six.moves.urllib_response"""
419
419
420
420
421 _urllib_response_moved_attributes = [
421 _urllib_response_moved_attributes = [
422 MovedAttribute("addbase", "urllib", "urllib.response"),
422 MovedAttribute("addbase", "urllib", "urllib.response"),
423 MovedAttribute("addclosehook", "urllib", "urllib.response"),
423 MovedAttribute("addclosehook", "urllib", "urllib.response"),
424 MovedAttribute("addinfo", "urllib", "urllib.response"),
424 MovedAttribute("addinfo", "urllib", "urllib.response"),
425 MovedAttribute("addinfourl", "urllib", "urllib.response"),
425 MovedAttribute("addinfourl", "urllib", "urllib.response"),
426 ]
426 ]
427 for attr in _urllib_response_moved_attributes:
427 for attr in _urllib_response_moved_attributes:
428 setattr(Module_six_moves_urllib_response, attr.name, attr)
428 setattr(Module_six_moves_urllib_response, attr.name, attr)
429 del attr
429 del attr
430
430
431 Module_six_moves_urllib_response._moved_attributes = _urllib_response_moved_attributes
431 Module_six_moves_urllib_response._moved_attributes = _urllib_response_moved_attributes
432
432
433 _importer._add_module(Module_six_moves_urllib_response(__name__ + ".moves.urllib.response"),
433 _importer._add_module(Module_six_moves_urllib_response(__name__ + ".moves.urllib.response"),
434 "moves.urllib_response", "moves.urllib.response")
434 "moves.urllib_response", "moves.urllib.response")
435
435
436
436
437 class Module_six_moves_urllib_robotparser(_LazyModule):
437 class Module_six_moves_urllib_robotparser(_LazyModule):
438 """Lazy loading of moved objects in six.moves.urllib_robotparser"""
438 """Lazy loading of moved objects in six.moves.urllib_robotparser"""
439
439
440
440
441 _urllib_robotparser_moved_attributes = [
441 _urllib_robotparser_moved_attributes = [
442 MovedAttribute("RobotFileParser", "robotparser", "urllib.robotparser"),
442 MovedAttribute("RobotFileParser", "robotparser", "urllib.robotparser"),
443 ]
443 ]
444 for attr in _urllib_robotparser_moved_attributes:
444 for attr in _urllib_robotparser_moved_attributes:
445 setattr(Module_six_moves_urllib_robotparser, attr.name, attr)
445 setattr(Module_six_moves_urllib_robotparser, attr.name, attr)
446 del attr
446 del attr
447
447
448 Module_six_moves_urllib_robotparser._moved_attributes = _urllib_robotparser_moved_attributes
448 Module_six_moves_urllib_robotparser._moved_attributes = _urllib_robotparser_moved_attributes
449
449
450 _importer._add_module(Module_six_moves_urllib_robotparser(__name__ + ".moves.urllib.robotparser"),
450 _importer._add_module(Module_six_moves_urllib_robotparser(__name__ + ".moves.urllib.robotparser"),
451 "moves.urllib_robotparser", "moves.urllib.robotparser")
451 "moves.urllib_robotparser", "moves.urllib.robotparser")
452
452
453
453
454 class Module_six_moves_urllib(types.ModuleType):
454 class Module_six_moves_urllib(types.ModuleType):
455 """Create a six.moves.urllib namespace that resembles the Python 3 namespace"""
455 """Create a six.moves.urllib namespace that resembles the Python 3 namespace"""
456 __path__ = [] # mark as package
456 __path__ = [] # mark as package
457 parse = _importer._get_module("moves.urllib_parse")
457 parse = _importer._get_module("moves.urllib_parse")
458 error = _importer._get_module("moves.urllib_error")
458 error = _importer._get_module("moves.urllib_error")
459 request = _importer._get_module("moves.urllib_request")
459 request = _importer._get_module("moves.urllib_request")
460 response = _importer._get_module("moves.urllib_response")
460 response = _importer._get_module("moves.urllib_response")
461 robotparser = _importer._get_module("moves.urllib_robotparser")
461 robotparser = _importer._get_module("moves.urllib_robotparser")
462
462
463 def __dir__(self):
463 def __dir__(self):
464 return ['parse', 'error', 'request', 'response', 'robotparser']
464 return ['parse', 'error', 'request', 'response', 'robotparser']
465
465
466 _importer._add_module(Module_six_moves_urllib(__name__ + ".moves.urllib"),
466 _importer._add_module(Module_six_moves_urllib(__name__ + ".moves.urllib"),
467 "moves.urllib")
467 "moves.urllib")
468
468
469
469
470 def add_move(move):
470 def add_move(move):
471 """Add an item to six.moves."""
471 """Add an item to six.moves."""
472 setattr(_MovedItems, move.name, move)
472 setattr(_MovedItems, move.name, move)
473
473
474
474
475 def remove_move(name):
475 def remove_move(name):
476 """Remove item from six.moves."""
476 """Remove item from six.moves."""
477 try:
477 try:
478 delattr(_MovedItems, name)
478 delattr(_MovedItems, name)
479 except AttributeError:
479 except AttributeError:
480 try:
480 try:
481 del moves.__dict__[name]
481 del moves.__dict__[name]
482 except KeyError:
482 except KeyError:
483 raise AttributeError("no such move, %r" % (name,))
483 raise AttributeError("no such move, %r" % (name,))
484
484
485
485
486 if PY3:
486 if PY3:
487 _meth_func = "__func__"
487 _meth_func = "__func__"
488 _meth_self = "__self__"
488 _meth_self = "__self__"
489
489
490 _func_closure = "__closure__"
490 _func_closure = "__closure__"
491 _func_code = "__code__"
491 _func_code = "__code__"
492 _func_defaults = "__defaults__"
492 _func_defaults = "__defaults__"
493 _func_globals = "__globals__"
493 _func_globals = "__globals__"
494 else:
494 else:
495 _meth_func = "im_func"
495 _meth_func = "im_func"
496 _meth_self = "im_self"
496 _meth_self = "im_self"
497
497
498 _func_closure = "func_closure"
498 _func_closure = "func_closure"
499 _func_code = "func_code"
499 _func_code = "func_code"
500 _func_defaults = "func_defaults"
500 _func_defaults = "func_defaults"
501 _func_globals = "func_globals"
501 _func_globals = "func_globals"
502
502
503
503
504 try:
504 try:
505 advance_iterator = next
505 advance_iterator = next
506 except NameError:
506 except NameError:
507 def advance_iterator(it):
507 def advance_iterator(it):
508 return it.next()
508 return next(it)
509 next = advance_iterator
509 next = advance_iterator
510
510
511
511
512 try:
512 try:
513 callable = callable
513 callable = callable
514 except NameError:
514 except NameError:
515 def callable(obj):
515 def callable(obj):
516 return any("__call__" in klass.__dict__ for klass in type(obj).__mro__)
516 return any("__call__" in klass.__dict__ for klass in type(obj).__mro__)
517
517
518
518
519 if PY3:
519 if PY3:
520 def get_unbound_function(unbound):
520 def get_unbound_function(unbound):
521 return unbound
521 return unbound
522
522
523 create_bound_method = types.MethodType
523 create_bound_method = types.MethodType
524
524
525 Iterator = object
525 Iterator = object
526 else:
526 else:
527 def get_unbound_function(unbound):
527 def get_unbound_function(unbound):
528 return unbound.im_func
528 return unbound.im_func
529
529
530 def create_bound_method(func, obj):
530 def create_bound_method(func, obj):
531 return types.MethodType(func, obj, obj.__class__)
531 return types.MethodType(func, obj, obj.__class__)
532
532
533 class Iterator(object):
533 class Iterator(object):
534
534
535 def next(self):
535 def next(self):
536 return type(self).__next__(self)
536 return type(self).__next__(self)
537
537
538 callable = callable
538 callable = callable
539 _add_doc(get_unbound_function,
539 _add_doc(get_unbound_function,
540 """Get the function out of a possibly unbound function""")
540 """Get the function out of a possibly unbound function""")
541
541
542
542
543 get_method_function = operator.attrgetter(_meth_func)
543 get_method_function = operator.attrgetter(_meth_func)
544 get_method_self = operator.attrgetter(_meth_self)
544 get_method_self = operator.attrgetter(_meth_self)
545 get_function_closure = operator.attrgetter(_func_closure)
545 get_function_closure = operator.attrgetter(_func_closure)
546 get_function_code = operator.attrgetter(_func_code)
546 get_function_code = operator.attrgetter(_func_code)
547 get_function_defaults = operator.attrgetter(_func_defaults)
547 get_function_defaults = operator.attrgetter(_func_defaults)
548 get_function_globals = operator.attrgetter(_func_globals)
548 get_function_globals = operator.attrgetter(_func_globals)
549
549
550
550
551 if PY3:
551 if PY3:
552 def iterkeys(d, **kw):
552 def iterkeys(d, **kw):
553 return iter(d.keys(**kw))
553 return iter(d.keys(**kw))
554
554
555 def itervalues(d, **kw):
555 def itervalues(d, **kw):
556 return iter(d.values(**kw))
556 return iter(d.values(**kw))
557
557
558 def iteritems(d, **kw):
558 def iteritems(d, **kw):
559 return iter(d.items(**kw))
559 return iter(d.items(**kw))
560
560
561 def iterlists(d, **kw):
561 def iterlists(d, **kw):
562 return iter(d.lists(**kw))
562 return iter(d.lists(**kw))
563
563
564 viewkeys = operator.methodcaller("keys")
564 viewkeys = operator.methodcaller("keys")
565
565
566 viewvalues = operator.methodcaller("values")
566 viewvalues = operator.methodcaller("values")
567
567
568 viewitems = operator.methodcaller("items")
568 viewitems = operator.methodcaller("items")
569 else:
569 else:
570 def iterkeys(d, **kw):
570 def iterkeys(d, **kw):
571 return iter(d.iterkeys(**kw))
571 return iter(d.iterkeys(**kw))
572
572
573 def itervalues(d, **kw):
573 def itervalues(d, **kw):
574 return iter(d.itervalues(**kw))
574 return iter(d.itervalues(**kw))
575
575
576 def iteritems(d, **kw):
576 def iteritems(d, **kw):
577 return iter(d.iteritems(**kw))
577 return iter(d.iteritems(**kw))
578
578
579 def iterlists(d, **kw):
579 def iterlists(d, **kw):
580 return iter(d.iterlists(**kw))
580 return iter(d.iterlists(**kw))
581
581
582 viewkeys = operator.methodcaller("viewkeys")
582 viewkeys = operator.methodcaller("viewkeys")
583
583
584 viewvalues = operator.methodcaller("viewvalues")
584 viewvalues = operator.methodcaller("viewvalues")
585
585
586 viewitems = operator.methodcaller("viewitems")
586 viewitems = operator.methodcaller("viewitems")
587
587
588 _add_doc(iterkeys, "Return an iterator over the keys of a dictionary.")
588 _add_doc(iterkeys, "Return an iterator over the keys of a dictionary.")
589 _add_doc(itervalues, "Return an iterator over the values of a dictionary.")
589 _add_doc(itervalues, "Return an iterator over the values of a dictionary.")
590 _add_doc(iteritems,
590 _add_doc(iteritems,
591 "Return an iterator over the (key, value) pairs of a dictionary.")
591 "Return an iterator over the (key, value) pairs of a dictionary.")
592 _add_doc(iterlists,
592 _add_doc(iterlists,
593 "Return an iterator over the (key, [values]) pairs of a dictionary.")
593 "Return an iterator over the (key, [values]) pairs of a dictionary.")
594
594
595
595
596 if PY3:
596 if PY3:
597 def b(s):
597 def b(s):
598 return s.encode("latin-1")
598 return s.encode("latin-1")
599 def u(s):
599 def u(s):
600 return s
600 return s
601 unichr = chr
601 unichr = chr
602 if sys.version_info[1] <= 1:
602 if sys.version_info[1] <= 1:
603 def int2byte(i):
603 def int2byte(i):
604 return bytes((i,))
604 return bytes((i,))
605 else:
605 else:
606 # This is about 2x faster than the implementation above on 3.2+
606 # This is about 2x faster than the implementation above on 3.2+
607 int2byte = operator.methodcaller("to_bytes", 1, "big")
607 int2byte = operator.methodcaller("to_bytes", 1, "big")
608 byte2int = operator.itemgetter(0)
608 byte2int = operator.itemgetter(0)
609 indexbytes = operator.getitem
609 indexbytes = operator.getitem
610 iterbytes = iter
610 iterbytes = iter
611 import io
611 import io
612 StringIO = io.StringIO
612 StringIO = io.StringIO
613 BytesIO = io.BytesIO
613 BytesIO = io.BytesIO
614 _assertCountEqual = "assertCountEqual"
614 _assertCountEqual = "assertCountEqual"
615 _assertRaisesRegex = "assertRaisesRegex"
615 _assertRaisesRegex = "assertRaisesRegex"
616 _assertRegex = "assertRegex"
616 _assertRegex = "assertRegex"
617 else:
617 else:
618 def b(s):
618 def b(s):
619 return s
619 return s
620 # Workaround for standalone backslash
620 # Workaround for standalone backslash
621 def u(s):
621 def u(s):
622 return unicode(s.replace(r'\\', r'\\\\'), "unicode_escape")
622 return unicode(s.replace(r'\\', r'\\\\'), "unicode_escape")
623 unichr = unichr
623 unichr = unichr
624 int2byte = chr
624 int2byte = chr
625 def byte2int(bs):
625 def byte2int(bs):
626 return ord(bs[0])
626 return ord(bs[0])
627 def indexbytes(buf, i):
627 def indexbytes(buf, i):
628 return ord(buf[i])
628 return ord(buf[i])
629 iterbytes = functools.partial(itertools.imap, ord)
629 iterbytes = functools.partial(itertools.imap, ord)
630 from io import StringIO
630 from io import StringIO
631 StringIO = BytesIO = StringIO.StringIO
631 StringIO = BytesIO = StringIO.StringIO
632 _assertCountEqual = "assertItemsEqual"
632 _assertCountEqual = "assertItemsEqual"
633 _assertRaisesRegex = "assertRaisesRegexp"
633 _assertRaisesRegex = "assertRaisesRegexp"
634 _assertRegex = "assertRegexpMatches"
634 _assertRegex = "assertRegexpMatches"
635 _add_doc(b, """Byte literal""")
635 _add_doc(b, """Byte literal""")
636 _add_doc(u, """Text literal""")
636 _add_doc(u, """Text literal""")
637
637
638
638
639 def assertCountEqual(self, *args, **kwargs):
639 def assertCountEqual(self, *args, **kwargs):
640 return getattr(self, _assertCountEqual)(*args, **kwargs)
640 return getattr(self, _assertCountEqual)(*args, **kwargs)
641
641
642
642
643 def assertRaisesRegex(self, *args, **kwargs):
643 def assertRaisesRegex(self, *args, **kwargs):
644 return getattr(self, _assertRaisesRegex)(*args, **kwargs)
644 return getattr(self, _assertRaisesRegex)(*args, **kwargs)
645
645
646
646
647 def assertRegex(self, *args, **kwargs):
647 def assertRegex(self, *args, **kwargs):
648 return getattr(self, _assertRegex)(*args, **kwargs)
648 return getattr(self, _assertRegex)(*args, **kwargs)
649
649
650
650
651 if PY3:
651 if PY3:
652 exec_ = getattr(moves.builtins, "exec")
652 exec_ = getattr(moves.builtins, "exec")
653
653
654
654
655 def reraise(tp, value, tb=None):
655 def reraise(tp, value, tb=None):
656 if value is None:
656 if value is None:
657 value = tp()
657 value = tp()
658 if value.__traceback__ is not tb:
658 if value.__traceback__ is not tb:
659 raise value.with_traceback(tb)
659 raise value.with_traceback(tb)
660 raise value
660 raise value
661
661
662 else:
662 else:
663 def exec_(_code_, _globs_=None, _locs_=None):
663 def exec_(_code_, _globs_=None, _locs_=None):
664 """Execute code in a namespace."""
664 """Execute code in a namespace."""
665 if _globs_ is None:
665 if _globs_ is None:
666 frame = sys._getframe(1)
666 frame = sys._getframe(1)
667 _globs_ = frame.f_globals
667 _globs_ = frame.f_globals
668 if _locs_ is None:
668 if _locs_ is None:
669 _locs_ = frame.f_locals
669 _locs_ = frame.f_locals
670 del frame
670 del frame
671 elif _locs_ is None:
671 elif _locs_ is None:
672 _locs_ = _globs_
672 _locs_ = _globs_
673 exec("""exec _code_ in _globs_, _locs_""")
673 exec("""exec _code_ in _globs_, _locs_""")
674
674
675
675
676 exec_("""def reraise(tp, value, tb=None):
676 exec_("""def reraise(tp, value, tb=None):
677 raise tp, value, tb
677 raise tp, value, tb
678 """)
678 """)
679
679
680
680
681 if sys.version_info[:2] == (3, 2):
681 if sys.version_info[:2] == (3, 2):
682 exec_("""def raise_from(value, from_value):
682 exec_("""def raise_from(value, from_value):
683 if from_value is None:
683 if from_value is None:
684 raise value
684 raise value
685 raise value from from_value
685 raise value from from_value
686 """)
686 """)
687 elif sys.version_info[:2] > (3, 2):
687 elif sys.version_info[:2] > (3, 2):
688 exec_("""def raise_from(value, from_value):
688 exec_("""def raise_from(value, from_value):
689 raise value from from_value
689 raise value from from_value
690 """)
690 """)
691 else:
691 else:
692 def raise_from(value, from_value):
692 def raise_from(value, from_value):
693 raise value
693 raise value
694
694
695
695
696 print_ = getattr(moves.builtins, "print", None)
696 print_ = getattr(moves.builtins, "print", None)
697 if print_ is None:
697 if print_ is None:
698 def print_(*args, **kwargs):
698 def print_(*args, **kwargs):
699 """The new-style print function for Python 2.4 and 2.5."""
699 """The new-style print function for Python 2.4 and 2.5."""
700 fp = kwargs.pop("file", sys.stdout)
700 fp = kwargs.pop("file", sys.stdout)
701 if fp is None:
701 if fp is None:
702 return
702 return
703 def write(data):
703 def write(data):
704 if not isinstance(data, str):
704 if not isinstance(data, str):
705 data = str(data)
705 data = str(data)
706 # If the file has an encoding, encode unicode with it.
706 # If the file has an encoding, encode unicode with it.
707 if (isinstance(fp, file) and
707 if (isinstance(fp, file) and
708 isinstance(data, unicode) and
708 isinstance(data, unicode) and
709 fp.encoding is not None):
709 fp.encoding is not None):
710 errors = getattr(fp, "errors", None)
710 errors = getattr(fp, "errors", None)
711 if errors is None:
711 if errors is None:
712 errors = "strict"
712 errors = "strict"
713 data = data.encode(fp.encoding, errors)
713 data = data.encode(fp.encoding, errors)
714 fp.write(data)
714 fp.write(data)
715 want_unicode = False
715 want_unicode = False
716 sep = kwargs.pop("sep", None)
716 sep = kwargs.pop("sep", None)
717 if sep is not None:
717 if sep is not None:
718 if isinstance(sep, unicode):
718 if isinstance(sep, unicode):
719 want_unicode = True
719 want_unicode = True
720 elif not isinstance(sep, str):
720 elif not isinstance(sep, str):
721 raise TypeError("sep must be None or a string")
721 raise TypeError("sep must be None or a string")
722 end = kwargs.pop("end", None)
722 end = kwargs.pop("end", None)
723 if end is not None:
723 if end is not None:
724 if isinstance(end, unicode):
724 if isinstance(end, unicode):
725 want_unicode = True
725 want_unicode = True
726 elif not isinstance(end, str):
726 elif not isinstance(end, str):
727 raise TypeError("end must be None or a string")
727 raise TypeError("end must be None or a string")
728 if kwargs:
728 if kwargs:
729 raise TypeError("invalid keyword arguments to print()")
729 raise TypeError("invalid keyword arguments to print()")
730 if not want_unicode:
730 if not want_unicode:
731 for arg in args:
731 for arg in args:
732 if isinstance(arg, unicode):
732 if isinstance(arg, unicode):
733 want_unicode = True
733 want_unicode = True
734 break
734 break
735 if want_unicode:
735 if want_unicode:
736 newline = unicode("\n")
736 newline = unicode("\n")
737 space = unicode(" ")
737 space = unicode(" ")
738 else:
738 else:
739 newline = "\n"
739 newline = "\n"
740 space = " "
740 space = " "
741 if sep is None:
741 if sep is None:
742 sep = space
742 sep = space
743 if end is None:
743 if end is None:
744 end = newline
744 end = newline
745 for i, arg in enumerate(args):
745 for i, arg in enumerate(args):
746 if i:
746 if i:
747 write(sep)
747 write(sep)
748 write(arg)
748 write(arg)
749 write(end)
749 write(end)
750 if sys.version_info[:2] < (3, 3):
750 if sys.version_info[:2] < (3, 3):
751 _print = print_
751 _print = print_
752 def print_(*args, **kwargs):
752 def print_(*args, **kwargs):
753 fp = kwargs.get("file", sys.stdout)
753 fp = kwargs.get("file", sys.stdout)
754 flush = kwargs.pop("flush", False)
754 flush = kwargs.pop("flush", False)
755 _print(*args, **kwargs)
755 _print(*args, **kwargs)
756 if flush and fp is not None:
756 if flush and fp is not None:
757 fp.flush()
757 fp.flush()
758
758
759 _add_doc(reraise, """Reraise an exception.""")
759 _add_doc(reraise, """Reraise an exception.""")
760
760
761 if sys.version_info[0:2] < (3, 4):
761 if sys.version_info[0:2] < (3, 4):
762 def wraps(wrapped, assigned=functools.WRAPPER_ASSIGNMENTS,
762 def wraps(wrapped, assigned=functools.WRAPPER_ASSIGNMENTS,
763 updated=functools.WRAPPER_UPDATES):
763 updated=functools.WRAPPER_UPDATES):
764 def wrapper(f):
764 def wrapper(f):
765 f = functools.wraps(wrapped, assigned, updated)(f)
765 f = functools.wraps(wrapped, assigned, updated)(f)
766 f.__wrapped__ = wrapped
766 f.__wrapped__ = wrapped
767 return f
767 return f
768 return wrapper
768 return wrapper
769 else:
769 else:
770 wraps = functools.wraps
770 wraps = functools.wraps
771
771
772 def with_metaclass(meta, *bases):
772 def with_metaclass(meta, *bases):
773 """Create a base class with a metaclass."""
773 """Create a base class with a metaclass."""
774 # This requires a bit of explanation: the basic idea is to make a dummy
774 # This requires a bit of explanation: the basic idea is to make a dummy
775 # metaclass for one level of class instantiation that replaces itself with
775 # metaclass for one level of class instantiation that replaces itself with
776 # the actual metaclass.
776 # the actual metaclass.
777 class metaclass(meta):
777 class metaclass(meta):
778 def __new__(cls, name, this_bases, d):
778 def __new__(cls, name, this_bases, d):
779 return meta(name, bases, d)
779 return meta(name, bases, d)
780 return type.__new__(metaclass, 'temporary_class', (), {})
780 return type.__new__(metaclass, 'temporary_class', (), {})
781
781
782
782
783 def add_metaclass(metaclass):
783 def add_metaclass(metaclass):
784 """Class decorator for creating a class with a metaclass."""
784 """Class decorator for creating a class with a metaclass."""
785 def wrapper(cls):
785 def wrapper(cls):
786 orig_vars = cls.__dict__.copy()
786 orig_vars = cls.__dict__.copy()
787 slots = orig_vars.get('__slots__')
787 slots = orig_vars.get('__slots__')
788 if slots is not None:
788 if slots is not None:
789 if isinstance(slots, str):
789 if isinstance(slots, str):
790 slots = [slots]
790 slots = [slots]
791 for slots_var in slots:
791 for slots_var in slots:
792 orig_vars.pop(slots_var)
792 orig_vars.pop(slots_var)
793 orig_vars.pop('__dict__', None)
793 orig_vars.pop('__dict__', None)
794 orig_vars.pop('__weakref__', None)
794 orig_vars.pop('__weakref__', None)
795 return metaclass(cls.__name__, cls.__bases__, orig_vars)
795 return metaclass(cls.__name__, cls.__bases__, orig_vars)
796 return wrapper
796 return wrapper
797
797
798
798
799 def python_2_unicode_compatible(klass):
799 def python_2_unicode_compatible(klass):
800 """
800 """
801 A decorator that defines __unicode__ and __str__ methods under Python 2.
801 A decorator that defines __unicode__ and __str__ methods under Python 2.
802 Under Python 3 it does nothing.
802 Under Python 3 it does nothing.
803
803
804 To support Python 2 and 3 with a single code base, define a __str__ method
804 To support Python 2 and 3 with a single code base, define a __str__ method
805 returning text and apply this decorator to the class.
805 returning text and apply this decorator to the class.
806 """
806 """
807 if PY2:
807 if PY2:
808 if '__str__' not in klass.__dict__:
808 if '__str__' not in klass.__dict__:
809 raise ValueError("@python_2_unicode_compatible cannot be applied "
809 raise ValueError("@python_2_unicode_compatible cannot be applied "
810 "to %s because it doesn't define __str__()." %
810 "to %s because it doesn't define __str__()." %
811 klass.__name__)
811 klass.__name__)
812 klass.__unicode__ = klass.__str__
812 klass.__unicode__ = klass.__str__
813 klass.__str__ = lambda self: self.__unicode__().encode('utf-8')
813 klass.__str__ = lambda self: self.__unicode__().encode('utf-8')
814 return klass
814 return klass
815
815
816
816
817 # Complete the moves implementation.
817 # Complete the moves implementation.
818 # This code is at the end of this module to speed up module loading.
818 # This code is at the end of this module to speed up module loading.
819 # Turn this module into a package.
819 # Turn this module into a package.
820 __path__ = [] # required for PEP 302 and PEP 451
820 __path__ = [] # required for PEP 302 and PEP 451
821 __package__ = __name__ # see PEP 366 @ReservedAssignment
821 __package__ = __name__ # see PEP 366 @ReservedAssignment
822 if globals().get("__spec__") is not None:
822 if globals().get("__spec__") is not None:
823 __spec__.submodule_search_locations = [] # PEP 451 @UndefinedVariable
823 __spec__.submodule_search_locations = [] # PEP 451 @UndefinedVariable
824 # Remove other six meta path importers, since they cause problems. This can
824 # Remove other six meta path importers, since they cause problems. This can
825 # happen if six is removed from sys.modules and then reloaded. (Setuptools does
825 # happen if six is removed from sys.modules and then reloaded. (Setuptools does
826 # this for some reason.)
826 # this for some reason.)
827 if sys.meta_path:
827 if sys.meta_path:
828 for i, importer in enumerate(sys.meta_path):
828 for i, importer in enumerate(sys.meta_path):
829 # Here's some real nastiness: Another "instance" of the six module might
829 # Here's some real nastiness: Another "instance" of the six module might
830 # be floating around. Therefore, we can't use isinstance() to check for
830 # be floating around. Therefore, we can't use isinstance() to check for
831 # the six meta path importer, since the other six instance will have
831 # the six meta path importer, since the other six instance will have
832 # inserted an importer with different class.
832 # inserted an importer with different class.
833 if (type(importer).__name__ == "_SixMetaPathImporter" and
833 if (type(importer).__name__ == "_SixMetaPathImporter" and
834 importer.name == __name__):
834 importer.name == __name__):
835 del sys.meta_path[i]
835 del sys.meta_path[i]
836 break
836 break
837 del i, importer
837 del i, importer
838 # Finally, add the importer to the meta path import hook.
838 # Finally, add the importer to the meta path import hook.
839 sys.meta_path.append(_importer)
839 sys.meta_path.append(_importer)
@@ -1,2155 +1,2155 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2020 RhodeCode GmbH
3 # Copyright (C) 2010-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 """
21 """
22 Helper functions
22 Helper functions
23
23
24 Consists of functions to typically be used within templates, but also
24 Consists of functions to typically be used within templates, but also
25 available to Controllers. This module is available to both as 'h'.
25 available to Controllers. This module is available to both as 'h'.
26 """
26 """
27 import base64
27 import base64
28 import collections
28 import collections
29
29
30 import os
30 import os
31 import random
31 import random
32 import hashlib
32 import hashlib
33 from io import StringIO
33 from io import StringIO
34 import textwrap
34 import textwrap
35 import urllib.request, urllib.parse, urllib.error
35 import urllib.request, urllib.parse, urllib.error
36 import math
36 import math
37 import logging
37 import logging
38 import re
38 import re
39 import time
39 import time
40 import string
40 import string
41 import hashlib
41 import hashlib
42 import regex
42 import regex
43 from collections import OrderedDict
43 from collections import OrderedDict
44
44
45 import pygments
45 import pygments
46 import itertools
46 import itertools
47 import fnmatch
47 import fnmatch
48 import bleach
48 import bleach
49
49
50 from datetime import datetime
50 from datetime import datetime
51 from functools import partial
51 from functools import partial
52 from pygments.formatters.html import HtmlFormatter
52 from pygments.formatters.html import HtmlFormatter
53 from pygments.lexers import (
53 from pygments.lexers import (
54 get_lexer_by_name, get_lexer_for_filename, get_lexer_for_mimetype)
54 get_lexer_by_name, get_lexer_for_filename, get_lexer_for_mimetype)
55
55
56 from pyramid.threadlocal import get_current_request
56 from pyramid.threadlocal import get_current_request
57 from tempita import looper
57 from tempita import looper
58 from webhelpers2.html import literal, HTML, escape
58 from webhelpers2.html import literal, HTML, escape
59 from webhelpers2.html._autolink import _auto_link_urls
59 from webhelpers2.html._autolink import _auto_link_urls
60 from webhelpers2.html.tools import (
60 from webhelpers2.html.tools import (
61 button_to, highlight, js_obfuscate, strip_links, strip_tags)
61 button_to, highlight, js_obfuscate, strip_links, strip_tags)
62
62
63 from webhelpers2.text import (
63 from webhelpers2.text import (
64 chop_at, collapse, convert_accented_entities,
64 chop_at, collapse, convert_accented_entities,
65 convert_misc_entities, lchop, plural, rchop, remove_formatting,
65 convert_misc_entities, lchop, plural, rchop, remove_formatting,
66 replace_whitespace, urlify, truncate, wrap_paragraphs)
66 replace_whitespace, urlify, truncate, wrap_paragraphs)
67 from webhelpers2.date import time_ago_in_words
67 from webhelpers2.date import time_ago_in_words
68
68
69 from webhelpers2.html.tags import (
69 from webhelpers2.html.tags import (
70 _input, NotGiven, _make_safe_id_component as safeid,
70 _input, NotGiven, _make_safe_id_component as safeid,
71 form as insecure_form,
71 form as insecure_form,
72 auto_discovery_link, checkbox, end_form, file,
72 auto_discovery_link, checkbox, end_form, file,
73 hidden, image, javascript_link, link_to, link_to_if, link_to_unless, ol,
73 hidden, image, javascript_link, link_to, link_to_if, link_to_unless, ol,
74 select as raw_select, stylesheet_link, submit, text, password, textarea,
74 select as raw_select, stylesheet_link, submit, text, password, textarea,
75 ul, radio, Options)
75 ul, radio, Options)
76
76
77 from webhelpers2.number import format_byte_size
77 from webhelpers2.number import format_byte_size
78
78
79 from rhodecode.lib.action_parser import action_parser
79 from rhodecode.lib.action_parser import action_parser
80 from rhodecode.lib.pagination import Page, RepoPage, SqlPage
80 from rhodecode.lib.pagination import Page, RepoPage, SqlPage
81 from rhodecode.lib.ext_json import json
81 from rhodecode.lib.ext_json import json
82 from rhodecode.lib.utils import repo_name_slug, get_custom_lexer
82 from rhodecode.lib.utils import repo_name_slug, get_custom_lexer
83 from rhodecode.lib.utils2 import (
83 from rhodecode.lib.utils2 import (
84 str2bool, safe_unicode, safe_str,
84 str2bool, safe_unicode, safe_str,
85 get_commit_safe, datetime_to_time, time_to_datetime, time_to_utcdatetime,
85 get_commit_safe, datetime_to_time, time_to_datetime, time_to_utcdatetime,
86 AttributeDict, safe_int, md5, md5_safe, get_host_info)
86 AttributeDict, safe_int, md5, md5_safe, get_host_info)
87 from rhodecode.lib.markup_renderer import MarkupRenderer, relative_links
87 from rhodecode.lib.markup_renderer import MarkupRenderer, relative_links
88 from rhodecode.lib.vcs.exceptions import CommitDoesNotExistError
88 from rhodecode.lib.vcs.exceptions import CommitDoesNotExistError
89 from rhodecode.lib.vcs.backends.base import BaseChangeset, EmptyCommit
89 from rhodecode.lib.vcs.backends.base import BaseChangeset, EmptyCommit
90 from rhodecode.lib.vcs.conf.settings import ARCHIVE_SPECS
90 from rhodecode.lib.vcs.conf.settings import ARCHIVE_SPECS
91 from rhodecode.lib.index.search_utils import get_matching_line_offsets
91 from rhodecode.lib.index.search_utils import get_matching_line_offsets
92 from rhodecode.config.conf import DATE_FORMAT, DATETIME_FORMAT
92 from rhodecode.config.conf import DATE_FORMAT, DATETIME_FORMAT
93 from rhodecode.model.changeset_status import ChangesetStatusModel
93 from rhodecode.model.changeset_status import ChangesetStatusModel
94 from rhodecode.model.db import Permission, User, Repository, UserApiKeys, FileStore
94 from rhodecode.model.db import Permission, User, Repository, UserApiKeys, FileStore
95 from rhodecode.model.repo_group import RepoGroupModel
95 from rhodecode.model.repo_group import RepoGroupModel
96 from rhodecode.model.settings import IssueTrackerSettingsModel
96 from rhodecode.model.settings import IssueTrackerSettingsModel
97
97
98
98
99 log = logging.getLogger(__name__)
99 log = logging.getLogger(__name__)
100
100
101
101
102 DEFAULT_USER = User.DEFAULT_USER
102 DEFAULT_USER = User.DEFAULT_USER
103 DEFAULT_USER_EMAIL = User.DEFAULT_USER_EMAIL
103 DEFAULT_USER_EMAIL = User.DEFAULT_USER_EMAIL
104
104
105
105
106 def asset(path, ver=None, **kwargs):
106 def asset(path, ver=None, **kwargs):
107 """
107 """
108 Helper to generate a static asset file path for rhodecode assets
108 Helper to generate a static asset file path for rhodecode assets
109
109
110 eg. h.asset('images/image.png', ver='3923')
110 eg. h.asset('images/image.png', ver='3923')
111
111
112 :param path: path of asset
112 :param path: path of asset
113 :param ver: optional version query param to append as ?ver=
113 :param ver: optional version query param to append as ?ver=
114 """
114 """
115 request = get_current_request()
115 request = get_current_request()
116 query = {}
116 query = {}
117 query.update(kwargs)
117 query.update(kwargs)
118 if ver:
118 if ver:
119 query = {'ver': ver}
119 query = {'ver': ver}
120 return request.static_path(
120 return request.static_path(
121 'rhodecode:public/{}'.format(path), _query=query)
121 'rhodecode:public/{}'.format(path), _query=query)
122
122
123
123
124 default_html_escape_table = {
124 default_html_escape_table = {
125 ord('&'): u'&amp;',
125 ord('&'): u'&amp;',
126 ord('<'): u'&lt;',
126 ord('<'): u'&lt;',
127 ord('>'): u'&gt;',
127 ord('>'): u'&gt;',
128 ord('"'): u'&quot;',
128 ord('"'): u'&quot;',
129 ord("'"): u'&#39;',
129 ord("'"): u'&#39;',
130 }
130 }
131
131
132
132
133 def html_escape(text, html_escape_table=default_html_escape_table):
133 def html_escape(text, html_escape_table=default_html_escape_table):
134 """Produce entities within text."""
134 """Produce entities within text."""
135 return text.translate(html_escape_table)
135 return text.translate(html_escape_table)
136
136
137
137
138 def chop_at_smart(s, sub, inclusive=False, suffix_if_chopped=None):
138 def chop_at_smart(s, sub, inclusive=False, suffix_if_chopped=None):
139 """
139 """
140 Truncate string ``s`` at the first occurrence of ``sub``.
140 Truncate string ``s`` at the first occurrence of ``sub``.
141
141
142 If ``inclusive`` is true, truncate just after ``sub`` rather than at it.
142 If ``inclusive`` is true, truncate just after ``sub`` rather than at it.
143 """
143 """
144 suffix_if_chopped = suffix_if_chopped or ''
144 suffix_if_chopped = suffix_if_chopped or ''
145 pos = s.find(sub)
145 pos = s.find(sub)
146 if pos == -1:
146 if pos == -1:
147 return s
147 return s
148
148
149 if inclusive:
149 if inclusive:
150 pos += len(sub)
150 pos += len(sub)
151
151
152 chopped = s[:pos]
152 chopped = s[:pos]
153 left = s[pos:].strip()
153 left = s[pos:].strip()
154
154
155 if left and suffix_if_chopped:
155 if left and suffix_if_chopped:
156 chopped += suffix_if_chopped
156 chopped += suffix_if_chopped
157
157
158 return chopped
158 return chopped
159
159
160
160
161 def shorter(text, size=20, prefix=False):
161 def shorter(text, size=20, prefix=False):
162 postfix = '...'
162 postfix = '...'
163 if len(text) > size:
163 if len(text) > size:
164 if prefix:
164 if prefix:
165 # shorten in front
165 # shorten in front
166 return postfix + text[-(size - len(postfix)):]
166 return postfix + text[-(size - len(postfix)):]
167 else:
167 else:
168 return text[:size - len(postfix)] + postfix
168 return text[:size - len(postfix)] + postfix
169 return text
169 return text
170
170
171
171
172 def reset(name, value=None, id=NotGiven, type="reset", **attrs):
172 def reset(name, value=None, id=NotGiven, type="reset", **attrs):
173 """
173 """
174 Reset button
174 Reset button
175 """
175 """
176 return _input(type, name, value, id, attrs)
176 return _input(type, name, value, id, attrs)
177
177
178
178
179 def select(name, selected_values, options, id=NotGiven, **attrs):
179 def select(name, selected_values, options, id=NotGiven, **attrs):
180
180
181 if isinstance(options, (list, tuple)):
181 if isinstance(options, (list, tuple)):
182 options_iter = options
182 options_iter = options
183 # Handle old value,label lists ... where value also can be value,label lists
183 # Handle old value,label lists ... where value also can be value,label lists
184 options = Options()
184 options = Options()
185 for opt in options_iter:
185 for opt in options_iter:
186 if isinstance(opt, tuple) and len(opt) == 2:
186 if isinstance(opt, tuple) and len(opt) == 2:
187 value, label = opt
187 value, label = opt
188 elif isinstance(opt, str):
188 elif isinstance(opt, str):
189 value = label = opt
189 value = label = opt
190 else:
190 else:
191 raise ValueError('invalid select option type %r' % type(opt))
191 raise ValueError('invalid select option type %r' % type(opt))
192
192
193 if isinstance(value, (list, tuple)):
193 if isinstance(value, (list, tuple)):
194 option_group = options.add_optgroup(label)
194 option_group = options.add_optgroup(label)
195 for opt2 in value:
195 for opt2 in value:
196 if isinstance(opt2, tuple) and len(opt2) == 2:
196 if isinstance(opt2, tuple) and len(opt2) == 2:
197 group_value, group_label = opt2
197 group_value, group_label = opt2
198 elif isinstance(opt2, str):
198 elif isinstance(opt2, str):
199 group_value = group_label = opt2
199 group_value = group_label = opt2
200 else:
200 else:
201 raise ValueError('invalid select option type %r' % type(opt2))
201 raise ValueError('invalid select option type %r' % type(opt2))
202
202
203 option_group.add_option(group_label, group_value)
203 option_group.add_option(group_label, group_value)
204 else:
204 else:
205 options.add_option(label, value)
205 options.add_option(label, value)
206
206
207 return raw_select(name, selected_values, options, id=id, **attrs)
207 return raw_select(name, selected_values, options, id=id, **attrs)
208
208
209
209
210 def branding(name, length=40):
210 def branding(name, length=40):
211 return truncate(name, length, indicator="")
211 return truncate(name, length, indicator="")
212
212
213
213
214 def FID(raw_id, path):
214 def FID(raw_id, path):
215 """
215 """
216 Creates a unique ID for filenode based on it's hash of path and commit
216 Creates a unique ID for filenode based on it's hash of path and commit
217 it's safe to use in urls
217 it's safe to use in urls
218
218
219 :param raw_id:
219 :param raw_id:
220 :param path:
220 :param path:
221 """
221 """
222
222
223 return 'c-%s-%s' % (short_id(raw_id), md5_safe(path)[:12])
223 return 'c-%s-%s' % (short_id(raw_id), md5_safe(path)[:12])
224
224
225
225
226 class _GetError(object):
226 class _GetError(object):
227 """Get error from form_errors, and represent it as span wrapped error
227 """Get error from form_errors, and represent it as span wrapped error
228 message
228 message
229
229
230 :param field_name: field to fetch errors for
230 :param field_name: field to fetch errors for
231 :param form_errors: form errors dict
231 :param form_errors: form errors dict
232 """
232 """
233
233
234 def __call__(self, field_name, form_errors):
234 def __call__(self, field_name, form_errors):
235 tmpl = """<span class="error_msg">%s</span>"""
235 tmpl = """<span class="error_msg">%s</span>"""
236 if form_errors and field_name in form_errors:
236 if form_errors and field_name in form_errors:
237 return literal(tmpl % form_errors.get(field_name))
237 return literal(tmpl % form_errors.get(field_name))
238
238
239
239
240 get_error = _GetError()
240 get_error = _GetError()
241
241
242
242
243 class _ToolTip(object):
243 class _ToolTip(object):
244
244
245 def __call__(self, tooltip_title, trim_at=50):
245 def __call__(self, tooltip_title, trim_at=50):
246 """
246 """
247 Special function just to wrap our text into nice formatted
247 Special function just to wrap our text into nice formatted
248 autowrapped text
248 autowrapped text
249
249
250 :param tooltip_title:
250 :param tooltip_title:
251 """
251 """
252 tooltip_title = escape(tooltip_title)
252 tooltip_title = escape(tooltip_title)
253 tooltip_title = tooltip_title.replace('<', '&lt;').replace('>', '&gt;')
253 tooltip_title = tooltip_title.replace('<', '&lt;').replace('>', '&gt;')
254 return tooltip_title
254 return tooltip_title
255
255
256
256
257 tooltip = _ToolTip()
257 tooltip = _ToolTip()
258
258
259 files_icon = u'<i class="file-breadcrumb-copy tooltip icon-clipboard clipboard-action" data-clipboard-text="{}" title="Copy file path"></i>'
259 files_icon = u'<i class="file-breadcrumb-copy tooltip icon-clipboard clipboard-action" data-clipboard-text="{}" title="Copy file path"></i>'
260
260
261
261
262 def files_breadcrumbs(repo_name, repo_type, commit_id, file_path, landing_ref_name=None, at_ref=None,
262 def files_breadcrumbs(repo_name, repo_type, commit_id, file_path, landing_ref_name=None, at_ref=None,
263 limit_items=False, linkify_last_item=False, hide_last_item=False,
263 limit_items=False, linkify_last_item=False, hide_last_item=False,
264 copy_path_icon=True):
264 copy_path_icon=True):
265 if isinstance(file_path, str):
265 if isinstance(file_path, str):
266 file_path = safe_unicode(file_path)
266 file_path = safe_unicode(file_path)
267
267
268 if at_ref:
268 if at_ref:
269 route_qry = {'at': at_ref}
269 route_qry = {'at': at_ref}
270 default_landing_ref = at_ref or landing_ref_name or commit_id
270 default_landing_ref = at_ref or landing_ref_name or commit_id
271 else:
271 else:
272 route_qry = None
272 route_qry = None
273 default_landing_ref = commit_id
273 default_landing_ref = commit_id
274
274
275 # first segment is a `HOME` link to repo files root location
275 # first segment is a `HOME` link to repo files root location
276 root_name = literal(u'<i class="icon-home"></i>')
276 root_name = literal(u'<i class="icon-home"></i>')
277
277
278 url_segments = [
278 url_segments = [
279 link_to(
279 link_to(
280 root_name,
280 root_name,
281 repo_files_by_ref_url(
281 repo_files_by_ref_url(
282 repo_name,
282 repo_name,
283 repo_type,
283 repo_type,
284 f_path=None, # None here is a special case for SVN repos,
284 f_path=None, # None here is a special case for SVN repos,
285 # that won't prefix with a ref
285 # that won't prefix with a ref
286 ref_name=default_landing_ref,
286 ref_name=default_landing_ref,
287 commit_id=commit_id,
287 commit_id=commit_id,
288 query=route_qry
288 query=route_qry
289 )
289 )
290 )]
290 )]
291
291
292 path_segments = file_path.split('/')
292 path_segments = file_path.split('/')
293 last_cnt = len(path_segments) - 1
293 last_cnt = len(path_segments) - 1
294 for cnt, segment in enumerate(path_segments):
294 for cnt, segment in enumerate(path_segments):
295 if not segment:
295 if not segment:
296 continue
296 continue
297 segment_html = escape(segment)
297 segment_html = escape(segment)
298
298
299 last_item = cnt == last_cnt
299 last_item = cnt == last_cnt
300
300
301 if last_item and hide_last_item:
301 if last_item and hide_last_item:
302 # iterate over and hide last element
302 # iterate over and hide last element
303 continue
303 continue
304
304
305 if last_item and linkify_last_item is False:
305 if last_item and linkify_last_item is False:
306 # plain version
306 # plain version
307 url_segments.append(segment_html)
307 url_segments.append(segment_html)
308 else:
308 else:
309 url_segments.append(
309 url_segments.append(
310 link_to(
310 link_to(
311 segment_html,
311 segment_html,
312 repo_files_by_ref_url(
312 repo_files_by_ref_url(
313 repo_name,
313 repo_name,
314 repo_type,
314 repo_type,
315 f_path='/'.join(path_segments[:cnt + 1]),
315 f_path='/'.join(path_segments[:cnt + 1]),
316 ref_name=default_landing_ref,
316 ref_name=default_landing_ref,
317 commit_id=commit_id,
317 commit_id=commit_id,
318 query=route_qry
318 query=route_qry
319 ),
319 ),
320 ))
320 ))
321
321
322 limited_url_segments = url_segments[:1] + ['...'] + url_segments[-5:]
322 limited_url_segments = url_segments[:1] + ['...'] + url_segments[-5:]
323 if limit_items and len(limited_url_segments) < len(url_segments):
323 if limit_items and len(limited_url_segments) < len(url_segments):
324 url_segments = limited_url_segments
324 url_segments = limited_url_segments
325
325
326 full_path = file_path
326 full_path = file_path
327 if copy_path_icon:
327 if copy_path_icon:
328 icon = files_icon.format(escape(full_path))
328 icon = files_icon.format(escape(full_path))
329 else:
329 else:
330 icon = ''
330 icon = ''
331
331
332 if file_path == '':
332 if file_path == '':
333 return root_name
333 return root_name
334 else:
334 else:
335 return literal(' / '.join(url_segments) + icon)
335 return literal(' / '.join(url_segments) + icon)
336
336
337
337
338 def files_url_data(request):
338 def files_url_data(request):
339 import urllib.request, urllib.parse, urllib.error
339 import urllib.request, urllib.parse, urllib.error
340 matchdict = request.matchdict
340 matchdict = request.matchdict
341
341
342 if 'f_path' not in matchdict:
342 if 'f_path' not in matchdict:
343 matchdict['f_path'] = ''
343 matchdict['f_path'] = ''
344 else:
344 else:
345 matchdict['f_path'] = urllib.parse.quote(safe_str(matchdict['f_path']))
345 matchdict['f_path'] = urllib.parse.quote(safe_str(matchdict['f_path']))
346 if 'commit_id' not in matchdict:
346 if 'commit_id' not in matchdict:
347 matchdict['commit_id'] = 'tip'
347 matchdict['commit_id'] = 'tip'
348
348
349 return json.dumps(matchdict)
349 return json.dumps(matchdict)
350
350
351
351
352 def repo_files_by_ref_url(db_repo_name, db_repo_type, f_path, ref_name, commit_id, query=None, ):
352 def repo_files_by_ref_url(db_repo_name, db_repo_type, f_path, ref_name, commit_id, query=None, ):
353 _is_svn = is_svn(db_repo_type)
353 _is_svn = is_svn(db_repo_type)
354 final_f_path = f_path
354 final_f_path = f_path
355
355
356 if _is_svn:
356 if _is_svn:
357 """
357 """
358 For SVN the ref_name cannot be used as a commit_id, it needs to be prefixed with
358 For SVN the ref_name cannot be used as a commit_id, it needs to be prefixed with
359 actually commit_id followed by the ref_name. This should be done only in case
359 actually commit_id followed by the ref_name. This should be done only in case
360 This is a initial landing url, without additional paths.
360 This is a initial landing url, without additional paths.
361
361
362 like: /1000/tags/1.0.0/?at=tags/1.0.0
362 like: /1000/tags/1.0.0/?at=tags/1.0.0
363 """
363 """
364
364
365 if ref_name and ref_name != 'tip':
365 if ref_name and ref_name != 'tip':
366 # NOTE(marcink): for svn the ref_name is actually the stored path, so we prefix it
366 # NOTE(marcink): for svn the ref_name is actually the stored path, so we prefix it
367 # for SVN we only do this magic prefix if it's root, .eg landing revision
367 # for SVN we only do this magic prefix if it's root, .eg landing revision
368 # of files link. If we are in the tree we don't need this since we traverse the url
368 # of files link. If we are in the tree we don't need this since we traverse the url
369 # that has everything stored
369 # that has everything stored
370 if f_path in ['', '/']:
370 if f_path in ['', '/']:
371 final_f_path = '/'.join([ref_name, f_path])
371 final_f_path = '/'.join([ref_name, f_path])
372
372
373 # SVN always needs a commit_id explicitly, without a named REF
373 # SVN always needs a commit_id explicitly, without a named REF
374 default_commit_id = commit_id
374 default_commit_id = commit_id
375 else:
375 else:
376 """
376 """
377 For git and mercurial we construct a new URL using the names instead of commit_id
377 For git and mercurial we construct a new URL using the names instead of commit_id
378 like: /master/some_path?at=master
378 like: /master/some_path?at=master
379 """
379 """
380 # We currently do not support branches with slashes
380 # We currently do not support branches with slashes
381 if '/' in ref_name:
381 if '/' in ref_name:
382 default_commit_id = commit_id
382 default_commit_id = commit_id
383 else:
383 else:
384 default_commit_id = ref_name
384 default_commit_id = ref_name
385
385
386 # sometimes we pass f_path as None, to indicate explicit no prefix,
386 # sometimes we pass f_path as None, to indicate explicit no prefix,
387 # we translate it to string to not have None
387 # we translate it to string to not have None
388 final_f_path = final_f_path or ''
388 final_f_path = final_f_path or ''
389
389
390 files_url = route_path(
390 files_url = route_path(
391 'repo_files',
391 'repo_files',
392 repo_name=db_repo_name,
392 repo_name=db_repo_name,
393 commit_id=default_commit_id,
393 commit_id=default_commit_id,
394 f_path=final_f_path,
394 f_path=final_f_path,
395 _query=query
395 _query=query
396 )
396 )
397 return files_url
397 return files_url
398
398
399
399
400 def code_highlight(code, lexer, formatter, use_hl_filter=False):
400 def code_highlight(code, lexer, formatter, use_hl_filter=False):
401 """
401 """
402 Lex ``code`` with ``lexer`` and format it with the formatter ``formatter``.
402 Lex ``code`` with ``lexer`` and format it with the formatter ``formatter``.
403
403
404 If ``outfile`` is given and a valid file object (an object
404 If ``outfile`` is given and a valid file object (an object
405 with a ``write`` method), the result will be written to it, otherwise
405 with a ``write`` method), the result will be written to it, otherwise
406 it is returned as a string.
406 it is returned as a string.
407 """
407 """
408 if use_hl_filter:
408 if use_hl_filter:
409 # add HL filter
409 # add HL filter
410 from rhodecode.lib.index import search_utils
410 from rhodecode.lib.index import search_utils
411 lexer.add_filter(search_utils.ElasticSearchHLFilter())
411 lexer.add_filter(search_utils.ElasticSearchHLFilter())
412 return pygments.format(pygments.lex(code, lexer), formatter)
412 return pygments.format(pygments.lex(code, lexer), formatter)
413
413
414
414
415 class CodeHtmlFormatter(HtmlFormatter):
415 class CodeHtmlFormatter(HtmlFormatter):
416 """
416 """
417 My code Html Formatter for source codes
417 My code Html Formatter for source codes
418 """
418 """
419
419
420 def wrap(self, source, outfile):
420 def wrap(self, source, outfile):
421 return self._wrap_div(self._wrap_pre(self._wrap_code(source)))
421 return self._wrap_div(self._wrap_pre(self._wrap_code(source)))
422
422
423 def _wrap_code(self, source):
423 def _wrap_code(self, source):
424 for cnt, it in enumerate(source):
424 for cnt, it in enumerate(source):
425 i, t = it
425 i, t = it
426 t = '<div id="L%s">%s</div>' % (cnt + 1, t)
426 t = '<div id="L%s">%s</div>' % (cnt + 1, t)
427 yield i, t
427 yield i, t
428
428
429 def _wrap_tablelinenos(self, inner):
429 def _wrap_tablelinenos(self, inner):
430 dummyoutfile = StringIO.StringIO()
430 dummyoutfile = StringIO.StringIO()
431 lncount = 0
431 lncount = 0
432 for t, line in inner:
432 for t, line in inner:
433 if t:
433 if t:
434 lncount += 1
434 lncount += 1
435 dummyoutfile.write(line)
435 dummyoutfile.write(line)
436
436
437 fl = self.linenostart
437 fl = self.linenostart
438 mw = len(str(lncount + fl - 1))
438 mw = len(str(lncount + fl - 1))
439 sp = self.linenospecial
439 sp = self.linenospecial
440 st = self.linenostep
440 st = self.linenostep
441 la = self.lineanchors
441 la = self.lineanchors
442 aln = self.anchorlinenos
442 aln = self.anchorlinenos
443 nocls = self.noclasses
443 nocls = self.noclasses
444 if sp:
444 if sp:
445 lines = []
445 lines = []
446
446
447 for i in range(fl, fl + lncount):
447 for i in range(fl, fl + lncount):
448 if i % st == 0:
448 if i % st == 0:
449 if i % sp == 0:
449 if i % sp == 0:
450 if aln:
450 if aln:
451 lines.append('<a href="#%s%d" class="special">%*d</a>' %
451 lines.append('<a href="#%s%d" class="special">%*d</a>' %
452 (la, i, mw, i))
452 (la, i, mw, i))
453 else:
453 else:
454 lines.append('<span class="special">%*d</span>' % (mw, i))
454 lines.append('<span class="special">%*d</span>' % (mw, i))
455 else:
455 else:
456 if aln:
456 if aln:
457 lines.append('<a href="#%s%d">%*d</a>' % (la, i, mw, i))
457 lines.append('<a href="#%s%d">%*d</a>' % (la, i, mw, i))
458 else:
458 else:
459 lines.append('%*d' % (mw, i))
459 lines.append('%*d' % (mw, i))
460 else:
460 else:
461 lines.append('')
461 lines.append('')
462 ls = '\n'.join(lines)
462 ls = '\n'.join(lines)
463 else:
463 else:
464 lines = []
464 lines = []
465 for i in range(fl, fl + lncount):
465 for i in range(fl, fl + lncount):
466 if i % st == 0:
466 if i % st == 0:
467 if aln:
467 if aln:
468 lines.append('<a href="#%s%d">%*d</a>' % (la, i, mw, i))
468 lines.append('<a href="#%s%d">%*d</a>' % (la, i, mw, i))
469 else:
469 else:
470 lines.append('%*d' % (mw, i))
470 lines.append('%*d' % (mw, i))
471 else:
471 else:
472 lines.append('')
472 lines.append('')
473 ls = '\n'.join(lines)
473 ls = '\n'.join(lines)
474
474
475 # in case you wonder about the seemingly redundant <div> here: since the
475 # in case you wonder about the seemingly redundant <div> here: since the
476 # content in the other cell also is wrapped in a div, some browsers in
476 # content in the other cell also is wrapped in a div, some browsers in
477 # some configurations seem to mess up the formatting...
477 # some configurations seem to mess up the formatting...
478 if nocls:
478 if nocls:
479 yield 0, ('<table class="%stable">' % self.cssclass +
479 yield 0, ('<table class="%stable">' % self.cssclass +
480 '<tr><td><div class="linenodiv" '
480 '<tr><td><div class="linenodiv" '
481 'style="background-color: #f0f0f0; padding-right: 10px">'
481 'style="background-color: #f0f0f0; padding-right: 10px">'
482 '<pre style="line-height: 125%">' +
482 '<pre style="line-height: 125%">' +
483 ls + '</pre></div></td><td id="hlcode" class="code">')
483 ls + '</pre></div></td><td id="hlcode" class="code">')
484 else:
484 else:
485 yield 0, ('<table class="%stable">' % self.cssclass +
485 yield 0, ('<table class="%stable">' % self.cssclass +
486 '<tr><td class="linenos"><div class="linenodiv"><pre>' +
486 '<tr><td class="linenos"><div class="linenodiv"><pre>' +
487 ls + '</pre></div></td><td id="hlcode" class="code">')
487 ls + '</pre></div></td><td id="hlcode" class="code">')
488 yield 0, dummyoutfile.getvalue()
488 yield 0, dummyoutfile.getvalue()
489 yield 0, '</td></tr></table>'
489 yield 0, '</td></tr></table>'
490
490
491
491
492 class SearchContentCodeHtmlFormatter(CodeHtmlFormatter):
492 class SearchContentCodeHtmlFormatter(CodeHtmlFormatter):
493 def __init__(self, **kw):
493 def __init__(self, **kw):
494 # only show these line numbers if set
494 # only show these line numbers if set
495 self.only_lines = kw.pop('only_line_numbers', [])
495 self.only_lines = kw.pop('only_line_numbers', [])
496 self.query_terms = kw.pop('query_terms', [])
496 self.query_terms = kw.pop('query_terms', [])
497 self.max_lines = kw.pop('max_lines', 5)
497 self.max_lines = kw.pop('max_lines', 5)
498 self.line_context = kw.pop('line_context', 3)
498 self.line_context = kw.pop('line_context', 3)
499 self.url = kw.pop('url', None)
499 self.url = kw.pop('url', None)
500
500
501 super(CodeHtmlFormatter, self).__init__(**kw)
501 super(CodeHtmlFormatter, self).__init__(**kw)
502
502
503 def _wrap_code(self, source):
503 def _wrap_code(self, source):
504 for cnt, it in enumerate(source):
504 for cnt, it in enumerate(source):
505 i, t = it
505 i, t = it
506 t = '<pre>%s</pre>' % t
506 t = '<pre>%s</pre>' % t
507 yield i, t
507 yield i, t
508
508
509 def _wrap_tablelinenos(self, inner):
509 def _wrap_tablelinenos(self, inner):
510 yield 0, '<table class="code-highlight %stable">' % self.cssclass
510 yield 0, '<table class="code-highlight %stable">' % self.cssclass
511
511
512 last_shown_line_number = 0
512 last_shown_line_number = 0
513 current_line_number = 1
513 current_line_number = 1
514
514
515 for t, line in inner:
515 for t, line in inner:
516 if not t:
516 if not t:
517 yield t, line
517 yield t, line
518 continue
518 continue
519
519
520 if current_line_number in self.only_lines:
520 if current_line_number in self.only_lines:
521 if last_shown_line_number + 1 != current_line_number:
521 if last_shown_line_number + 1 != current_line_number:
522 yield 0, '<tr>'
522 yield 0, '<tr>'
523 yield 0, '<td class="line">...</td>'
523 yield 0, '<td class="line">...</td>'
524 yield 0, '<td id="hlcode" class="code"></td>'
524 yield 0, '<td id="hlcode" class="code"></td>'
525 yield 0, '</tr>'
525 yield 0, '</tr>'
526
526
527 yield 0, '<tr>'
527 yield 0, '<tr>'
528 if self.url:
528 if self.url:
529 yield 0, '<td class="line"><a href="%s#L%i">%i</a></td>' % (
529 yield 0, '<td class="line"><a href="%s#L%i">%i</a></td>' % (
530 self.url, current_line_number, current_line_number)
530 self.url, current_line_number, current_line_number)
531 else:
531 else:
532 yield 0, '<td class="line"><a href="">%i</a></td>' % (
532 yield 0, '<td class="line"><a href="">%i</a></td>' % (
533 current_line_number)
533 current_line_number)
534 yield 0, '<td id="hlcode" class="code">' + line + '</td>'
534 yield 0, '<td id="hlcode" class="code">' + line + '</td>'
535 yield 0, '</tr>'
535 yield 0, '</tr>'
536
536
537 last_shown_line_number = current_line_number
537 last_shown_line_number = current_line_number
538
538
539 current_line_number += 1
539 current_line_number += 1
540
540
541 yield 0, '</table>'
541 yield 0, '</table>'
542
542
543
543
544 def hsv_to_rgb(h, s, v):
544 def hsv_to_rgb(h, s, v):
545 """ Convert hsv color values to rgb """
545 """ Convert hsv color values to rgb """
546
546
547 if s == 0.0:
547 if s == 0.0:
548 return v, v, v
548 return v, v, v
549 i = int(h * 6.0) # XXX assume int() truncates!
549 i = int(h * 6.0) # XXX assume int() truncates!
550 f = (h * 6.0) - i
550 f = (h * 6.0) - i
551 p = v * (1.0 - s)
551 p = v * (1.0 - s)
552 q = v * (1.0 - s * f)
552 q = v * (1.0 - s * f)
553 t = v * (1.0 - s * (1.0 - f))
553 t = v * (1.0 - s * (1.0 - f))
554 i = i % 6
554 i = i % 6
555 if i == 0:
555 if i == 0:
556 return v, t, p
556 return v, t, p
557 if i == 1:
557 if i == 1:
558 return q, v, p
558 return q, v, p
559 if i == 2:
559 if i == 2:
560 return p, v, t
560 return p, v, t
561 if i == 3:
561 if i == 3:
562 return p, q, v
562 return p, q, v
563 if i == 4:
563 if i == 4:
564 return t, p, v
564 return t, p, v
565 if i == 5:
565 if i == 5:
566 return v, p, q
566 return v, p, q
567
567
568
568
569 def unique_color_generator(n=10000, saturation=0.10, lightness=0.95):
569 def unique_color_generator(n=10000, saturation=0.10, lightness=0.95):
570 """
570 """
571 Generator for getting n of evenly distributed colors using
571 Generator for getting n of evenly distributed colors using
572 hsv color and golden ratio. It always return same order of colors
572 hsv color and golden ratio. It always return same order of colors
573
573
574 :param n: number of colors to generate
574 :param n: number of colors to generate
575 :param saturation: saturation of returned colors
575 :param saturation: saturation of returned colors
576 :param lightness: lightness of returned colors
576 :param lightness: lightness of returned colors
577 :returns: RGB tuple
577 :returns: RGB tuple
578 """
578 """
579
579
580 golden_ratio = 0.618033988749895
580 golden_ratio = 0.618033988749895
581 h = 0.22717784590367374
581 h = 0.22717784590367374
582
582
583 for _ in range(n):
583 for _ in range(n):
584 h += golden_ratio
584 h += golden_ratio
585 h %= 1
585 h %= 1
586 HSV_tuple = [h, saturation, lightness]
586 HSV_tuple = [h, saturation, lightness]
587 RGB_tuple = hsv_to_rgb(*HSV_tuple)
587 RGB_tuple = hsv_to_rgb(*HSV_tuple)
588 yield map(lambda x: str(int(x * 256)), RGB_tuple)
588 yield map(lambda x: str(int(x * 256)), RGB_tuple)
589
589
590
590
591 def color_hasher(n=10000, saturation=0.10, lightness=0.95):
591 def color_hasher(n=10000, saturation=0.10, lightness=0.95):
592 """
592 """
593 Returns a function which when called with an argument returns a unique
593 Returns a function which when called with an argument returns a unique
594 color for that argument, eg.
594 color for that argument, eg.
595
595
596 :param n: number of colors to generate
596 :param n: number of colors to generate
597 :param saturation: saturation of returned colors
597 :param saturation: saturation of returned colors
598 :param lightness: lightness of returned colors
598 :param lightness: lightness of returned colors
599 :returns: css RGB string
599 :returns: css RGB string
600
600
601 >>> color_hash = color_hasher()
601 >>> color_hash = color_hasher()
602 >>> color_hash('hello')
602 >>> color_hash('hello')
603 'rgb(34, 12, 59)'
603 'rgb(34, 12, 59)'
604 >>> color_hash('hello')
604 >>> color_hash('hello')
605 'rgb(34, 12, 59)'
605 'rgb(34, 12, 59)'
606 >>> color_hash('other')
606 >>> color_hash('other')
607 'rgb(90, 224, 159)'
607 'rgb(90, 224, 159)'
608 """
608 """
609
609
610 color_dict = {}
610 color_dict = {}
611 cgenerator = unique_color_generator(
611 cgenerator = unique_color_generator(
612 saturation=saturation, lightness=lightness)
612 saturation=saturation, lightness=lightness)
613
613
614 def get_color_string(thing):
614 def get_color_string(thing):
615 if thing in color_dict:
615 if thing in color_dict:
616 col = color_dict[thing]
616 col = color_dict[thing]
617 else:
617 else:
618 col = color_dict[thing] = cgenerator.next()
618 col = color_dict[thing] = next(cgenerator)
619 return "rgb(%s)" % (', '.join(col))
619 return "rgb(%s)" % (', '.join(col))
620
620
621 return get_color_string
621 return get_color_string
622
622
623
623
624 def get_lexer_safe(mimetype=None, filepath=None):
624 def get_lexer_safe(mimetype=None, filepath=None):
625 """
625 """
626 Tries to return a relevant pygments lexer using mimetype/filepath name,
626 Tries to return a relevant pygments lexer using mimetype/filepath name,
627 defaulting to plain text if none could be found
627 defaulting to plain text if none could be found
628 """
628 """
629 lexer = None
629 lexer = None
630 try:
630 try:
631 if mimetype:
631 if mimetype:
632 lexer = get_lexer_for_mimetype(mimetype)
632 lexer = get_lexer_for_mimetype(mimetype)
633 if not lexer:
633 if not lexer:
634 lexer = get_lexer_for_filename(filepath)
634 lexer = get_lexer_for_filename(filepath)
635 except pygments.util.ClassNotFound:
635 except pygments.util.ClassNotFound:
636 pass
636 pass
637
637
638 if not lexer:
638 if not lexer:
639 lexer = get_lexer_by_name('text')
639 lexer = get_lexer_by_name('text')
640
640
641 return lexer
641 return lexer
642
642
643
643
644 def get_lexer_for_filenode(filenode):
644 def get_lexer_for_filenode(filenode):
645 lexer = get_custom_lexer(filenode.extension) or filenode.lexer
645 lexer = get_custom_lexer(filenode.extension) or filenode.lexer
646 return lexer
646 return lexer
647
647
648
648
649 def pygmentize(filenode, **kwargs):
649 def pygmentize(filenode, **kwargs):
650 """
650 """
651 pygmentize function using pygments
651 pygmentize function using pygments
652
652
653 :param filenode:
653 :param filenode:
654 """
654 """
655 lexer = get_lexer_for_filenode(filenode)
655 lexer = get_lexer_for_filenode(filenode)
656 return literal(code_highlight(filenode.content, lexer,
656 return literal(code_highlight(filenode.content, lexer,
657 CodeHtmlFormatter(**kwargs)))
657 CodeHtmlFormatter(**kwargs)))
658
658
659
659
660 def is_following_repo(repo_name, user_id):
660 def is_following_repo(repo_name, user_id):
661 from rhodecode.model.scm import ScmModel
661 from rhodecode.model.scm import ScmModel
662 return ScmModel().is_following_repo(repo_name, user_id)
662 return ScmModel().is_following_repo(repo_name, user_id)
663
663
664
664
665 class _Message(object):
665 class _Message(object):
666 """A message returned by ``Flash.pop_messages()``.
666 """A message returned by ``Flash.pop_messages()``.
667
667
668 Converting the message to a string returns the message text. Instances
668 Converting the message to a string returns the message text. Instances
669 also have the following attributes:
669 also have the following attributes:
670
670
671 * ``message``: the message text.
671 * ``message``: the message text.
672 * ``category``: the category specified when the message was created.
672 * ``category``: the category specified when the message was created.
673 """
673 """
674
674
675 def __init__(self, category, message, sub_data=None):
675 def __init__(self, category, message, sub_data=None):
676 self.category = category
676 self.category = category
677 self.message = message
677 self.message = message
678 self.sub_data = sub_data or {}
678 self.sub_data = sub_data or {}
679
679
680 def __str__(self):
680 def __str__(self):
681 return self.message
681 return self.message
682
682
683 __unicode__ = __str__
683 __unicode__ = __str__
684
684
685 def __html__(self):
685 def __html__(self):
686 return escape(safe_unicode(self.message))
686 return escape(safe_unicode(self.message))
687
687
688
688
689 class Flash(object):
689 class Flash(object):
690 # List of allowed categories. If None, allow any category.
690 # List of allowed categories. If None, allow any category.
691 categories = ["warning", "notice", "error", "success"]
691 categories = ["warning", "notice", "error", "success"]
692
692
693 # Default category if none is specified.
693 # Default category if none is specified.
694 default_category = "notice"
694 default_category = "notice"
695
695
696 def __init__(self, session_key="flash", categories=None,
696 def __init__(self, session_key="flash", categories=None,
697 default_category=None):
697 default_category=None):
698 """
698 """
699 Instantiate a ``Flash`` object.
699 Instantiate a ``Flash`` object.
700
700
701 ``session_key`` is the key to save the messages under in the user's
701 ``session_key`` is the key to save the messages under in the user's
702 session.
702 session.
703
703
704 ``categories`` is an optional list which overrides the default list
704 ``categories`` is an optional list which overrides the default list
705 of categories.
705 of categories.
706
706
707 ``default_category`` overrides the default category used for messages
707 ``default_category`` overrides the default category used for messages
708 when none is specified.
708 when none is specified.
709 """
709 """
710 self.session_key = session_key
710 self.session_key = session_key
711 if categories is not None:
711 if categories is not None:
712 self.categories = categories
712 self.categories = categories
713 if default_category is not None:
713 if default_category is not None:
714 self.default_category = default_category
714 self.default_category = default_category
715 if self.categories and self.default_category not in self.categories:
715 if self.categories and self.default_category not in self.categories:
716 raise ValueError(
716 raise ValueError(
717 "unrecognized default category %r" % (self.default_category,))
717 "unrecognized default category %r" % (self.default_category,))
718
718
719 def pop_messages(self, session=None, request=None):
719 def pop_messages(self, session=None, request=None):
720 """
720 """
721 Return all accumulated messages and delete them from the session.
721 Return all accumulated messages and delete them from the session.
722
722
723 The return value is a list of ``Message`` objects.
723 The return value is a list of ``Message`` objects.
724 """
724 """
725 messages = []
725 messages = []
726
726
727 if not session:
727 if not session:
728 if not request:
728 if not request:
729 request = get_current_request()
729 request = get_current_request()
730 session = request.session
730 session = request.session
731
731
732 # Pop the 'old' pylons flash messages. They are tuples of the form
732 # Pop the 'old' pylons flash messages. They are tuples of the form
733 # (category, message)
733 # (category, message)
734 for cat, msg in session.pop(self.session_key, []):
734 for cat, msg in session.pop(self.session_key, []):
735 messages.append(_Message(cat, msg))
735 messages.append(_Message(cat, msg))
736
736
737 # Pop the 'new' pyramid flash messages for each category as list
737 # Pop the 'new' pyramid flash messages for each category as list
738 # of strings.
738 # of strings.
739 for cat in self.categories:
739 for cat in self.categories:
740 for msg in session.pop_flash(queue=cat):
740 for msg in session.pop_flash(queue=cat):
741 sub_data = {}
741 sub_data = {}
742 if hasattr(msg, 'rsplit'):
742 if hasattr(msg, 'rsplit'):
743 flash_data = msg.rsplit('|DELIM|', 1)
743 flash_data = msg.rsplit('|DELIM|', 1)
744 org_message = flash_data[0]
744 org_message = flash_data[0]
745 if len(flash_data) > 1:
745 if len(flash_data) > 1:
746 sub_data = json.loads(flash_data[1])
746 sub_data = json.loads(flash_data[1])
747 else:
747 else:
748 org_message = msg
748 org_message = msg
749
749
750 messages.append(_Message(cat, org_message, sub_data=sub_data))
750 messages.append(_Message(cat, org_message, sub_data=sub_data))
751
751
752 # Map messages from the default queue to the 'notice' category.
752 # Map messages from the default queue to the 'notice' category.
753 for msg in session.pop_flash():
753 for msg in session.pop_flash():
754 messages.append(_Message('notice', msg))
754 messages.append(_Message('notice', msg))
755
755
756 session.save()
756 session.save()
757 return messages
757 return messages
758
758
759 def json_alerts(self, session=None, request=None):
759 def json_alerts(self, session=None, request=None):
760 payloads = []
760 payloads = []
761 messages = flash.pop_messages(session=session, request=request) or []
761 messages = flash.pop_messages(session=session, request=request) or []
762 for message in messages:
762 for message in messages:
763 payloads.append({
763 payloads.append({
764 'message': {
764 'message': {
765 'message': u'{}'.format(message.message),
765 'message': u'{}'.format(message.message),
766 'level': message.category,
766 'level': message.category,
767 'force': True,
767 'force': True,
768 'subdata': message.sub_data
768 'subdata': message.sub_data
769 }
769 }
770 })
770 })
771 return json.dumps(payloads)
771 return json.dumps(payloads)
772
772
773 def __call__(self, message, category=None, ignore_duplicate=True,
773 def __call__(self, message, category=None, ignore_duplicate=True,
774 session=None, request=None):
774 session=None, request=None):
775
775
776 if not session:
776 if not session:
777 if not request:
777 if not request:
778 request = get_current_request()
778 request = get_current_request()
779 session = request.session
779 session = request.session
780
780
781 session.flash(
781 session.flash(
782 message, queue=category, allow_duplicate=not ignore_duplicate)
782 message, queue=category, allow_duplicate=not ignore_duplicate)
783
783
784
784
785 flash = Flash()
785 flash = Flash()
786
786
787 #==============================================================================
787 #==============================================================================
788 # SCM FILTERS available via h.
788 # SCM FILTERS available via h.
789 #==============================================================================
789 #==============================================================================
790 from rhodecode.lib.vcs.utils import author_name, author_email
790 from rhodecode.lib.vcs.utils import author_name, author_email
791 from rhodecode.lib.utils2 import age, age_from_seconds
791 from rhodecode.lib.utils2 import age, age_from_seconds
792 from rhodecode.model.db import User, ChangesetStatus
792 from rhodecode.model.db import User, ChangesetStatus
793
793
794
794
795 email = author_email
795 email = author_email
796
796
797
797
798 def capitalize(raw_text):
798 def capitalize(raw_text):
799 return raw_text.capitalize()
799 return raw_text.capitalize()
800
800
801
801
802 def short_id(long_id):
802 def short_id(long_id):
803 return long_id[:12]
803 return long_id[:12]
804
804
805
805
806 def hide_credentials(url):
806 def hide_credentials(url):
807 from rhodecode.lib.utils2 import credentials_filter
807 from rhodecode.lib.utils2 import credentials_filter
808 return credentials_filter(url)
808 return credentials_filter(url)
809
809
810
810
811 import pytz
811 import pytz
812 import tzlocal
812 import tzlocal
813 local_timezone = tzlocal.get_localzone()
813 local_timezone = tzlocal.get_localzone()
814
814
815
815
816 def get_timezone(datetime_iso, time_is_local=False):
816 def get_timezone(datetime_iso, time_is_local=False):
817 tzinfo = '+00:00'
817 tzinfo = '+00:00'
818
818
819 # detect if we have a timezone info, otherwise, add it
819 # detect if we have a timezone info, otherwise, add it
820 if time_is_local and isinstance(datetime_iso, datetime) and not datetime_iso.tzinfo:
820 if time_is_local and isinstance(datetime_iso, datetime) and not datetime_iso.tzinfo:
821 force_timezone = os.environ.get('RC_TIMEZONE', '')
821 force_timezone = os.environ.get('RC_TIMEZONE', '')
822 if force_timezone:
822 if force_timezone:
823 force_timezone = pytz.timezone(force_timezone)
823 force_timezone = pytz.timezone(force_timezone)
824 timezone = force_timezone or local_timezone
824 timezone = force_timezone or local_timezone
825 offset = timezone.localize(datetime_iso).strftime('%z')
825 offset = timezone.localize(datetime_iso).strftime('%z')
826 tzinfo = '{}:{}'.format(offset[:-2], offset[-2:])
826 tzinfo = '{}:{}'.format(offset[:-2], offset[-2:])
827 return tzinfo
827 return tzinfo
828
828
829
829
830 def age_component(datetime_iso, value=None, time_is_local=False, tooltip=True):
830 def age_component(datetime_iso, value=None, time_is_local=False, tooltip=True):
831 title = value or format_date(datetime_iso)
831 title = value or format_date(datetime_iso)
832 tzinfo = get_timezone(datetime_iso, time_is_local=time_is_local)
832 tzinfo = get_timezone(datetime_iso, time_is_local=time_is_local)
833
833
834 return literal(
834 return literal(
835 '<time class="timeago {cls}" title="{tt_title}" datetime="{dt}{tzinfo}">{title}</time>'.format(
835 '<time class="timeago {cls}" title="{tt_title}" datetime="{dt}{tzinfo}">{title}</time>'.format(
836 cls='tooltip' if tooltip else '',
836 cls='tooltip' if tooltip else '',
837 tt_title=('{title}{tzinfo}'.format(title=title, tzinfo=tzinfo)) if tooltip else '',
837 tt_title=('{title}{tzinfo}'.format(title=title, tzinfo=tzinfo)) if tooltip else '',
838 title=title, dt=datetime_iso, tzinfo=tzinfo
838 title=title, dt=datetime_iso, tzinfo=tzinfo
839 ))
839 ))
840
840
841
841
842 def _shorten_commit_id(commit_id, commit_len=None):
842 def _shorten_commit_id(commit_id, commit_len=None):
843 if commit_len is None:
843 if commit_len is None:
844 request = get_current_request()
844 request = get_current_request()
845 commit_len = request.call_context.visual.show_sha_length
845 commit_len = request.call_context.visual.show_sha_length
846 return commit_id[:commit_len]
846 return commit_id[:commit_len]
847
847
848
848
849 def show_id(commit, show_idx=None, commit_len=None):
849 def show_id(commit, show_idx=None, commit_len=None):
850 """
850 """
851 Configurable function that shows ID
851 Configurable function that shows ID
852 by default it's r123:fffeeefffeee
852 by default it's r123:fffeeefffeee
853
853
854 :param commit: commit instance
854 :param commit: commit instance
855 """
855 """
856 if show_idx is None:
856 if show_idx is None:
857 request = get_current_request()
857 request = get_current_request()
858 show_idx = request.call_context.visual.show_revision_number
858 show_idx = request.call_context.visual.show_revision_number
859
859
860 raw_id = _shorten_commit_id(commit.raw_id, commit_len=commit_len)
860 raw_id = _shorten_commit_id(commit.raw_id, commit_len=commit_len)
861 if show_idx:
861 if show_idx:
862 return 'r%s:%s' % (commit.idx, raw_id)
862 return 'r%s:%s' % (commit.idx, raw_id)
863 else:
863 else:
864 return '%s' % (raw_id, )
864 return '%s' % (raw_id, )
865
865
866
866
867 def format_date(date):
867 def format_date(date):
868 """
868 """
869 use a standardized formatting for dates used in RhodeCode
869 use a standardized formatting for dates used in RhodeCode
870
870
871 :param date: date/datetime object
871 :param date: date/datetime object
872 :return: formatted date
872 :return: formatted date
873 """
873 """
874
874
875 if date:
875 if date:
876 _fmt = "%a, %d %b %Y %H:%M:%S"
876 _fmt = "%a, %d %b %Y %H:%M:%S"
877 return safe_unicode(date.strftime(_fmt))
877 return safe_unicode(date.strftime(_fmt))
878
878
879 return u""
879 return u""
880
880
881
881
882 class _RepoChecker(object):
882 class _RepoChecker(object):
883
883
884 def __init__(self, backend_alias):
884 def __init__(self, backend_alias):
885 self._backend_alias = backend_alias
885 self._backend_alias = backend_alias
886
886
887 def __call__(self, repository):
887 def __call__(self, repository):
888 if hasattr(repository, 'alias'):
888 if hasattr(repository, 'alias'):
889 _type = repository.alias
889 _type = repository.alias
890 elif hasattr(repository, 'repo_type'):
890 elif hasattr(repository, 'repo_type'):
891 _type = repository.repo_type
891 _type = repository.repo_type
892 else:
892 else:
893 _type = repository
893 _type = repository
894 return _type == self._backend_alias
894 return _type == self._backend_alias
895
895
896
896
897 is_git = _RepoChecker('git')
897 is_git = _RepoChecker('git')
898 is_hg = _RepoChecker('hg')
898 is_hg = _RepoChecker('hg')
899 is_svn = _RepoChecker('svn')
899 is_svn = _RepoChecker('svn')
900
900
901
901
902 def get_repo_type_by_name(repo_name):
902 def get_repo_type_by_name(repo_name):
903 repo = Repository.get_by_repo_name(repo_name)
903 repo = Repository.get_by_repo_name(repo_name)
904 if repo:
904 if repo:
905 return repo.repo_type
905 return repo.repo_type
906
906
907
907
908 def is_svn_without_proxy(repository):
908 def is_svn_without_proxy(repository):
909 if is_svn(repository):
909 if is_svn(repository):
910 from rhodecode.model.settings import VcsSettingsModel
910 from rhodecode.model.settings import VcsSettingsModel
911 conf = VcsSettingsModel().get_ui_settings_as_config_obj()
911 conf = VcsSettingsModel().get_ui_settings_as_config_obj()
912 return not str2bool(conf.get('vcs_svn_proxy', 'http_requests_enabled'))
912 return not str2bool(conf.get('vcs_svn_proxy', 'http_requests_enabled'))
913 return False
913 return False
914
914
915
915
916 def discover_user(author):
916 def discover_user(author):
917 """
917 """
918 Tries to discover RhodeCode User based on the author string. Author string
918 Tries to discover RhodeCode User based on the author string. Author string
919 is typically `FirstName LastName <email@address.com>`
919 is typically `FirstName LastName <email@address.com>`
920 """
920 """
921
921
922 # if author is already an instance use it for extraction
922 # if author is already an instance use it for extraction
923 if isinstance(author, User):
923 if isinstance(author, User):
924 return author
924 return author
925
925
926 # Valid email in the attribute passed, see if they're in the system
926 # Valid email in the attribute passed, see if they're in the system
927 _email = author_email(author)
927 _email = author_email(author)
928 if _email != '':
928 if _email != '':
929 user = User.get_by_email(_email, case_insensitive=True, cache=True)
929 user = User.get_by_email(_email, case_insensitive=True, cache=True)
930 if user is not None:
930 if user is not None:
931 return user
931 return user
932
932
933 # Maybe it's a username, we try to extract it and fetch by username ?
933 # Maybe it's a username, we try to extract it and fetch by username ?
934 _author = author_name(author)
934 _author = author_name(author)
935 user = User.get_by_username(_author, case_insensitive=True, cache=True)
935 user = User.get_by_username(_author, case_insensitive=True, cache=True)
936 if user is not None:
936 if user is not None:
937 return user
937 return user
938
938
939 return None
939 return None
940
940
941
941
942 def email_or_none(author):
942 def email_or_none(author):
943 # extract email from the commit string
943 # extract email from the commit string
944 _email = author_email(author)
944 _email = author_email(author)
945
945
946 # If we have an email, use it, otherwise
946 # If we have an email, use it, otherwise
947 # see if it contains a username we can get an email from
947 # see if it contains a username we can get an email from
948 if _email != '':
948 if _email != '':
949 return _email
949 return _email
950 else:
950 else:
951 user = User.get_by_username(
951 user = User.get_by_username(
952 author_name(author), case_insensitive=True, cache=True)
952 author_name(author), case_insensitive=True, cache=True)
953
953
954 if user is not None:
954 if user is not None:
955 return user.email
955 return user.email
956
956
957 # No valid email, not a valid user in the system, none!
957 # No valid email, not a valid user in the system, none!
958 return None
958 return None
959
959
960
960
961 def link_to_user(author, length=0, **kwargs):
961 def link_to_user(author, length=0, **kwargs):
962 user = discover_user(author)
962 user = discover_user(author)
963 # user can be None, but if we have it already it means we can re-use it
963 # user can be None, but if we have it already it means we can re-use it
964 # in the person() function, so we save 1 intensive-query
964 # in the person() function, so we save 1 intensive-query
965 if user:
965 if user:
966 author = user
966 author = user
967
967
968 display_person = person(author, 'username_or_name_or_email')
968 display_person = person(author, 'username_or_name_or_email')
969 if length:
969 if length:
970 display_person = shorter(display_person, length)
970 display_person = shorter(display_person, length)
971
971
972 if user and user.username != user.DEFAULT_USER:
972 if user and user.username != user.DEFAULT_USER:
973 return link_to(
973 return link_to(
974 escape(display_person),
974 escape(display_person),
975 route_path('user_profile', username=user.username),
975 route_path('user_profile', username=user.username),
976 **kwargs)
976 **kwargs)
977 else:
977 else:
978 return escape(display_person)
978 return escape(display_person)
979
979
980
980
981 def link_to_group(users_group_name, **kwargs):
981 def link_to_group(users_group_name, **kwargs):
982 return link_to(
982 return link_to(
983 escape(users_group_name),
983 escape(users_group_name),
984 route_path('user_group_profile', user_group_name=users_group_name),
984 route_path('user_group_profile', user_group_name=users_group_name),
985 **kwargs)
985 **kwargs)
986
986
987
987
988 def person(author, show_attr="username_and_name"):
988 def person(author, show_attr="username_and_name"):
989 user = discover_user(author)
989 user = discover_user(author)
990 if user:
990 if user:
991 return getattr(user, show_attr)
991 return getattr(user, show_attr)
992 else:
992 else:
993 _author = author_name(author)
993 _author = author_name(author)
994 _email = email(author)
994 _email = email(author)
995 return _author or _email
995 return _author or _email
996
996
997
997
998 def author_string(email):
998 def author_string(email):
999 if email:
999 if email:
1000 user = User.get_by_email(email, case_insensitive=True, cache=True)
1000 user = User.get_by_email(email, case_insensitive=True, cache=True)
1001 if user:
1001 if user:
1002 if user.first_name or user.last_name:
1002 if user.first_name or user.last_name:
1003 return '%s %s &lt;%s&gt;' % (
1003 return '%s %s &lt;%s&gt;' % (
1004 user.first_name, user.last_name, email)
1004 user.first_name, user.last_name, email)
1005 else:
1005 else:
1006 return email
1006 return email
1007 else:
1007 else:
1008 return email
1008 return email
1009 else:
1009 else:
1010 return None
1010 return None
1011
1011
1012
1012
1013 def person_by_id(id_, show_attr="username_and_name"):
1013 def person_by_id(id_, show_attr="username_and_name"):
1014 # attr to return from fetched user
1014 # attr to return from fetched user
1015 person_getter = lambda usr: getattr(usr, show_attr)
1015 person_getter = lambda usr: getattr(usr, show_attr)
1016
1016
1017 #maybe it's an ID ?
1017 #maybe it's an ID ?
1018 if str(id_).isdigit() or isinstance(id_, int):
1018 if str(id_).isdigit() or isinstance(id_, int):
1019 id_ = int(id_)
1019 id_ = int(id_)
1020 user = User.get(id_)
1020 user = User.get(id_)
1021 if user is not None:
1021 if user is not None:
1022 return person_getter(user)
1022 return person_getter(user)
1023 return id_
1023 return id_
1024
1024
1025
1025
1026 def gravatar_with_user(request, author, show_disabled=False, tooltip=False):
1026 def gravatar_with_user(request, author, show_disabled=False, tooltip=False):
1027 _render = request.get_partial_renderer('rhodecode:templates/base/base.mako')
1027 _render = request.get_partial_renderer('rhodecode:templates/base/base.mako')
1028 return _render('gravatar_with_user', author, show_disabled=show_disabled, tooltip=tooltip)
1028 return _render('gravatar_with_user', author, show_disabled=show_disabled, tooltip=tooltip)
1029
1029
1030
1030
1031 tags_paterns = OrderedDict((
1031 tags_paterns = OrderedDict((
1032 ('lang', (re.compile(r'\[(lang|language)\ \=\&gt;\ *([a-zA-Z\-\/\#\+\.]*)\]'),
1032 ('lang', (re.compile(r'\[(lang|language)\ \=\&gt;\ *([a-zA-Z\-\/\#\+\.]*)\]'),
1033 '<div class="metatag" tag="lang">\\2</div>')),
1033 '<div class="metatag" tag="lang">\\2</div>')),
1034
1034
1035 ('see', (re.compile(r'\[see\ \=\&gt;\ *([a-zA-Z0-9\/\=\?\&amp;\ \:\/\.\-]*)\]'),
1035 ('see', (re.compile(r'\[see\ \=\&gt;\ *([a-zA-Z0-9\/\=\?\&amp;\ \:\/\.\-]*)\]'),
1036 '<div class="metatag" tag="see">see: \\1 </div>')),
1036 '<div class="metatag" tag="see">see: \\1 </div>')),
1037
1037
1038 ('url', (re.compile(r'\[url\ \=\&gt;\ \[([a-zA-Z0-9\ \.\-\_]+)\]\((http://|https://|/)(.*?)\)\]'),
1038 ('url', (re.compile(r'\[url\ \=\&gt;\ \[([a-zA-Z0-9\ \.\-\_]+)\]\((http://|https://|/)(.*?)\)\]'),
1039 '<div class="metatag" tag="url"> <a href="\\2\\3">\\1</a> </div>')),
1039 '<div class="metatag" tag="url"> <a href="\\2\\3">\\1</a> </div>')),
1040
1040
1041 ('license', (re.compile(r'\[license\ \=\&gt;\ *([a-zA-Z0-9\/\=\?\&amp;\ \:\/\.\-]*)\]'),
1041 ('license', (re.compile(r'\[license\ \=\&gt;\ *([a-zA-Z0-9\/\=\?\&amp;\ \:\/\.\-]*)\]'),
1042 '<div class="metatag" tag="license"><a href="http:\/\/www.opensource.org/licenses/\\1">\\1</a></div>')),
1042 '<div class="metatag" tag="license"><a href="http:\/\/www.opensource.org/licenses/\\1">\\1</a></div>')),
1043
1043
1044 ('ref', (re.compile(r'\[(requires|recommends|conflicts|base)\ \=\&gt;\ *([a-zA-Z0-9\-\/]*)\]'),
1044 ('ref', (re.compile(r'\[(requires|recommends|conflicts|base)\ \=\&gt;\ *([a-zA-Z0-9\-\/]*)\]'),
1045 '<div class="metatag" tag="ref \\1">\\1: <a href="/\\2">\\2</a></div>')),
1045 '<div class="metatag" tag="ref \\1">\\1: <a href="/\\2">\\2</a></div>')),
1046
1046
1047 ('state', (re.compile(r'\[(stable|featured|stale|dead|dev|deprecated)\]'),
1047 ('state', (re.compile(r'\[(stable|featured|stale|dead|dev|deprecated)\]'),
1048 '<div class="metatag" tag="state \\1">\\1</div>')),
1048 '<div class="metatag" tag="state \\1">\\1</div>')),
1049
1049
1050 # label in grey
1050 # label in grey
1051 ('label', (re.compile(r'\[([a-z]+)\]'),
1051 ('label', (re.compile(r'\[([a-z]+)\]'),
1052 '<div class="metatag" tag="label">\\1</div>')),
1052 '<div class="metatag" tag="label">\\1</div>')),
1053
1053
1054 # generic catch all in grey
1054 # generic catch all in grey
1055 ('generic', (re.compile(r'\[([a-zA-Z0-9\.\-\_]+)\]'),
1055 ('generic', (re.compile(r'\[([a-zA-Z0-9\.\-\_]+)\]'),
1056 '<div class="metatag" tag="generic">\\1</div>')),
1056 '<div class="metatag" tag="generic">\\1</div>')),
1057 ))
1057 ))
1058
1058
1059
1059
1060 def extract_metatags(value):
1060 def extract_metatags(value):
1061 """
1061 """
1062 Extract supported meta-tags from given text value
1062 Extract supported meta-tags from given text value
1063 """
1063 """
1064 tags = []
1064 tags = []
1065 if not value:
1065 if not value:
1066 return tags, ''
1066 return tags, ''
1067
1067
1068 for key, val in tags_paterns.items():
1068 for key, val in tags_paterns.items():
1069 pat, replace_html = val
1069 pat, replace_html = val
1070 tags.extend([(key, x.group()) for x in pat.finditer(value)])
1070 tags.extend([(key, x.group()) for x in pat.finditer(value)])
1071 value = pat.sub('', value)
1071 value = pat.sub('', value)
1072
1072
1073 return tags, value
1073 return tags, value
1074
1074
1075
1075
1076 def style_metatag(tag_type, value):
1076 def style_metatag(tag_type, value):
1077 """
1077 """
1078 converts tags from value into html equivalent
1078 converts tags from value into html equivalent
1079 """
1079 """
1080 if not value:
1080 if not value:
1081 return ''
1081 return ''
1082
1082
1083 html_value = value
1083 html_value = value
1084 tag_data = tags_paterns.get(tag_type)
1084 tag_data = tags_paterns.get(tag_type)
1085 if tag_data:
1085 if tag_data:
1086 pat, replace_html = tag_data
1086 pat, replace_html = tag_data
1087 # convert to plain `unicode` instead of a markup tag to be used in
1087 # convert to plain `unicode` instead of a markup tag to be used in
1088 # regex expressions. safe_unicode doesn't work here
1088 # regex expressions. safe_unicode doesn't work here
1089 html_value = pat.sub(replace_html, unicode(value))
1089 html_value = pat.sub(replace_html, unicode(value))
1090
1090
1091 return html_value
1091 return html_value
1092
1092
1093
1093
1094 def bool2icon(value, show_at_false=True):
1094 def bool2icon(value, show_at_false=True):
1095 """
1095 """
1096 Returns boolean value of a given value, represented as html element with
1096 Returns boolean value of a given value, represented as html element with
1097 classes that will represent icons
1097 classes that will represent icons
1098
1098
1099 :param value: given value to convert to html node
1099 :param value: given value to convert to html node
1100 """
1100 """
1101
1101
1102 if value: # does bool conversion
1102 if value: # does bool conversion
1103 return HTML.tag('i', class_="icon-true", title='True')
1103 return HTML.tag('i', class_="icon-true", title='True')
1104 else: # not true as bool
1104 else: # not true as bool
1105 if show_at_false:
1105 if show_at_false:
1106 return HTML.tag('i', class_="icon-false", title='False')
1106 return HTML.tag('i', class_="icon-false", title='False')
1107 return HTML.tag('i')
1107 return HTML.tag('i')
1108
1108
1109
1109
1110 def b64(inp):
1110 def b64(inp):
1111 return base64.b64encode(inp)
1111 return base64.b64encode(inp)
1112
1112
1113 #==============================================================================
1113 #==============================================================================
1114 # PERMS
1114 # PERMS
1115 #==============================================================================
1115 #==============================================================================
1116 from rhodecode.lib.auth import (
1116 from rhodecode.lib.auth import (
1117 HasPermissionAny, HasPermissionAll,
1117 HasPermissionAny, HasPermissionAll,
1118 HasRepoPermissionAny, HasRepoPermissionAll, HasRepoGroupPermissionAll,
1118 HasRepoPermissionAny, HasRepoPermissionAll, HasRepoGroupPermissionAll,
1119 HasRepoGroupPermissionAny, HasRepoPermissionAnyApi, get_csrf_token,
1119 HasRepoGroupPermissionAny, HasRepoPermissionAnyApi, get_csrf_token,
1120 csrf_token_key, AuthUser)
1120 csrf_token_key, AuthUser)
1121
1121
1122
1122
1123 #==============================================================================
1123 #==============================================================================
1124 # GRAVATAR URL
1124 # GRAVATAR URL
1125 #==============================================================================
1125 #==============================================================================
1126 class InitialsGravatar(object):
1126 class InitialsGravatar(object):
1127 def __init__(self, email_address, first_name, last_name, size=30,
1127 def __init__(self, email_address, first_name, last_name, size=30,
1128 background=None, text_color='#fff'):
1128 background=None, text_color='#fff'):
1129 self.size = size
1129 self.size = size
1130 self.first_name = first_name
1130 self.first_name = first_name
1131 self.last_name = last_name
1131 self.last_name = last_name
1132 self.email_address = email_address
1132 self.email_address = email_address
1133 self.background = background or self.str2color(email_address)
1133 self.background = background or self.str2color(email_address)
1134 self.text_color = text_color
1134 self.text_color = text_color
1135
1135
1136 def get_color_bank(self):
1136 def get_color_bank(self):
1137 """
1137 """
1138 returns a predefined list of colors that gravatars can use.
1138 returns a predefined list of colors that gravatars can use.
1139 Those are randomized distinct colors that guarantee readability and
1139 Those are randomized distinct colors that guarantee readability and
1140 uniqueness.
1140 uniqueness.
1141
1141
1142 generated with: http://phrogz.net/css/distinct-colors.html
1142 generated with: http://phrogz.net/css/distinct-colors.html
1143 """
1143 """
1144 return [
1144 return [
1145 '#bf3030', '#a67f53', '#00ff00', '#5989b3', '#392040', '#d90000',
1145 '#bf3030', '#a67f53', '#00ff00', '#5989b3', '#392040', '#d90000',
1146 '#402910', '#204020', '#79baf2', '#a700b3', '#bf6060', '#7f5320',
1146 '#402910', '#204020', '#79baf2', '#a700b3', '#bf6060', '#7f5320',
1147 '#008000', '#003059', '#ee00ff', '#ff0000', '#8c4b00', '#007300',
1147 '#008000', '#003059', '#ee00ff', '#ff0000', '#8c4b00', '#007300',
1148 '#005fb3', '#de73e6', '#ff4040', '#ffaa00', '#3df255', '#203140',
1148 '#005fb3', '#de73e6', '#ff4040', '#ffaa00', '#3df255', '#203140',
1149 '#47004d', '#591616', '#664400', '#59b365', '#0d2133', '#83008c',
1149 '#47004d', '#591616', '#664400', '#59b365', '#0d2133', '#83008c',
1150 '#592d2d', '#bf9f60', '#73e682', '#1d3f73', '#73006b', '#402020',
1150 '#592d2d', '#bf9f60', '#73e682', '#1d3f73', '#73006b', '#402020',
1151 '#b2862d', '#397341', '#597db3', '#e600d6', '#a60000', '#736039',
1151 '#b2862d', '#397341', '#597db3', '#e600d6', '#a60000', '#736039',
1152 '#00b318', '#79aaf2', '#330d30', '#ff8080', '#403010', '#16591f',
1152 '#00b318', '#79aaf2', '#330d30', '#ff8080', '#403010', '#16591f',
1153 '#002459', '#8c4688', '#e50000', '#ffbf40', '#00732e', '#102340',
1153 '#002459', '#8c4688', '#e50000', '#ffbf40', '#00732e', '#102340',
1154 '#bf60ac', '#8c4646', '#cc8800', '#00a642', '#1d3473', '#b32d98',
1154 '#bf60ac', '#8c4646', '#cc8800', '#00a642', '#1d3473', '#b32d98',
1155 '#660e00', '#ffd580', '#80ffb2', '#7391e6', '#733967', '#d97b6c',
1155 '#660e00', '#ffd580', '#80ffb2', '#7391e6', '#733967', '#d97b6c',
1156 '#8c5e00', '#59b389', '#3967e6', '#590047', '#73281d', '#665200',
1156 '#8c5e00', '#59b389', '#3967e6', '#590047', '#73281d', '#665200',
1157 '#00e67a', '#2d50b3', '#8c2377', '#734139', '#b2982d', '#16593a',
1157 '#00e67a', '#2d50b3', '#8c2377', '#734139', '#b2982d', '#16593a',
1158 '#001859', '#ff00aa', '#a65e53', '#ffcc00', '#0d3321', '#2d3959',
1158 '#001859', '#ff00aa', '#a65e53', '#ffcc00', '#0d3321', '#2d3959',
1159 '#731d56', '#401610', '#4c3d00', '#468c6c', '#002ca6', '#d936a3',
1159 '#731d56', '#401610', '#4c3d00', '#468c6c', '#002ca6', '#d936a3',
1160 '#d94c36', '#403920', '#36d9a3', '#0d1733', '#592d4a', '#993626',
1160 '#d94c36', '#403920', '#36d9a3', '#0d1733', '#592d4a', '#993626',
1161 '#cca300', '#00734d', '#46598c', '#8c005e', '#7f1100', '#8c7000',
1161 '#cca300', '#00734d', '#46598c', '#8c005e', '#7f1100', '#8c7000',
1162 '#00a66f', '#7382e6', '#b32d74', '#d9896c', '#ffe680', '#1d7362',
1162 '#00a66f', '#7382e6', '#b32d74', '#d9896c', '#ffe680', '#1d7362',
1163 '#364cd9', '#73003d', '#d93a00', '#998a4d', '#59b3a1', '#5965b3',
1163 '#364cd9', '#73003d', '#d93a00', '#998a4d', '#59b3a1', '#5965b3',
1164 '#e5007a', '#73341d', '#665f00', '#00b38f', '#0018b3', '#59163a',
1164 '#e5007a', '#73341d', '#665f00', '#00b38f', '#0018b3', '#59163a',
1165 '#b2502d', '#bfb960', '#00ffcc', '#23318c', '#a6537f', '#734939',
1165 '#b2502d', '#bfb960', '#00ffcc', '#23318c', '#a6537f', '#734939',
1166 '#b2a700', '#104036', '#3d3df2', '#402031', '#e56739', '#736f39',
1166 '#b2a700', '#104036', '#3d3df2', '#402031', '#e56739', '#736f39',
1167 '#79f2ea', '#000059', '#401029', '#4c1400', '#ffee00', '#005953',
1167 '#79f2ea', '#000059', '#401029', '#4c1400', '#ffee00', '#005953',
1168 '#101040', '#990052', '#402820', '#403d10', '#00ffee', '#0000d9',
1168 '#101040', '#990052', '#402820', '#403d10', '#00ffee', '#0000d9',
1169 '#ff80c4', '#a66953', '#eeff00', '#00ccbe', '#8080ff', '#e673a1',
1169 '#ff80c4', '#a66953', '#eeff00', '#00ccbe', '#8080ff', '#e673a1',
1170 '#a62c00', '#474d00', '#1a3331', '#46468c', '#733950', '#662900',
1170 '#a62c00', '#474d00', '#1a3331', '#46468c', '#733950', '#662900',
1171 '#858c23', '#238c85', '#0f0073', '#b20047', '#d9986c', '#becc00',
1171 '#858c23', '#238c85', '#0f0073', '#b20047', '#d9986c', '#becc00',
1172 '#396f73', '#281d73', '#ff0066', '#ff6600', '#dee673', '#59adb3',
1172 '#396f73', '#281d73', '#ff0066', '#ff6600', '#dee673', '#59adb3',
1173 '#6559b3', '#590024', '#b2622d', '#98b32d', '#36ced9', '#332d59',
1173 '#6559b3', '#590024', '#b2622d', '#98b32d', '#36ced9', '#332d59',
1174 '#40001a', '#733f1d', '#526600', '#005359', '#242040', '#bf6079',
1174 '#40001a', '#733f1d', '#526600', '#005359', '#242040', '#bf6079',
1175 '#735039', '#cef23d', '#007780', '#5630bf', '#66001b', '#b24700',
1175 '#735039', '#cef23d', '#007780', '#5630bf', '#66001b', '#b24700',
1176 '#acbf60', '#1d6273', '#25008c', '#731d34', '#a67453', '#50592d',
1176 '#acbf60', '#1d6273', '#25008c', '#731d34', '#a67453', '#50592d',
1177 '#00ccff', '#6600ff', '#ff0044', '#4c1f00', '#8a994d', '#79daf2',
1177 '#00ccff', '#6600ff', '#ff0044', '#4c1f00', '#8a994d', '#79daf2',
1178 '#a173e6', '#d93662', '#402310', '#aaff00', '#2d98b3', '#8c40ff',
1178 '#a173e6', '#d93662', '#402310', '#aaff00', '#2d98b3', '#8c40ff',
1179 '#592d39', '#ff8c40', '#354020', '#103640', '#1a0040', '#331a20',
1179 '#592d39', '#ff8c40', '#354020', '#103640', '#1a0040', '#331a20',
1180 '#331400', '#334d00', '#1d5673', '#583973', '#7f0022', '#4c3626',
1180 '#331400', '#334d00', '#1d5673', '#583973', '#7f0022', '#4c3626',
1181 '#88cc00', '#36a3d9', '#3d0073', '#d9364c', '#33241a', '#698c23',
1181 '#88cc00', '#36a3d9', '#3d0073', '#d9364c', '#33241a', '#698c23',
1182 '#5995b3', '#300059', '#e57382', '#7f3300', '#366600', '#00aaff',
1182 '#5995b3', '#300059', '#e57382', '#7f3300', '#366600', '#00aaff',
1183 '#3a1659', '#733941', '#663600', '#74b32d', '#003c59', '#7f53a6',
1183 '#3a1659', '#733941', '#663600', '#74b32d', '#003c59', '#7f53a6',
1184 '#73000f', '#ff8800', '#baf279', '#79caf2', '#291040', '#a6293a',
1184 '#73000f', '#ff8800', '#baf279', '#79caf2', '#291040', '#a6293a',
1185 '#b2742d', '#587339', '#0077b3', '#632699', '#400009', '#d9a66c',
1185 '#b2742d', '#587339', '#0077b3', '#632699', '#400009', '#d9a66c',
1186 '#294010', '#2d4a59', '#aa00ff', '#4c131b', '#b25f00', '#5ce600',
1186 '#294010', '#2d4a59', '#aa00ff', '#4c131b', '#b25f00', '#5ce600',
1187 '#267399', '#a336d9', '#990014', '#664e33', '#86bf60', '#0088ff',
1187 '#267399', '#a336d9', '#990014', '#664e33', '#86bf60', '#0088ff',
1188 '#7700b3', '#593a16', '#073300', '#1d4b73', '#ac60bf', '#e59539',
1188 '#7700b3', '#593a16', '#073300', '#1d4b73', '#ac60bf', '#e59539',
1189 '#4f8c46', '#368dd9', '#5c0073'
1189 '#4f8c46', '#368dd9', '#5c0073'
1190 ]
1190 ]
1191
1191
1192 def rgb_to_hex_color(self, rgb_tuple):
1192 def rgb_to_hex_color(self, rgb_tuple):
1193 """
1193 """
1194 Converts an rgb_tuple passed to an hex color.
1194 Converts an rgb_tuple passed to an hex color.
1195
1195
1196 :param rgb_tuple: tuple with 3 ints represents rgb color space
1196 :param rgb_tuple: tuple with 3 ints represents rgb color space
1197 """
1197 """
1198 return '#' + ("".join(map(chr, rgb_tuple)).encode('hex'))
1198 return '#' + ("".join(map(chr, rgb_tuple)).encode('hex'))
1199
1199
1200 def email_to_int_list(self, email_str):
1200 def email_to_int_list(self, email_str):
1201 """
1201 """
1202 Get every byte of the hex digest value of email and turn it to integer.
1202 Get every byte of the hex digest value of email and turn it to integer.
1203 It's going to be always between 0-255
1203 It's going to be always between 0-255
1204 """
1204 """
1205 digest = md5_safe(email_str.lower())
1205 digest = md5_safe(email_str.lower())
1206 return [int(digest[i * 2:i * 2 + 2], 16) for i in range(16)]
1206 return [int(digest[i * 2:i * 2 + 2], 16) for i in range(16)]
1207
1207
1208 def pick_color_bank_index(self, email_str, color_bank):
1208 def pick_color_bank_index(self, email_str, color_bank):
1209 return self.email_to_int_list(email_str)[0] % len(color_bank)
1209 return self.email_to_int_list(email_str)[0] % len(color_bank)
1210
1210
1211 def str2color(self, email_str):
1211 def str2color(self, email_str):
1212 """
1212 """
1213 Tries to map in a stable algorithm an email to color
1213 Tries to map in a stable algorithm an email to color
1214
1214
1215 :param email_str:
1215 :param email_str:
1216 """
1216 """
1217 color_bank = self.get_color_bank()
1217 color_bank = self.get_color_bank()
1218 # pick position (module it's length so we always find it in the
1218 # pick position (module it's length so we always find it in the
1219 # bank even if it's smaller than 256 values
1219 # bank even if it's smaller than 256 values
1220 pos = self.pick_color_bank_index(email_str, color_bank)
1220 pos = self.pick_color_bank_index(email_str, color_bank)
1221 return color_bank[pos]
1221 return color_bank[pos]
1222
1222
1223 def normalize_email(self, email_address):
1223 def normalize_email(self, email_address):
1224 import unicodedata
1224 import unicodedata
1225 # default host used to fill in the fake/missing email
1225 # default host used to fill in the fake/missing email
1226 default_host = u'localhost'
1226 default_host = u'localhost'
1227
1227
1228 if not email_address:
1228 if not email_address:
1229 email_address = u'%s@%s' % (User.DEFAULT_USER, default_host)
1229 email_address = u'%s@%s' % (User.DEFAULT_USER, default_host)
1230
1230
1231 email_address = safe_unicode(email_address)
1231 email_address = safe_unicode(email_address)
1232
1232
1233 if u'@' not in email_address:
1233 if u'@' not in email_address:
1234 email_address = u'%s@%s' % (email_address, default_host)
1234 email_address = u'%s@%s' % (email_address, default_host)
1235
1235
1236 if email_address.endswith(u'@'):
1236 if email_address.endswith(u'@'):
1237 email_address = u'%s%s' % (email_address, default_host)
1237 email_address = u'%s%s' % (email_address, default_host)
1238
1238
1239 email_address = unicodedata.normalize('NFKD', email_address)\
1239 email_address = unicodedata.normalize('NFKD', email_address)\
1240 .encode('ascii', 'ignore')
1240 .encode('ascii', 'ignore')
1241 return email_address
1241 return email_address
1242
1242
1243 def get_initials(self):
1243 def get_initials(self):
1244 """
1244 """
1245 Returns 2 letter initials calculated based on the input.
1245 Returns 2 letter initials calculated based on the input.
1246 The algorithm picks first given email address, and takes first letter
1246 The algorithm picks first given email address, and takes first letter
1247 of part before @, and then the first letter of server name. In case
1247 of part before @, and then the first letter of server name. In case
1248 the part before @ is in a format of `somestring.somestring2` it replaces
1248 the part before @ is in a format of `somestring.somestring2` it replaces
1249 the server letter with first letter of somestring2
1249 the server letter with first letter of somestring2
1250
1250
1251 In case function was initialized with both first and lastname, this
1251 In case function was initialized with both first and lastname, this
1252 overrides the extraction from email by first letter of the first and
1252 overrides the extraction from email by first letter of the first and
1253 last name. We add special logic to that functionality, In case Full name
1253 last name. We add special logic to that functionality, In case Full name
1254 is compound, like Guido Von Rossum, we use last part of the last name
1254 is compound, like Guido Von Rossum, we use last part of the last name
1255 (Von Rossum) picking `R`.
1255 (Von Rossum) picking `R`.
1256
1256
1257 Function also normalizes the non-ascii characters to they ascii
1257 Function also normalizes the non-ascii characters to they ascii
1258 representation, eg Δ„ => A
1258 representation, eg Δ„ => A
1259 """
1259 """
1260 import unicodedata
1260 import unicodedata
1261 # replace non-ascii to ascii
1261 # replace non-ascii to ascii
1262 first_name = unicodedata.normalize(
1262 first_name = unicodedata.normalize(
1263 'NFKD', safe_unicode(self.first_name)).encode('ascii', 'ignore')
1263 'NFKD', safe_unicode(self.first_name)).encode('ascii', 'ignore')
1264 last_name = unicodedata.normalize(
1264 last_name = unicodedata.normalize(
1265 'NFKD', safe_unicode(self.last_name)).encode('ascii', 'ignore')
1265 'NFKD', safe_unicode(self.last_name)).encode('ascii', 'ignore')
1266
1266
1267 # do NFKD encoding, and also make sure email has proper format
1267 # do NFKD encoding, and also make sure email has proper format
1268 email_address = self.normalize_email(self.email_address)
1268 email_address = self.normalize_email(self.email_address)
1269
1269
1270 # first push the email initials
1270 # first push the email initials
1271 prefix, server = email_address.split('@', 1)
1271 prefix, server = email_address.split('@', 1)
1272
1272
1273 # check if prefix is maybe a 'first_name.last_name' syntax
1273 # check if prefix is maybe a 'first_name.last_name' syntax
1274 _dot_split = prefix.rsplit('.', 1)
1274 _dot_split = prefix.rsplit('.', 1)
1275 if len(_dot_split) == 2 and _dot_split[1]:
1275 if len(_dot_split) == 2 and _dot_split[1]:
1276 initials = [_dot_split[0][0], _dot_split[1][0]]
1276 initials = [_dot_split[0][0], _dot_split[1][0]]
1277 else:
1277 else:
1278 initials = [prefix[0], server[0]]
1278 initials = [prefix[0], server[0]]
1279
1279
1280 # then try to replace either first_name or last_name
1280 # then try to replace either first_name or last_name
1281 fn_letter = (first_name or " ")[0].strip()
1281 fn_letter = (first_name or " ")[0].strip()
1282 ln_letter = (last_name.split(' ', 1)[-1] or " ")[0].strip()
1282 ln_letter = (last_name.split(' ', 1)[-1] or " ")[0].strip()
1283
1283
1284 if fn_letter:
1284 if fn_letter:
1285 initials[0] = fn_letter
1285 initials[0] = fn_letter
1286
1286
1287 if ln_letter:
1287 if ln_letter:
1288 initials[1] = ln_letter
1288 initials[1] = ln_letter
1289
1289
1290 return ''.join(initials).upper()
1290 return ''.join(initials).upper()
1291
1291
1292 def get_img_data_by_type(self, font_family, img_type):
1292 def get_img_data_by_type(self, font_family, img_type):
1293 default_user = """
1293 default_user = """
1294 <svg xmlns="http://www.w3.org/2000/svg"
1294 <svg xmlns="http://www.w3.org/2000/svg"
1295 version="1.1" x="0px" y="0px" width="{size}" height="{size}"
1295 version="1.1" x="0px" y="0px" width="{size}" height="{size}"
1296 viewBox="-15 -10 439.165 429.164"
1296 viewBox="-15 -10 439.165 429.164"
1297
1297
1298 xml:space="preserve"
1298 xml:space="preserve"
1299 style="background:{background};" >
1299 style="background:{background};" >
1300
1300
1301 <path d="M204.583,216.671c50.664,0,91.74-48.075,
1301 <path d="M204.583,216.671c50.664,0,91.74-48.075,
1302 91.74-107.378c0-82.237-41.074-107.377-91.74-107.377
1302 91.74-107.378c0-82.237-41.074-107.377-91.74-107.377
1303 c-50.668,0-91.74,25.14-91.74,107.377C112.844,
1303 c-50.668,0-91.74,25.14-91.74,107.377C112.844,
1304 168.596,153.916,216.671,
1304 168.596,153.916,216.671,
1305 204.583,216.671z" fill="{text_color}"/>
1305 204.583,216.671z" fill="{text_color}"/>
1306 <path d="M407.164,374.717L360.88,
1306 <path d="M407.164,374.717L360.88,
1307 270.454c-2.117-4.771-5.836-8.728-10.465-11.138l-71.83-37.392
1307 270.454c-2.117-4.771-5.836-8.728-10.465-11.138l-71.83-37.392
1308 c-1.584-0.823-3.502-0.663-4.926,0.415c-20.316,
1308 c-1.584-0.823-3.502-0.663-4.926,0.415c-20.316,
1309 15.366-44.203,23.488-69.076,23.488c-24.877,
1309 15.366-44.203,23.488-69.076,23.488c-24.877,
1310 0-48.762-8.122-69.078-23.488
1310 0-48.762-8.122-69.078-23.488
1311 c-1.428-1.078-3.346-1.238-4.93-0.415L58.75,
1311 c-1.428-1.078-3.346-1.238-4.93-0.415L58.75,
1312 259.316c-4.631,2.41-8.346,6.365-10.465,11.138L2.001,374.717
1312 259.316c-4.631,2.41-8.346,6.365-10.465,11.138L2.001,374.717
1313 c-3.191,7.188-2.537,15.412,1.75,22.005c4.285,
1313 c-3.191,7.188-2.537,15.412,1.75,22.005c4.285,
1314 6.592,11.537,10.526,19.4,10.526h362.861c7.863,0,15.117-3.936,
1314 6.592,11.537,10.526,19.4,10.526h362.861c7.863,0,15.117-3.936,
1315 19.402-10.527 C409.699,390.129,
1315 19.402-10.527 C409.699,390.129,
1316 410.355,381.902,407.164,374.717z" fill="{text_color}"/>
1316 410.355,381.902,407.164,374.717z" fill="{text_color}"/>
1317 </svg>""".format(
1317 </svg>""".format(
1318 size=self.size,
1318 size=self.size,
1319 background='#979797', # @grey4
1319 background='#979797', # @grey4
1320 text_color=self.text_color,
1320 text_color=self.text_color,
1321 font_family=font_family)
1321 font_family=font_family)
1322
1322
1323 return {
1323 return {
1324 "default_user": default_user
1324 "default_user": default_user
1325 }[img_type]
1325 }[img_type]
1326
1326
1327 def get_img_data(self, svg_type=None):
1327 def get_img_data(self, svg_type=None):
1328 """
1328 """
1329 generates the svg metadata for image
1329 generates the svg metadata for image
1330 """
1330 """
1331 fonts = [
1331 fonts = [
1332 '-apple-system',
1332 '-apple-system',
1333 'BlinkMacSystemFont',
1333 'BlinkMacSystemFont',
1334 'Segoe UI',
1334 'Segoe UI',
1335 'Roboto',
1335 'Roboto',
1336 'Oxygen-Sans',
1336 'Oxygen-Sans',
1337 'Ubuntu',
1337 'Ubuntu',
1338 'Cantarell',
1338 'Cantarell',
1339 'Helvetica Neue',
1339 'Helvetica Neue',
1340 'sans-serif'
1340 'sans-serif'
1341 ]
1341 ]
1342 font_family = ','.join(fonts)
1342 font_family = ','.join(fonts)
1343 if svg_type:
1343 if svg_type:
1344 return self.get_img_data_by_type(font_family, svg_type)
1344 return self.get_img_data_by_type(font_family, svg_type)
1345
1345
1346 initials = self.get_initials()
1346 initials = self.get_initials()
1347 img_data = """
1347 img_data = """
1348 <svg xmlns="http://www.w3.org/2000/svg" pointer-events="none"
1348 <svg xmlns="http://www.w3.org/2000/svg" pointer-events="none"
1349 width="{size}" height="{size}"
1349 width="{size}" height="{size}"
1350 style="width: 100%; height: 100%; background-color: {background}"
1350 style="width: 100%; height: 100%; background-color: {background}"
1351 viewBox="0 0 {size} {size}">
1351 viewBox="0 0 {size} {size}">
1352 <text text-anchor="middle" y="50%" x="50%" dy="0.35em"
1352 <text text-anchor="middle" y="50%" x="50%" dy="0.35em"
1353 pointer-events="auto" fill="{text_color}"
1353 pointer-events="auto" fill="{text_color}"
1354 font-family="{font_family}"
1354 font-family="{font_family}"
1355 style="font-weight: 400; font-size: {f_size}px;">{text}
1355 style="font-weight: 400; font-size: {f_size}px;">{text}
1356 </text>
1356 </text>
1357 </svg>""".format(
1357 </svg>""".format(
1358 size=self.size,
1358 size=self.size,
1359 f_size=self.size/2.05, # scale the text inside the box nicely
1359 f_size=self.size/2.05, # scale the text inside the box nicely
1360 background=self.background,
1360 background=self.background,
1361 text_color=self.text_color,
1361 text_color=self.text_color,
1362 text=initials.upper(),
1362 text=initials.upper(),
1363 font_family=font_family)
1363 font_family=font_family)
1364
1364
1365 return img_data
1365 return img_data
1366
1366
1367 def generate_svg(self, svg_type=None):
1367 def generate_svg(self, svg_type=None):
1368 img_data = self.get_img_data(svg_type)
1368 img_data = self.get_img_data(svg_type)
1369 return "data:image/svg+xml;base64,%s" % base64.b64encode(img_data)
1369 return "data:image/svg+xml;base64,%s" % base64.b64encode(img_data)
1370
1370
1371
1371
1372 def initials_gravatar(request, email_address, first_name, last_name, size=30, store_on_disk=False):
1372 def initials_gravatar(request, email_address, first_name, last_name, size=30, store_on_disk=False):
1373
1373
1374 svg_type = None
1374 svg_type = None
1375 if email_address == User.DEFAULT_USER_EMAIL:
1375 if email_address == User.DEFAULT_USER_EMAIL:
1376 svg_type = 'default_user'
1376 svg_type = 'default_user'
1377
1377
1378 klass = InitialsGravatar(email_address, first_name, last_name, size)
1378 klass = InitialsGravatar(email_address, first_name, last_name, size)
1379
1379
1380 if store_on_disk:
1380 if store_on_disk:
1381 from rhodecode.apps.file_store import utils as store_utils
1381 from rhodecode.apps.file_store import utils as store_utils
1382 from rhodecode.apps.file_store.exceptions import FileNotAllowedException, \
1382 from rhodecode.apps.file_store.exceptions import FileNotAllowedException, \
1383 FileOverSizeException
1383 FileOverSizeException
1384 from rhodecode.model.db import Session
1384 from rhodecode.model.db import Session
1385
1385
1386 image_key = md5_safe(email_address.lower()
1386 image_key = md5_safe(email_address.lower()
1387 + first_name.lower() + last_name.lower())
1387 + first_name.lower() + last_name.lower())
1388
1388
1389 storage = store_utils.get_file_storage(request.registry.settings)
1389 storage = store_utils.get_file_storage(request.registry.settings)
1390 filename = '{}.svg'.format(image_key)
1390 filename = '{}.svg'.format(image_key)
1391 subdir = 'gravatars'
1391 subdir = 'gravatars'
1392 # since final name has a counter, we apply the 0
1392 # since final name has a counter, we apply the 0
1393 uid = storage.apply_counter(0, store_utils.uid_filename(filename, randomized=False))
1393 uid = storage.apply_counter(0, store_utils.uid_filename(filename, randomized=False))
1394 store_uid = os.path.join(subdir, uid)
1394 store_uid = os.path.join(subdir, uid)
1395
1395
1396 db_entry = FileStore.get_by_store_uid(store_uid)
1396 db_entry = FileStore.get_by_store_uid(store_uid)
1397 if db_entry:
1397 if db_entry:
1398 return request.route_path('download_file', fid=store_uid)
1398 return request.route_path('download_file', fid=store_uid)
1399
1399
1400 img_data = klass.get_img_data(svg_type=svg_type)
1400 img_data = klass.get_img_data(svg_type=svg_type)
1401 img_file = store_utils.bytes_to_file_obj(img_data)
1401 img_file = store_utils.bytes_to_file_obj(img_data)
1402
1402
1403 try:
1403 try:
1404 store_uid, metadata = storage.save_file(
1404 store_uid, metadata = storage.save_file(
1405 img_file, filename, directory=subdir,
1405 img_file, filename, directory=subdir,
1406 extensions=['.svg'], randomized_name=False)
1406 extensions=['.svg'], randomized_name=False)
1407 except (FileNotAllowedException, FileOverSizeException):
1407 except (FileNotAllowedException, FileOverSizeException):
1408 raise
1408 raise
1409
1409
1410 try:
1410 try:
1411 entry = FileStore.create(
1411 entry = FileStore.create(
1412 file_uid=store_uid, filename=metadata["filename"],
1412 file_uid=store_uid, filename=metadata["filename"],
1413 file_hash=metadata["sha256"], file_size=metadata["size"],
1413 file_hash=metadata["sha256"], file_size=metadata["size"],
1414 file_display_name=filename,
1414 file_display_name=filename,
1415 file_description=u'user gravatar `{}`'.format(safe_unicode(filename)),
1415 file_description=u'user gravatar `{}`'.format(safe_unicode(filename)),
1416 hidden=True, check_acl=False, user_id=1
1416 hidden=True, check_acl=False, user_id=1
1417 )
1417 )
1418 Session().add(entry)
1418 Session().add(entry)
1419 Session().commit()
1419 Session().commit()
1420 log.debug('Stored upload in DB as %s', entry)
1420 log.debug('Stored upload in DB as %s', entry)
1421 except Exception:
1421 except Exception:
1422 raise
1422 raise
1423
1423
1424 return request.route_path('download_file', fid=store_uid)
1424 return request.route_path('download_file', fid=store_uid)
1425
1425
1426 else:
1426 else:
1427 return klass.generate_svg(svg_type=svg_type)
1427 return klass.generate_svg(svg_type=svg_type)
1428
1428
1429
1429
1430 def gravatar_external(request, gravatar_url_tmpl, email_address, size=30):
1430 def gravatar_external(request, gravatar_url_tmpl, email_address, size=30):
1431 return safe_str(gravatar_url_tmpl)\
1431 return safe_str(gravatar_url_tmpl)\
1432 .replace('{email}', email_address) \
1432 .replace('{email}', email_address) \
1433 .replace('{md5email}', md5_safe(email_address.lower())) \
1433 .replace('{md5email}', md5_safe(email_address.lower())) \
1434 .replace('{netloc}', request.host) \
1434 .replace('{netloc}', request.host) \
1435 .replace('{scheme}', request.scheme) \
1435 .replace('{scheme}', request.scheme) \
1436 .replace('{size}', safe_str(size))
1436 .replace('{size}', safe_str(size))
1437
1437
1438
1438
1439 def gravatar_url(email_address, size=30, request=None):
1439 def gravatar_url(email_address, size=30, request=None):
1440 request = request or get_current_request()
1440 request = request or get_current_request()
1441 _use_gravatar = request.call_context.visual.use_gravatar
1441 _use_gravatar = request.call_context.visual.use_gravatar
1442
1442
1443 email_address = email_address or User.DEFAULT_USER_EMAIL
1443 email_address = email_address or User.DEFAULT_USER_EMAIL
1444 if isinstance(email_address, unicode):
1444 if isinstance(email_address, unicode):
1445 # hashlib crashes on unicode items
1445 # hashlib crashes on unicode items
1446 email_address = safe_str(email_address)
1446 email_address = safe_str(email_address)
1447
1447
1448 # empty email or default user
1448 # empty email or default user
1449 if not email_address or email_address == User.DEFAULT_USER_EMAIL:
1449 if not email_address or email_address == User.DEFAULT_USER_EMAIL:
1450 return initials_gravatar(request, User.DEFAULT_USER_EMAIL, '', '', size=size)
1450 return initials_gravatar(request, User.DEFAULT_USER_EMAIL, '', '', size=size)
1451
1451
1452 if _use_gravatar:
1452 if _use_gravatar:
1453 gravatar_url_tmpl = request.call_context.visual.gravatar_url \
1453 gravatar_url_tmpl = request.call_context.visual.gravatar_url \
1454 or User.DEFAULT_GRAVATAR_URL
1454 or User.DEFAULT_GRAVATAR_URL
1455 return gravatar_external(request, gravatar_url_tmpl, email_address, size=size)
1455 return gravatar_external(request, gravatar_url_tmpl, email_address, size=size)
1456
1456
1457 else:
1457 else:
1458 return initials_gravatar(request, email_address, '', '', size=size)
1458 return initials_gravatar(request, email_address, '', '', size=size)
1459
1459
1460
1460
1461 def breadcrumb_repo_link(repo):
1461 def breadcrumb_repo_link(repo):
1462 """
1462 """
1463 Makes a breadcrumbs path link to repo
1463 Makes a breadcrumbs path link to repo
1464
1464
1465 ex::
1465 ex::
1466 group >> subgroup >> repo
1466 group >> subgroup >> repo
1467
1467
1468 :param repo: a Repository instance
1468 :param repo: a Repository instance
1469 """
1469 """
1470
1470
1471 path = [
1471 path = [
1472 link_to(group.name, route_path('repo_group_home', repo_group_name=group.group_name),
1472 link_to(group.name, route_path('repo_group_home', repo_group_name=group.group_name),
1473 title='last change:{}'.format(format_date(group.last_commit_change)))
1473 title='last change:{}'.format(format_date(group.last_commit_change)))
1474 for group in repo.groups_with_parents
1474 for group in repo.groups_with_parents
1475 ] + [
1475 ] + [
1476 link_to(repo.just_name, route_path('repo_summary', repo_name=repo.repo_name),
1476 link_to(repo.just_name, route_path('repo_summary', repo_name=repo.repo_name),
1477 title='last change:{}'.format(format_date(repo.last_commit_change)))
1477 title='last change:{}'.format(format_date(repo.last_commit_change)))
1478 ]
1478 ]
1479
1479
1480 return literal(' &raquo; '.join(path))
1480 return literal(' &raquo; '.join(path))
1481
1481
1482
1482
1483 def breadcrumb_repo_group_link(repo_group):
1483 def breadcrumb_repo_group_link(repo_group):
1484 """
1484 """
1485 Makes a breadcrumbs path link to repo
1485 Makes a breadcrumbs path link to repo
1486
1486
1487 ex::
1487 ex::
1488 group >> subgroup
1488 group >> subgroup
1489
1489
1490 :param repo_group: a Repository Group instance
1490 :param repo_group: a Repository Group instance
1491 """
1491 """
1492
1492
1493 path = [
1493 path = [
1494 link_to(group.name,
1494 link_to(group.name,
1495 route_path('repo_group_home', repo_group_name=group.group_name),
1495 route_path('repo_group_home', repo_group_name=group.group_name),
1496 title='last change:{}'.format(format_date(group.last_commit_change)))
1496 title='last change:{}'.format(format_date(group.last_commit_change)))
1497 for group in repo_group.parents
1497 for group in repo_group.parents
1498 ] + [
1498 ] + [
1499 link_to(repo_group.name,
1499 link_to(repo_group.name,
1500 route_path('repo_group_home', repo_group_name=repo_group.group_name),
1500 route_path('repo_group_home', repo_group_name=repo_group.group_name),
1501 title='last change:{}'.format(format_date(repo_group.last_commit_change)))
1501 title='last change:{}'.format(format_date(repo_group.last_commit_change)))
1502 ]
1502 ]
1503
1503
1504 return literal(' &raquo; '.join(path))
1504 return literal(' &raquo; '.join(path))
1505
1505
1506
1506
1507 def format_byte_size_binary(file_size):
1507 def format_byte_size_binary(file_size):
1508 """
1508 """
1509 Formats file/folder sizes to standard.
1509 Formats file/folder sizes to standard.
1510 """
1510 """
1511 if file_size is None:
1511 if file_size is None:
1512 file_size = 0
1512 file_size = 0
1513
1513
1514 formatted_size = format_byte_size(file_size, binary=True)
1514 formatted_size = format_byte_size(file_size, binary=True)
1515 return formatted_size
1515 return formatted_size
1516
1516
1517
1517
1518 def urlify_text(text_, safe=True, **href_attrs):
1518 def urlify_text(text_, safe=True, **href_attrs):
1519 """
1519 """
1520 Extract urls from text and make html links out of them
1520 Extract urls from text and make html links out of them
1521 """
1521 """
1522
1522
1523 url_pat = re.compile(r'''(http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@#.&+]'''
1523 url_pat = re.compile(r'''(http[s]?://(?:[a-zA-Z]|[0-9]|[$-_@#.&+]'''
1524 '''|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+)''')
1524 '''|[!*\(\),]|(?:%[0-9a-fA-F][0-9a-fA-F]))+)''')
1525
1525
1526 def url_func(match_obj):
1526 def url_func(match_obj):
1527 url_full = match_obj.groups()[0]
1527 url_full = match_obj.groups()[0]
1528 a_options = dict(href_attrs)
1528 a_options = dict(href_attrs)
1529 a_options['href'] = url_full
1529 a_options['href'] = url_full
1530 a_text = url_full
1530 a_text = url_full
1531 return HTML.tag("a", a_text, **a_options)
1531 return HTML.tag("a", a_text, **a_options)
1532
1532
1533 _new_text = url_pat.sub(url_func, text_)
1533 _new_text = url_pat.sub(url_func, text_)
1534
1534
1535 if safe:
1535 if safe:
1536 return literal(_new_text)
1536 return literal(_new_text)
1537 return _new_text
1537 return _new_text
1538
1538
1539
1539
1540 def urlify_commits(text_, repo_name):
1540 def urlify_commits(text_, repo_name):
1541 """
1541 """
1542 Extract commit ids from text and make link from them
1542 Extract commit ids from text and make link from them
1543
1543
1544 :param text_:
1544 :param text_:
1545 :param repo_name: repo name to build the URL with
1545 :param repo_name: repo name to build the URL with
1546 """
1546 """
1547
1547
1548 url_pat = re.compile(r'(^|\s)([0-9a-fA-F]{12,40})($|\s)')
1548 url_pat = re.compile(r'(^|\s)([0-9a-fA-F]{12,40})($|\s)')
1549
1549
1550 def url_func(match_obj):
1550 def url_func(match_obj):
1551 commit_id = match_obj.groups()[1]
1551 commit_id = match_obj.groups()[1]
1552 pref = match_obj.groups()[0]
1552 pref = match_obj.groups()[0]
1553 suf = match_obj.groups()[2]
1553 suf = match_obj.groups()[2]
1554
1554
1555 tmpl = (
1555 tmpl = (
1556 '%(pref)s<a class="tooltip-hovercard %(cls)s" href="%(url)s" data-hovercard-alt="%(hovercard_alt)s" data-hovercard-url="%(hovercard_url)s">'
1556 '%(pref)s<a class="tooltip-hovercard %(cls)s" href="%(url)s" data-hovercard-alt="%(hovercard_alt)s" data-hovercard-url="%(hovercard_url)s">'
1557 '%(commit_id)s</a>%(suf)s'
1557 '%(commit_id)s</a>%(suf)s'
1558 )
1558 )
1559 return tmpl % {
1559 return tmpl % {
1560 'pref': pref,
1560 'pref': pref,
1561 'cls': 'revision-link',
1561 'cls': 'revision-link',
1562 'url': route_url(
1562 'url': route_url(
1563 'repo_commit', repo_name=repo_name, commit_id=commit_id),
1563 'repo_commit', repo_name=repo_name, commit_id=commit_id),
1564 'commit_id': commit_id,
1564 'commit_id': commit_id,
1565 'suf': suf,
1565 'suf': suf,
1566 'hovercard_alt': 'Commit: {}'.format(commit_id),
1566 'hovercard_alt': 'Commit: {}'.format(commit_id),
1567 'hovercard_url': route_url(
1567 'hovercard_url': route_url(
1568 'hovercard_repo_commit', repo_name=repo_name, commit_id=commit_id)
1568 'hovercard_repo_commit', repo_name=repo_name, commit_id=commit_id)
1569 }
1569 }
1570
1570
1571 new_text = url_pat.sub(url_func, text_)
1571 new_text = url_pat.sub(url_func, text_)
1572
1572
1573 return new_text
1573 return new_text
1574
1574
1575
1575
1576 def _process_url_func(match_obj, repo_name, uid, entry,
1576 def _process_url_func(match_obj, repo_name, uid, entry,
1577 return_raw_data=False, link_format='html'):
1577 return_raw_data=False, link_format='html'):
1578 pref = ''
1578 pref = ''
1579 if match_obj.group().startswith(' '):
1579 if match_obj.group().startswith(' '):
1580 pref = ' '
1580 pref = ' '
1581
1581
1582 issue_id = ''.join(match_obj.groups())
1582 issue_id = ''.join(match_obj.groups())
1583
1583
1584 if link_format == 'html':
1584 if link_format == 'html':
1585 tmpl = (
1585 tmpl = (
1586 '%(pref)s<a class="tooltip %(cls)s" href="%(url)s" title="%(title)s">'
1586 '%(pref)s<a class="tooltip %(cls)s" href="%(url)s" title="%(title)s">'
1587 '%(issue-prefix)s%(id-repr)s'
1587 '%(issue-prefix)s%(id-repr)s'
1588 '</a>')
1588 '</a>')
1589 elif link_format == 'html+hovercard':
1589 elif link_format == 'html+hovercard':
1590 tmpl = (
1590 tmpl = (
1591 '%(pref)s<a class="tooltip-hovercard %(cls)s" href="%(url)s" data-hovercard-url="%(hovercard_url)s">'
1591 '%(pref)s<a class="tooltip-hovercard %(cls)s" href="%(url)s" data-hovercard-url="%(hovercard_url)s">'
1592 '%(issue-prefix)s%(id-repr)s'
1592 '%(issue-prefix)s%(id-repr)s'
1593 '</a>')
1593 '</a>')
1594 elif link_format in ['rst', 'rst+hovercard']:
1594 elif link_format in ['rst', 'rst+hovercard']:
1595 tmpl = '`%(issue-prefix)s%(id-repr)s <%(url)s>`_'
1595 tmpl = '`%(issue-prefix)s%(id-repr)s <%(url)s>`_'
1596 elif link_format in ['markdown', 'markdown+hovercard']:
1596 elif link_format in ['markdown', 'markdown+hovercard']:
1597 tmpl = '[%(pref)s%(issue-prefix)s%(id-repr)s](%(url)s)'
1597 tmpl = '[%(pref)s%(issue-prefix)s%(id-repr)s](%(url)s)'
1598 else:
1598 else:
1599 raise ValueError('Bad link_format:{}'.format(link_format))
1599 raise ValueError('Bad link_format:{}'.format(link_format))
1600
1600
1601 (repo_name_cleaned,
1601 (repo_name_cleaned,
1602 parent_group_name) = RepoGroupModel()._get_group_name_and_parent(repo_name)
1602 parent_group_name) = RepoGroupModel()._get_group_name_and_parent(repo_name)
1603
1603
1604 # variables replacement
1604 # variables replacement
1605 named_vars = {
1605 named_vars = {
1606 'id': issue_id,
1606 'id': issue_id,
1607 'repo': repo_name,
1607 'repo': repo_name,
1608 'repo_name': repo_name_cleaned,
1608 'repo_name': repo_name_cleaned,
1609 'group_name': parent_group_name,
1609 'group_name': parent_group_name,
1610 # set dummy keys so we always have them
1610 # set dummy keys so we always have them
1611 'hostname': '',
1611 'hostname': '',
1612 'netloc': '',
1612 'netloc': '',
1613 'scheme': ''
1613 'scheme': ''
1614 }
1614 }
1615
1615
1616 request = get_current_request()
1616 request = get_current_request()
1617 if request:
1617 if request:
1618 # exposes, hostname, netloc, scheme
1618 # exposes, hostname, netloc, scheme
1619 host_data = get_host_info(request)
1619 host_data = get_host_info(request)
1620 named_vars.update(host_data)
1620 named_vars.update(host_data)
1621
1621
1622 # named regex variables
1622 # named regex variables
1623 named_vars.update(match_obj.groupdict())
1623 named_vars.update(match_obj.groupdict())
1624 _url = string.Template(entry['url']).safe_substitute(**named_vars)
1624 _url = string.Template(entry['url']).safe_substitute(**named_vars)
1625 desc = string.Template(escape(entry['desc'])).safe_substitute(**named_vars)
1625 desc = string.Template(escape(entry['desc'])).safe_substitute(**named_vars)
1626 hovercard_url = string.Template(entry.get('hovercard_url', '')).safe_substitute(**named_vars)
1626 hovercard_url = string.Template(entry.get('hovercard_url', '')).safe_substitute(**named_vars)
1627
1627
1628 def quote_cleaner(input_str):
1628 def quote_cleaner(input_str):
1629 """Remove quotes as it's HTML"""
1629 """Remove quotes as it's HTML"""
1630 return input_str.replace('"', '')
1630 return input_str.replace('"', '')
1631
1631
1632 data = {
1632 data = {
1633 'pref': pref,
1633 'pref': pref,
1634 'cls': quote_cleaner('issue-tracker-link'),
1634 'cls': quote_cleaner('issue-tracker-link'),
1635 'url': quote_cleaner(_url),
1635 'url': quote_cleaner(_url),
1636 'id-repr': issue_id,
1636 'id-repr': issue_id,
1637 'issue-prefix': entry['pref'],
1637 'issue-prefix': entry['pref'],
1638 'serv': entry['url'],
1638 'serv': entry['url'],
1639 'title': bleach.clean(desc, strip=True),
1639 'title': bleach.clean(desc, strip=True),
1640 'hovercard_url': hovercard_url
1640 'hovercard_url': hovercard_url
1641 }
1641 }
1642
1642
1643 if return_raw_data:
1643 if return_raw_data:
1644 return {
1644 return {
1645 'id': issue_id,
1645 'id': issue_id,
1646 'url': _url
1646 'url': _url
1647 }
1647 }
1648 return tmpl % data
1648 return tmpl % data
1649
1649
1650
1650
1651 def get_active_pattern_entries(repo_name):
1651 def get_active_pattern_entries(repo_name):
1652 repo = None
1652 repo = None
1653 if repo_name:
1653 if repo_name:
1654 # Retrieving repo_name to avoid invalid repo_name to explode on
1654 # Retrieving repo_name to avoid invalid repo_name to explode on
1655 # IssueTrackerSettingsModel but still passing invalid name further down
1655 # IssueTrackerSettingsModel but still passing invalid name further down
1656 repo = Repository.get_by_repo_name(repo_name, cache=True)
1656 repo = Repository.get_by_repo_name(repo_name, cache=True)
1657
1657
1658 settings_model = IssueTrackerSettingsModel(repo=repo)
1658 settings_model = IssueTrackerSettingsModel(repo=repo)
1659 active_entries = settings_model.get_settings(cache=True)
1659 active_entries = settings_model.get_settings(cache=True)
1660 return active_entries
1660 return active_entries
1661
1661
1662
1662
1663 pr_pattern_re = regex.compile(r'(?:(?:^!)|(?: !))(\d+)')
1663 pr_pattern_re = regex.compile(r'(?:(?:^!)|(?: !))(\d+)')
1664
1664
1665 allowed_link_formats = [
1665 allowed_link_formats = [
1666 'html', 'rst', 'markdown', 'html+hovercard', 'rst+hovercard', 'markdown+hovercard']
1666 'html', 'rst', 'markdown', 'html+hovercard', 'rst+hovercard', 'markdown+hovercard']
1667
1667
1668 compile_cache = {
1668 compile_cache = {
1669
1669
1670 }
1670 }
1671
1671
1672
1672
1673 def process_patterns(text_string, repo_name, link_format='html', active_entries=None):
1673 def process_patterns(text_string, repo_name, link_format='html', active_entries=None):
1674
1674
1675 if link_format not in allowed_link_formats:
1675 if link_format not in allowed_link_formats:
1676 raise ValueError('Link format can be only one of:{} got {}'.format(
1676 raise ValueError('Link format can be only one of:{} got {}'.format(
1677 allowed_link_formats, link_format))
1677 allowed_link_formats, link_format))
1678 issues_data = []
1678 issues_data = []
1679 errors = []
1679 errors = []
1680 new_text = text_string
1680 new_text = text_string
1681
1681
1682 if active_entries is None:
1682 if active_entries is None:
1683 log.debug('Fetch active issue tracker patterns for repo: %s', repo_name)
1683 log.debug('Fetch active issue tracker patterns for repo: %s', repo_name)
1684 active_entries = get_active_pattern_entries(repo_name)
1684 active_entries = get_active_pattern_entries(repo_name)
1685
1685
1686 log.debug('Got %s pattern entries to process', len(active_entries))
1686 log.debug('Got %s pattern entries to process', len(active_entries))
1687
1687
1688 for uid, entry in active_entries.items():
1688 for uid, entry in active_entries.items():
1689
1689
1690 if not (entry['pat'] and entry['url']):
1690 if not (entry['pat'] and entry['url']):
1691 log.debug('skipping due to missing data')
1691 log.debug('skipping due to missing data')
1692 continue
1692 continue
1693
1693
1694 log.debug('issue tracker entry: uid: `%s` PAT:%s URL:%s PREFIX:%s',
1694 log.debug('issue tracker entry: uid: `%s` PAT:%s URL:%s PREFIX:%s',
1695 uid, entry['pat'], entry['url'], entry['pref'])
1695 uid, entry['pat'], entry['url'], entry['pref'])
1696
1696
1697 if entry.get('pat_compiled'):
1697 if entry.get('pat_compiled'):
1698 pattern = entry['pat_compiled']
1698 pattern = entry['pat_compiled']
1699 elif entry['pat'] in compile_cache:
1699 elif entry['pat'] in compile_cache:
1700 pattern = compile_cache[entry['pat']]
1700 pattern = compile_cache[entry['pat']]
1701 else:
1701 else:
1702 try:
1702 try:
1703 pattern = regex.compile(r'%s' % entry['pat'])
1703 pattern = regex.compile(r'%s' % entry['pat'])
1704 except regex.error as e:
1704 except regex.error as e:
1705 regex_err = ValueError('{}:{}'.format(entry['pat'], e))
1705 regex_err = ValueError('{}:{}'.format(entry['pat'], e))
1706 log.exception('issue tracker pattern: `%s` failed to compile', regex_err)
1706 log.exception('issue tracker pattern: `%s` failed to compile', regex_err)
1707 errors.append(regex_err)
1707 errors.append(regex_err)
1708 continue
1708 continue
1709 compile_cache[entry['pat']] = pattern
1709 compile_cache[entry['pat']] = pattern
1710
1710
1711 data_func = partial(
1711 data_func = partial(
1712 _process_url_func, repo_name=repo_name, entry=entry, uid=uid,
1712 _process_url_func, repo_name=repo_name, entry=entry, uid=uid,
1713 return_raw_data=True)
1713 return_raw_data=True)
1714
1714
1715 for match_obj in pattern.finditer(text_string):
1715 for match_obj in pattern.finditer(text_string):
1716 issues_data.append(data_func(match_obj))
1716 issues_data.append(data_func(match_obj))
1717
1717
1718 url_func = partial(
1718 url_func = partial(
1719 _process_url_func, repo_name=repo_name, entry=entry, uid=uid,
1719 _process_url_func, repo_name=repo_name, entry=entry, uid=uid,
1720 link_format=link_format)
1720 link_format=link_format)
1721
1721
1722 new_text = pattern.sub(url_func, new_text)
1722 new_text = pattern.sub(url_func, new_text)
1723 log.debug('processed prefix:uid `%s`', uid)
1723 log.debug('processed prefix:uid `%s`', uid)
1724
1724
1725 # finally use global replace, eg !123 -> pr-link, those will not catch
1725 # finally use global replace, eg !123 -> pr-link, those will not catch
1726 # if already similar pattern exists
1726 # if already similar pattern exists
1727 server_url = '${scheme}://${netloc}'
1727 server_url = '${scheme}://${netloc}'
1728 pr_entry = {
1728 pr_entry = {
1729 'pref': '!',
1729 'pref': '!',
1730 'url': server_url + '/_admin/pull-requests/${id}',
1730 'url': server_url + '/_admin/pull-requests/${id}',
1731 'desc': 'Pull Request !${id}',
1731 'desc': 'Pull Request !${id}',
1732 'hovercard_url': server_url + '/_hovercard/pull_request/${id}'
1732 'hovercard_url': server_url + '/_hovercard/pull_request/${id}'
1733 }
1733 }
1734 pr_url_func = partial(
1734 pr_url_func = partial(
1735 _process_url_func, repo_name=repo_name, entry=pr_entry, uid=None,
1735 _process_url_func, repo_name=repo_name, entry=pr_entry, uid=None,
1736 link_format=link_format+'+hovercard')
1736 link_format=link_format+'+hovercard')
1737 new_text = pr_pattern_re.sub(pr_url_func, new_text)
1737 new_text = pr_pattern_re.sub(pr_url_func, new_text)
1738 log.debug('processed !pr pattern')
1738 log.debug('processed !pr pattern')
1739
1739
1740 return new_text, issues_data, errors
1740 return new_text, issues_data, errors
1741
1741
1742
1742
1743 def urlify_commit_message(commit_text, repository=None, active_pattern_entries=None,
1743 def urlify_commit_message(commit_text, repository=None, active_pattern_entries=None,
1744 issues_container_callback=None, error_container=None):
1744 issues_container_callback=None, error_container=None):
1745 """
1745 """
1746 Parses given text message and makes proper links.
1746 Parses given text message and makes proper links.
1747 issues are linked to given issue-server, and rest is a commit link
1747 issues are linked to given issue-server, and rest is a commit link
1748 """
1748 """
1749
1749
1750 def escaper(_text):
1750 def escaper(_text):
1751 return _text.replace('<', '&lt;').replace('>', '&gt;')
1751 return _text.replace('<', '&lt;').replace('>', '&gt;')
1752
1752
1753 new_text = escaper(commit_text)
1753 new_text = escaper(commit_text)
1754
1754
1755 # extract http/https links and make them real urls
1755 # extract http/https links and make them real urls
1756 new_text = urlify_text(new_text, safe=False)
1756 new_text = urlify_text(new_text, safe=False)
1757
1757
1758 # urlify commits - extract commit ids and make link out of them, if we have
1758 # urlify commits - extract commit ids and make link out of them, if we have
1759 # the scope of repository present.
1759 # the scope of repository present.
1760 if repository:
1760 if repository:
1761 new_text = urlify_commits(new_text, repository)
1761 new_text = urlify_commits(new_text, repository)
1762
1762
1763 # process issue tracker patterns
1763 # process issue tracker patterns
1764 new_text, issues, errors = process_patterns(
1764 new_text, issues, errors = process_patterns(
1765 new_text, repository or '', active_entries=active_pattern_entries)
1765 new_text, repository or '', active_entries=active_pattern_entries)
1766
1766
1767 if issues_container_callback is not None:
1767 if issues_container_callback is not None:
1768 for issue in issues:
1768 for issue in issues:
1769 issues_container_callback(issue)
1769 issues_container_callback(issue)
1770
1770
1771 if error_container is not None:
1771 if error_container is not None:
1772 error_container.extend(errors)
1772 error_container.extend(errors)
1773
1773
1774 return literal(new_text)
1774 return literal(new_text)
1775
1775
1776
1776
1777 def render_binary(repo_name, file_obj):
1777 def render_binary(repo_name, file_obj):
1778 """
1778 """
1779 Choose how to render a binary file
1779 Choose how to render a binary file
1780 """
1780 """
1781
1781
1782 # unicode
1782 # unicode
1783 filename = file_obj.name
1783 filename = file_obj.name
1784
1784
1785 # images
1785 # images
1786 for ext in ['*.png', '*.jpeg', '*.jpg', '*.ico', '*.gif']:
1786 for ext in ['*.png', '*.jpeg', '*.jpg', '*.ico', '*.gif']:
1787 if fnmatch.fnmatch(filename, pat=ext):
1787 if fnmatch.fnmatch(filename, pat=ext):
1788 src = route_path(
1788 src = route_path(
1789 'repo_file_raw', repo_name=repo_name,
1789 'repo_file_raw', repo_name=repo_name,
1790 commit_id=file_obj.commit.raw_id,
1790 commit_id=file_obj.commit.raw_id,
1791 f_path=file_obj.path)
1791 f_path=file_obj.path)
1792
1792
1793 return literal(
1793 return literal(
1794 '<img class="rendered-binary" alt="rendered-image" src="{}">'.format(src))
1794 '<img class="rendered-binary" alt="rendered-image" src="{}">'.format(src))
1795
1795
1796
1796
1797 def renderer_from_filename(filename, exclude=None):
1797 def renderer_from_filename(filename, exclude=None):
1798 """
1798 """
1799 choose a renderer based on filename, this works only for text based files
1799 choose a renderer based on filename, this works only for text based files
1800 """
1800 """
1801
1801
1802 # ipython
1802 # ipython
1803 for ext in ['*.ipynb']:
1803 for ext in ['*.ipynb']:
1804 if fnmatch.fnmatch(filename, pat=ext):
1804 if fnmatch.fnmatch(filename, pat=ext):
1805 return 'jupyter'
1805 return 'jupyter'
1806
1806
1807 is_markup = MarkupRenderer.renderer_from_filename(filename, exclude=exclude)
1807 is_markup = MarkupRenderer.renderer_from_filename(filename, exclude=exclude)
1808 if is_markup:
1808 if is_markup:
1809 return is_markup
1809 return is_markup
1810 return None
1810 return None
1811
1811
1812
1812
1813 def render(source, renderer='rst', mentions=False, relative_urls=None,
1813 def render(source, renderer='rst', mentions=False, relative_urls=None,
1814 repo_name=None, active_pattern_entries=None, issues_container_callback=None):
1814 repo_name=None, active_pattern_entries=None, issues_container_callback=None):
1815
1815
1816 def maybe_convert_relative_links(html_source):
1816 def maybe_convert_relative_links(html_source):
1817 if relative_urls:
1817 if relative_urls:
1818 return relative_links(html_source, relative_urls)
1818 return relative_links(html_source, relative_urls)
1819 return html_source
1819 return html_source
1820
1820
1821 if renderer == 'plain':
1821 if renderer == 'plain':
1822 return literal(
1822 return literal(
1823 MarkupRenderer.plain(source, leading_newline=False))
1823 MarkupRenderer.plain(source, leading_newline=False))
1824
1824
1825 elif renderer == 'rst':
1825 elif renderer == 'rst':
1826 if repo_name:
1826 if repo_name:
1827 # process patterns on comments if we pass in repo name
1827 # process patterns on comments if we pass in repo name
1828 source, issues, errors = process_patterns(
1828 source, issues, errors = process_patterns(
1829 source, repo_name, link_format='rst',
1829 source, repo_name, link_format='rst',
1830 active_entries=active_pattern_entries)
1830 active_entries=active_pattern_entries)
1831 if issues_container_callback is not None:
1831 if issues_container_callback is not None:
1832 for issue in issues:
1832 for issue in issues:
1833 issues_container_callback(issue)
1833 issues_container_callback(issue)
1834
1834
1835 return literal(
1835 return literal(
1836 '<div class="rst-block">%s</div>' %
1836 '<div class="rst-block">%s</div>' %
1837 maybe_convert_relative_links(
1837 maybe_convert_relative_links(
1838 MarkupRenderer.rst(source, mentions=mentions)))
1838 MarkupRenderer.rst(source, mentions=mentions)))
1839
1839
1840 elif renderer == 'markdown':
1840 elif renderer == 'markdown':
1841 if repo_name:
1841 if repo_name:
1842 # process patterns on comments if we pass in repo name
1842 # process patterns on comments if we pass in repo name
1843 source, issues, errors = process_patterns(
1843 source, issues, errors = process_patterns(
1844 source, repo_name, link_format='markdown',
1844 source, repo_name, link_format='markdown',
1845 active_entries=active_pattern_entries)
1845 active_entries=active_pattern_entries)
1846 if issues_container_callback is not None:
1846 if issues_container_callback is not None:
1847 for issue in issues:
1847 for issue in issues:
1848 issues_container_callback(issue)
1848 issues_container_callback(issue)
1849
1849
1850
1850
1851 return literal(
1851 return literal(
1852 '<div class="markdown-block">%s</div>' %
1852 '<div class="markdown-block">%s</div>' %
1853 maybe_convert_relative_links(
1853 maybe_convert_relative_links(
1854 MarkupRenderer.markdown(source, flavored=True,
1854 MarkupRenderer.markdown(source, flavored=True,
1855 mentions=mentions)))
1855 mentions=mentions)))
1856
1856
1857 elif renderer == 'jupyter':
1857 elif renderer == 'jupyter':
1858 return literal(
1858 return literal(
1859 '<div class="ipynb">%s</div>' %
1859 '<div class="ipynb">%s</div>' %
1860 maybe_convert_relative_links(
1860 maybe_convert_relative_links(
1861 MarkupRenderer.jupyter(source)))
1861 MarkupRenderer.jupyter(source)))
1862
1862
1863 # None means just show the file-source
1863 # None means just show the file-source
1864 return None
1864 return None
1865
1865
1866
1866
1867 def commit_status(repo, commit_id):
1867 def commit_status(repo, commit_id):
1868 return ChangesetStatusModel().get_status(repo, commit_id)
1868 return ChangesetStatusModel().get_status(repo, commit_id)
1869
1869
1870
1870
1871 def commit_status_lbl(commit_status):
1871 def commit_status_lbl(commit_status):
1872 return dict(ChangesetStatus.STATUSES).get(commit_status)
1872 return dict(ChangesetStatus.STATUSES).get(commit_status)
1873
1873
1874
1874
1875 def commit_time(repo_name, commit_id):
1875 def commit_time(repo_name, commit_id):
1876 repo = Repository.get_by_repo_name(repo_name)
1876 repo = Repository.get_by_repo_name(repo_name)
1877 commit = repo.get_commit(commit_id=commit_id)
1877 commit = repo.get_commit(commit_id=commit_id)
1878 return commit.date
1878 return commit.date
1879
1879
1880
1880
1881 def get_permission_name(key):
1881 def get_permission_name(key):
1882 return dict(Permission.PERMS).get(key)
1882 return dict(Permission.PERMS).get(key)
1883
1883
1884
1884
1885 def journal_filter_help(request):
1885 def journal_filter_help(request):
1886 _ = request.translate
1886 _ = request.translate
1887 from rhodecode.lib.audit_logger import ACTIONS
1887 from rhodecode.lib.audit_logger import ACTIONS
1888 actions = '\n'.join(textwrap.wrap(', '.join(sorted(ACTIONS.keys())), 80))
1888 actions = '\n'.join(textwrap.wrap(', '.join(sorted(ACTIONS.keys())), 80))
1889
1889
1890 return _(
1890 return _(
1891 'Example filter terms:\n' +
1891 'Example filter terms:\n' +
1892 ' repository:vcs\n' +
1892 ' repository:vcs\n' +
1893 ' username:marcin\n' +
1893 ' username:marcin\n' +
1894 ' username:(NOT marcin)\n' +
1894 ' username:(NOT marcin)\n' +
1895 ' action:*push*\n' +
1895 ' action:*push*\n' +
1896 ' ip:127.0.0.1\n' +
1896 ' ip:127.0.0.1\n' +
1897 ' date:20120101\n' +
1897 ' date:20120101\n' +
1898 ' date:[20120101100000 TO 20120102]\n' +
1898 ' date:[20120101100000 TO 20120102]\n' +
1899 '\n' +
1899 '\n' +
1900 'Actions: {actions}\n' +
1900 'Actions: {actions}\n' +
1901 '\n' +
1901 '\n' +
1902 'Generate wildcards using \'*\' character:\n' +
1902 'Generate wildcards using \'*\' character:\n' +
1903 ' "repository:vcs*" - search everything starting with \'vcs\'\n' +
1903 ' "repository:vcs*" - search everything starting with \'vcs\'\n' +
1904 ' "repository:*vcs*" - search for repository containing \'vcs\'\n' +
1904 ' "repository:*vcs*" - search for repository containing \'vcs\'\n' +
1905 '\n' +
1905 '\n' +
1906 'Optional AND / OR operators in queries\n' +
1906 'Optional AND / OR operators in queries\n' +
1907 ' "repository:vcs OR repository:test"\n' +
1907 ' "repository:vcs OR repository:test"\n' +
1908 ' "username:test AND repository:test*"\n'
1908 ' "username:test AND repository:test*"\n'
1909 ).format(actions=actions)
1909 ).format(actions=actions)
1910
1910
1911
1911
1912 def not_mapped_error(repo_name):
1912 def not_mapped_error(repo_name):
1913 from rhodecode.translation import _
1913 from rhodecode.translation import _
1914 flash(_('%s repository is not mapped to db perhaps'
1914 flash(_('%s repository is not mapped to db perhaps'
1915 ' it was created or renamed from the filesystem'
1915 ' it was created or renamed from the filesystem'
1916 ' please run the application again'
1916 ' please run the application again'
1917 ' in order to rescan repositories') % repo_name, category='error')
1917 ' in order to rescan repositories') % repo_name, category='error')
1918
1918
1919
1919
1920 def ip_range(ip_addr):
1920 def ip_range(ip_addr):
1921 from rhodecode.model.db import UserIpMap
1921 from rhodecode.model.db import UserIpMap
1922 s, e = UserIpMap._get_ip_range(ip_addr)
1922 s, e = UserIpMap._get_ip_range(ip_addr)
1923 return '%s - %s' % (s, e)
1923 return '%s - %s' % (s, e)
1924
1924
1925
1925
1926 def form(url, method='post', needs_csrf_token=True, **attrs):
1926 def form(url, method='post', needs_csrf_token=True, **attrs):
1927 """Wrapper around webhelpers.tags.form to prevent CSRF attacks."""
1927 """Wrapper around webhelpers.tags.form to prevent CSRF attacks."""
1928 if method.lower() != 'get' and needs_csrf_token:
1928 if method.lower() != 'get' and needs_csrf_token:
1929 raise Exception(
1929 raise Exception(
1930 'Forms to POST/PUT/DELETE endpoints should have (in general) a ' +
1930 'Forms to POST/PUT/DELETE endpoints should have (in general) a ' +
1931 'CSRF token. If the endpoint does not require such token you can ' +
1931 'CSRF token. If the endpoint does not require such token you can ' +
1932 'explicitly set the parameter needs_csrf_token to false.')
1932 'explicitly set the parameter needs_csrf_token to false.')
1933
1933
1934 return insecure_form(url, method=method, **attrs)
1934 return insecure_form(url, method=method, **attrs)
1935
1935
1936
1936
1937 def secure_form(form_url, method="POST", multipart=False, **attrs):
1937 def secure_form(form_url, method="POST", multipart=False, **attrs):
1938 """Start a form tag that points the action to an url. This
1938 """Start a form tag that points the action to an url. This
1939 form tag will also include the hidden field containing
1939 form tag will also include the hidden field containing
1940 the auth token.
1940 the auth token.
1941
1941
1942 The url options should be given either as a string, or as a
1942 The url options should be given either as a string, or as a
1943 ``url()`` function. The method for the form defaults to POST.
1943 ``url()`` function. The method for the form defaults to POST.
1944
1944
1945 Options:
1945 Options:
1946
1946
1947 ``multipart``
1947 ``multipart``
1948 If set to True, the enctype is set to "multipart/form-data".
1948 If set to True, the enctype is set to "multipart/form-data".
1949 ``method``
1949 ``method``
1950 The method to use when submitting the form, usually either
1950 The method to use when submitting the form, usually either
1951 "GET" or "POST". If "PUT", "DELETE", or another verb is used, a
1951 "GET" or "POST". If "PUT", "DELETE", or another verb is used, a
1952 hidden input with name _method is added to simulate the verb
1952 hidden input with name _method is added to simulate the verb
1953 over POST.
1953 over POST.
1954
1954
1955 """
1955 """
1956
1956
1957 if 'request' in attrs:
1957 if 'request' in attrs:
1958 session = attrs['request'].session
1958 session = attrs['request'].session
1959 del attrs['request']
1959 del attrs['request']
1960 else:
1960 else:
1961 raise ValueError(
1961 raise ValueError(
1962 'Calling this form requires request= to be passed as argument')
1962 'Calling this form requires request= to be passed as argument')
1963
1963
1964 _form = insecure_form(form_url, method, multipart, **attrs)
1964 _form = insecure_form(form_url, method, multipart, **attrs)
1965 token = literal(
1965 token = literal(
1966 '<input type="hidden" name="{}" value="{}">'.format(
1966 '<input type="hidden" name="{}" value="{}">'.format(
1967 csrf_token_key, get_csrf_token(session)))
1967 csrf_token_key, get_csrf_token(session)))
1968
1968
1969 return literal("%s\n%s" % (_form, token))
1969 return literal("%s\n%s" % (_form, token))
1970
1970
1971
1971
1972 def dropdownmenu(name, selected, options, enable_filter=False, **attrs):
1972 def dropdownmenu(name, selected, options, enable_filter=False, **attrs):
1973 select_html = select(name, selected, options, **attrs)
1973 select_html = select(name, selected, options, **attrs)
1974
1974
1975 select2 = """
1975 select2 = """
1976 <script>
1976 <script>
1977 $(document).ready(function() {
1977 $(document).ready(function() {
1978 $('#%s').select2({
1978 $('#%s').select2({
1979 containerCssClass: 'drop-menu %s',
1979 containerCssClass: 'drop-menu %s',
1980 dropdownCssClass: 'drop-menu-dropdown',
1980 dropdownCssClass: 'drop-menu-dropdown',
1981 dropdownAutoWidth: true%s
1981 dropdownAutoWidth: true%s
1982 });
1982 });
1983 });
1983 });
1984 </script>
1984 </script>
1985 """
1985 """
1986
1986
1987 filter_option = """,
1987 filter_option = """,
1988 minimumResultsForSearch: -1
1988 minimumResultsForSearch: -1
1989 """
1989 """
1990 input_id = attrs.get('id') or name
1990 input_id = attrs.get('id') or name
1991 extra_classes = ' '.join(attrs.pop('extra_classes', []))
1991 extra_classes = ' '.join(attrs.pop('extra_classes', []))
1992 filter_enabled = "" if enable_filter else filter_option
1992 filter_enabled = "" if enable_filter else filter_option
1993 select_script = literal(select2 % (input_id, extra_classes, filter_enabled))
1993 select_script = literal(select2 % (input_id, extra_classes, filter_enabled))
1994
1994
1995 return literal(select_html+select_script)
1995 return literal(select_html+select_script)
1996
1996
1997
1997
1998 def get_visual_attr(tmpl_context_var, attr_name):
1998 def get_visual_attr(tmpl_context_var, attr_name):
1999 """
1999 """
2000 A safe way to get a variable from visual variable of template context
2000 A safe way to get a variable from visual variable of template context
2001
2001
2002 :param tmpl_context_var: instance of tmpl_context, usually present as `c`
2002 :param tmpl_context_var: instance of tmpl_context, usually present as `c`
2003 :param attr_name: name of the attribute we fetch from the c.visual
2003 :param attr_name: name of the attribute we fetch from the c.visual
2004 """
2004 """
2005 visual = getattr(tmpl_context_var, 'visual', None)
2005 visual = getattr(tmpl_context_var, 'visual', None)
2006 if not visual:
2006 if not visual:
2007 return
2007 return
2008 else:
2008 else:
2009 return getattr(visual, attr_name, None)
2009 return getattr(visual, attr_name, None)
2010
2010
2011
2011
2012 def get_last_path_part(file_node):
2012 def get_last_path_part(file_node):
2013 if not file_node.path:
2013 if not file_node.path:
2014 return u'/'
2014 return u'/'
2015
2015
2016 path = safe_unicode(file_node.path.split('/')[-1])
2016 path = safe_unicode(file_node.path.split('/')[-1])
2017 return u'../' + path
2017 return u'../' + path
2018
2018
2019
2019
2020 def route_url(*args, **kwargs):
2020 def route_url(*args, **kwargs):
2021 """
2021 """
2022 Wrapper around pyramids `route_url` (fully qualified url) function.
2022 Wrapper around pyramids `route_url` (fully qualified url) function.
2023 """
2023 """
2024 req = get_current_request()
2024 req = get_current_request()
2025 return req.route_url(*args, **kwargs)
2025 return req.route_url(*args, **kwargs)
2026
2026
2027
2027
2028 def route_path(*args, **kwargs):
2028 def route_path(*args, **kwargs):
2029 """
2029 """
2030 Wrapper around pyramids `route_path` function.
2030 Wrapper around pyramids `route_path` function.
2031 """
2031 """
2032 req = get_current_request()
2032 req = get_current_request()
2033 return req.route_path(*args, **kwargs)
2033 return req.route_path(*args, **kwargs)
2034
2034
2035
2035
2036 def route_path_or_none(*args, **kwargs):
2036 def route_path_or_none(*args, **kwargs):
2037 try:
2037 try:
2038 return route_path(*args, **kwargs)
2038 return route_path(*args, **kwargs)
2039 except KeyError:
2039 except KeyError:
2040 return None
2040 return None
2041
2041
2042
2042
2043 def current_route_path(request, **kw):
2043 def current_route_path(request, **kw):
2044 new_args = request.GET.mixed()
2044 new_args = request.GET.mixed()
2045 new_args.update(kw)
2045 new_args.update(kw)
2046 return request.current_route_path(_query=new_args)
2046 return request.current_route_path(_query=new_args)
2047
2047
2048
2048
2049 def curl_api_example(method, args):
2049 def curl_api_example(method, args):
2050 args_json = json.dumps(OrderedDict([
2050 args_json = json.dumps(OrderedDict([
2051 ('id', 1),
2051 ('id', 1),
2052 ('auth_token', 'SECRET'),
2052 ('auth_token', 'SECRET'),
2053 ('method', method),
2053 ('method', method),
2054 ('args', args)
2054 ('args', args)
2055 ]))
2055 ]))
2056
2056
2057 return "curl {api_url} -X POST -H 'content-type:text/plain' --data-binary '{args_json}'".format(
2057 return "curl {api_url} -X POST -H 'content-type:text/plain' --data-binary '{args_json}'".format(
2058 api_url=route_url('apiv2'),
2058 api_url=route_url('apiv2'),
2059 args_json=args_json
2059 args_json=args_json
2060 )
2060 )
2061
2061
2062
2062
2063 def api_call_example(method, args):
2063 def api_call_example(method, args):
2064 """
2064 """
2065 Generates an API call example via CURL
2065 Generates an API call example via CURL
2066 """
2066 """
2067 curl_call = curl_api_example(method, args)
2067 curl_call = curl_api_example(method, args)
2068
2068
2069 return literal(
2069 return literal(
2070 curl_call +
2070 curl_call +
2071 "<br/><br/>SECRET can be found in <a href=\"{token_url}\">auth-tokens</a> page, "
2071 "<br/><br/>SECRET can be found in <a href=\"{token_url}\">auth-tokens</a> page, "
2072 "and needs to be of `api calls` role."
2072 "and needs to be of `api calls` role."
2073 .format(token_url=route_url('my_account_auth_tokens')))
2073 .format(token_url=route_url('my_account_auth_tokens')))
2074
2074
2075
2075
2076 def notification_description(notification, request):
2076 def notification_description(notification, request):
2077 """
2077 """
2078 Generate notification human readable description based on notification type
2078 Generate notification human readable description based on notification type
2079 """
2079 """
2080 from rhodecode.model.notification import NotificationModel
2080 from rhodecode.model.notification import NotificationModel
2081 return NotificationModel().make_description(
2081 return NotificationModel().make_description(
2082 notification, translate=request.translate)
2082 notification, translate=request.translate)
2083
2083
2084
2084
2085 def go_import_header(request, db_repo=None):
2085 def go_import_header(request, db_repo=None):
2086 """
2086 """
2087 Creates a header for go-import functionality in Go Lang
2087 Creates a header for go-import functionality in Go Lang
2088 """
2088 """
2089
2089
2090 if not db_repo:
2090 if not db_repo:
2091 return
2091 return
2092 if 'go-get' not in request.GET:
2092 if 'go-get' not in request.GET:
2093 return
2093 return
2094
2094
2095 clone_url = db_repo.clone_url()
2095 clone_url = db_repo.clone_url()
2096 prefix = re.split(r'^https?:\/\/', clone_url)[-1]
2096 prefix = re.split(r'^https?:\/\/', clone_url)[-1]
2097 # we have a repo and go-get flag,
2097 # we have a repo and go-get flag,
2098 return literal('<meta name="go-import" content="{} {} {}">'.format(
2098 return literal('<meta name="go-import" content="{} {} {}">'.format(
2099 prefix, db_repo.repo_type, clone_url))
2099 prefix, db_repo.repo_type, clone_url))
2100
2100
2101
2101
2102 def reviewer_as_json(*args, **kwargs):
2102 def reviewer_as_json(*args, **kwargs):
2103 from rhodecode.apps.repository.utils import reviewer_as_json as _reviewer_as_json
2103 from rhodecode.apps.repository.utils import reviewer_as_json as _reviewer_as_json
2104 return _reviewer_as_json(*args, **kwargs)
2104 return _reviewer_as_json(*args, **kwargs)
2105
2105
2106
2106
2107 def get_repo_view_type(request):
2107 def get_repo_view_type(request):
2108 route_name = request.matched_route.name
2108 route_name = request.matched_route.name
2109 route_to_view_type = {
2109 route_to_view_type = {
2110 'repo_changelog': 'commits',
2110 'repo_changelog': 'commits',
2111 'repo_commits': 'commits',
2111 'repo_commits': 'commits',
2112 'repo_files': 'files',
2112 'repo_files': 'files',
2113 'repo_summary': 'summary',
2113 'repo_summary': 'summary',
2114 'repo_commit': 'commit'
2114 'repo_commit': 'commit'
2115 }
2115 }
2116
2116
2117 return route_to_view_type.get(route_name)
2117 return route_to_view_type.get(route_name)
2118
2118
2119
2119
2120 def is_active(menu_entry, selected):
2120 def is_active(menu_entry, selected):
2121 """
2121 """
2122 Returns active class for selecting menus in templates
2122 Returns active class for selecting menus in templates
2123 <li class=${h.is_active('settings', current_active)}></li>
2123 <li class=${h.is_active('settings', current_active)}></li>
2124 """
2124 """
2125 if not isinstance(menu_entry, list):
2125 if not isinstance(menu_entry, list):
2126 menu_entry = [menu_entry]
2126 menu_entry = [menu_entry]
2127
2127
2128 if selected in menu_entry:
2128 if selected in menu_entry:
2129 return "active"
2129 return "active"
2130
2130
2131
2131
2132 class IssuesRegistry(object):
2132 class IssuesRegistry(object):
2133 """
2133 """
2134 issue_registry = IssuesRegistry()
2134 issue_registry = IssuesRegistry()
2135 some_func(issues_callback=issues_registry(...))
2135 some_func(issues_callback=issues_registry(...))
2136 """
2136 """
2137
2137
2138 def __init__(self):
2138 def __init__(self):
2139 self.issues = []
2139 self.issues = []
2140 self.unique_issues = collections.defaultdict(lambda: [])
2140 self.unique_issues = collections.defaultdict(lambda: [])
2141
2141
2142 def __call__(self, commit_dict=None):
2142 def __call__(self, commit_dict=None):
2143 def callback(issue):
2143 def callback(issue):
2144 if commit_dict and issue:
2144 if commit_dict and issue:
2145 issue['commit'] = commit_dict
2145 issue['commit'] = commit_dict
2146 self.issues.append(issue)
2146 self.issues.append(issue)
2147 self.unique_issues[issue['id']].append(issue)
2147 self.unique_issues[issue['id']].append(issue)
2148 return callback
2148 return callback
2149
2149
2150 def get_issues(self):
2150 def get_issues(self):
2151 return self.issues
2151 return self.issues
2152
2152
2153 @property
2153 @property
2154 def issues_unique_count(self):
2154 def issues_unique_count(self):
2155 return len(set(i['id'] for i in self.issues))
2155 return len(set(i['id'] for i in self.issues))
@@ -1,679 +1,679 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2014-2020 RhodeCode GmbH
3 # Copyright (C) 2014-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 """
21 """
22 SimpleVCS middleware for handling protocol request (push/clone etc.)
22 SimpleVCS middleware for handling protocol request (push/clone etc.)
23 It's implemented with basic auth function
23 It's implemented with basic auth function
24 """
24 """
25
25
26 import os
26 import os
27 import re
27 import re
28 import logging
28 import logging
29 import importlib
29 import importlib
30 from functools import wraps
30 from functools import wraps
31 from io import StringIO
31 from io import StringIO
32 from lxml import etree
32 from lxml import etree
33
33
34 import time
34 import time
35 from paste.httpheaders import REMOTE_USER, AUTH_TYPE
35 from paste.httpheaders import REMOTE_USER, AUTH_TYPE
36
36
37 from pyramid.httpexceptions import (
37 from pyramid.httpexceptions import (
38 HTTPNotFound, HTTPForbidden, HTTPNotAcceptable, HTTPInternalServerError)
38 HTTPNotFound, HTTPForbidden, HTTPNotAcceptable, HTTPInternalServerError)
39 from zope.cachedescriptors.property import Lazy as LazyProperty
39 from zope.cachedescriptors.property import Lazy as LazyProperty
40
40
41 import rhodecode
41 import rhodecode
42 from rhodecode.authentication.base import authenticate, VCS_TYPE, loadplugin
42 from rhodecode.authentication.base import authenticate, VCS_TYPE, loadplugin
43 from rhodecode.lib import rc_cache
43 from rhodecode.lib import rc_cache
44 from rhodecode.lib.auth import AuthUser, HasPermissionAnyMiddleware
44 from rhodecode.lib.auth import AuthUser, HasPermissionAnyMiddleware
45 from rhodecode.lib.base import (
45 from rhodecode.lib.base import (
46 BasicAuth, get_ip_addr, get_user_agent, vcs_operation_context)
46 BasicAuth, get_ip_addr, get_user_agent, vcs_operation_context)
47 from rhodecode.lib.exceptions import (UserCreationError, NotAllowedToCreateUserError)
47 from rhodecode.lib.exceptions import (UserCreationError, NotAllowedToCreateUserError)
48 from rhodecode.lib.hooks_daemon import prepare_callback_daemon
48 from rhodecode.lib.hooks_daemon import prepare_callback_daemon
49 from rhodecode.lib.middleware import appenlight
49 from rhodecode.lib.middleware import appenlight
50 from rhodecode.lib.middleware.utils import scm_app_http
50 from rhodecode.lib.middleware.utils import scm_app_http
51 from rhodecode.lib.utils import is_valid_repo, SLUG_RE
51 from rhodecode.lib.utils import is_valid_repo, SLUG_RE
52 from rhodecode.lib.utils2 import safe_str, fix_PATH, str2bool, safe_unicode
52 from rhodecode.lib.utils2 import safe_str, fix_PATH, str2bool, safe_unicode
53 from rhodecode.lib.vcs.conf import settings as vcs_settings
53 from rhodecode.lib.vcs.conf import settings as vcs_settings
54 from rhodecode.lib.vcs.backends import base
54 from rhodecode.lib.vcs.backends import base
55
55
56 from rhodecode.model import meta
56 from rhodecode.model import meta
57 from rhodecode.model.db import User, Repository, PullRequest
57 from rhodecode.model.db import User, Repository, PullRequest
58 from rhodecode.model.scm import ScmModel
58 from rhodecode.model.scm import ScmModel
59 from rhodecode.model.pull_request import PullRequestModel
59 from rhodecode.model.pull_request import PullRequestModel
60 from rhodecode.model.settings import SettingsModel, VcsSettingsModel
60 from rhodecode.model.settings import SettingsModel, VcsSettingsModel
61
61
62 log = logging.getLogger(__name__)
62 log = logging.getLogger(__name__)
63
63
64
64
65 def extract_svn_txn_id(acl_repo_name, data):
65 def extract_svn_txn_id(acl_repo_name, data):
66 """
66 """
67 Helper method for extraction of svn txn_id from submitted XML data during
67 Helper method for extraction of svn txn_id from submitted XML data during
68 POST operations
68 POST operations
69 """
69 """
70 try:
70 try:
71 root = etree.fromstring(data)
71 root = etree.fromstring(data)
72 pat = re.compile(r'/txn/(?P<txn_id>.*)')
72 pat = re.compile(r'/txn/(?P<txn_id>.*)')
73 for el in root:
73 for el in root:
74 if el.tag == '{DAV:}source':
74 if el.tag == '{DAV:}source':
75 for sub_el in el:
75 for sub_el in el:
76 if sub_el.tag == '{DAV:}href':
76 if sub_el.tag == '{DAV:}href':
77 match = pat.search(sub_el.text)
77 match = pat.search(sub_el.text)
78 if match:
78 if match:
79 svn_tx_id = match.groupdict()['txn_id']
79 svn_tx_id = match.groupdict()['txn_id']
80 txn_id = rc_cache.utils.compute_key_from_params(
80 txn_id = rc_cache.utils.compute_key_from_params(
81 acl_repo_name, svn_tx_id)
81 acl_repo_name, svn_tx_id)
82 return txn_id
82 return txn_id
83 except Exception:
83 except Exception:
84 log.exception('Failed to extract txn_id')
84 log.exception('Failed to extract txn_id')
85
85
86
86
87 def initialize_generator(factory):
87 def initialize_generator(factory):
88 """
88 """
89 Initializes the returned generator by draining its first element.
89 Initializes the returned generator by draining its first element.
90
90
91 This can be used to give a generator an initializer, which is the code
91 This can be used to give a generator an initializer, which is the code
92 up to the first yield statement. This decorator enforces that the first
92 up to the first yield statement. This decorator enforces that the first
93 produced element has the value ``"__init__"`` to make its special
93 produced element has the value ``"__init__"`` to make its special
94 purpose very explicit in the using code.
94 purpose very explicit in the using code.
95 """
95 """
96
96
97 @wraps(factory)
97 @wraps(factory)
98 def wrapper(*args, **kwargs):
98 def wrapper(*args, **kwargs):
99 gen = factory(*args, **kwargs)
99 gen = factory(*args, **kwargs)
100 try:
100 try:
101 init = gen.next()
101 init = next(gen)
102 except StopIteration:
102 except StopIteration:
103 raise ValueError('Generator must yield at least one element.')
103 raise ValueError('Generator must yield at least one element.')
104 if init != "__init__":
104 if init != "__init__":
105 raise ValueError('First yielded element must be "__init__".')
105 raise ValueError('First yielded element must be "__init__".')
106 return gen
106 return gen
107 return wrapper
107 return wrapper
108
108
109
109
110 class SimpleVCS(object):
110 class SimpleVCS(object):
111 """Common functionality for SCM HTTP handlers."""
111 """Common functionality for SCM HTTP handlers."""
112
112
113 SCM = 'unknown'
113 SCM = 'unknown'
114
114
115 acl_repo_name = None
115 acl_repo_name = None
116 url_repo_name = None
116 url_repo_name = None
117 vcs_repo_name = None
117 vcs_repo_name = None
118 rc_extras = {}
118 rc_extras = {}
119
119
120 # We have to handle requests to shadow repositories different than requests
120 # We have to handle requests to shadow repositories different than requests
121 # to normal repositories. Therefore we have to distinguish them. To do this
121 # to normal repositories. Therefore we have to distinguish them. To do this
122 # we use this regex which will match only on URLs pointing to shadow
122 # we use this regex which will match only on URLs pointing to shadow
123 # repositories.
123 # repositories.
124 shadow_repo_re = re.compile(
124 shadow_repo_re = re.compile(
125 '(?P<groups>(?:{slug_pat}/)*)' # repo groups
125 '(?P<groups>(?:{slug_pat}/)*)' # repo groups
126 '(?P<target>{slug_pat})/' # target repo
126 '(?P<target>{slug_pat})/' # target repo
127 'pull-request/(?P<pr_id>\d+)/' # pull request
127 'pull-request/(?P<pr_id>\d+)/' # pull request
128 'repository$' # shadow repo
128 'repository$' # shadow repo
129 .format(slug_pat=SLUG_RE.pattern))
129 .format(slug_pat=SLUG_RE.pattern))
130
130
131 def __init__(self, config, registry):
131 def __init__(self, config, registry):
132 self.registry = registry
132 self.registry = registry
133 self.config = config
133 self.config = config
134 # re-populated by specialized middleware
134 # re-populated by specialized middleware
135 self.repo_vcs_config = base.Config()
135 self.repo_vcs_config = base.Config()
136
136
137 rc_settings = SettingsModel().get_all_settings(cache=True, from_request=False)
137 rc_settings = SettingsModel().get_all_settings(cache=True, from_request=False)
138 realm = rc_settings.get('rhodecode_realm') or 'RhodeCode AUTH'
138 realm = rc_settings.get('rhodecode_realm') or 'RhodeCode AUTH'
139
139
140 # authenticate this VCS request using authfunc
140 # authenticate this VCS request using authfunc
141 auth_ret_code_detection = \
141 auth_ret_code_detection = \
142 str2bool(self.config.get('auth_ret_code_detection', False))
142 str2bool(self.config.get('auth_ret_code_detection', False))
143 self.authenticate = BasicAuth(
143 self.authenticate = BasicAuth(
144 '', authenticate, registry, config.get('auth_ret_code'),
144 '', authenticate, registry, config.get('auth_ret_code'),
145 auth_ret_code_detection, rc_realm=realm)
145 auth_ret_code_detection, rc_realm=realm)
146 self.ip_addr = '0.0.0.0'
146 self.ip_addr = '0.0.0.0'
147
147
148 @LazyProperty
148 @LazyProperty
149 def global_vcs_config(self):
149 def global_vcs_config(self):
150 try:
150 try:
151 return VcsSettingsModel().get_ui_settings_as_config_obj()
151 return VcsSettingsModel().get_ui_settings_as_config_obj()
152 except Exception:
152 except Exception:
153 return base.Config()
153 return base.Config()
154
154
155 @property
155 @property
156 def base_path(self):
156 def base_path(self):
157 settings_path = self.repo_vcs_config.get(*VcsSettingsModel.PATH_SETTING)
157 settings_path = self.repo_vcs_config.get(*VcsSettingsModel.PATH_SETTING)
158
158
159 if not settings_path:
159 if not settings_path:
160 settings_path = self.global_vcs_config.get(*VcsSettingsModel.PATH_SETTING)
160 settings_path = self.global_vcs_config.get(*VcsSettingsModel.PATH_SETTING)
161
161
162 if not settings_path:
162 if not settings_path:
163 # try, maybe we passed in explicitly as config option
163 # try, maybe we passed in explicitly as config option
164 settings_path = self.config.get('base_path')
164 settings_path = self.config.get('base_path')
165
165
166 if not settings_path:
166 if not settings_path:
167 raise ValueError('FATAL: base_path is empty')
167 raise ValueError('FATAL: base_path is empty')
168 return settings_path
168 return settings_path
169
169
170 def set_repo_names(self, environ):
170 def set_repo_names(self, environ):
171 """
171 """
172 This will populate the attributes acl_repo_name, url_repo_name,
172 This will populate the attributes acl_repo_name, url_repo_name,
173 vcs_repo_name and is_shadow_repo. In case of requests to normal (non
173 vcs_repo_name and is_shadow_repo. In case of requests to normal (non
174 shadow) repositories all names are equal. In case of requests to a
174 shadow) repositories all names are equal. In case of requests to a
175 shadow repository the acl-name points to the target repo of the pull
175 shadow repository the acl-name points to the target repo of the pull
176 request and the vcs-name points to the shadow repo file system path.
176 request and the vcs-name points to the shadow repo file system path.
177 The url-name is always the URL used by the vcs client program.
177 The url-name is always the URL used by the vcs client program.
178
178
179 Example in case of a shadow repo:
179 Example in case of a shadow repo:
180 acl_repo_name = RepoGroup/MyRepo
180 acl_repo_name = RepoGroup/MyRepo
181 url_repo_name = RepoGroup/MyRepo/pull-request/3/repository
181 url_repo_name = RepoGroup/MyRepo/pull-request/3/repository
182 vcs_repo_name = /repo/base/path/RepoGroup/.__shadow_MyRepo_pr-3'
182 vcs_repo_name = /repo/base/path/RepoGroup/.__shadow_MyRepo_pr-3'
183 """
183 """
184 # First we set the repo name from URL for all attributes. This is the
184 # First we set the repo name from URL for all attributes. This is the
185 # default if handling normal (non shadow) repo requests.
185 # default if handling normal (non shadow) repo requests.
186 self.url_repo_name = self._get_repository_name(environ)
186 self.url_repo_name = self._get_repository_name(environ)
187 self.acl_repo_name = self.vcs_repo_name = self.url_repo_name
187 self.acl_repo_name = self.vcs_repo_name = self.url_repo_name
188 self.is_shadow_repo = False
188 self.is_shadow_repo = False
189
189
190 # Check if this is a request to a shadow repository.
190 # Check if this is a request to a shadow repository.
191 match = self.shadow_repo_re.match(self.url_repo_name)
191 match = self.shadow_repo_re.match(self.url_repo_name)
192 if match:
192 if match:
193 match_dict = match.groupdict()
193 match_dict = match.groupdict()
194
194
195 # Build acl repo name from regex match.
195 # Build acl repo name from regex match.
196 acl_repo_name = safe_unicode('{groups}{target}'.format(
196 acl_repo_name = safe_unicode('{groups}{target}'.format(
197 groups=match_dict['groups'] or '',
197 groups=match_dict['groups'] or '',
198 target=match_dict['target']))
198 target=match_dict['target']))
199
199
200 # Retrieve pull request instance by ID from regex match.
200 # Retrieve pull request instance by ID from regex match.
201 pull_request = PullRequest.get(match_dict['pr_id'])
201 pull_request = PullRequest.get(match_dict['pr_id'])
202
202
203 # Only proceed if we got a pull request and if acl repo name from
203 # Only proceed if we got a pull request and if acl repo name from
204 # URL equals the target repo name of the pull request.
204 # URL equals the target repo name of the pull request.
205 if pull_request and (acl_repo_name == pull_request.target_repo.repo_name):
205 if pull_request and (acl_repo_name == pull_request.target_repo.repo_name):
206
206
207 # Get file system path to shadow repository.
207 # Get file system path to shadow repository.
208 workspace_id = PullRequestModel()._workspace_id(pull_request)
208 workspace_id = PullRequestModel()._workspace_id(pull_request)
209 vcs_repo_name = pull_request.target_repo.get_shadow_repository_path(workspace_id)
209 vcs_repo_name = pull_request.target_repo.get_shadow_repository_path(workspace_id)
210
210
211 # Store names for later usage.
211 # Store names for later usage.
212 self.vcs_repo_name = vcs_repo_name
212 self.vcs_repo_name = vcs_repo_name
213 self.acl_repo_name = acl_repo_name
213 self.acl_repo_name = acl_repo_name
214 self.is_shadow_repo = True
214 self.is_shadow_repo = True
215
215
216 log.debug('Setting all VCS repository names: %s', {
216 log.debug('Setting all VCS repository names: %s', {
217 'acl_repo_name': self.acl_repo_name,
217 'acl_repo_name': self.acl_repo_name,
218 'url_repo_name': self.url_repo_name,
218 'url_repo_name': self.url_repo_name,
219 'vcs_repo_name': self.vcs_repo_name,
219 'vcs_repo_name': self.vcs_repo_name,
220 })
220 })
221
221
222 @property
222 @property
223 def scm_app(self):
223 def scm_app(self):
224 custom_implementation = self.config['vcs.scm_app_implementation']
224 custom_implementation = self.config['vcs.scm_app_implementation']
225 if custom_implementation == 'http':
225 if custom_implementation == 'http':
226 log.debug('Using HTTP implementation of scm app.')
226 log.debug('Using HTTP implementation of scm app.')
227 scm_app_impl = scm_app_http
227 scm_app_impl = scm_app_http
228 else:
228 else:
229 log.debug('Using custom implementation of scm_app: "{}"'.format(
229 log.debug('Using custom implementation of scm_app: "{}"'.format(
230 custom_implementation))
230 custom_implementation))
231 scm_app_impl = importlib.import_module(custom_implementation)
231 scm_app_impl = importlib.import_module(custom_implementation)
232 return scm_app_impl
232 return scm_app_impl
233
233
234 def _get_by_id(self, repo_name):
234 def _get_by_id(self, repo_name):
235 """
235 """
236 Gets a special pattern _<ID> from clone url and tries to replace it
236 Gets a special pattern _<ID> from clone url and tries to replace it
237 with a repository_name for support of _<ID> non changeable urls
237 with a repository_name for support of _<ID> non changeable urls
238 """
238 """
239
239
240 data = repo_name.split('/')
240 data = repo_name.split('/')
241 if len(data) >= 2:
241 if len(data) >= 2:
242 from rhodecode.model.repo import RepoModel
242 from rhodecode.model.repo import RepoModel
243 by_id_match = RepoModel().get_repo_by_id(repo_name)
243 by_id_match = RepoModel().get_repo_by_id(repo_name)
244 if by_id_match:
244 if by_id_match:
245 data[1] = by_id_match.repo_name
245 data[1] = by_id_match.repo_name
246
246
247 return safe_str('/'.join(data))
247 return safe_str('/'.join(data))
248
248
249 def _invalidate_cache(self, repo_name):
249 def _invalidate_cache(self, repo_name):
250 """
250 """
251 Set's cache for this repository for invalidation on next access
251 Set's cache for this repository for invalidation on next access
252
252
253 :param repo_name: full repo name, also a cache key
253 :param repo_name: full repo name, also a cache key
254 """
254 """
255 ScmModel().mark_for_invalidation(repo_name)
255 ScmModel().mark_for_invalidation(repo_name)
256
256
257 def is_valid_and_existing_repo(self, repo_name, base_path, scm_type):
257 def is_valid_and_existing_repo(self, repo_name, base_path, scm_type):
258 db_repo = Repository.get_by_repo_name(repo_name)
258 db_repo = Repository.get_by_repo_name(repo_name)
259 if not db_repo:
259 if not db_repo:
260 log.debug('Repository `%s` not found inside the database.',
260 log.debug('Repository `%s` not found inside the database.',
261 repo_name)
261 repo_name)
262 return False
262 return False
263
263
264 if db_repo.repo_type != scm_type:
264 if db_repo.repo_type != scm_type:
265 log.warning(
265 log.warning(
266 'Repository `%s` have incorrect scm_type, expected %s got %s',
266 'Repository `%s` have incorrect scm_type, expected %s got %s',
267 repo_name, db_repo.repo_type, scm_type)
267 repo_name, db_repo.repo_type, scm_type)
268 return False
268 return False
269
269
270 config = db_repo._config
270 config = db_repo._config
271 config.set('extensions', 'largefiles', '')
271 config.set('extensions', 'largefiles', '')
272 return is_valid_repo(
272 return is_valid_repo(
273 repo_name, base_path,
273 repo_name, base_path,
274 explicit_scm=scm_type, expect_scm=scm_type, config=config)
274 explicit_scm=scm_type, expect_scm=scm_type, config=config)
275
275
276 def valid_and_active_user(self, user):
276 def valid_and_active_user(self, user):
277 """
277 """
278 Checks if that user is not empty, and if it's actually object it checks
278 Checks if that user is not empty, and if it's actually object it checks
279 if he's active.
279 if he's active.
280
280
281 :param user: user object or None
281 :param user: user object or None
282 :return: boolean
282 :return: boolean
283 """
283 """
284 if user is None:
284 if user is None:
285 return False
285 return False
286
286
287 elif user.active:
287 elif user.active:
288 return True
288 return True
289
289
290 return False
290 return False
291
291
292 @property
292 @property
293 def is_shadow_repo_dir(self):
293 def is_shadow_repo_dir(self):
294 return os.path.isdir(self.vcs_repo_name)
294 return os.path.isdir(self.vcs_repo_name)
295
295
296 def _check_permission(self, action, user, auth_user, repo_name, ip_addr=None,
296 def _check_permission(self, action, user, auth_user, repo_name, ip_addr=None,
297 plugin_id='', plugin_cache_active=False, cache_ttl=0):
297 plugin_id='', plugin_cache_active=False, cache_ttl=0):
298 """
298 """
299 Checks permissions using action (push/pull) user and repository
299 Checks permissions using action (push/pull) user and repository
300 name. If plugin_cache and ttl is set it will use the plugin which
300 name. If plugin_cache and ttl is set it will use the plugin which
301 authenticated the user to store the cached permissions result for N
301 authenticated the user to store the cached permissions result for N
302 amount of seconds as in cache_ttl
302 amount of seconds as in cache_ttl
303
303
304 :param action: push or pull action
304 :param action: push or pull action
305 :param user: user instance
305 :param user: user instance
306 :param repo_name: repository name
306 :param repo_name: repository name
307 """
307 """
308
308
309 log.debug('AUTH_CACHE_TTL for permissions `%s` active: %s (TTL: %s)',
309 log.debug('AUTH_CACHE_TTL for permissions `%s` active: %s (TTL: %s)',
310 plugin_id, plugin_cache_active, cache_ttl)
310 plugin_id, plugin_cache_active, cache_ttl)
311
311
312 user_id = user.user_id
312 user_id = user.user_id
313 cache_namespace_uid = 'cache_user_auth.{}'.format(user_id)
313 cache_namespace_uid = 'cache_user_auth.{}'.format(user_id)
314 region = rc_cache.get_or_create_region('cache_perms', cache_namespace_uid)
314 region = rc_cache.get_or_create_region('cache_perms', cache_namespace_uid)
315
315
316 @region.conditional_cache_on_arguments(namespace=cache_namespace_uid,
316 @region.conditional_cache_on_arguments(namespace=cache_namespace_uid,
317 expiration_time=cache_ttl,
317 expiration_time=cache_ttl,
318 condition=plugin_cache_active)
318 condition=plugin_cache_active)
319 def compute_perm_vcs(
319 def compute_perm_vcs(
320 cache_name, plugin_id, action, user_id, repo_name, ip_addr):
320 cache_name, plugin_id, action, user_id, repo_name, ip_addr):
321
321
322 log.debug('auth: calculating permission access now...')
322 log.debug('auth: calculating permission access now...')
323 # check IP
323 # check IP
324 inherit = user.inherit_default_permissions
324 inherit = user.inherit_default_permissions
325 ip_allowed = AuthUser.check_ip_allowed(
325 ip_allowed = AuthUser.check_ip_allowed(
326 user_id, ip_addr, inherit_from_default=inherit)
326 user_id, ip_addr, inherit_from_default=inherit)
327 if ip_allowed:
327 if ip_allowed:
328 log.info('Access for IP:%s allowed', ip_addr)
328 log.info('Access for IP:%s allowed', ip_addr)
329 else:
329 else:
330 return False
330 return False
331
331
332 if action == 'push':
332 if action == 'push':
333 perms = ('repository.write', 'repository.admin')
333 perms = ('repository.write', 'repository.admin')
334 if not HasPermissionAnyMiddleware(*perms)(auth_user, repo_name):
334 if not HasPermissionAnyMiddleware(*perms)(auth_user, repo_name):
335 return False
335 return False
336
336
337 else:
337 else:
338 # any other action need at least read permission
338 # any other action need at least read permission
339 perms = (
339 perms = (
340 'repository.read', 'repository.write', 'repository.admin')
340 'repository.read', 'repository.write', 'repository.admin')
341 if not HasPermissionAnyMiddleware(*perms)(auth_user, repo_name):
341 if not HasPermissionAnyMiddleware(*perms)(auth_user, repo_name):
342 return False
342 return False
343
343
344 return True
344 return True
345
345
346 start = time.time()
346 start = time.time()
347 log.debug('Running plugin `%s` permissions check', plugin_id)
347 log.debug('Running plugin `%s` permissions check', plugin_id)
348
348
349 # for environ based auth, password can be empty, but then the validation is
349 # for environ based auth, password can be empty, but then the validation is
350 # on the server that fills in the env data needed for authentication
350 # on the server that fills in the env data needed for authentication
351 perm_result = compute_perm_vcs(
351 perm_result = compute_perm_vcs(
352 'vcs_permissions', plugin_id, action, user.user_id, repo_name, ip_addr)
352 'vcs_permissions', plugin_id, action, user.user_id, repo_name, ip_addr)
353
353
354 auth_time = time.time() - start
354 auth_time = time.time() - start
355 log.debug('Permissions for plugin `%s` completed in %.4fs, '
355 log.debug('Permissions for plugin `%s` completed in %.4fs, '
356 'expiration time of fetched cache %.1fs.',
356 'expiration time of fetched cache %.1fs.',
357 plugin_id, auth_time, cache_ttl)
357 plugin_id, auth_time, cache_ttl)
358
358
359 return perm_result
359 return perm_result
360
360
361 def _get_http_scheme(self, environ):
361 def _get_http_scheme(self, environ):
362 try:
362 try:
363 return environ['wsgi.url_scheme']
363 return environ['wsgi.url_scheme']
364 except Exception:
364 except Exception:
365 log.exception('Failed to read http scheme')
365 log.exception('Failed to read http scheme')
366 return 'http'
366 return 'http'
367
367
368 def _check_ssl(self, environ, start_response):
368 def _check_ssl(self, environ, start_response):
369 """
369 """
370 Checks the SSL check flag and returns False if SSL is not present
370 Checks the SSL check flag and returns False if SSL is not present
371 and required True otherwise
371 and required True otherwise
372 """
372 """
373 org_proto = environ['wsgi._org_proto']
373 org_proto = environ['wsgi._org_proto']
374 # check if we have SSL required ! if not it's a bad request !
374 # check if we have SSL required ! if not it's a bad request !
375 require_ssl = str2bool(self.repo_vcs_config.get('web', 'push_ssl'))
375 require_ssl = str2bool(self.repo_vcs_config.get('web', 'push_ssl'))
376 if require_ssl and org_proto == 'http':
376 if require_ssl and org_proto == 'http':
377 log.debug(
377 log.debug(
378 'Bad request: detected protocol is `%s` and '
378 'Bad request: detected protocol is `%s` and '
379 'SSL/HTTPS is required.', org_proto)
379 'SSL/HTTPS is required.', org_proto)
380 return False
380 return False
381 return True
381 return True
382
382
383 def _get_default_cache_ttl(self):
383 def _get_default_cache_ttl(self):
384 # take AUTH_CACHE_TTL from the `rhodecode` auth plugin
384 # take AUTH_CACHE_TTL from the `rhodecode` auth plugin
385 plugin = loadplugin('egg:rhodecode-enterprise-ce#rhodecode')
385 plugin = loadplugin('egg:rhodecode-enterprise-ce#rhodecode')
386 plugin_settings = plugin.get_settings()
386 plugin_settings = plugin.get_settings()
387 plugin_cache_active, cache_ttl = plugin.get_ttl_cache(
387 plugin_cache_active, cache_ttl = plugin.get_ttl_cache(
388 plugin_settings) or (False, 0)
388 plugin_settings) or (False, 0)
389 return plugin_cache_active, cache_ttl
389 return plugin_cache_active, cache_ttl
390
390
391 def __call__(self, environ, start_response):
391 def __call__(self, environ, start_response):
392 try:
392 try:
393 return self._handle_request(environ, start_response)
393 return self._handle_request(environ, start_response)
394 except Exception:
394 except Exception:
395 log.exception("Exception while handling request")
395 log.exception("Exception while handling request")
396 appenlight.track_exception(environ)
396 appenlight.track_exception(environ)
397 return HTTPInternalServerError()(environ, start_response)
397 return HTTPInternalServerError()(environ, start_response)
398 finally:
398 finally:
399 meta.Session.remove()
399 meta.Session.remove()
400
400
401 def _handle_request(self, environ, start_response):
401 def _handle_request(self, environ, start_response):
402 if not self._check_ssl(environ, start_response):
402 if not self._check_ssl(environ, start_response):
403 reason = ('SSL required, while RhodeCode was unable '
403 reason = ('SSL required, while RhodeCode was unable '
404 'to detect this as SSL request')
404 'to detect this as SSL request')
405 log.debug('User not allowed to proceed, %s', reason)
405 log.debug('User not allowed to proceed, %s', reason)
406 return HTTPNotAcceptable(reason)(environ, start_response)
406 return HTTPNotAcceptable(reason)(environ, start_response)
407
407
408 if not self.url_repo_name:
408 if not self.url_repo_name:
409 log.warning('Repository name is empty: %s', self.url_repo_name)
409 log.warning('Repository name is empty: %s', self.url_repo_name)
410 # failed to get repo name, we fail now
410 # failed to get repo name, we fail now
411 return HTTPNotFound()(environ, start_response)
411 return HTTPNotFound()(environ, start_response)
412 log.debug('Extracted repo name is %s', self.url_repo_name)
412 log.debug('Extracted repo name is %s', self.url_repo_name)
413
413
414 ip_addr = get_ip_addr(environ)
414 ip_addr = get_ip_addr(environ)
415 user_agent = get_user_agent(environ)
415 user_agent = get_user_agent(environ)
416 username = None
416 username = None
417
417
418 # skip passing error to error controller
418 # skip passing error to error controller
419 environ['pylons.status_code_redirect'] = True
419 environ['pylons.status_code_redirect'] = True
420
420
421 # ======================================================================
421 # ======================================================================
422 # GET ACTION PULL or PUSH
422 # GET ACTION PULL or PUSH
423 # ======================================================================
423 # ======================================================================
424 action = self._get_action(environ)
424 action = self._get_action(environ)
425
425
426 # ======================================================================
426 # ======================================================================
427 # Check if this is a request to a shadow repository of a pull request.
427 # Check if this is a request to a shadow repository of a pull request.
428 # In this case only pull action is allowed.
428 # In this case only pull action is allowed.
429 # ======================================================================
429 # ======================================================================
430 if self.is_shadow_repo and action != 'pull':
430 if self.is_shadow_repo and action != 'pull':
431 reason = 'Only pull action is allowed for shadow repositories.'
431 reason = 'Only pull action is allowed for shadow repositories.'
432 log.debug('User not allowed to proceed, %s', reason)
432 log.debug('User not allowed to proceed, %s', reason)
433 return HTTPNotAcceptable(reason)(environ, start_response)
433 return HTTPNotAcceptable(reason)(environ, start_response)
434
434
435 # Check if the shadow repo actually exists, in case someone refers
435 # Check if the shadow repo actually exists, in case someone refers
436 # to it, and it has been deleted because of successful merge.
436 # to it, and it has been deleted because of successful merge.
437 if self.is_shadow_repo and not self.is_shadow_repo_dir:
437 if self.is_shadow_repo and not self.is_shadow_repo_dir:
438 log.debug(
438 log.debug(
439 'Shadow repo detected, and shadow repo dir `%s` is missing',
439 'Shadow repo detected, and shadow repo dir `%s` is missing',
440 self.is_shadow_repo_dir)
440 self.is_shadow_repo_dir)
441 return HTTPNotFound()(environ, start_response)
441 return HTTPNotFound()(environ, start_response)
442
442
443 # ======================================================================
443 # ======================================================================
444 # CHECK ANONYMOUS PERMISSION
444 # CHECK ANONYMOUS PERMISSION
445 # ======================================================================
445 # ======================================================================
446 detect_force_push = False
446 detect_force_push = False
447 check_branch_perms = False
447 check_branch_perms = False
448 if action in ['pull', 'push']:
448 if action in ['pull', 'push']:
449 user_obj = anonymous_user = User.get_default_user()
449 user_obj = anonymous_user = User.get_default_user()
450 auth_user = user_obj.AuthUser()
450 auth_user = user_obj.AuthUser()
451 username = anonymous_user.username
451 username = anonymous_user.username
452 if anonymous_user.active:
452 if anonymous_user.active:
453 plugin_cache_active, cache_ttl = self._get_default_cache_ttl()
453 plugin_cache_active, cache_ttl = self._get_default_cache_ttl()
454 # ONLY check permissions if the user is activated
454 # ONLY check permissions if the user is activated
455 anonymous_perm = self._check_permission(
455 anonymous_perm = self._check_permission(
456 action, anonymous_user, auth_user, self.acl_repo_name, ip_addr,
456 action, anonymous_user, auth_user, self.acl_repo_name, ip_addr,
457 plugin_id='anonymous_access',
457 plugin_id='anonymous_access',
458 plugin_cache_active=plugin_cache_active,
458 plugin_cache_active=plugin_cache_active,
459 cache_ttl=cache_ttl,
459 cache_ttl=cache_ttl,
460 )
460 )
461 else:
461 else:
462 anonymous_perm = False
462 anonymous_perm = False
463
463
464 if not anonymous_user.active or not anonymous_perm:
464 if not anonymous_user.active or not anonymous_perm:
465 if not anonymous_user.active:
465 if not anonymous_user.active:
466 log.debug('Anonymous access is disabled, running '
466 log.debug('Anonymous access is disabled, running '
467 'authentication')
467 'authentication')
468
468
469 if not anonymous_perm:
469 if not anonymous_perm:
470 log.debug('Not enough credentials to access this '
470 log.debug('Not enough credentials to access this '
471 'repository as anonymous user')
471 'repository as anonymous user')
472
472
473 username = None
473 username = None
474 # ==============================================================
474 # ==============================================================
475 # DEFAULT PERM FAILED OR ANONYMOUS ACCESS IS DISABLED SO WE
475 # DEFAULT PERM FAILED OR ANONYMOUS ACCESS IS DISABLED SO WE
476 # NEED TO AUTHENTICATE AND ASK FOR AUTH USER PERMISSIONS
476 # NEED TO AUTHENTICATE AND ASK FOR AUTH USER PERMISSIONS
477 # ==============================================================
477 # ==============================================================
478
478
479 # try to auth based on environ, container auth methods
479 # try to auth based on environ, container auth methods
480 log.debug('Running PRE-AUTH for container based authentication')
480 log.debug('Running PRE-AUTH for container based authentication')
481 pre_auth = authenticate(
481 pre_auth = authenticate(
482 '', '', environ, VCS_TYPE, registry=self.registry,
482 '', '', environ, VCS_TYPE, registry=self.registry,
483 acl_repo_name=self.acl_repo_name)
483 acl_repo_name=self.acl_repo_name)
484 if pre_auth and pre_auth.get('username'):
484 if pre_auth and pre_auth.get('username'):
485 username = pre_auth['username']
485 username = pre_auth['username']
486 log.debug('PRE-AUTH got %s as username', username)
486 log.debug('PRE-AUTH got %s as username', username)
487 if pre_auth:
487 if pre_auth:
488 log.debug('PRE-AUTH successful from %s',
488 log.debug('PRE-AUTH successful from %s',
489 pre_auth.get('auth_data', {}).get('_plugin'))
489 pre_auth.get('auth_data', {}).get('_plugin'))
490
490
491 # If not authenticated by the container, running basic auth
491 # If not authenticated by the container, running basic auth
492 # before inject the calling repo_name for special scope checks
492 # before inject the calling repo_name for special scope checks
493 self.authenticate.acl_repo_name = self.acl_repo_name
493 self.authenticate.acl_repo_name = self.acl_repo_name
494
494
495 plugin_cache_active, cache_ttl = False, 0
495 plugin_cache_active, cache_ttl = False, 0
496 plugin = None
496 plugin = None
497 if not username:
497 if not username:
498 self.authenticate.realm = self.authenticate.get_rc_realm()
498 self.authenticate.realm = self.authenticate.get_rc_realm()
499
499
500 try:
500 try:
501 auth_result = self.authenticate(environ)
501 auth_result = self.authenticate(environ)
502 except (UserCreationError, NotAllowedToCreateUserError) as e:
502 except (UserCreationError, NotAllowedToCreateUserError) as e:
503 log.error(e)
503 log.error(e)
504 reason = safe_str(e)
504 reason = safe_str(e)
505 return HTTPNotAcceptable(reason)(environ, start_response)
505 return HTTPNotAcceptable(reason)(environ, start_response)
506
506
507 if isinstance(auth_result, dict):
507 if isinstance(auth_result, dict):
508 AUTH_TYPE.update(environ, 'basic')
508 AUTH_TYPE.update(environ, 'basic')
509 REMOTE_USER.update(environ, auth_result['username'])
509 REMOTE_USER.update(environ, auth_result['username'])
510 username = auth_result['username']
510 username = auth_result['username']
511 plugin = auth_result.get('auth_data', {}).get('_plugin')
511 plugin = auth_result.get('auth_data', {}).get('_plugin')
512 log.info(
512 log.info(
513 'MAIN-AUTH successful for user `%s` from %s plugin',
513 'MAIN-AUTH successful for user `%s` from %s plugin',
514 username, plugin)
514 username, plugin)
515
515
516 plugin_cache_active, cache_ttl = auth_result.get(
516 plugin_cache_active, cache_ttl = auth_result.get(
517 'auth_data', {}).get('_ttl_cache') or (False, 0)
517 'auth_data', {}).get('_ttl_cache') or (False, 0)
518 else:
518 else:
519 return auth_result.wsgi_application(environ, start_response)
519 return auth_result.wsgi_application(environ, start_response)
520
520
521 # ==============================================================
521 # ==============================================================
522 # CHECK PERMISSIONS FOR THIS REQUEST USING GIVEN USERNAME
522 # CHECK PERMISSIONS FOR THIS REQUEST USING GIVEN USERNAME
523 # ==============================================================
523 # ==============================================================
524 user = User.get_by_username(username)
524 user = User.get_by_username(username)
525 if not self.valid_and_active_user(user):
525 if not self.valid_and_active_user(user):
526 return HTTPForbidden()(environ, start_response)
526 return HTTPForbidden()(environ, start_response)
527 username = user.username
527 username = user.username
528 user_id = user.user_id
528 user_id = user.user_id
529
529
530 # check user attributes for password change flag
530 # check user attributes for password change flag
531 user_obj = user
531 user_obj = user
532 auth_user = user_obj.AuthUser()
532 auth_user = user_obj.AuthUser()
533 if user_obj and user_obj.username != User.DEFAULT_USER and \
533 if user_obj and user_obj.username != User.DEFAULT_USER and \
534 user_obj.user_data.get('force_password_change'):
534 user_obj.user_data.get('force_password_change'):
535 reason = 'password change required'
535 reason = 'password change required'
536 log.debug('User not allowed to authenticate, %s', reason)
536 log.debug('User not allowed to authenticate, %s', reason)
537 return HTTPNotAcceptable(reason)(environ, start_response)
537 return HTTPNotAcceptable(reason)(environ, start_response)
538
538
539 # check permissions for this repository
539 # check permissions for this repository
540 perm = self._check_permission(
540 perm = self._check_permission(
541 action, user, auth_user, self.acl_repo_name, ip_addr,
541 action, user, auth_user, self.acl_repo_name, ip_addr,
542 plugin, plugin_cache_active, cache_ttl)
542 plugin, plugin_cache_active, cache_ttl)
543 if not perm:
543 if not perm:
544 return HTTPForbidden()(environ, start_response)
544 return HTTPForbidden()(environ, start_response)
545 environ['rc_auth_user_id'] = user_id
545 environ['rc_auth_user_id'] = user_id
546
546
547 if action == 'push':
547 if action == 'push':
548 perms = auth_user.get_branch_permissions(self.acl_repo_name)
548 perms = auth_user.get_branch_permissions(self.acl_repo_name)
549 if perms:
549 if perms:
550 check_branch_perms = True
550 check_branch_perms = True
551 detect_force_push = True
551 detect_force_push = True
552
552
553 # extras are injected into UI object and later available
553 # extras are injected into UI object and later available
554 # in hooks executed by RhodeCode
554 # in hooks executed by RhodeCode
555 check_locking = _should_check_locking(environ.get('QUERY_STRING'))
555 check_locking = _should_check_locking(environ.get('QUERY_STRING'))
556
556
557 extras = vcs_operation_context(
557 extras = vcs_operation_context(
558 environ, repo_name=self.acl_repo_name, username=username,
558 environ, repo_name=self.acl_repo_name, username=username,
559 action=action, scm=self.SCM, check_locking=check_locking,
559 action=action, scm=self.SCM, check_locking=check_locking,
560 is_shadow_repo=self.is_shadow_repo, check_branch_perms=check_branch_perms,
560 is_shadow_repo=self.is_shadow_repo, check_branch_perms=check_branch_perms,
561 detect_force_push=detect_force_push
561 detect_force_push=detect_force_push
562 )
562 )
563
563
564 # ======================================================================
564 # ======================================================================
565 # REQUEST HANDLING
565 # REQUEST HANDLING
566 # ======================================================================
566 # ======================================================================
567 repo_path = os.path.join(
567 repo_path = os.path.join(
568 safe_str(self.base_path), safe_str(self.vcs_repo_name))
568 safe_str(self.base_path), safe_str(self.vcs_repo_name))
569 log.debug('Repository path is %s', repo_path)
569 log.debug('Repository path is %s', repo_path)
570
570
571 fix_PATH()
571 fix_PATH()
572
572
573 log.info(
573 log.info(
574 '%s action on %s repo "%s" by "%s" from %s %s',
574 '%s action on %s repo "%s" by "%s" from %s %s',
575 action, self.SCM, safe_str(self.url_repo_name),
575 action, self.SCM, safe_str(self.url_repo_name),
576 safe_str(username), ip_addr, user_agent)
576 safe_str(username), ip_addr, user_agent)
577
577
578 return self._generate_vcs_response(
578 return self._generate_vcs_response(
579 environ, start_response, repo_path, extras, action)
579 environ, start_response, repo_path, extras, action)
580
580
581 @initialize_generator
581 @initialize_generator
582 def _generate_vcs_response(
582 def _generate_vcs_response(
583 self, environ, start_response, repo_path, extras, action):
583 self, environ, start_response, repo_path, extras, action):
584 """
584 """
585 Returns a generator for the response content.
585 Returns a generator for the response content.
586
586
587 This method is implemented as a generator, so that it can trigger
587 This method is implemented as a generator, so that it can trigger
588 the cache validation after all content sent back to the client. It
588 the cache validation after all content sent back to the client. It
589 also handles the locking exceptions which will be triggered when
589 also handles the locking exceptions which will be triggered when
590 the first chunk is produced by the underlying WSGI application.
590 the first chunk is produced by the underlying WSGI application.
591 """
591 """
592 txn_id = ''
592 txn_id = ''
593 if 'CONTENT_LENGTH' in environ and environ['REQUEST_METHOD'] == 'MERGE':
593 if 'CONTENT_LENGTH' in environ and environ['REQUEST_METHOD'] == 'MERGE':
594 # case for SVN, we want to re-use the callback daemon port
594 # case for SVN, we want to re-use the callback daemon port
595 # so we use the txn_id, for this we peek the body, and still save
595 # so we use the txn_id, for this we peek the body, and still save
596 # it as wsgi.input
596 # it as wsgi.input
597 data = environ['wsgi.input'].read()
597 data = environ['wsgi.input'].read()
598 environ['wsgi.input'] = StringIO(data)
598 environ['wsgi.input'] = StringIO(data)
599 txn_id = extract_svn_txn_id(self.acl_repo_name, data)
599 txn_id = extract_svn_txn_id(self.acl_repo_name, data)
600
600
601 callback_daemon, extras = self._prepare_callback_daemon(
601 callback_daemon, extras = self._prepare_callback_daemon(
602 extras, environ, action, txn_id=txn_id)
602 extras, environ, action, txn_id=txn_id)
603 log.debug('HOOKS extras is %s', extras)
603 log.debug('HOOKS extras is %s', extras)
604
604
605 http_scheme = self._get_http_scheme(environ)
605 http_scheme = self._get_http_scheme(environ)
606
606
607 config = self._create_config(extras, self.acl_repo_name, scheme=http_scheme)
607 config = self._create_config(extras, self.acl_repo_name, scheme=http_scheme)
608 app = self._create_wsgi_app(repo_path, self.url_repo_name, config)
608 app = self._create_wsgi_app(repo_path, self.url_repo_name, config)
609 with callback_daemon:
609 with callback_daemon:
610 app.rc_extras = extras
610 app.rc_extras = extras
611
611
612 try:
612 try:
613 response = app(environ, start_response)
613 response = app(environ, start_response)
614 finally:
614 finally:
615 # This statement works together with the decorator
615 # This statement works together with the decorator
616 # "initialize_generator" above. The decorator ensures that
616 # "initialize_generator" above. The decorator ensures that
617 # we hit the first yield statement before the generator is
617 # we hit the first yield statement before the generator is
618 # returned back to the WSGI server. This is needed to
618 # returned back to the WSGI server. This is needed to
619 # ensure that the call to "app" above triggers the
619 # ensure that the call to "app" above triggers the
620 # needed callback to "start_response" before the
620 # needed callback to "start_response" before the
621 # generator is actually used.
621 # generator is actually used.
622 yield "__init__"
622 yield "__init__"
623
623
624 # iter content
624 # iter content
625 for chunk in response:
625 for chunk in response:
626 yield chunk
626 yield chunk
627
627
628 try:
628 try:
629 # invalidate cache on push
629 # invalidate cache on push
630 if action == 'push':
630 if action == 'push':
631 self._invalidate_cache(self.url_repo_name)
631 self._invalidate_cache(self.url_repo_name)
632 finally:
632 finally:
633 meta.Session.remove()
633 meta.Session.remove()
634
634
635 def _get_repository_name(self, environ):
635 def _get_repository_name(self, environ):
636 """Get repository name out of the environmnent
636 """Get repository name out of the environmnent
637
637
638 :param environ: WSGI environment
638 :param environ: WSGI environment
639 """
639 """
640 raise NotImplementedError()
640 raise NotImplementedError()
641
641
642 def _get_action(self, environ):
642 def _get_action(self, environ):
643 """Map request commands into a pull or push command.
643 """Map request commands into a pull or push command.
644
644
645 :param environ: WSGI environment
645 :param environ: WSGI environment
646 """
646 """
647 raise NotImplementedError()
647 raise NotImplementedError()
648
648
649 def _create_wsgi_app(self, repo_path, repo_name, config):
649 def _create_wsgi_app(self, repo_path, repo_name, config):
650 """Return the WSGI app that will finally handle the request."""
650 """Return the WSGI app that will finally handle the request."""
651 raise NotImplementedError()
651 raise NotImplementedError()
652
652
653 def _create_config(self, extras, repo_name, scheme='http'):
653 def _create_config(self, extras, repo_name, scheme='http'):
654 """Create a safe config representation."""
654 """Create a safe config representation."""
655 raise NotImplementedError()
655 raise NotImplementedError()
656
656
657 def _should_use_callback_daemon(self, extras, environ, action):
657 def _should_use_callback_daemon(self, extras, environ, action):
658 if extras.get('is_shadow_repo'):
658 if extras.get('is_shadow_repo'):
659 # we don't want to execute hooks, and callback daemon for shadow repos
659 # we don't want to execute hooks, and callback daemon for shadow repos
660 return False
660 return False
661 return True
661 return True
662
662
663 def _prepare_callback_daemon(self, extras, environ, action, txn_id=None):
663 def _prepare_callback_daemon(self, extras, environ, action, txn_id=None):
664 direct_calls = vcs_settings.HOOKS_DIRECT_CALLS
664 direct_calls = vcs_settings.HOOKS_DIRECT_CALLS
665 if not self._should_use_callback_daemon(extras, environ, action):
665 if not self._should_use_callback_daemon(extras, environ, action):
666 # disable callback daemon for actions that don't require it
666 # disable callback daemon for actions that don't require it
667 direct_calls = True
667 direct_calls = True
668
668
669 return prepare_callback_daemon(
669 return prepare_callback_daemon(
670 extras, protocol=vcs_settings.HOOKS_PROTOCOL,
670 extras, protocol=vcs_settings.HOOKS_PROTOCOL,
671 host=vcs_settings.HOOKS_HOST, use_direct_calls=direct_calls, txn_id=txn_id)
671 host=vcs_settings.HOOKS_HOST, use_direct_calls=direct_calls, txn_id=txn_id)
672
672
673
673
674 def _should_check_locking(query_string):
674 def _should_check_locking(query_string):
675 # this is kind of hacky, but due to how mercurial handles client-server
675 # this is kind of hacky, but due to how mercurial handles client-server
676 # server see all operation on commit; bookmarks, phases and
676 # server see all operation on commit; bookmarks, phases and
677 # obsolescence marker in different transaction, we don't want to check
677 # obsolescence marker in different transaction, we don't want to check
678 # locking on those
678 # locking on those
679 return query_string not in ['cmd=listkeys']
679 return query_string not in ['cmd=listkeys']
@@ -1,412 +1,412 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2016-2020 RhodeCode GmbH
3 # Copyright (C) 2016-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 """
21 """
22 Client for the VCSServer implemented based on HTTP.
22 Client for the VCSServer implemented based on HTTP.
23 """
23 """
24
24
25 import copy
25 import copy
26 import logging
26 import logging
27 import threading
27 import threading
28 import time
28 import time
29 import urllib.request, urllib.error, urllib.parse
29 import urllib.request, urllib.error, urllib.parse
30 import urllib.parse
30 import urllib.parse
31 import uuid
31 import uuid
32 import traceback
32 import traceback
33
33
34 import pycurl
34 import pycurl
35 import msgpack
35 import msgpack
36 import requests
36 import requests
37 from requests.packages.urllib3.util.retry import Retry
37 from requests.packages.urllib3.util.retry import Retry
38
38
39 import rhodecode
39 import rhodecode
40 from rhodecode.lib import rc_cache
40 from rhodecode.lib import rc_cache
41 from rhodecode.lib.rc_cache.utils import compute_key_from_params
41 from rhodecode.lib.rc_cache.utils import compute_key_from_params
42 from rhodecode.lib.system_info import get_cert_path
42 from rhodecode.lib.system_info import get_cert_path
43 from rhodecode.lib.vcs import exceptions, CurlSession
43 from rhodecode.lib.vcs import exceptions, CurlSession
44 from rhodecode.lib.utils2 import str2bool
44 from rhodecode.lib.utils2 import str2bool
45
45
46 log = logging.getLogger(__name__)
46 log = logging.getLogger(__name__)
47
47
48
48
49 # TODO: mikhail: Keep it in sync with vcsserver's
49 # TODO: mikhail: Keep it in sync with vcsserver's
50 # HTTPApplication.ALLOWED_EXCEPTIONS
50 # HTTPApplication.ALLOWED_EXCEPTIONS
51 EXCEPTIONS_MAP = {
51 EXCEPTIONS_MAP = {
52 'KeyError': KeyError,
52 'KeyError': KeyError,
53 'URLError': urllib.error.URLError,
53 'URLError': urllib.error.URLError,
54 }
54 }
55
55
56
56
57 def _remote_call(url, payload, exceptions_map, session):
57 def _remote_call(url, payload, exceptions_map, session):
58 try:
58 try:
59 headers = {
59 headers = {
60 'X-RC-Method': payload.get('method'),
60 'X-RC-Method': payload.get('method'),
61 'X-RC-Repo-Name': payload.get('_repo_name')
61 'X-RC-Repo-Name': payload.get('_repo_name')
62 }
62 }
63 response = session.post(url, data=msgpack.packb(payload), headers=headers)
63 response = session.post(url, data=msgpack.packb(payload), headers=headers)
64 except pycurl.error as e:
64 except pycurl.error as e:
65 msg = '{}. \npycurl traceback: {}'.format(e, traceback.format_exc())
65 msg = '{}. \npycurl traceback: {}'.format(e, traceback.format_exc())
66 raise exceptions.HttpVCSCommunicationError(msg)
66 raise exceptions.HttpVCSCommunicationError(msg)
67 except Exception as e:
67 except Exception as e:
68 message = getattr(e, 'message', '')
68 message = getattr(e, 'message', '')
69 if 'Failed to connect' in message:
69 if 'Failed to connect' in message:
70 # gevent doesn't return proper pycurl errors
70 # gevent doesn't return proper pycurl errors
71 raise exceptions.HttpVCSCommunicationError(e)
71 raise exceptions.HttpVCSCommunicationError(e)
72 else:
72 else:
73 raise
73 raise
74
74
75 if response.status_code >= 400:
75 if response.status_code >= 400:
76 log.error('Call to %s returned non 200 HTTP code: %s',
76 log.error('Call to %s returned non 200 HTTP code: %s',
77 url, response.status_code)
77 url, response.status_code)
78 raise exceptions.HttpVCSCommunicationError(repr(response.content))
78 raise exceptions.HttpVCSCommunicationError(repr(response.content))
79
79
80 try:
80 try:
81 response = msgpack.unpackb(response.content)
81 response = msgpack.unpackb(response.content)
82 except Exception:
82 except Exception:
83 log.exception('Failed to decode response %r', response.content)
83 log.exception('Failed to decode response %r', response.content)
84 raise
84 raise
85
85
86 error = response.get('error')
86 error = response.get('error')
87 if error:
87 if error:
88 type_ = error.get('type', 'Exception')
88 type_ = error.get('type', 'Exception')
89 exc = exceptions_map.get(type_, Exception)
89 exc = exceptions_map.get(type_, Exception)
90 exc = exc(error.get('message'))
90 exc = exc(error.get('message'))
91 try:
91 try:
92 exc._vcs_kind = error['_vcs_kind']
92 exc._vcs_kind = error['_vcs_kind']
93 except KeyError:
93 except KeyError:
94 pass
94 pass
95
95
96 try:
96 try:
97 exc._vcs_server_traceback = error['traceback']
97 exc._vcs_server_traceback = error['traceback']
98 exc._vcs_server_org_exc_name = error['org_exc']
98 exc._vcs_server_org_exc_name = error['org_exc']
99 exc._vcs_server_org_exc_tb = error['org_exc_tb']
99 exc._vcs_server_org_exc_tb = error['org_exc_tb']
100 except KeyError:
100 except KeyError:
101 pass
101 pass
102
102
103 raise exc
103 raise exc
104 return response.get('result')
104 return response.get('result')
105
105
106
106
107 def _streaming_remote_call(url, payload, exceptions_map, session, chunk_size):
107 def _streaming_remote_call(url, payload, exceptions_map, session, chunk_size):
108 try:
108 try:
109 headers = {
109 headers = {
110 'X-RC-Method': payload.get('method'),
110 'X-RC-Method': payload.get('method'),
111 'X-RC-Repo-Name': payload.get('_repo_name')
111 'X-RC-Repo-Name': payload.get('_repo_name')
112 }
112 }
113 response = session.post(url, data=msgpack.packb(payload), headers=headers)
113 response = session.post(url, data=msgpack.packb(payload), headers=headers)
114 except pycurl.error as e:
114 except pycurl.error as e:
115 msg = '{}. \npycurl traceback: {}'.format(e, traceback.format_exc())
115 msg = '{}. \npycurl traceback: {}'.format(e, traceback.format_exc())
116 raise exceptions.HttpVCSCommunicationError(msg)
116 raise exceptions.HttpVCSCommunicationError(msg)
117 except Exception as e:
117 except Exception as e:
118 message = getattr(e, 'message', '')
118 message = getattr(e, 'message', '')
119 if 'Failed to connect' in message:
119 if 'Failed to connect' in message:
120 # gevent doesn't return proper pycurl errors
120 # gevent doesn't return proper pycurl errors
121 raise exceptions.HttpVCSCommunicationError(e)
121 raise exceptions.HttpVCSCommunicationError(e)
122 else:
122 else:
123 raise
123 raise
124
124
125 if response.status_code >= 400:
125 if response.status_code >= 400:
126 log.error('Call to %s returned non 200 HTTP code: %s',
126 log.error('Call to %s returned non 200 HTTP code: %s',
127 url, response.status_code)
127 url, response.status_code)
128 raise exceptions.HttpVCSCommunicationError(repr(response.content))
128 raise exceptions.HttpVCSCommunicationError(repr(response.content))
129
129
130 return response.iter_content(chunk_size=chunk_size)
130 return response.iter_content(chunk_size=chunk_size)
131
131
132
132
133 class ServiceConnection(object):
133 class ServiceConnection(object):
134 def __init__(self, server_and_port, backend_endpoint, session_factory):
134 def __init__(self, server_and_port, backend_endpoint, session_factory):
135 self.url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint)
135 self.url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint)
136 self._session_factory = session_factory
136 self._session_factory = session_factory
137
137
138 def __getattr__(self, name):
138 def __getattr__(self, name):
139 def f(*args, **kwargs):
139 def f(*args, **kwargs):
140 return self._call(name, *args, **kwargs)
140 return self._call(name, *args, **kwargs)
141 return f
141 return f
142
142
143 @exceptions.map_vcs_exceptions
143 @exceptions.map_vcs_exceptions
144 def _call(self, name, *args, **kwargs):
144 def _call(self, name, *args, **kwargs):
145 payload = {
145 payload = {
146 'id': str(uuid.uuid4()),
146 'id': str(uuid.uuid4()),
147 'method': name,
147 'method': name,
148 'params': {'args': args, 'kwargs': kwargs}
148 'params': {'args': args, 'kwargs': kwargs}
149 }
149 }
150 return _remote_call(
150 return _remote_call(
151 self.url, payload, EXCEPTIONS_MAP, self._session_factory())
151 self.url, payload, EXCEPTIONS_MAP, self._session_factory())
152
152
153
153
154 class RemoteVCSMaker(object):
154 class RemoteVCSMaker(object):
155
155
156 def __init__(self, server_and_port, backend_endpoint, backend_type, session_factory):
156 def __init__(self, server_and_port, backend_endpoint, backend_type, session_factory):
157 self.url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint)
157 self.url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint)
158 self.stream_url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint+'/stream')
158 self.stream_url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint+'/stream')
159
159
160 self._session_factory = session_factory
160 self._session_factory = session_factory
161 self.backend_type = backend_type
161 self.backend_type = backend_type
162
162
163 @classmethod
163 @classmethod
164 def init_cache_region(cls, repo_id):
164 def init_cache_region(cls, repo_id):
165 cache_namespace_uid = 'cache_repo.{}'.format(repo_id)
165 cache_namespace_uid = 'cache_repo.{}'.format(repo_id)
166 region = rc_cache.get_or_create_region('cache_repo', cache_namespace_uid)
166 region = rc_cache.get_or_create_region('cache_repo', cache_namespace_uid)
167 return region, cache_namespace_uid
167 return region, cache_namespace_uid
168
168
169 def __call__(self, path, repo_id, config, with_wire=None):
169 def __call__(self, path, repo_id, config, with_wire=None):
170 log.debug('%s RepoMaker call on %s', self.backend_type.upper(), path)
170 log.debug('%s RepoMaker call on %s', self.backend_type.upper(), path)
171 return RemoteRepo(path, repo_id, config, self, with_wire=with_wire)
171 return RemoteRepo(path, repo_id, config, self, with_wire=with_wire)
172
172
173 def __getattr__(self, name):
173 def __getattr__(self, name):
174 def remote_attr(*args, **kwargs):
174 def remote_attr(*args, **kwargs):
175 return self._call(name, *args, **kwargs)
175 return self._call(name, *args, **kwargs)
176 return remote_attr
176 return remote_attr
177
177
178 @exceptions.map_vcs_exceptions
178 @exceptions.map_vcs_exceptions
179 def _call(self, func_name, *args, **kwargs):
179 def _call(self, func_name, *args, **kwargs):
180 payload = {
180 payload = {
181 'id': str(uuid.uuid4()),
181 'id': str(uuid.uuid4()),
182 'method': func_name,
182 'method': func_name,
183 'backend': self.backend_type,
183 'backend': self.backend_type,
184 'params': {'args': args, 'kwargs': kwargs}
184 'params': {'args': args, 'kwargs': kwargs}
185 }
185 }
186 url = self.url
186 url = self.url
187 return _remote_call(url, payload, EXCEPTIONS_MAP, self._session_factory())
187 return _remote_call(url, payload, EXCEPTIONS_MAP, self._session_factory())
188
188
189
189
190 class RemoteRepo(object):
190 class RemoteRepo(object):
191 CHUNK_SIZE = 16384
191 CHUNK_SIZE = 16384
192
192
193 def __init__(self, path, repo_id, config, remote_maker, with_wire=None):
193 def __init__(self, path, repo_id, config, remote_maker, with_wire=None):
194 self.url = remote_maker.url
194 self.url = remote_maker.url
195 self.stream_url = remote_maker.stream_url
195 self.stream_url = remote_maker.stream_url
196 self._session = remote_maker._session_factory()
196 self._session = remote_maker._session_factory()
197
197
198 cache_repo_id = self._repo_id_sanitizer(repo_id)
198 cache_repo_id = self._repo_id_sanitizer(repo_id)
199 _repo_name = self._get_repo_name(config, path)
199 _repo_name = self._get_repo_name(config, path)
200 self._cache_region, self._cache_namespace = \
200 self._cache_region, self._cache_namespace = \
201 remote_maker.init_cache_region(cache_repo_id)
201 remote_maker.init_cache_region(cache_repo_id)
202
202
203 with_wire = with_wire or {}
203 with_wire = with_wire or {}
204
204
205 repo_state_uid = with_wire.get('repo_state_uid') or 'state'
205 repo_state_uid = with_wire.get('repo_state_uid') or 'state'
206
206
207 self._wire = {
207 self._wire = {
208 "_repo_name": _repo_name,
208 "_repo_name": _repo_name,
209 "path": path, # repo path
209 "path": path, # repo path
210 "repo_id": repo_id,
210 "repo_id": repo_id,
211 "cache_repo_id": cache_repo_id,
211 "cache_repo_id": cache_repo_id,
212 "config": config,
212 "config": config,
213 "repo_state_uid": repo_state_uid,
213 "repo_state_uid": repo_state_uid,
214 "context": self._create_vcs_cache_context(path, repo_state_uid)
214 "context": self._create_vcs_cache_context(path, repo_state_uid)
215 }
215 }
216
216
217 if with_wire:
217 if with_wire:
218 self._wire.update(with_wire)
218 self._wire.update(with_wire)
219
219
220 # NOTE(johbo): Trading complexity for performance. Avoiding the call to
220 # NOTE(johbo): Trading complexity for performance. Avoiding the call to
221 # log.debug brings a few percent gain even if is is not active.
221 # log.debug brings a few percent gain even if is is not active.
222 if log.isEnabledFor(logging.DEBUG):
222 if log.isEnabledFor(logging.DEBUG):
223 self._call_with_logging = True
223 self._call_with_logging = True
224
224
225 self.cert_dir = get_cert_path(rhodecode.CONFIG.get('__file__'))
225 self.cert_dir = get_cert_path(rhodecode.CONFIG.get('__file__'))
226
226
227 def _get_repo_name(self, config, path):
227 def _get_repo_name(self, config, path):
228 repo_store = config.get('paths', '/')
228 repo_store = config.get('paths', '/')
229 return path.split(repo_store)[-1].lstrip('/')
229 return path.split(repo_store)[-1].lstrip('/')
230
230
231 def _repo_id_sanitizer(self, repo_id):
231 def _repo_id_sanitizer(self, repo_id):
232 pathless = repo_id.replace('/', '__').replace('-', '_')
232 pathless = repo_id.replace('/', '__').replace('-', '_')
233 return ''.join(char if ord(char) < 128 else '_{}_'.format(ord(char)) for char in pathless)
233 return ''.join(char if ord(char) < 128 else '_{}_'.format(ord(char)) for char in pathless)
234
234
235 def __getattr__(self, name):
235 def __getattr__(self, name):
236
236
237 if name.startswith('stream:'):
237 if name.startswith('stream:'):
238 def repo_remote_attr(*args, **kwargs):
238 def repo_remote_attr(*args, **kwargs):
239 return self._call_stream(name, *args, **kwargs)
239 return self._call_stream(name, *args, **kwargs)
240 else:
240 else:
241 def repo_remote_attr(*args, **kwargs):
241 def repo_remote_attr(*args, **kwargs):
242 return self._call(name, *args, **kwargs)
242 return self._call(name, *args, **kwargs)
243
243
244 return repo_remote_attr
244 return repo_remote_attr
245
245
246 def _base_call(self, name, *args, **kwargs):
246 def _base_call(self, name, *args, **kwargs):
247 # TODO: oliver: This is currently necessary pre-call since the
247 # TODO: oliver: This is currently necessary pre-call since the
248 # config object is being changed for hooking scenarios
248 # config object is being changed for hooking scenarios
249 wire = copy.deepcopy(self._wire)
249 wire = copy.deepcopy(self._wire)
250 wire["config"] = wire["config"].serialize()
250 wire["config"] = wire["config"].serialize()
251 wire["config"].append(('vcs', 'ssl_dir', self.cert_dir))
251 wire["config"].append(('vcs', 'ssl_dir', self.cert_dir))
252
252
253 payload = {
253 payload = {
254 'id': str(uuid.uuid4()),
254 'id': str(uuid.uuid4()),
255 'method': name,
255 'method': name,
256 "_repo_name": wire['_repo_name'],
256 "_repo_name": wire['_repo_name'],
257 'params': {'wire': wire, 'args': args, 'kwargs': kwargs}
257 'params': {'wire': wire, 'args': args, 'kwargs': kwargs}
258 }
258 }
259
259
260 context_uid = wire.get('context')
260 context_uid = wire.get('context')
261 return context_uid, payload
261 return context_uid, payload
262
262
263 def get_local_cache(self, name, args):
263 def get_local_cache(self, name, args):
264 cache_on = False
264 cache_on = False
265 cache_key = ''
265 cache_key = ''
266 local_cache_on = str2bool(rhodecode.CONFIG.get('vcs.methods.cache'))
266 local_cache_on = str2bool(rhodecode.CONFIG.get('vcs.methods.cache'))
267
267
268 cache_methods = [
268 cache_methods = [
269 'branches', 'tags', 'bookmarks',
269 'branches', 'tags', 'bookmarks',
270 'is_large_file', 'is_binary',
270 'is_large_file', 'is_binary',
271 'fctx_size', 'stream:fctx_node_data', 'blob_raw_length',
271 'fctx_size', 'stream:fctx_node_data', 'blob_raw_length',
272 'node_history',
272 'node_history',
273 'revision', 'tree_items',
273 'revision', 'tree_items',
274 'ctx_list', 'ctx_branch', 'ctx_description',
274 'ctx_list', 'ctx_branch', 'ctx_description',
275 'bulk_request',
275 'bulk_request',
276 'assert_correct_path'
276 'assert_correct_path'
277 ]
277 ]
278
278
279 if local_cache_on and name in cache_methods:
279 if local_cache_on and name in cache_methods:
280 cache_on = True
280 cache_on = True
281 repo_state_uid = self._wire['repo_state_uid']
281 repo_state_uid = self._wire['repo_state_uid']
282 call_args = [a for a in args]
282 call_args = [a for a in args]
283 cache_key = compute_key_from_params(repo_state_uid, name, *call_args)
283 cache_key = compute_key_from_params(repo_state_uid, name, *call_args)
284
284
285 return cache_on, cache_key
285 return cache_on, cache_key
286
286
287 @exceptions.map_vcs_exceptions
287 @exceptions.map_vcs_exceptions
288 def _call(self, name, *args, **kwargs):
288 def _call(self, name, *args, **kwargs):
289 context_uid, payload = self._base_call(name, *args, **kwargs)
289 context_uid, payload = self._base_call(name, *args, **kwargs)
290 url = self.url
290 url = self.url
291
291
292 start = time.time()
292 start = time.time()
293 cache_on, cache_key = self.get_local_cache(name, args)
293 cache_on, cache_key = self.get_local_cache(name, args)
294
294
295 @self._cache_region.conditional_cache_on_arguments(
295 @self._cache_region.conditional_cache_on_arguments(
296 namespace=self._cache_namespace, condition=cache_on and cache_key)
296 namespace=self._cache_namespace, condition=cache_on and cache_key)
297 def remote_call(_cache_key):
297 def remote_call(_cache_key):
298 if self._call_with_logging:
298 if self._call_with_logging:
299 log.debug('Calling %s@%s with args:%.10240r. wire_context: %s cache_on: %s',
299 log.debug('Calling %s@%s with args:%.10240r. wire_context: %s cache_on: %s',
300 url, name, args, context_uid, cache_on)
300 url, name, args, context_uid, cache_on)
301 return _remote_call(url, payload, EXCEPTIONS_MAP, self._session)
301 return _remote_call(url, payload, EXCEPTIONS_MAP, self._session)
302
302
303 result = remote_call(cache_key)
303 result = remote_call(cache_key)
304 if self._call_with_logging:
304 if self._call_with_logging:
305 log.debug('Call %s@%s took: %.4fs. wire_context: %s',
305 log.debug('Call %s@%s took: %.4fs. wire_context: %s',
306 url, name, time.time()-start, context_uid)
306 url, name, time.time()-start, context_uid)
307 return result
307 return result
308
308
309 @exceptions.map_vcs_exceptions
309 @exceptions.map_vcs_exceptions
310 def _call_stream(self, name, *args, **kwargs):
310 def _call_stream(self, name, *args, **kwargs):
311 context_uid, payload = self._base_call(name, *args, **kwargs)
311 context_uid, payload = self._base_call(name, *args, **kwargs)
312 payload['chunk_size'] = self.CHUNK_SIZE
312 payload['chunk_size'] = self.CHUNK_SIZE
313 url = self.stream_url
313 url = self.stream_url
314
314
315 start = time.time()
315 start = time.time()
316 cache_on, cache_key = self.get_local_cache(name, args)
316 cache_on, cache_key = self.get_local_cache(name, args)
317
317
318 # Cache is a problem because this is a stream
318 # Cache is a problem because this is a stream
319 def streaming_remote_call(_cache_key):
319 def streaming_remote_call(_cache_key):
320 if self._call_with_logging:
320 if self._call_with_logging:
321 log.debug('Calling %s@%s with args:%.10240r. wire_context: %s cache_on: %s',
321 log.debug('Calling %s@%s with args:%.10240r. wire_context: %s cache_on: %s',
322 url, name, args, context_uid, cache_on)
322 url, name, args, context_uid, cache_on)
323 return _streaming_remote_call(url, payload, EXCEPTIONS_MAP, self._session, self.CHUNK_SIZE)
323 return _streaming_remote_call(url, payload, EXCEPTIONS_MAP, self._session, self.CHUNK_SIZE)
324
324
325 result = streaming_remote_call(cache_key)
325 result = streaming_remote_call(cache_key)
326 if self._call_with_logging:
326 if self._call_with_logging:
327 log.debug('Call %s@%s took: %.4fs. wire_context: %s',
327 log.debug('Call %s@%s took: %.4fs. wire_context: %s',
328 url, name, time.time()-start, context_uid)
328 url, name, time.time()-start, context_uid)
329 return result
329 return result
330
330
331 def __getitem__(self, key):
331 def __getitem__(self, key):
332 return self.revision(key)
332 return self.revision(key)
333
333
334 def _create_vcs_cache_context(self, *args):
334 def _create_vcs_cache_context(self, *args):
335 """
335 """
336 Creates a unique string which is passed to the VCSServer on every
336 Creates a unique string which is passed to the VCSServer on every
337 remote call. It is used as cache key in the VCSServer.
337 remote call. It is used as cache key in the VCSServer.
338 """
338 """
339 hash_key = '-'.join(map(str, args))
339 hash_key = '-'.join(map(str, args))
340 return str(uuid.uuid5(uuid.NAMESPACE_URL, hash_key))
340 return str(uuid.uuid5(uuid.NAMESPACE_URL, hash_key))
341
341
342 def invalidate_vcs_cache(self):
342 def invalidate_vcs_cache(self):
343 """
343 """
344 This invalidates the context which is sent to the VCSServer on every
344 This invalidates the context which is sent to the VCSServer on every
345 call to a remote method. It forces the VCSServer to create a fresh
345 call to a remote method. It forces the VCSServer to create a fresh
346 repository instance on the next call to a remote method.
346 repository instance on the next call to a remote method.
347 """
347 """
348 self._wire['context'] = str(uuid.uuid4())
348 self._wire['context'] = str(uuid.uuid4())
349
349
350
350
351 class VcsHttpProxy(object):
351 class VcsHttpProxy(object):
352
352
353 CHUNK_SIZE = 16384
353 CHUNK_SIZE = 16384
354
354
355 def __init__(self, server_and_port, backend_endpoint):
355 def __init__(self, server_and_port, backend_endpoint):
356 retries = Retry(total=5, connect=None, read=None, redirect=None)
356 retries = Retry(total=5, connect=None, read=None, redirect=None)
357
357
358 adapter = requests.adapters.HTTPAdapter(max_retries=retries)
358 adapter = requests.adapters.HTTPAdapter(max_retries=retries)
359 self.base_url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint)
359 self.base_url = urllib.parse.urlparse.urljoin('http://%s' % server_and_port, backend_endpoint)
360 self.session = requests.Session()
360 self.session = requests.Session()
361 self.session.mount('http://', adapter)
361 self.session.mount('http://', adapter)
362
362
363 def handle(self, environment, input_data, *args, **kwargs):
363 def handle(self, environment, input_data, *args, **kwargs):
364 data = {
364 data = {
365 'environment': environment,
365 'environment': environment,
366 'input_data': input_data,
366 'input_data': input_data,
367 'args': args,
367 'args': args,
368 'kwargs': kwargs
368 'kwargs': kwargs
369 }
369 }
370 result = self.session.post(
370 result = self.session.post(
371 self.base_url, msgpack.packb(data), stream=True)
371 self.base_url, msgpack.packb(data), stream=True)
372 return self._get_result(result)
372 return self._get_result(result)
373
373
374 def _deserialize_and_raise(self, error):
374 def _deserialize_and_raise(self, error):
375 exception = Exception(error['message'])
375 exception = Exception(error['message'])
376 try:
376 try:
377 exception._vcs_kind = error['_vcs_kind']
377 exception._vcs_kind = error['_vcs_kind']
378 except KeyError:
378 except KeyError:
379 pass
379 pass
380 raise exception
380 raise exception
381
381
382 def _iterate(self, result):
382 def _iterate(self, result):
383 unpacker = msgpack.Unpacker()
383 unpacker = msgpack.Unpacker()
384 for line in result.iter_content(chunk_size=self.CHUNK_SIZE):
384 for line in result.iter_content(chunk_size=self.CHUNK_SIZE):
385 unpacker.feed(line)
385 unpacker.feed(line)
386 for chunk in unpacker:
386 for chunk in unpacker:
387 yield chunk
387 yield chunk
388
388
389 def _get_result(self, result):
389 def _get_result(self, result):
390 iterator = self._iterate(result)
390 iterator = self._iterate(result)
391 error = iterator.next()
391 error = next(iterator)
392 if error:
392 if error:
393 self._deserialize_and_raise(error)
393 self._deserialize_and_raise(error)
394
394
395 status = iterator.next()
395 status = next(iterator)
396 headers = iterator.next()
396 headers = next(iterator)
397
397
398 return iterator, status, headers
398 return iterator, status, headers
399
399
400
400
401 class ThreadlocalSessionFactory(object):
401 class ThreadlocalSessionFactory(object):
402 """
402 """
403 Creates one CurlSession per thread on demand.
403 Creates one CurlSession per thread on demand.
404 """
404 """
405
405
406 def __init__(self):
406 def __init__(self):
407 self._thread_local = threading.local()
407 self._thread_local = threading.local()
408
408
409 def __call__(self):
409 def __call__(self):
410 if not hasattr(self._thread_local, 'curl_session'):
410 if not hasattr(self._thread_local, 'curl_session'):
411 self._thread_local.curl_session = CurlSession()
411 self._thread_local.curl_session = CurlSession()
412 return self._thread_local.curl_session
412 return self._thread_local.curl_session
@@ -1,135 +1,135 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2020 RhodeCode GmbH
3 # Copyright (C) 2010-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21
21
22 import logging
22 import logging
23
23
24 import rhodecode
24 import rhodecode
25 from rhodecode.model import meta, db
25 from rhodecode.model import meta, db
26 from rhodecode.lib.utils2 import obfuscate_url_pw, get_encryption_key
26 from rhodecode.lib.utils2 import obfuscate_url_pw, get_encryption_key
27
27
28 log = logging.getLogger(__name__)
28 log = logging.getLogger(__name__)
29
29
30
30
31 def init_model(engine, encryption_key=None):
31 def init_model(engine, encryption_key=None):
32 """
32 """
33 Initializes db session, bind the engine with the metadata,
33 Initializes db session, bind the engine with the metadata,
34 Call this before using any of the tables or classes in the model,
34 Call this before using any of the tables or classes in the model,
35 preferably once in application start
35 preferably once in application start
36
36
37 :param engine: engine to bind to
37 :param engine: engine to bind to
38 """
38 """
39 engine_str = obfuscate_url_pw(str(engine.url))
39 engine_str = obfuscate_url_pw(str(engine.url))
40 log.info("RhodeCode %s initializing db for %s", rhodecode.__version__, engine_str)
40 log.info("RhodeCode %s initializing db for %s", rhodecode.__version__, engine_str)
41 meta.Base.metadata.bind = engine
41 meta.Base.metadata.bind = engine
42 db.ENCRYPTION_KEY = encryption_key
42 db.ENCRYPTION_KEY = encryption_key
43
43
44
44
45 def init_model_encryption(migration_models, config=None):
45 def init_model_encryption(migration_models, config=None):
46 from pyramid.threadlocal import get_current_registry
46 from pyramid.threadlocal import get_current_registry
47 config = config or get_current_registry().settings
47 config = config or get_current_registry().settings
48 migration_models.ENCRYPTION_KEY = get_encryption_key(config)
48 migration_models.ENCRYPTION_KEY = get_encryption_key(config)
49 db.ENCRYPTION_KEY = get_encryption_key(config)
49 db.ENCRYPTION_KEY = get_encryption_key(config)
50
50
51
51
52 class BaseModel(object):
52 class BaseModel(object):
53 """
53 """
54 Base Model for all RhodeCode models, it adds sql alchemy session
54 Base Model for all RhodeCode models, it adds sql alchemy session
55 into instance of model
55 into instance of model
56
56
57 :param sa: If passed it reuses this session instead of creating a new one
57 :param sa: If passed it reuses this session instead of creating a new one
58 """
58 """
59
59
60 cls = None # override in child class
60 cls = None # override in child class
61
61
62 def __init__(self, sa=None):
62 def __init__(self, sa=None):
63 if sa is not None:
63 if sa is not None:
64 self.sa = sa
64 self.sa = sa
65 else:
65 else:
66 self.sa = meta.Session()
66 self.sa = meta.Session()
67
67
68 def _get_instance(self, cls, instance, callback=None):
68 def _get_instance(self, cls, instance, callback=None):
69 """
69 """
70 Gets instance of given cls using some simple lookup mechanism.
70 Gets instance of given cls using some simple lookup mechanism.
71
71
72 :param cls: classes to fetch
72 :param cls: classes to fetch
73 :param instance: int or Instance
73 :param instance: int or Instance
74 :param callback: callback to call if all lookups failed
74 :param callback: callback to call if all lookups failed
75 """
75 """
76
76
77 if isinstance(instance, cls):
77 if isinstance(instance, cls):
78 return instance
78 return instance
79 elif isinstance(instance, int):
79 elif isinstance(instance, int):
80 if isinstance(cls, tuple):
80 if isinstance(cls, tuple):
81 # if we pass multi instances we pick first to .get()
81 # if we pass multi instances we pick first to .get()
82 cls = cls[0]
82 cls = cls[0]
83 return cls.get(instance)
83 return cls.get(instance)
84 else:
84 else:
85 if instance:
85 if instance:
86 if callback is None:
86 if callback is None:
87 raise Exception(
87 raise Exception(
88 'given object must be int, long or Instance of %s '
88 'given object must be int or Instance of %s '
89 'got %s, no callback provided' % (cls, type(instance))
89 'got %s, no callback provided' % (cls, type(instance))
90 )
90 )
91 else:
91 else:
92 return callback(instance)
92 return callback(instance)
93
93
94 def _get_user(self, user):
94 def _get_user(self, user):
95 """
95 """
96 Helper method to get user by ID, or username fallback
96 Helper method to get user by ID, or username fallback
97
97
98 :param user: UserID, username, or User instance
98 :param user: UserID, username, or User instance
99 """
99 """
100 return self._get_instance(
100 return self._get_instance(
101 db.User, user, callback=db.User.get_by_username)
101 db.User, user, callback=db.User.get_by_username)
102
102
103 def _get_user_group(self, user_group):
103 def _get_user_group(self, user_group):
104 """
104 """
105 Helper method to get user by ID, or username fallback
105 Helper method to get user by ID, or username fallback
106
106
107 :param user_group: UserGroupID, user_group_name, or UserGroup instance
107 :param user_group: UserGroupID, user_group_name, or UserGroup instance
108 """
108 """
109 return self._get_instance(
109 return self._get_instance(
110 db.UserGroup, user_group, callback=db.UserGroup.get_by_group_name)
110 db.UserGroup, user_group, callback=db.UserGroup.get_by_group_name)
111
111
112 def _get_repo(self, repository):
112 def _get_repo(self, repository):
113 """
113 """
114 Helper method to get repository by ID, or repository name
114 Helper method to get repository by ID, or repository name
115
115
116 :param repository: RepoID, repository name or Repository Instance
116 :param repository: RepoID, repository name or Repository Instance
117 """
117 """
118 return self._get_instance(
118 return self._get_instance(
119 db.Repository, repository, callback=db.Repository.get_by_repo_name)
119 db.Repository, repository, callback=db.Repository.get_by_repo_name)
120
120
121 def _get_perm(self, permission):
121 def _get_perm(self, permission):
122 """
122 """
123 Helper method to get permission by ID, or permission name
123 Helper method to get permission by ID, or permission name
124
124
125 :param permission: PermissionID, permission_name or Permission instance
125 :param permission: PermissionID, permission_name or Permission instance
126 """
126 """
127 return self._get_instance(
127 return self._get_instance(
128 db.Permission, permission, callback=db.Permission.get_by_key)
128 db.Permission, permission, callback=db.Permission.get_by_key)
129
129
130 @classmethod
130 @classmethod
131 def get_all(cls):
131 def get_all(cls):
132 """
132 """
133 Returns all instances of what is defined in `cls` class variable
133 Returns all instances of what is defined in `cls` class variable
134 """
134 """
135 return cls.cls.getAll()
135 return cls.cls.getAll()
1 NO CONTENT: modified file
NO CONTENT: modified file
The requested commit or file is too big and content was truncated. Show full diff
@@ -1,918 +1,918 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2020 RhodeCode GmbH
3 # Copyright (C) 2010-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 import os
21 import os
22 import re
22 import re
23 import hashlib
23 import hashlib
24 import logging
24 import logging
25 import time
25 import time
26 from collections import namedtuple
26 from collections import namedtuple
27 from functools import wraps
27 from functools import wraps
28 import bleach
28 import bleach
29 from pyramid.threadlocal import get_current_request, get_current_registry
29 from pyramid.threadlocal import get_current_request, get_current_registry
30
30
31 from rhodecode.lib import rc_cache
31 from rhodecode.lib import rc_cache
32 from rhodecode.lib.utils2 import (
32 from rhodecode.lib.utils2 import (
33 Optional, AttributeDict, safe_str, remove_prefix, str2bool)
33 Optional, AttributeDict, safe_str, remove_prefix, str2bool)
34 from rhodecode.lib.vcs.backends import base
34 from rhodecode.lib.vcs.backends import base
35 from rhodecode.lib.statsd_client import StatsdClient
35 from rhodecode.lib.statsd_client import StatsdClient
36 from rhodecode.model import BaseModel
36 from rhodecode.model import BaseModel
37 from rhodecode.model.db import (
37 from rhodecode.model.db import (
38 RepoRhodeCodeUi, RepoRhodeCodeSetting, RhodeCodeUi, RhodeCodeSetting)
38 RepoRhodeCodeUi, RepoRhodeCodeSetting, RhodeCodeUi, RhodeCodeSetting)
39 from rhodecode.model.meta import Session
39 from rhodecode.model.meta import Session
40
40
41
41
42 log = logging.getLogger(__name__)
42 log = logging.getLogger(__name__)
43
43
44
44
45 UiSetting = namedtuple(
45 UiSetting = namedtuple(
46 'UiSetting', ['section', 'key', 'value', 'active'])
46 'UiSetting', ['section', 'key', 'value', 'active'])
47
47
48 SOCIAL_PLUGINS_LIST = ['github', 'bitbucket', 'twitter', 'google']
48 SOCIAL_PLUGINS_LIST = ['github', 'bitbucket', 'twitter', 'google']
49
49
50
50
51 class SettingNotFound(Exception):
51 class SettingNotFound(Exception):
52 def __init__(self, setting_id):
52 def __init__(self, setting_id):
53 msg = 'Setting `{}` is not found'.format(setting_id)
53 msg = 'Setting `{}` is not found'.format(setting_id)
54 super(SettingNotFound, self).__init__(msg)
54 super(SettingNotFound, self).__init__(msg)
55
55
56
56
57 class SettingsModel(BaseModel):
57 class SettingsModel(BaseModel):
58 BUILTIN_HOOKS = (
58 BUILTIN_HOOKS = (
59 RhodeCodeUi.HOOK_REPO_SIZE, RhodeCodeUi.HOOK_PUSH,
59 RhodeCodeUi.HOOK_REPO_SIZE, RhodeCodeUi.HOOK_PUSH,
60 RhodeCodeUi.HOOK_PRE_PUSH, RhodeCodeUi.HOOK_PRETX_PUSH,
60 RhodeCodeUi.HOOK_PRE_PUSH, RhodeCodeUi.HOOK_PRETX_PUSH,
61 RhodeCodeUi.HOOK_PULL, RhodeCodeUi.HOOK_PRE_PULL,
61 RhodeCodeUi.HOOK_PULL, RhodeCodeUi.HOOK_PRE_PULL,
62 RhodeCodeUi.HOOK_PUSH_KEY,)
62 RhodeCodeUi.HOOK_PUSH_KEY,)
63 HOOKS_SECTION = 'hooks'
63 HOOKS_SECTION = 'hooks'
64
64
65 def __init__(self, sa=None, repo=None):
65 def __init__(self, sa=None, repo=None):
66 self.repo = repo
66 self.repo = repo
67 self.UiDbModel = RepoRhodeCodeUi if repo else RhodeCodeUi
67 self.UiDbModel = RepoRhodeCodeUi if repo else RhodeCodeUi
68 self.SettingsDbModel = (
68 self.SettingsDbModel = (
69 RepoRhodeCodeSetting if repo else RhodeCodeSetting)
69 RepoRhodeCodeSetting if repo else RhodeCodeSetting)
70 super(SettingsModel, self).__init__(sa)
70 super(SettingsModel, self).__init__(sa)
71
71
72 def get_ui_by_key(self, key):
72 def get_ui_by_key(self, key):
73 q = self.UiDbModel.query()
73 q = self.UiDbModel.query()
74 q = q.filter(self.UiDbModel.ui_key == key)
74 q = q.filter(self.UiDbModel.ui_key == key)
75 q = self._filter_by_repo(RepoRhodeCodeUi, q)
75 q = self._filter_by_repo(RepoRhodeCodeUi, q)
76 return q.scalar()
76 return q.scalar()
77
77
78 def get_ui_by_section(self, section):
78 def get_ui_by_section(self, section):
79 q = self.UiDbModel.query()
79 q = self.UiDbModel.query()
80 q = q.filter(self.UiDbModel.ui_section == section)
80 q = q.filter(self.UiDbModel.ui_section == section)
81 q = self._filter_by_repo(RepoRhodeCodeUi, q)
81 q = self._filter_by_repo(RepoRhodeCodeUi, q)
82 return q.all()
82 return q.all()
83
83
84 def get_ui_by_section_and_key(self, section, key):
84 def get_ui_by_section_and_key(self, section, key):
85 q = self.UiDbModel.query()
85 q = self.UiDbModel.query()
86 q = q.filter(self.UiDbModel.ui_section == section)
86 q = q.filter(self.UiDbModel.ui_section == section)
87 q = q.filter(self.UiDbModel.ui_key == key)
87 q = q.filter(self.UiDbModel.ui_key == key)
88 q = self._filter_by_repo(RepoRhodeCodeUi, q)
88 q = self._filter_by_repo(RepoRhodeCodeUi, q)
89 return q.scalar()
89 return q.scalar()
90
90
91 def get_ui(self, section=None, key=None):
91 def get_ui(self, section=None, key=None):
92 q = self.UiDbModel.query()
92 q = self.UiDbModel.query()
93 q = self._filter_by_repo(RepoRhodeCodeUi, q)
93 q = self._filter_by_repo(RepoRhodeCodeUi, q)
94
94
95 if section:
95 if section:
96 q = q.filter(self.UiDbModel.ui_section == section)
96 q = q.filter(self.UiDbModel.ui_section == section)
97 if key:
97 if key:
98 q = q.filter(self.UiDbModel.ui_key == key)
98 q = q.filter(self.UiDbModel.ui_key == key)
99
99
100 # TODO: mikhail: add caching
100 # TODO: mikhail: add caching
101 result = [
101 result = [
102 UiSetting(
102 UiSetting(
103 section=safe_str(r.ui_section), key=safe_str(r.ui_key),
103 section=safe_str(r.ui_section), key=safe_str(r.ui_key),
104 value=safe_str(r.ui_value), active=r.ui_active
104 value=safe_str(r.ui_value), active=r.ui_active
105 )
105 )
106 for r in q.all()
106 for r in q.all()
107 ]
107 ]
108 return result
108 return result
109
109
110 def get_builtin_hooks(self):
110 def get_builtin_hooks(self):
111 q = self.UiDbModel.query()
111 q = self.UiDbModel.query()
112 q = q.filter(self.UiDbModel.ui_key.in_(self.BUILTIN_HOOKS))
112 q = q.filter(self.UiDbModel.ui_key.in_(self.BUILTIN_HOOKS))
113 return self._get_hooks(q)
113 return self._get_hooks(q)
114
114
115 def get_custom_hooks(self):
115 def get_custom_hooks(self):
116 q = self.UiDbModel.query()
116 q = self.UiDbModel.query()
117 q = q.filter(~self.UiDbModel.ui_key.in_(self.BUILTIN_HOOKS))
117 q = q.filter(~self.UiDbModel.ui_key.in_(self.BUILTIN_HOOKS))
118 return self._get_hooks(q)
118 return self._get_hooks(q)
119
119
120 def create_ui_section_value(self, section, val, key=None, active=True):
120 def create_ui_section_value(self, section, val, key=None, active=True):
121 new_ui = self.UiDbModel()
121 new_ui = self.UiDbModel()
122 new_ui.ui_section = section
122 new_ui.ui_section = section
123 new_ui.ui_value = val
123 new_ui.ui_value = val
124 new_ui.ui_active = active
124 new_ui.ui_active = active
125
125
126 repository_id = ''
126 repository_id = ''
127 if self.repo:
127 if self.repo:
128 repo = self._get_repo(self.repo)
128 repo = self._get_repo(self.repo)
129 repository_id = repo.repo_id
129 repository_id = repo.repo_id
130 new_ui.repository_id = repository_id
130 new_ui.repository_id = repository_id
131
131
132 if not key:
132 if not key:
133 # keys are unique so they need appended info
133 # keys are unique so they need appended info
134 if self.repo:
134 if self.repo:
135 key = hashlib.sha1(
135 key = hashlib.sha1(
136 '{}{}{}'.format(section, val, repository_id)).hexdigest()
136 '{}{}{}'.format(section, val, repository_id)).hexdigest()
137 else:
137 else:
138 key = hashlib.sha1('{}{}'.format(section, val)).hexdigest()
138 key = hashlib.sha1('{}{}'.format(section, val)).hexdigest()
139
139
140 new_ui.ui_key = key
140 new_ui.ui_key = key
141
141
142 Session().add(new_ui)
142 Session().add(new_ui)
143 return new_ui
143 return new_ui
144
144
145 def create_or_update_hook(self, key, value):
145 def create_or_update_hook(self, key, value):
146 ui = (
146 ui = (
147 self.get_ui_by_section_and_key(self.HOOKS_SECTION, key) or
147 self.get_ui_by_section_and_key(self.HOOKS_SECTION, key) or
148 self.UiDbModel())
148 self.UiDbModel())
149 ui.ui_section = self.HOOKS_SECTION
149 ui.ui_section = self.HOOKS_SECTION
150 ui.ui_active = True
150 ui.ui_active = True
151 ui.ui_key = key
151 ui.ui_key = key
152 ui.ui_value = value
152 ui.ui_value = value
153
153
154 if self.repo:
154 if self.repo:
155 repo = self._get_repo(self.repo)
155 repo = self._get_repo(self.repo)
156 repository_id = repo.repo_id
156 repository_id = repo.repo_id
157 ui.repository_id = repository_id
157 ui.repository_id = repository_id
158
158
159 Session().add(ui)
159 Session().add(ui)
160 return ui
160 return ui
161
161
162 def delete_ui(self, id_):
162 def delete_ui(self, id_):
163 ui = self.UiDbModel.get(id_)
163 ui = self.UiDbModel.get(id_)
164 if not ui:
164 if not ui:
165 raise SettingNotFound(id_)
165 raise SettingNotFound(id_)
166 Session().delete(ui)
166 Session().delete(ui)
167
167
168 def get_setting_by_name(self, name):
168 def get_setting_by_name(self, name):
169 q = self._get_settings_query()
169 q = self._get_settings_query()
170 q = q.filter(self.SettingsDbModel.app_settings_name == name)
170 q = q.filter(self.SettingsDbModel.app_settings_name == name)
171 return q.scalar()
171 return q.scalar()
172
172
173 def create_or_update_setting(
173 def create_or_update_setting(
174 self, name, val=Optional(''), type_=Optional('unicode')):
174 self, name, val=Optional(''), type_=Optional('unicode')):
175 """
175 """
176 Creates or updates RhodeCode setting. If updates is triggered it will
176 Creates or updates RhodeCode setting. If updates is triggered it will
177 only update parameters that are explicitly set Optional instance will
177 only update parameters that are explicitly set Optional instance will
178 be skipped
178 be skipped
179
179
180 :param name:
180 :param name:
181 :param val:
181 :param val:
182 :param type_:
182 :param type_:
183 :return:
183 :return:
184 """
184 """
185
185
186 res = self.get_setting_by_name(name)
186 res = self.get_setting_by_name(name)
187 repo = self._get_repo(self.repo) if self.repo else None
187 repo = self._get_repo(self.repo) if self.repo else None
188
188
189 if not res:
189 if not res:
190 val = Optional.extract(val)
190 val = Optional.extract(val)
191 type_ = Optional.extract(type_)
191 type_ = Optional.extract(type_)
192
192
193 args = (
193 args = (
194 (repo.repo_id, name, val, type_)
194 (repo.repo_id, name, val, type_)
195 if repo else (name, val, type_))
195 if repo else (name, val, type_))
196 res = self.SettingsDbModel(*args)
196 res = self.SettingsDbModel(*args)
197
197
198 else:
198 else:
199 if self.repo:
199 if self.repo:
200 res.repository_id = repo.repo_id
200 res.repository_id = repo.repo_id
201
201
202 res.app_settings_name = name
202 res.app_settings_name = name
203 if not isinstance(type_, Optional):
203 if not isinstance(type_, Optional):
204 # update if set
204 # update if set
205 res.app_settings_type = type_
205 res.app_settings_type = type_
206 if not isinstance(val, Optional):
206 if not isinstance(val, Optional):
207 # update if set
207 # update if set
208 res.app_settings_value = val
208 res.app_settings_value = val
209
209
210 Session().add(res)
210 Session().add(res)
211 return res
211 return res
212
212
213 def get_cache_region(self):
213 def get_cache_region(self):
214 repo = self._get_repo(self.repo) if self.repo else None
214 repo = self._get_repo(self.repo) if self.repo else None
215 cache_key = "repo.{}".format(repo.repo_id) if repo else "general_settings"
215 cache_key = "repo.{}".format(repo.repo_id) if repo else "general_settings"
216 cache_namespace_uid = 'cache_settings.{}'.format(cache_key)
216 cache_namespace_uid = 'cache_settings.{}'.format(cache_key)
217 region = rc_cache.get_or_create_region('cache_general', cache_namespace_uid)
217 region = rc_cache.get_or_create_region('cache_general', cache_namespace_uid)
218 return region, cache_key
218 return region, cache_key
219
219
220 def invalidate_settings_cache(self):
220 def invalidate_settings_cache(self):
221 region, cache_key = self.get_cache_region()
221 region, cache_key = self.get_cache_region()
222 log.debug('Invalidation cache region %s for cache_key: %s', region, cache_key)
222 log.debug('Invalidation cache region %s for cache_key: %s', region, cache_key)
223 region.invalidate()
223 region.invalidate()
224
224
225 def get_all_settings(self, cache=False, from_request=True):
225 def get_all_settings(self, cache=False, from_request=True):
226 # defines if we use GLOBAL, or PER_REPO
226 # defines if we use GLOBAL, or PER_REPO
227 repo = self._get_repo(self.repo) if self.repo else None
227 repo = self._get_repo(self.repo) if self.repo else None
228
228
229 # initially try the requests context, this is the fastest
229 # initially try the requests context, this is the fastest
230 # we only fetch global config
230 # we only fetch global config
231 if from_request:
231 if from_request:
232 request = get_current_request()
232 request = get_current_request()
233
233
234 if request and not repo and hasattr(request, 'call_context') and hasattr(request.call_context, 'rc_config'):
234 if request and not repo and hasattr(request, 'call_context') and hasattr(request.call_context, 'rc_config'):
235 rc_config = request.call_context.rc_config
235 rc_config = request.call_context.rc_config
236 if rc_config:
236 if rc_config:
237 return rc_config
237 return rc_config
238
238
239 region, cache_key = self.get_cache_region()
239 region, cache_key = self.get_cache_region()
240
240
241 @region.conditional_cache_on_arguments(condition=cache)
241 @region.conditional_cache_on_arguments(condition=cache)
242 def _get_all_settings(name, key):
242 def _get_all_settings(name, key):
243 q = self._get_settings_query()
243 q = self._get_settings_query()
244 if not q:
244 if not q:
245 raise Exception('Could not get application settings !')
245 raise Exception('Could not get application settings !')
246
246
247 settings = {
247 settings = {
248 'rhodecode_' + res.app_settings_name: res.app_settings_value
248 'rhodecode_' + res.app_settings_name: res.app_settings_value
249 for res in q
249 for res in q
250 }
250 }
251 return settings
251 return settings
252
252
253 start = time.time()
253 start = time.time()
254 result = _get_all_settings('rhodecode_settings', cache_key)
254 result = _get_all_settings('rhodecode_settings', cache_key)
255 compute_time = time.time() - start
255 compute_time = time.time() - start
256 log.debug('cached method:%s took %.4fs', _get_all_settings.func_name, compute_time)
256 log.debug('cached method:%s took %.4fs', _get_all_settings.__name__, compute_time)
257
257
258 statsd = StatsdClient.statsd
258 statsd = StatsdClient.statsd
259 if statsd:
259 if statsd:
260 elapsed_time_ms = round(1000.0 * compute_time) # use ms only
260 elapsed_time_ms = round(1000.0 * compute_time) # use ms only
261 statsd.timing("rhodecode_settings_timing.histogram", elapsed_time_ms,
261 statsd.timing("rhodecode_settings_timing.histogram", elapsed_time_ms,
262 use_decimals=False)
262 use_decimals=False)
263
263
264 log.debug('Fetching app settings for key: %s took: %.4fs: cache: %s', cache_key, compute_time, cache)
264 log.debug('Fetching app settings for key: %s took: %.4fs: cache: %s', cache_key, compute_time, cache)
265
265
266 return result
266 return result
267
267
268 def get_auth_settings(self):
268 def get_auth_settings(self):
269 q = self._get_settings_query()
269 q = self._get_settings_query()
270 q = q.filter(
270 q = q.filter(
271 self.SettingsDbModel.app_settings_name.startswith('auth_'))
271 self.SettingsDbModel.app_settings_name.startswith('auth_'))
272 rows = q.all()
272 rows = q.all()
273 auth_settings = {
273 auth_settings = {
274 row.app_settings_name: row.app_settings_value for row in rows}
274 row.app_settings_name: row.app_settings_value for row in rows}
275 return auth_settings
275 return auth_settings
276
276
277 def get_auth_plugins(self):
277 def get_auth_plugins(self):
278 auth_plugins = self.get_setting_by_name("auth_plugins")
278 auth_plugins = self.get_setting_by_name("auth_plugins")
279 return auth_plugins.app_settings_value
279 return auth_plugins.app_settings_value
280
280
281 def get_default_repo_settings(self, strip_prefix=False):
281 def get_default_repo_settings(self, strip_prefix=False):
282 q = self._get_settings_query()
282 q = self._get_settings_query()
283 q = q.filter(
283 q = q.filter(
284 self.SettingsDbModel.app_settings_name.startswith('default_'))
284 self.SettingsDbModel.app_settings_name.startswith('default_'))
285 rows = q.all()
285 rows = q.all()
286
286
287 result = {}
287 result = {}
288 for row in rows:
288 for row in rows:
289 key = row.app_settings_name
289 key = row.app_settings_name
290 if strip_prefix:
290 if strip_prefix:
291 key = remove_prefix(key, prefix='default_')
291 key = remove_prefix(key, prefix='default_')
292 result.update({key: row.app_settings_value})
292 result.update({key: row.app_settings_value})
293 return result
293 return result
294
294
295 def get_repo(self):
295 def get_repo(self):
296 repo = self._get_repo(self.repo)
296 repo = self._get_repo(self.repo)
297 if not repo:
297 if not repo:
298 raise Exception(
298 raise Exception(
299 'Repository `{}` cannot be found inside the database'.format(
299 'Repository `{}` cannot be found inside the database'.format(
300 self.repo))
300 self.repo))
301 return repo
301 return repo
302
302
303 def _filter_by_repo(self, model, query):
303 def _filter_by_repo(self, model, query):
304 if self.repo:
304 if self.repo:
305 repo = self.get_repo()
305 repo = self.get_repo()
306 query = query.filter(model.repository_id == repo.repo_id)
306 query = query.filter(model.repository_id == repo.repo_id)
307 return query
307 return query
308
308
309 def _get_hooks(self, query):
309 def _get_hooks(self, query):
310 query = query.filter(self.UiDbModel.ui_section == self.HOOKS_SECTION)
310 query = query.filter(self.UiDbModel.ui_section == self.HOOKS_SECTION)
311 query = self._filter_by_repo(RepoRhodeCodeUi, query)
311 query = self._filter_by_repo(RepoRhodeCodeUi, query)
312 return query.all()
312 return query.all()
313
313
314 def _get_settings_query(self):
314 def _get_settings_query(self):
315 q = self.SettingsDbModel.query()
315 q = self.SettingsDbModel.query()
316 return self._filter_by_repo(RepoRhodeCodeSetting, q)
316 return self._filter_by_repo(RepoRhodeCodeSetting, q)
317
317
318 def list_enabled_social_plugins(self, settings):
318 def list_enabled_social_plugins(self, settings):
319 enabled = []
319 enabled = []
320 for plug in SOCIAL_PLUGINS_LIST:
320 for plug in SOCIAL_PLUGINS_LIST:
321 if str2bool(settings.get('rhodecode_auth_{}_enabled'.format(plug)
321 if str2bool(settings.get('rhodecode_auth_{}_enabled'.format(plug)
322 )):
322 )):
323 enabled.append(plug)
323 enabled.append(plug)
324 return enabled
324 return enabled
325
325
326
326
327 def assert_repo_settings(func):
327 def assert_repo_settings(func):
328 @wraps(func)
328 @wraps(func)
329 def _wrapper(self, *args, **kwargs):
329 def _wrapper(self, *args, **kwargs):
330 if not self.repo_settings:
330 if not self.repo_settings:
331 raise Exception('Repository is not specified')
331 raise Exception('Repository is not specified')
332 return func(self, *args, **kwargs)
332 return func(self, *args, **kwargs)
333 return _wrapper
333 return _wrapper
334
334
335
335
336 class IssueTrackerSettingsModel(object):
336 class IssueTrackerSettingsModel(object):
337 INHERIT_SETTINGS = 'inherit_issue_tracker_settings'
337 INHERIT_SETTINGS = 'inherit_issue_tracker_settings'
338 SETTINGS_PREFIX = 'issuetracker_'
338 SETTINGS_PREFIX = 'issuetracker_'
339
339
340 def __init__(self, sa=None, repo=None):
340 def __init__(self, sa=None, repo=None):
341 self.global_settings = SettingsModel(sa=sa)
341 self.global_settings = SettingsModel(sa=sa)
342 self.repo_settings = SettingsModel(sa=sa, repo=repo) if repo else None
342 self.repo_settings = SettingsModel(sa=sa, repo=repo) if repo else None
343
343
344 @property
344 @property
345 def inherit_global_settings(self):
345 def inherit_global_settings(self):
346 if not self.repo_settings:
346 if not self.repo_settings:
347 return True
347 return True
348 setting = self.repo_settings.get_setting_by_name(self.INHERIT_SETTINGS)
348 setting = self.repo_settings.get_setting_by_name(self.INHERIT_SETTINGS)
349 return setting.app_settings_value if setting else True
349 return setting.app_settings_value if setting else True
350
350
351 @inherit_global_settings.setter
351 @inherit_global_settings.setter
352 def inherit_global_settings(self, value):
352 def inherit_global_settings(self, value):
353 if self.repo_settings:
353 if self.repo_settings:
354 settings = self.repo_settings.create_or_update_setting(
354 settings = self.repo_settings.create_or_update_setting(
355 self.INHERIT_SETTINGS, value, type_='bool')
355 self.INHERIT_SETTINGS, value, type_='bool')
356 Session().add(settings)
356 Session().add(settings)
357
357
358 def _get_keyname(self, key, uid, prefix=''):
358 def _get_keyname(self, key, uid, prefix=''):
359 return '{0}{1}{2}_{3}'.format(
359 return '{0}{1}{2}_{3}'.format(
360 prefix, self.SETTINGS_PREFIX, key, uid)
360 prefix, self.SETTINGS_PREFIX, key, uid)
361
361
362 def _make_dict_for_settings(self, qs):
362 def _make_dict_for_settings(self, qs):
363 prefix_match = self._get_keyname('pat', '', 'rhodecode_')
363 prefix_match = self._get_keyname('pat', '', 'rhodecode_')
364
364
365 issuetracker_entries = {}
365 issuetracker_entries = {}
366 # create keys
366 # create keys
367 for k, v in qs.items():
367 for k, v in qs.items():
368 if k.startswith(prefix_match):
368 if k.startswith(prefix_match):
369 uid = k[len(prefix_match):]
369 uid = k[len(prefix_match):]
370 issuetracker_entries[uid] = None
370 issuetracker_entries[uid] = None
371
371
372 def url_cleaner(input_str):
372 def url_cleaner(input_str):
373 input_str = input_str.replace('"', '').replace("'", '')
373 input_str = input_str.replace('"', '').replace("'", '')
374 input_str = bleach.clean(input_str, strip=True)
374 input_str = bleach.clean(input_str, strip=True)
375 return input_str
375 return input_str
376
376
377 # populate
377 # populate
378 for uid in issuetracker_entries:
378 for uid in issuetracker_entries:
379 url_data = qs.get(self._get_keyname('url', uid, 'rhodecode_'))
379 url_data = qs.get(self._get_keyname('url', uid, 'rhodecode_'))
380
380
381 pat = qs.get(self._get_keyname('pat', uid, 'rhodecode_'))
381 pat = qs.get(self._get_keyname('pat', uid, 'rhodecode_'))
382 try:
382 try:
383 pat_compiled = re.compile(r'%s' % pat)
383 pat_compiled = re.compile(r'%s' % pat)
384 except re.error:
384 except re.error:
385 pat_compiled = None
385 pat_compiled = None
386
386
387 issuetracker_entries[uid] = AttributeDict({
387 issuetracker_entries[uid] = AttributeDict({
388 'pat': pat,
388 'pat': pat,
389 'pat_compiled': pat_compiled,
389 'pat_compiled': pat_compiled,
390 'url': url_cleaner(
390 'url': url_cleaner(
391 qs.get(self._get_keyname('url', uid, 'rhodecode_')) or ''),
391 qs.get(self._get_keyname('url', uid, 'rhodecode_')) or ''),
392 'pref': bleach.clean(
392 'pref': bleach.clean(
393 qs.get(self._get_keyname('pref', uid, 'rhodecode_')) or ''),
393 qs.get(self._get_keyname('pref', uid, 'rhodecode_')) or ''),
394 'desc': qs.get(
394 'desc': qs.get(
395 self._get_keyname('desc', uid, 'rhodecode_')),
395 self._get_keyname('desc', uid, 'rhodecode_')),
396 })
396 })
397
397
398 return issuetracker_entries
398 return issuetracker_entries
399
399
400 def get_global_settings(self, cache=False):
400 def get_global_settings(self, cache=False):
401 """
401 """
402 Returns list of global issue tracker settings
402 Returns list of global issue tracker settings
403 """
403 """
404 defaults = self.global_settings.get_all_settings(cache=cache)
404 defaults = self.global_settings.get_all_settings(cache=cache)
405 settings = self._make_dict_for_settings(defaults)
405 settings = self._make_dict_for_settings(defaults)
406 return settings
406 return settings
407
407
408 def get_repo_settings(self, cache=False):
408 def get_repo_settings(self, cache=False):
409 """
409 """
410 Returns list of issue tracker settings per repository
410 Returns list of issue tracker settings per repository
411 """
411 """
412 if not self.repo_settings:
412 if not self.repo_settings:
413 raise Exception('Repository is not specified')
413 raise Exception('Repository is not specified')
414 all_settings = self.repo_settings.get_all_settings(cache=cache)
414 all_settings = self.repo_settings.get_all_settings(cache=cache)
415 settings = self._make_dict_for_settings(all_settings)
415 settings = self._make_dict_for_settings(all_settings)
416 return settings
416 return settings
417
417
418 def get_settings(self, cache=False):
418 def get_settings(self, cache=False):
419 if self.inherit_global_settings:
419 if self.inherit_global_settings:
420 return self.get_global_settings(cache=cache)
420 return self.get_global_settings(cache=cache)
421 else:
421 else:
422 return self.get_repo_settings(cache=cache)
422 return self.get_repo_settings(cache=cache)
423
423
424 def delete_entries(self, uid):
424 def delete_entries(self, uid):
425 if self.repo_settings:
425 if self.repo_settings:
426 all_patterns = self.get_repo_settings()
426 all_patterns = self.get_repo_settings()
427 settings_model = self.repo_settings
427 settings_model = self.repo_settings
428 else:
428 else:
429 all_patterns = self.get_global_settings()
429 all_patterns = self.get_global_settings()
430 settings_model = self.global_settings
430 settings_model = self.global_settings
431 entries = all_patterns.get(uid, [])
431 entries = all_patterns.get(uid, [])
432
432
433 for del_key in entries:
433 for del_key in entries:
434 setting_name = self._get_keyname(del_key, uid)
434 setting_name = self._get_keyname(del_key, uid)
435 entry = settings_model.get_setting_by_name(setting_name)
435 entry = settings_model.get_setting_by_name(setting_name)
436 if entry:
436 if entry:
437 Session().delete(entry)
437 Session().delete(entry)
438
438
439 Session().commit()
439 Session().commit()
440
440
441 def create_or_update_setting(
441 def create_or_update_setting(
442 self, name, val=Optional(''), type_=Optional('unicode')):
442 self, name, val=Optional(''), type_=Optional('unicode')):
443 if self.repo_settings:
443 if self.repo_settings:
444 setting = self.repo_settings.create_or_update_setting(
444 setting = self.repo_settings.create_or_update_setting(
445 name, val, type_)
445 name, val, type_)
446 else:
446 else:
447 setting = self.global_settings.create_or_update_setting(
447 setting = self.global_settings.create_or_update_setting(
448 name, val, type_)
448 name, val, type_)
449 return setting
449 return setting
450
450
451
451
452 class VcsSettingsModel(object):
452 class VcsSettingsModel(object):
453
453
454 INHERIT_SETTINGS = 'inherit_vcs_settings'
454 INHERIT_SETTINGS = 'inherit_vcs_settings'
455 GENERAL_SETTINGS = (
455 GENERAL_SETTINGS = (
456 'use_outdated_comments',
456 'use_outdated_comments',
457 'pr_merge_enabled',
457 'pr_merge_enabled',
458 'hg_use_rebase_for_merging',
458 'hg_use_rebase_for_merging',
459 'hg_close_branch_before_merging',
459 'hg_close_branch_before_merging',
460 'git_use_rebase_for_merging',
460 'git_use_rebase_for_merging',
461 'git_close_branch_before_merging',
461 'git_close_branch_before_merging',
462 'diff_cache',
462 'diff_cache',
463 )
463 )
464
464
465 HOOKS_SETTINGS = (
465 HOOKS_SETTINGS = (
466 ('hooks', 'changegroup.repo_size'),
466 ('hooks', 'changegroup.repo_size'),
467 ('hooks', 'changegroup.push_logger'),
467 ('hooks', 'changegroup.push_logger'),
468 ('hooks', 'outgoing.pull_logger'),
468 ('hooks', 'outgoing.pull_logger'),
469 )
469 )
470 HG_SETTINGS = (
470 HG_SETTINGS = (
471 ('extensions', 'largefiles'),
471 ('extensions', 'largefiles'),
472 ('phases', 'publish'),
472 ('phases', 'publish'),
473 ('extensions', 'evolve'),
473 ('extensions', 'evolve'),
474 ('extensions', 'topic'),
474 ('extensions', 'topic'),
475 ('experimental', 'evolution'),
475 ('experimental', 'evolution'),
476 ('experimental', 'evolution.exchange'),
476 ('experimental', 'evolution.exchange'),
477 )
477 )
478 GIT_SETTINGS = (
478 GIT_SETTINGS = (
479 ('vcs_git_lfs', 'enabled'),
479 ('vcs_git_lfs', 'enabled'),
480 )
480 )
481 GLOBAL_HG_SETTINGS = (
481 GLOBAL_HG_SETTINGS = (
482 ('extensions', 'largefiles'),
482 ('extensions', 'largefiles'),
483 ('largefiles', 'usercache'),
483 ('largefiles', 'usercache'),
484 ('phases', 'publish'),
484 ('phases', 'publish'),
485 ('extensions', 'hgsubversion'),
485 ('extensions', 'hgsubversion'),
486 ('extensions', 'evolve'),
486 ('extensions', 'evolve'),
487 ('extensions', 'topic'),
487 ('extensions', 'topic'),
488 ('experimental', 'evolution'),
488 ('experimental', 'evolution'),
489 ('experimental', 'evolution.exchange'),
489 ('experimental', 'evolution.exchange'),
490 )
490 )
491
491
492 GLOBAL_GIT_SETTINGS = (
492 GLOBAL_GIT_SETTINGS = (
493 ('vcs_git_lfs', 'enabled'),
493 ('vcs_git_lfs', 'enabled'),
494 ('vcs_git_lfs', 'store_location')
494 ('vcs_git_lfs', 'store_location')
495 )
495 )
496
496
497 GLOBAL_SVN_SETTINGS = (
497 GLOBAL_SVN_SETTINGS = (
498 ('vcs_svn_proxy', 'http_requests_enabled'),
498 ('vcs_svn_proxy', 'http_requests_enabled'),
499 ('vcs_svn_proxy', 'http_server_url')
499 ('vcs_svn_proxy', 'http_server_url')
500 )
500 )
501
501
502 SVN_BRANCH_SECTION = 'vcs_svn_branch'
502 SVN_BRANCH_SECTION = 'vcs_svn_branch'
503 SVN_TAG_SECTION = 'vcs_svn_tag'
503 SVN_TAG_SECTION = 'vcs_svn_tag'
504 SSL_SETTING = ('web', 'push_ssl')
504 SSL_SETTING = ('web', 'push_ssl')
505 PATH_SETTING = ('paths', '/')
505 PATH_SETTING = ('paths', '/')
506
506
507 def __init__(self, sa=None, repo=None):
507 def __init__(self, sa=None, repo=None):
508 self.global_settings = SettingsModel(sa=sa)
508 self.global_settings = SettingsModel(sa=sa)
509 self.repo_settings = SettingsModel(sa=sa, repo=repo) if repo else None
509 self.repo_settings = SettingsModel(sa=sa, repo=repo) if repo else None
510 self._ui_settings = (
510 self._ui_settings = (
511 self.HG_SETTINGS + self.GIT_SETTINGS + self.HOOKS_SETTINGS)
511 self.HG_SETTINGS + self.GIT_SETTINGS + self.HOOKS_SETTINGS)
512 self._svn_sections = (self.SVN_BRANCH_SECTION, self.SVN_TAG_SECTION)
512 self._svn_sections = (self.SVN_BRANCH_SECTION, self.SVN_TAG_SECTION)
513
513
514 @property
514 @property
515 @assert_repo_settings
515 @assert_repo_settings
516 def inherit_global_settings(self):
516 def inherit_global_settings(self):
517 setting = self.repo_settings.get_setting_by_name(self.INHERIT_SETTINGS)
517 setting = self.repo_settings.get_setting_by_name(self.INHERIT_SETTINGS)
518 return setting.app_settings_value if setting else True
518 return setting.app_settings_value if setting else True
519
519
520 @inherit_global_settings.setter
520 @inherit_global_settings.setter
521 @assert_repo_settings
521 @assert_repo_settings
522 def inherit_global_settings(self, value):
522 def inherit_global_settings(self, value):
523 self.repo_settings.create_or_update_setting(
523 self.repo_settings.create_or_update_setting(
524 self.INHERIT_SETTINGS, value, type_='bool')
524 self.INHERIT_SETTINGS, value, type_='bool')
525
525
526 def get_global_svn_branch_patterns(self):
526 def get_global_svn_branch_patterns(self):
527 return self.global_settings.get_ui_by_section(self.SVN_BRANCH_SECTION)
527 return self.global_settings.get_ui_by_section(self.SVN_BRANCH_SECTION)
528
528
529 @assert_repo_settings
529 @assert_repo_settings
530 def get_repo_svn_branch_patterns(self):
530 def get_repo_svn_branch_patterns(self):
531 return self.repo_settings.get_ui_by_section(self.SVN_BRANCH_SECTION)
531 return self.repo_settings.get_ui_by_section(self.SVN_BRANCH_SECTION)
532
532
533 def get_global_svn_tag_patterns(self):
533 def get_global_svn_tag_patterns(self):
534 return self.global_settings.get_ui_by_section(self.SVN_TAG_SECTION)
534 return self.global_settings.get_ui_by_section(self.SVN_TAG_SECTION)
535
535
536 @assert_repo_settings
536 @assert_repo_settings
537 def get_repo_svn_tag_patterns(self):
537 def get_repo_svn_tag_patterns(self):
538 return self.repo_settings.get_ui_by_section(self.SVN_TAG_SECTION)
538 return self.repo_settings.get_ui_by_section(self.SVN_TAG_SECTION)
539
539
540 def get_global_settings(self):
540 def get_global_settings(self):
541 return self._collect_all_settings(global_=True)
541 return self._collect_all_settings(global_=True)
542
542
543 @assert_repo_settings
543 @assert_repo_settings
544 def get_repo_settings(self):
544 def get_repo_settings(self):
545 return self._collect_all_settings(global_=False)
545 return self._collect_all_settings(global_=False)
546
546
547 @assert_repo_settings
547 @assert_repo_settings
548 def get_repo_settings_inherited(self):
548 def get_repo_settings_inherited(self):
549 global_settings = self.get_global_settings()
549 global_settings = self.get_global_settings()
550 global_settings.update(self.get_repo_settings())
550 global_settings.update(self.get_repo_settings())
551 return global_settings
551 return global_settings
552
552
553 @assert_repo_settings
553 @assert_repo_settings
554 def create_or_update_repo_settings(
554 def create_or_update_repo_settings(
555 self, data, inherit_global_settings=False):
555 self, data, inherit_global_settings=False):
556 from rhodecode.model.scm import ScmModel
556 from rhodecode.model.scm import ScmModel
557
557
558 self.inherit_global_settings = inherit_global_settings
558 self.inherit_global_settings = inherit_global_settings
559
559
560 repo = self.repo_settings.get_repo()
560 repo = self.repo_settings.get_repo()
561 if not inherit_global_settings:
561 if not inherit_global_settings:
562 if repo.repo_type == 'svn':
562 if repo.repo_type == 'svn':
563 self.create_repo_svn_settings(data)
563 self.create_repo_svn_settings(data)
564 else:
564 else:
565 self.create_or_update_repo_hook_settings(data)
565 self.create_or_update_repo_hook_settings(data)
566 self.create_or_update_repo_pr_settings(data)
566 self.create_or_update_repo_pr_settings(data)
567
567
568 if repo.repo_type == 'hg':
568 if repo.repo_type == 'hg':
569 self.create_or_update_repo_hg_settings(data)
569 self.create_or_update_repo_hg_settings(data)
570
570
571 if repo.repo_type == 'git':
571 if repo.repo_type == 'git':
572 self.create_or_update_repo_git_settings(data)
572 self.create_or_update_repo_git_settings(data)
573
573
574 ScmModel().mark_for_invalidation(repo.repo_name, delete=True)
574 ScmModel().mark_for_invalidation(repo.repo_name, delete=True)
575
575
576 @assert_repo_settings
576 @assert_repo_settings
577 def create_or_update_repo_hook_settings(self, data):
577 def create_or_update_repo_hook_settings(self, data):
578 for section, key in self.HOOKS_SETTINGS:
578 for section, key in self.HOOKS_SETTINGS:
579 data_key = self._get_form_ui_key(section, key)
579 data_key = self._get_form_ui_key(section, key)
580 if data_key not in data:
580 if data_key not in data:
581 raise ValueError(
581 raise ValueError(
582 'The given data does not contain {} key'.format(data_key))
582 'The given data does not contain {} key'.format(data_key))
583
583
584 active = data.get(data_key)
584 active = data.get(data_key)
585 repo_setting = self.repo_settings.get_ui_by_section_and_key(
585 repo_setting = self.repo_settings.get_ui_by_section_and_key(
586 section, key)
586 section, key)
587 if not repo_setting:
587 if not repo_setting:
588 global_setting = self.global_settings.\
588 global_setting = self.global_settings.\
589 get_ui_by_section_and_key(section, key)
589 get_ui_by_section_and_key(section, key)
590 self.repo_settings.create_ui_section_value(
590 self.repo_settings.create_ui_section_value(
591 section, global_setting.ui_value, key=key, active=active)
591 section, global_setting.ui_value, key=key, active=active)
592 else:
592 else:
593 repo_setting.ui_active = active
593 repo_setting.ui_active = active
594 Session().add(repo_setting)
594 Session().add(repo_setting)
595
595
596 def update_global_hook_settings(self, data):
596 def update_global_hook_settings(self, data):
597 for section, key in self.HOOKS_SETTINGS:
597 for section, key in self.HOOKS_SETTINGS:
598 data_key = self._get_form_ui_key(section, key)
598 data_key = self._get_form_ui_key(section, key)
599 if data_key not in data:
599 if data_key not in data:
600 raise ValueError(
600 raise ValueError(
601 'The given data does not contain {} key'.format(data_key))
601 'The given data does not contain {} key'.format(data_key))
602 active = data.get(data_key)
602 active = data.get(data_key)
603 repo_setting = self.global_settings.get_ui_by_section_and_key(
603 repo_setting = self.global_settings.get_ui_by_section_and_key(
604 section, key)
604 section, key)
605 repo_setting.ui_active = active
605 repo_setting.ui_active = active
606 Session().add(repo_setting)
606 Session().add(repo_setting)
607
607
608 @assert_repo_settings
608 @assert_repo_settings
609 def create_or_update_repo_pr_settings(self, data):
609 def create_or_update_repo_pr_settings(self, data):
610 return self._create_or_update_general_settings(
610 return self._create_or_update_general_settings(
611 self.repo_settings, data)
611 self.repo_settings, data)
612
612
613 def create_or_update_global_pr_settings(self, data):
613 def create_or_update_global_pr_settings(self, data):
614 return self._create_or_update_general_settings(
614 return self._create_or_update_general_settings(
615 self.global_settings, data)
615 self.global_settings, data)
616
616
617 @assert_repo_settings
617 @assert_repo_settings
618 def create_repo_svn_settings(self, data):
618 def create_repo_svn_settings(self, data):
619 return self._create_svn_settings(self.repo_settings, data)
619 return self._create_svn_settings(self.repo_settings, data)
620
620
621 def _set_evolution(self, settings, is_enabled):
621 def _set_evolution(self, settings, is_enabled):
622 if is_enabled:
622 if is_enabled:
623 # if evolve is active set evolution=all
623 # if evolve is active set evolution=all
624
624
625 self._create_or_update_ui(
625 self._create_or_update_ui(
626 settings, *('experimental', 'evolution'), value='all',
626 settings, *('experimental', 'evolution'), value='all',
627 active=True)
627 active=True)
628 self._create_or_update_ui(
628 self._create_or_update_ui(
629 settings, *('experimental', 'evolution.exchange'), value='yes',
629 settings, *('experimental', 'evolution.exchange'), value='yes',
630 active=True)
630 active=True)
631 # if evolve is active set topics server support
631 # if evolve is active set topics server support
632 self._create_or_update_ui(
632 self._create_or_update_ui(
633 settings, *('extensions', 'topic'), value='',
633 settings, *('extensions', 'topic'), value='',
634 active=True)
634 active=True)
635
635
636 else:
636 else:
637 self._create_or_update_ui(
637 self._create_or_update_ui(
638 settings, *('experimental', 'evolution'), value='',
638 settings, *('experimental', 'evolution'), value='',
639 active=False)
639 active=False)
640 self._create_or_update_ui(
640 self._create_or_update_ui(
641 settings, *('experimental', 'evolution.exchange'), value='no',
641 settings, *('experimental', 'evolution.exchange'), value='no',
642 active=False)
642 active=False)
643 self._create_or_update_ui(
643 self._create_or_update_ui(
644 settings, *('extensions', 'topic'), value='',
644 settings, *('extensions', 'topic'), value='',
645 active=False)
645 active=False)
646
646
647 @assert_repo_settings
647 @assert_repo_settings
648 def create_or_update_repo_hg_settings(self, data):
648 def create_or_update_repo_hg_settings(self, data):
649 largefiles, phases, evolve = \
649 largefiles, phases, evolve = \
650 self.HG_SETTINGS[:3]
650 self.HG_SETTINGS[:3]
651 largefiles_key, phases_key, evolve_key = \
651 largefiles_key, phases_key, evolve_key = \
652 self._get_settings_keys(self.HG_SETTINGS[:3], data)
652 self._get_settings_keys(self.HG_SETTINGS[:3], data)
653
653
654 self._create_or_update_ui(
654 self._create_or_update_ui(
655 self.repo_settings, *largefiles, value='',
655 self.repo_settings, *largefiles, value='',
656 active=data[largefiles_key])
656 active=data[largefiles_key])
657 self._create_or_update_ui(
657 self._create_or_update_ui(
658 self.repo_settings, *evolve, value='',
658 self.repo_settings, *evolve, value='',
659 active=data[evolve_key])
659 active=data[evolve_key])
660 self._set_evolution(self.repo_settings, is_enabled=data[evolve_key])
660 self._set_evolution(self.repo_settings, is_enabled=data[evolve_key])
661
661
662 self._create_or_update_ui(
662 self._create_or_update_ui(
663 self.repo_settings, *phases, value=safe_str(data[phases_key]))
663 self.repo_settings, *phases, value=safe_str(data[phases_key]))
664
664
665 def create_or_update_global_hg_settings(self, data):
665 def create_or_update_global_hg_settings(self, data):
666 largefiles, largefiles_store, phases, hgsubversion, evolve \
666 largefiles, largefiles_store, phases, hgsubversion, evolve \
667 = self.GLOBAL_HG_SETTINGS[:5]
667 = self.GLOBAL_HG_SETTINGS[:5]
668 largefiles_key, largefiles_store_key, phases_key, subversion_key, evolve_key \
668 largefiles_key, largefiles_store_key, phases_key, subversion_key, evolve_key \
669 = self._get_settings_keys(self.GLOBAL_HG_SETTINGS[:5], data)
669 = self._get_settings_keys(self.GLOBAL_HG_SETTINGS[:5], data)
670
670
671 self._create_or_update_ui(
671 self._create_or_update_ui(
672 self.global_settings, *largefiles, value='',
672 self.global_settings, *largefiles, value='',
673 active=data[largefiles_key])
673 active=data[largefiles_key])
674 self._create_or_update_ui(
674 self._create_or_update_ui(
675 self.global_settings, *largefiles_store, value=data[largefiles_store_key])
675 self.global_settings, *largefiles_store, value=data[largefiles_store_key])
676 self._create_or_update_ui(
676 self._create_or_update_ui(
677 self.global_settings, *phases, value=safe_str(data[phases_key]))
677 self.global_settings, *phases, value=safe_str(data[phases_key]))
678 self._create_or_update_ui(
678 self._create_or_update_ui(
679 self.global_settings, *hgsubversion, active=data[subversion_key])
679 self.global_settings, *hgsubversion, active=data[subversion_key])
680 self._create_or_update_ui(
680 self._create_or_update_ui(
681 self.global_settings, *evolve, value='',
681 self.global_settings, *evolve, value='',
682 active=data[evolve_key])
682 active=data[evolve_key])
683 self._set_evolution(self.global_settings, is_enabled=data[evolve_key])
683 self._set_evolution(self.global_settings, is_enabled=data[evolve_key])
684
684
685 def create_or_update_repo_git_settings(self, data):
685 def create_or_update_repo_git_settings(self, data):
686 # NOTE(marcink): # comma makes unpack work properly
686 # NOTE(marcink): # comma makes unpack work properly
687 lfs_enabled, \
687 lfs_enabled, \
688 = self.GIT_SETTINGS
688 = self.GIT_SETTINGS
689
689
690 lfs_enabled_key, \
690 lfs_enabled_key, \
691 = self._get_settings_keys(self.GIT_SETTINGS, data)
691 = self._get_settings_keys(self.GIT_SETTINGS, data)
692
692
693 self._create_or_update_ui(
693 self._create_or_update_ui(
694 self.repo_settings, *lfs_enabled, value=data[lfs_enabled_key],
694 self.repo_settings, *lfs_enabled, value=data[lfs_enabled_key],
695 active=data[lfs_enabled_key])
695 active=data[lfs_enabled_key])
696
696
697 def create_or_update_global_git_settings(self, data):
697 def create_or_update_global_git_settings(self, data):
698 lfs_enabled, lfs_store_location \
698 lfs_enabled, lfs_store_location \
699 = self.GLOBAL_GIT_SETTINGS
699 = self.GLOBAL_GIT_SETTINGS
700 lfs_enabled_key, lfs_store_location_key \
700 lfs_enabled_key, lfs_store_location_key \
701 = self._get_settings_keys(self.GLOBAL_GIT_SETTINGS, data)
701 = self._get_settings_keys(self.GLOBAL_GIT_SETTINGS, data)
702
702
703 self._create_or_update_ui(
703 self._create_or_update_ui(
704 self.global_settings, *lfs_enabled, value=data[lfs_enabled_key],
704 self.global_settings, *lfs_enabled, value=data[lfs_enabled_key],
705 active=data[lfs_enabled_key])
705 active=data[lfs_enabled_key])
706 self._create_or_update_ui(
706 self._create_or_update_ui(
707 self.global_settings, *lfs_store_location,
707 self.global_settings, *lfs_store_location,
708 value=data[lfs_store_location_key])
708 value=data[lfs_store_location_key])
709
709
710 def create_or_update_global_svn_settings(self, data):
710 def create_or_update_global_svn_settings(self, data):
711 # branch/tags patterns
711 # branch/tags patterns
712 self._create_svn_settings(self.global_settings, data)
712 self._create_svn_settings(self.global_settings, data)
713
713
714 http_requests_enabled, http_server_url = self.GLOBAL_SVN_SETTINGS
714 http_requests_enabled, http_server_url = self.GLOBAL_SVN_SETTINGS
715 http_requests_enabled_key, http_server_url_key = self._get_settings_keys(
715 http_requests_enabled_key, http_server_url_key = self._get_settings_keys(
716 self.GLOBAL_SVN_SETTINGS, data)
716 self.GLOBAL_SVN_SETTINGS, data)
717
717
718 self._create_or_update_ui(
718 self._create_or_update_ui(
719 self.global_settings, *http_requests_enabled,
719 self.global_settings, *http_requests_enabled,
720 value=safe_str(data[http_requests_enabled_key]))
720 value=safe_str(data[http_requests_enabled_key]))
721 self._create_or_update_ui(
721 self._create_or_update_ui(
722 self.global_settings, *http_server_url,
722 self.global_settings, *http_server_url,
723 value=data[http_server_url_key])
723 value=data[http_server_url_key])
724
724
725 def update_global_ssl_setting(self, value):
725 def update_global_ssl_setting(self, value):
726 self._create_or_update_ui(
726 self._create_or_update_ui(
727 self.global_settings, *self.SSL_SETTING, value=value)
727 self.global_settings, *self.SSL_SETTING, value=value)
728
728
729 def update_global_path_setting(self, value):
729 def update_global_path_setting(self, value):
730 self._create_or_update_ui(
730 self._create_or_update_ui(
731 self.global_settings, *self.PATH_SETTING, value=value)
731 self.global_settings, *self.PATH_SETTING, value=value)
732
732
733 @assert_repo_settings
733 @assert_repo_settings
734 def delete_repo_svn_pattern(self, id_):
734 def delete_repo_svn_pattern(self, id_):
735 ui = self.repo_settings.UiDbModel.get(id_)
735 ui = self.repo_settings.UiDbModel.get(id_)
736 if ui and ui.repository.repo_name == self.repo_settings.repo:
736 if ui and ui.repository.repo_name == self.repo_settings.repo:
737 # only delete if it's the same repo as initialized settings
737 # only delete if it's the same repo as initialized settings
738 self.repo_settings.delete_ui(id_)
738 self.repo_settings.delete_ui(id_)
739 else:
739 else:
740 # raise error as if we wouldn't find this option
740 # raise error as if we wouldn't find this option
741 self.repo_settings.delete_ui(-1)
741 self.repo_settings.delete_ui(-1)
742
742
743 def delete_global_svn_pattern(self, id_):
743 def delete_global_svn_pattern(self, id_):
744 self.global_settings.delete_ui(id_)
744 self.global_settings.delete_ui(id_)
745
745
746 @assert_repo_settings
746 @assert_repo_settings
747 def get_repo_ui_settings(self, section=None, key=None):
747 def get_repo_ui_settings(self, section=None, key=None):
748 global_uis = self.global_settings.get_ui(section, key)
748 global_uis = self.global_settings.get_ui(section, key)
749 repo_uis = self.repo_settings.get_ui(section, key)
749 repo_uis = self.repo_settings.get_ui(section, key)
750
750
751 filtered_repo_uis = self._filter_ui_settings(repo_uis)
751 filtered_repo_uis = self._filter_ui_settings(repo_uis)
752 filtered_repo_uis_keys = [
752 filtered_repo_uis_keys = [
753 (s.section, s.key) for s in filtered_repo_uis]
753 (s.section, s.key) for s in filtered_repo_uis]
754
754
755 def _is_global_ui_filtered(ui):
755 def _is_global_ui_filtered(ui):
756 return (
756 return (
757 (ui.section, ui.key) in filtered_repo_uis_keys
757 (ui.section, ui.key) in filtered_repo_uis_keys
758 or ui.section in self._svn_sections)
758 or ui.section in self._svn_sections)
759
759
760 filtered_global_uis = [
760 filtered_global_uis = [
761 ui for ui in global_uis if not _is_global_ui_filtered(ui)]
761 ui for ui in global_uis if not _is_global_ui_filtered(ui)]
762
762
763 return filtered_global_uis + filtered_repo_uis
763 return filtered_global_uis + filtered_repo_uis
764
764
765 def get_global_ui_settings(self, section=None, key=None):
765 def get_global_ui_settings(self, section=None, key=None):
766 return self.global_settings.get_ui(section, key)
766 return self.global_settings.get_ui(section, key)
767
767
768 def get_ui_settings_as_config_obj(self, section=None, key=None):
768 def get_ui_settings_as_config_obj(self, section=None, key=None):
769 config = base.Config()
769 config = base.Config()
770
770
771 ui_settings = self.get_ui_settings(section=section, key=key)
771 ui_settings = self.get_ui_settings(section=section, key=key)
772
772
773 for entry in ui_settings:
773 for entry in ui_settings:
774 config.set(entry.section, entry.key, entry.value)
774 config.set(entry.section, entry.key, entry.value)
775
775
776 return config
776 return config
777
777
778 def get_ui_settings(self, section=None, key=None):
778 def get_ui_settings(self, section=None, key=None):
779 if not self.repo_settings or self.inherit_global_settings:
779 if not self.repo_settings or self.inherit_global_settings:
780 return self.get_global_ui_settings(section, key)
780 return self.get_global_ui_settings(section, key)
781 else:
781 else:
782 return self.get_repo_ui_settings(section, key)
782 return self.get_repo_ui_settings(section, key)
783
783
784 def get_svn_patterns(self, section=None):
784 def get_svn_patterns(self, section=None):
785 if not self.repo_settings:
785 if not self.repo_settings:
786 return self.get_global_ui_settings(section)
786 return self.get_global_ui_settings(section)
787 else:
787 else:
788 return self.get_repo_ui_settings(section)
788 return self.get_repo_ui_settings(section)
789
789
790 @assert_repo_settings
790 @assert_repo_settings
791 def get_repo_general_settings(self):
791 def get_repo_general_settings(self):
792 global_settings = self.global_settings.get_all_settings()
792 global_settings = self.global_settings.get_all_settings()
793 repo_settings = self.repo_settings.get_all_settings()
793 repo_settings = self.repo_settings.get_all_settings()
794 filtered_repo_settings = self._filter_general_settings(repo_settings)
794 filtered_repo_settings = self._filter_general_settings(repo_settings)
795 global_settings.update(filtered_repo_settings)
795 global_settings.update(filtered_repo_settings)
796 return global_settings
796 return global_settings
797
797
798 def get_global_general_settings(self):
798 def get_global_general_settings(self):
799 return self.global_settings.get_all_settings()
799 return self.global_settings.get_all_settings()
800
800
801 def get_general_settings(self):
801 def get_general_settings(self):
802 if not self.repo_settings or self.inherit_global_settings:
802 if not self.repo_settings or self.inherit_global_settings:
803 return self.get_global_general_settings()
803 return self.get_global_general_settings()
804 else:
804 else:
805 return self.get_repo_general_settings()
805 return self.get_repo_general_settings()
806
806
807 def get_repos_location(self):
807 def get_repos_location(self):
808 return self.global_settings.get_ui_by_key('/').ui_value
808 return self.global_settings.get_ui_by_key('/').ui_value
809
809
810 def _filter_ui_settings(self, settings):
810 def _filter_ui_settings(self, settings):
811 filtered_settings = [
811 filtered_settings = [
812 s for s in settings if self._should_keep_setting(s)]
812 s for s in settings if self._should_keep_setting(s)]
813 return filtered_settings
813 return filtered_settings
814
814
815 def _should_keep_setting(self, setting):
815 def _should_keep_setting(self, setting):
816 keep = (
816 keep = (
817 (setting.section, setting.key) in self._ui_settings or
817 (setting.section, setting.key) in self._ui_settings or
818 setting.section in self._svn_sections)
818 setting.section in self._svn_sections)
819 return keep
819 return keep
820
820
821 def _filter_general_settings(self, settings):
821 def _filter_general_settings(self, settings):
822 keys = ['rhodecode_{}'.format(key) for key in self.GENERAL_SETTINGS]
822 keys = ['rhodecode_{}'.format(key) for key in self.GENERAL_SETTINGS]
823 return {
823 return {
824 k: settings[k]
824 k: settings[k]
825 for k in settings if k in keys}
825 for k in settings if k in keys}
826
826
827 def _collect_all_settings(self, global_=False):
827 def _collect_all_settings(self, global_=False):
828 settings = self.global_settings if global_ else self.repo_settings
828 settings = self.global_settings if global_ else self.repo_settings
829 result = {}
829 result = {}
830
830
831 for section, key in self._ui_settings:
831 for section, key in self._ui_settings:
832 ui = settings.get_ui_by_section_and_key(section, key)
832 ui = settings.get_ui_by_section_and_key(section, key)
833 result_key = self._get_form_ui_key(section, key)
833 result_key = self._get_form_ui_key(section, key)
834
834
835 if ui:
835 if ui:
836 if section in ('hooks', 'extensions'):
836 if section in ('hooks', 'extensions'):
837 result[result_key] = ui.ui_active
837 result[result_key] = ui.ui_active
838 elif result_key in ['vcs_git_lfs_enabled']:
838 elif result_key in ['vcs_git_lfs_enabled']:
839 result[result_key] = ui.ui_active
839 result[result_key] = ui.ui_active
840 else:
840 else:
841 result[result_key] = ui.ui_value
841 result[result_key] = ui.ui_value
842
842
843 for name in self.GENERAL_SETTINGS:
843 for name in self.GENERAL_SETTINGS:
844 setting = settings.get_setting_by_name(name)
844 setting = settings.get_setting_by_name(name)
845 if setting:
845 if setting:
846 result_key = 'rhodecode_{}'.format(name)
846 result_key = 'rhodecode_{}'.format(name)
847 result[result_key] = setting.app_settings_value
847 result[result_key] = setting.app_settings_value
848
848
849 return result
849 return result
850
850
851 def _get_form_ui_key(self, section, key):
851 def _get_form_ui_key(self, section, key):
852 return '{section}_{key}'.format(
852 return '{section}_{key}'.format(
853 section=section, key=key.replace('.', '_'))
853 section=section, key=key.replace('.', '_'))
854
854
855 def _create_or_update_ui(
855 def _create_or_update_ui(
856 self, settings, section, key, value=None, active=None):
856 self, settings, section, key, value=None, active=None):
857 ui = settings.get_ui_by_section_and_key(section, key)
857 ui = settings.get_ui_by_section_and_key(section, key)
858 if not ui:
858 if not ui:
859 active = True if active is None else active
859 active = True if active is None else active
860 settings.create_ui_section_value(
860 settings.create_ui_section_value(
861 section, value, key=key, active=active)
861 section, value, key=key, active=active)
862 else:
862 else:
863 if active is not None:
863 if active is not None:
864 ui.ui_active = active
864 ui.ui_active = active
865 if value is not None:
865 if value is not None:
866 ui.ui_value = value
866 ui.ui_value = value
867 Session().add(ui)
867 Session().add(ui)
868
868
869 def _create_svn_settings(self, settings, data):
869 def _create_svn_settings(self, settings, data):
870 svn_settings = {
870 svn_settings = {
871 'new_svn_branch': self.SVN_BRANCH_SECTION,
871 'new_svn_branch': self.SVN_BRANCH_SECTION,
872 'new_svn_tag': self.SVN_TAG_SECTION
872 'new_svn_tag': self.SVN_TAG_SECTION
873 }
873 }
874 for key in svn_settings:
874 for key in svn_settings:
875 if data.get(key):
875 if data.get(key):
876 settings.create_ui_section_value(svn_settings[key], data[key])
876 settings.create_ui_section_value(svn_settings[key], data[key])
877
877
878 def _create_or_update_general_settings(self, settings, data):
878 def _create_or_update_general_settings(self, settings, data):
879 for name in self.GENERAL_SETTINGS:
879 for name in self.GENERAL_SETTINGS:
880 data_key = 'rhodecode_{}'.format(name)
880 data_key = 'rhodecode_{}'.format(name)
881 if data_key not in data:
881 if data_key not in data:
882 raise ValueError(
882 raise ValueError(
883 'The given data does not contain {} key'.format(data_key))
883 'The given data does not contain {} key'.format(data_key))
884 setting = settings.create_or_update_setting(
884 setting = settings.create_or_update_setting(
885 name, data[data_key], 'bool')
885 name, data[data_key], 'bool')
886 Session().add(setting)
886 Session().add(setting)
887
887
888 def _get_settings_keys(self, settings, data):
888 def _get_settings_keys(self, settings, data):
889 data_keys = [self._get_form_ui_key(*s) for s in settings]
889 data_keys = [self._get_form_ui_key(*s) for s in settings]
890 for data_key in data_keys:
890 for data_key in data_keys:
891 if data_key not in data:
891 if data_key not in data:
892 raise ValueError(
892 raise ValueError(
893 'The given data does not contain {} key'.format(data_key))
893 'The given data does not contain {} key'.format(data_key))
894 return data_keys
894 return data_keys
895
895
896 def create_largeobjects_dirs_if_needed(self, repo_store_path):
896 def create_largeobjects_dirs_if_needed(self, repo_store_path):
897 """
897 """
898 This is subscribed to the `pyramid.events.ApplicationCreated` event. It
898 This is subscribed to the `pyramid.events.ApplicationCreated` event. It
899 does a repository scan if enabled in the settings.
899 does a repository scan if enabled in the settings.
900 """
900 """
901
901
902 from rhodecode.lib.vcs.backends.hg import largefiles_store
902 from rhodecode.lib.vcs.backends.hg import largefiles_store
903 from rhodecode.lib.vcs.backends.git import lfs_store
903 from rhodecode.lib.vcs.backends.git import lfs_store
904
904
905 paths = [
905 paths = [
906 largefiles_store(repo_store_path),
906 largefiles_store(repo_store_path),
907 lfs_store(repo_store_path)]
907 lfs_store(repo_store_path)]
908
908
909 for path in paths:
909 for path in paths:
910 if os.path.isdir(path):
910 if os.path.isdir(path):
911 continue
911 continue
912 if os.path.isfile(path):
912 if os.path.isfile(path):
913 continue
913 continue
914 # not a file nor dir, we try to create it
914 # not a file nor dir, we try to create it
915 try:
915 try:
916 os.makedirs(path)
916 os.makedirs(path)
917 except Exception:
917 except Exception:
918 log.warning('Failed to create largefiles dir:%s', path)
918 log.warning('Failed to create largefiles dir:%s', path)
@@ -1,255 +1,255 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2020 RhodeCode GmbH
3 # Copyright (C) 2010-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 import os
21 import os
22 import time
22 import time
23 import logging
23 import logging
24 import datetime
24 import datetime
25 import hashlib
25 import hashlib
26 import tempfile
26 import tempfile
27 from os.path import join as jn
27 from os.path import join as jn
28
28
29 from tempfile import _RandomNameSequence
29 from tempfile import _RandomNameSequence
30
30
31 import pytest
31 import pytest
32
32
33 from rhodecode.model.db import User
33 from rhodecode.model.db import User
34 from rhodecode.lib import auth
34 from rhodecode.lib import auth
35 from rhodecode.lib import helpers as h
35 from rhodecode.lib import helpers as h
36 from rhodecode.lib.helpers import flash
36 from rhodecode.lib.helpers import flash
37 from rhodecode.lib.utils2 import safe_str
37 from rhodecode.lib.utils2 import safe_str
38
38
39
39
40 log = logging.getLogger(__name__)
40 log = logging.getLogger(__name__)
41
41
42 __all__ = [
42 __all__ = [
43 'get_new_dir', 'TestController', 'route_path_generator',
43 'get_new_dir', 'TestController', 'route_path_generator',
44 'clear_cache_regions',
44 'clear_cache_regions',
45 'assert_session_flash', 'login_user', 'no_newline_id_generator',
45 'assert_session_flash', 'login_user', 'no_newline_id_generator',
46 'TESTS_TMP_PATH', 'HG_REPO', 'GIT_REPO', 'SVN_REPO',
46 'TESTS_TMP_PATH', 'HG_REPO', 'GIT_REPO', 'SVN_REPO',
47 'NEW_HG_REPO', 'NEW_GIT_REPO',
47 'NEW_HG_REPO', 'NEW_GIT_REPO',
48 'HG_FORK', 'GIT_FORK', 'TEST_USER_ADMIN_LOGIN', 'TEST_USER_ADMIN_PASS',
48 'HG_FORK', 'GIT_FORK', 'TEST_USER_ADMIN_LOGIN', 'TEST_USER_ADMIN_PASS',
49 'TEST_USER_REGULAR_LOGIN', 'TEST_USER_REGULAR_PASS',
49 'TEST_USER_REGULAR_LOGIN', 'TEST_USER_REGULAR_PASS',
50 'TEST_USER_REGULAR_EMAIL', 'TEST_USER_REGULAR2_LOGIN',
50 'TEST_USER_REGULAR_EMAIL', 'TEST_USER_REGULAR2_LOGIN',
51 'TEST_USER_REGULAR2_PASS', 'TEST_USER_REGULAR2_EMAIL', 'TEST_HG_REPO',
51 'TEST_USER_REGULAR2_PASS', 'TEST_USER_REGULAR2_EMAIL', 'TEST_HG_REPO',
52 'TEST_HG_REPO_CLONE', 'TEST_HG_REPO_PULL', 'TEST_GIT_REPO',
52 'TEST_HG_REPO_CLONE', 'TEST_HG_REPO_PULL', 'TEST_GIT_REPO',
53 'TEST_GIT_REPO_CLONE', 'TEST_GIT_REPO_PULL', 'SCM_TESTS',
53 'TEST_GIT_REPO_CLONE', 'TEST_GIT_REPO_PULL', 'SCM_TESTS',
54 ]
54 ]
55
55
56
56
57 # SOME GLOBALS FOR TESTS
57 # SOME GLOBALS FOR TESTS
58 TEST_DIR = tempfile.gettempdir()
58 TEST_DIR = tempfile.gettempdir()
59
59
60 TESTS_TMP_PATH = jn(TEST_DIR, 'rc_test_%s' % _RandomNameSequence().next())
60 TESTS_TMP_PATH = jn(TEST_DIR, 'rc_test_{}'.format(next(_RandomNameSequence())))
61 TEST_USER_ADMIN_LOGIN = 'test_admin'
61 TEST_USER_ADMIN_LOGIN = 'test_admin'
62 TEST_USER_ADMIN_PASS = 'test12'
62 TEST_USER_ADMIN_PASS = 'test12'
63 TEST_USER_ADMIN_EMAIL = 'test_admin@mail.com'
63 TEST_USER_ADMIN_EMAIL = 'test_admin@mail.com'
64
64
65 TEST_USER_REGULAR_LOGIN = 'test_regular'
65 TEST_USER_REGULAR_LOGIN = 'test_regular'
66 TEST_USER_REGULAR_PASS = 'test12'
66 TEST_USER_REGULAR_PASS = 'test12'
67 TEST_USER_REGULAR_EMAIL = 'test_regular@mail.com'
67 TEST_USER_REGULAR_EMAIL = 'test_regular@mail.com'
68
68
69 TEST_USER_REGULAR2_LOGIN = 'test_regular2'
69 TEST_USER_REGULAR2_LOGIN = 'test_regular2'
70 TEST_USER_REGULAR2_PASS = 'test12'
70 TEST_USER_REGULAR2_PASS = 'test12'
71 TEST_USER_REGULAR2_EMAIL = 'test_regular2@mail.com'
71 TEST_USER_REGULAR2_EMAIL = 'test_regular2@mail.com'
72
72
73 HG_REPO = 'vcs_test_hg'
73 HG_REPO = 'vcs_test_hg'
74 GIT_REPO = 'vcs_test_git'
74 GIT_REPO = 'vcs_test_git'
75 SVN_REPO = 'vcs_test_svn'
75 SVN_REPO = 'vcs_test_svn'
76
76
77 NEW_HG_REPO = 'vcs_test_hg_new'
77 NEW_HG_REPO = 'vcs_test_hg_new'
78 NEW_GIT_REPO = 'vcs_test_git_new'
78 NEW_GIT_REPO = 'vcs_test_git_new'
79
79
80 HG_FORK = 'vcs_test_hg_fork'
80 HG_FORK = 'vcs_test_hg_fork'
81 GIT_FORK = 'vcs_test_git_fork'
81 GIT_FORK = 'vcs_test_git_fork'
82
82
83 ## VCS
83 ## VCS
84 SCM_TESTS = ['hg', 'git']
84 SCM_TESTS = ['hg', 'git']
85 uniq_suffix = str(int(time.mktime(datetime.datetime.now().timetuple())))
85 uniq_suffix = str(int(time.mktime(datetime.datetime.now().timetuple())))
86
86
87 TEST_GIT_REPO = jn(TESTS_TMP_PATH, GIT_REPO)
87 TEST_GIT_REPO = jn(TESTS_TMP_PATH, GIT_REPO)
88 TEST_GIT_REPO_CLONE = jn(TESTS_TMP_PATH, 'vcsgitclone%s' % uniq_suffix)
88 TEST_GIT_REPO_CLONE = jn(TESTS_TMP_PATH, 'vcsgitclone%s' % uniq_suffix)
89 TEST_GIT_REPO_PULL = jn(TESTS_TMP_PATH, 'vcsgitpull%s' % uniq_suffix)
89 TEST_GIT_REPO_PULL = jn(TESTS_TMP_PATH, 'vcsgitpull%s' % uniq_suffix)
90
90
91 TEST_HG_REPO = jn(TESTS_TMP_PATH, HG_REPO)
91 TEST_HG_REPO = jn(TESTS_TMP_PATH, HG_REPO)
92 TEST_HG_REPO_CLONE = jn(TESTS_TMP_PATH, 'vcshgclone%s' % uniq_suffix)
92 TEST_HG_REPO_CLONE = jn(TESTS_TMP_PATH, 'vcshgclone%s' % uniq_suffix)
93 TEST_HG_REPO_PULL = jn(TESTS_TMP_PATH, 'vcshgpull%s' % uniq_suffix)
93 TEST_HG_REPO_PULL = jn(TESTS_TMP_PATH, 'vcshgpull%s' % uniq_suffix)
94
94
95 TEST_REPO_PREFIX = 'vcs-test'
95 TEST_REPO_PREFIX = 'vcs-test'
96
96
97
97
98 def clear_cache_regions(regions=None):
98 def clear_cache_regions(regions=None):
99 # dogpile
99 # dogpile
100 from rhodecode.lib.rc_cache import region_meta
100 from rhodecode.lib.rc_cache import region_meta
101 for region_name, region in region_meta.dogpile_cache_regions.items():
101 for region_name, region in region_meta.dogpile_cache_regions.items():
102 if not regions or region_name in regions:
102 if not regions or region_name in regions:
103 region.invalidate()
103 region.invalidate()
104
104
105
105
106 def get_new_dir(title):
106 def get_new_dir(title):
107 """
107 """
108 Returns always new directory path.
108 Returns always new directory path.
109 """
109 """
110 from rhodecode.tests.vcs.utils import get_normalized_path
110 from rhodecode.tests.vcs.utils import get_normalized_path
111 name_parts = [TEST_REPO_PREFIX]
111 name_parts = [TEST_REPO_PREFIX]
112 if title:
112 if title:
113 name_parts.append(title)
113 name_parts.append(title)
114 hex_str = hashlib.sha1('%s %s' % (os.getpid(), time.time())).hexdigest()
114 hex_str = hashlib.sha1('%s %s' % (os.getpid(), time.time())).hexdigest()
115 name_parts.append(hex_str)
115 name_parts.append(hex_str)
116 name = '-'.join(name_parts)
116 name = '-'.join(name_parts)
117 path = os.path.join(TEST_DIR, name)
117 path = os.path.join(TEST_DIR, name)
118 return get_normalized_path(path)
118 return get_normalized_path(path)
119
119
120
120
121 def repo_id_generator(name):
121 def repo_id_generator(name):
122 numeric_hash = 0
122 numeric_hash = 0
123 for char in name:
123 for char in name:
124 numeric_hash += (ord(char))
124 numeric_hash += (ord(char))
125 return numeric_hash
125 return numeric_hash
126
126
127
127
128 @pytest.mark.usefixtures('app', 'index_location')
128 @pytest.mark.usefixtures('app', 'index_location')
129 class TestController(object):
129 class TestController(object):
130
130
131 maxDiff = None
131 maxDiff = None
132
132
133 def log_user(self, username=TEST_USER_ADMIN_LOGIN,
133 def log_user(self, username=TEST_USER_ADMIN_LOGIN,
134 password=TEST_USER_ADMIN_PASS):
134 password=TEST_USER_ADMIN_PASS):
135 self._logged_username = username
135 self._logged_username = username
136 self._session = login_user_session(self.app, username, password)
136 self._session = login_user_session(self.app, username, password)
137 self.csrf_token = auth.get_csrf_token(self._session)
137 self.csrf_token = auth.get_csrf_token(self._session)
138
138
139 return self._session['rhodecode_user']
139 return self._session['rhodecode_user']
140
140
141 def logout_user(self):
141 def logout_user(self):
142 logout_user_session(self.app, auth.get_csrf_token(self._session))
142 logout_user_session(self.app, auth.get_csrf_token(self._session))
143 self.csrf_token = None
143 self.csrf_token = None
144 self._logged_username = None
144 self._logged_username = None
145 self._session = None
145 self._session = None
146
146
147 def _get_logged_user(self):
147 def _get_logged_user(self):
148 return User.get_by_username(self._logged_username)
148 return User.get_by_username(self._logged_username)
149
149
150
150
151 def login_user_session(
151 def login_user_session(
152 app, username=TEST_USER_ADMIN_LOGIN, password=TEST_USER_ADMIN_PASS):
152 app, username=TEST_USER_ADMIN_LOGIN, password=TEST_USER_ADMIN_PASS):
153
153
154 response = app.post(
154 response = app.post(
155 h.route_path('login'),
155 h.route_path('login'),
156 {'username': username, 'password': password})
156 {'username': username, 'password': password})
157 if 'invalid user name' in response.body:
157 if 'invalid user name' in response.body:
158 pytest.fail('could not login using %s %s' % (username, password))
158 pytest.fail('could not login using %s %s' % (username, password))
159
159
160 assert response.status == '302 Found'
160 assert response.status == '302 Found'
161 response = response.follow()
161 response = response.follow()
162 assert response.status == '200 OK'
162 assert response.status == '200 OK'
163
163
164 session = response.get_session_from_response()
164 session = response.get_session_from_response()
165 assert 'rhodecode_user' in session
165 assert 'rhodecode_user' in session
166 rc_user = session['rhodecode_user']
166 rc_user = session['rhodecode_user']
167 assert rc_user.get('username') == username
167 assert rc_user.get('username') == username
168 assert rc_user.get('is_authenticated')
168 assert rc_user.get('is_authenticated')
169
169
170 return session
170 return session
171
171
172
172
173 def logout_user_session(app, csrf_token):
173 def logout_user_session(app, csrf_token):
174 app.post(h.route_path('logout'), {'csrf_token': csrf_token}, status=302)
174 app.post(h.route_path('logout'), {'csrf_token': csrf_token}, status=302)
175
175
176
176
177 def login_user(app, username=TEST_USER_ADMIN_LOGIN,
177 def login_user(app, username=TEST_USER_ADMIN_LOGIN,
178 password=TEST_USER_ADMIN_PASS):
178 password=TEST_USER_ADMIN_PASS):
179 return login_user_session(app, username, password)['rhodecode_user']
179 return login_user_session(app, username, password)['rhodecode_user']
180
180
181
181
182 def assert_session_flash(response, msg=None, category=None, no_=None):
182 def assert_session_flash(response, msg=None, category=None, no_=None):
183 """
183 """
184 Assert on a flash message in the current session.
184 Assert on a flash message in the current session.
185
185
186 :param response: Response from give calll, it will contain flash
186 :param response: Response from give calll, it will contain flash
187 messages or bound session with them.
187 messages or bound session with them.
188 :param msg: The expected message. Will be evaluated if a
188 :param msg: The expected message. Will be evaluated if a
189 :class:`LazyString` is passed in.
189 :class:`LazyString` is passed in.
190 :param category: Optional. If passed, the message category will be
190 :param category: Optional. If passed, the message category will be
191 checked as well.
191 checked as well.
192 :param no_: Optional. If passed, the message will be checked to NOT
192 :param no_: Optional. If passed, the message will be checked to NOT
193 be in the flash session
193 be in the flash session
194 """
194 """
195 if msg is None and no_ is None:
195 if msg is None and no_ is None:
196 raise ValueError("Parameter msg or no_ is required.")
196 raise ValueError("Parameter msg or no_ is required.")
197
197
198 if msg and no_:
198 if msg and no_:
199 raise ValueError("Please specify either msg or no_, but not both")
199 raise ValueError("Please specify either msg or no_, but not both")
200
200
201 session = response.get_session_from_response()
201 session = response.get_session_from_response()
202 messages = flash.pop_messages(session=session)
202 messages = flash.pop_messages(session=session)
203 msg = _eval_if_lazy(msg)
203 msg = _eval_if_lazy(msg)
204
204
205 if no_:
205 if no_:
206 error_msg = 'unable to detect no_ message `%s` in empty flash list' % no_
206 error_msg = 'unable to detect no_ message `%s` in empty flash list' % no_
207 else:
207 else:
208 error_msg = 'unable to find message `%s` in empty flash list' % msg
208 error_msg = 'unable to find message `%s` in empty flash list' % msg
209 assert messages, error_msg
209 assert messages, error_msg
210 message = messages[0]
210 message = messages[0]
211
211
212 message_text = _eval_if_lazy(message.message) or ''
212 message_text = _eval_if_lazy(message.message) or ''
213
213
214 if no_:
214 if no_:
215 if no_ in message_text:
215 if no_ in message_text:
216 msg = u'msg `%s` found in session flash.' % (no_,)
216 msg = u'msg `%s` found in session flash.' % (no_,)
217 pytest.fail(safe_str(msg))
217 pytest.fail(safe_str(msg))
218 else:
218 else:
219 if msg not in message_text:
219 if msg not in message_text:
220 fail_msg = u'msg `%s` not found in session ' \
220 fail_msg = u'msg `%s` not found in session ' \
221 u'flash: got `%s` (type:%s) instead' % (
221 u'flash: got `%s` (type:%s) instead' % (
222 msg, message_text, type(message_text))
222 msg, message_text, type(message_text))
223
223
224 pytest.fail(safe_str(fail_msg))
224 pytest.fail(safe_str(fail_msg))
225 if category:
225 if category:
226 assert category == message.category
226 assert category == message.category
227
227
228
228
229 def _eval_if_lazy(value):
229 def _eval_if_lazy(value):
230 return value.eval() if hasattr(value, 'eval') else value
230 return value.eval() if hasattr(value, 'eval') else value
231
231
232
232
233 def no_newline_id_generator(test_name):
233 def no_newline_id_generator(test_name):
234 """
234 """
235 Generates a test name without spaces or newlines characters. Used for
235 Generates a test name without spaces or newlines characters. Used for
236 nicer output of progress of test
236 nicer output of progress of test
237 """
237 """
238 org_name = test_name
238 org_name = test_name
239 test_name = safe_str(test_name)\
239 test_name = safe_str(test_name)\
240 .replace('\n', '_N') \
240 .replace('\n', '_N') \
241 .replace('\r', '_N') \
241 .replace('\r', '_N') \
242 .replace('\t', '_T') \
242 .replace('\t', '_T') \
243 .replace(' ', '_S')
243 .replace(' ', '_S')
244
244
245 return test_name or 'test-with-empty-name'
245 return test_name or 'test-with-empty-name'
246
246
247
247
248 def route_path_generator(url_defs, name, params=None, **kwargs):
248 def route_path_generator(url_defs, name, params=None, **kwargs):
249 import urllib.request, urllib.parse, urllib.error
249 import urllib.request, urllib.parse, urllib.error
250
250
251 base_url = url_defs[name].format(**kwargs)
251 base_url = url_defs[name].format(**kwargs)
252
252
253 if params:
253 if params:
254 base_url = '{}?{}'.format(base_url, urllib.parse.urlencode(params))
254 base_url = '{}?{}'.format(base_url, urllib.parse.urlencode(params))
255 return base_url
255 return base_url
@@ -1,203 +1,203 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2020 RhodeCode GmbH
3 # Copyright (C) 2010-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 """
21 """
22 Test suite for making push/pull operations
22 Test suite for making push/pull operations
23 """
23 """
24
24
25 import os
25 import os
26 import sys
26 import sys
27 import shutil
27 import shutil
28 import logging
28 import logging
29 from os.path import join as jn
29 from os.path import join as jn
30 from os.path import dirname as dn
30 from os.path import dirname as dn
31
31
32 from tempfile import _RandomNameSequence
32 from tempfile import _RandomNameSequence
33 from subprocess import Popen, PIPE
33 from subprocess import Popen, PIPE
34
34
35 from rhodecode.lib.utils2 import engine_from_config
35 from rhodecode.lib.utils2 import engine_from_config
36 from rhodecode.lib.auth import get_crypt_password
36 from rhodecode.lib.auth import get_crypt_password
37 from rhodecode.model import init_model
37 from rhodecode.model import init_model
38 from rhodecode.model import meta
38 from rhodecode.model import meta
39 from rhodecode.model.db import User, Repository
39 from rhodecode.model.db import User, Repository
40
40
41 from rhodecode.tests import TESTS_TMP_PATH, HG_REPO
41 from rhodecode.tests import TESTS_TMP_PATH, HG_REPO
42
42
43 rel_path = dn(dn(dn(dn(os.path.abspath(__file__)))))
43 rel_path = dn(dn(dn(dn(os.path.abspath(__file__)))))
44
44
45
45
46 USER = 'test_admin'
46 USER = 'test_admin'
47 PASS = 'test12'
47 PASS = 'test12'
48 HOST = 'rc.local'
48 HOST = 'rc.local'
49 METHOD = 'pull'
49 METHOD = 'pull'
50 DEBUG = True
50 DEBUG = True
51 log = logging.getLogger(__name__)
51 log = logging.getLogger(__name__)
52
52
53
53
54 class Command(object):
54 class Command(object):
55
55
56 def __init__(self, cwd):
56 def __init__(self, cwd):
57 self.cwd = cwd
57 self.cwd = cwd
58
58
59 def execute(self, cmd, *args):
59 def execute(self, cmd, *args):
60 """Runs command on the system with given ``args``.
60 """Runs command on the system with given ``args``.
61 """
61 """
62
62
63 command = cmd + ' ' + ' '.join(args)
63 command = cmd + ' ' + ' '.join(args)
64 log.debug('Executing %s', command)
64 log.debug('Executing %s', command)
65 if DEBUG:
65 if DEBUG:
66 print(command)
66 print(command)
67 p = Popen(command, shell=True, stdout=PIPE, stderr=PIPE, cwd=self.cwd)
67 p = Popen(command, shell=True, stdout=PIPE, stderr=PIPE, cwd=self.cwd)
68 stdout, stderr = p.communicate()
68 stdout, stderr = p.communicate()
69 if DEBUG:
69 if DEBUG:
70 print('{} {}'.format(stdout, stderr))
70 print('{} {}'.format(stdout, stderr))
71 return stdout, stderr
71 return stdout, stderr
72
72
73
73
74 def get_session():
74 def get_session():
75 conf = {}
75 conf = {}
76 engine = engine_from_config(conf, 'sqlalchemy.db1.')
76 engine = engine_from_config(conf, 'sqlalchemy.db1.')
77 init_model(engine)
77 init_model(engine)
78 sa = meta.Session
78 sa = meta.Session
79 return sa
79 return sa
80
80
81
81
82 def create_test_user(force=True):
82 def create_test_user(force=True):
83 print('creating test user')
83 print('creating test user')
84 sa = get_session()
84 sa = get_session()
85
85
86 user = sa.query(User).filter(User.username == USER).scalar()
86 user = sa.query(User).filter(User.username == USER).scalar()
87
87
88 if force and user is not None:
88 if force and user is not None:
89 print('removing current user')
89 print('removing current user')
90 for repo in sa.query(Repository).filter(Repository.user == user).all():
90 for repo in sa.query(Repository).filter(Repository.user == user).all():
91 sa.delete(repo)
91 sa.delete(repo)
92 sa.delete(user)
92 sa.delete(user)
93 sa.commit()
93 sa.commit()
94
94
95 if user is None or force:
95 if user is None or force:
96 print('creating new one')
96 print('creating new one')
97 new_usr = User()
97 new_usr = User()
98 new_usr.username = USER
98 new_usr.username = USER
99 new_usr.password = get_crypt_password(PASS)
99 new_usr.password = get_crypt_password(PASS)
100 new_usr.email = 'mail@mail.com'
100 new_usr.email = 'mail@mail.com'
101 new_usr.name = 'test'
101 new_usr.name = 'test'
102 new_usr.lastname = 'lasttestname'
102 new_usr.lastname = 'lasttestname'
103 new_usr.active = True
103 new_usr.active = True
104 new_usr.admin = True
104 new_usr.admin = True
105 sa.add(new_usr)
105 sa.add(new_usr)
106 sa.commit()
106 sa.commit()
107
107
108 print('done')
108 print('done')
109
109
110
110
111 def create_test_repo(force=True):
111 def create_test_repo(force=True):
112 print('creating test repo')
112 print('creating test repo')
113 from rhodecode.model.repo import RepoModel
113 from rhodecode.model.repo import RepoModel
114 sa = get_session()
114 sa = get_session()
115
115
116 user = sa.query(User).filter(User.username == USER).scalar()
116 user = sa.query(User).filter(User.username == USER).scalar()
117 if user is None:
117 if user is None:
118 raise Exception('user not found')
118 raise Exception('user not found')
119
119
120 repo = sa.query(Repository).filter(Repository.repo_name == HG_REPO).scalar()
120 repo = sa.query(Repository).filter(Repository.repo_name == HG_REPO).scalar()
121
121
122 if repo is None:
122 if repo is None:
123 print('repo not found creating')
123 print('repo not found creating')
124
124
125 form_data = {'repo_name': HG_REPO,
125 form_data = {'repo_name': HG_REPO,
126 'repo_type': 'hg',
126 'repo_type': 'hg',
127 'private':False,
127 'private':False,
128 'clone_uri': '' }
128 'clone_uri': '' }
129 rm = RepoModel(sa)
129 rm = RepoModel(sa)
130 rm.base_path = '/home/hg'
130 rm.base_path = '/home/hg'
131 rm.create(form_data, user)
131 rm.create(form_data, user)
132
132
133 print('done')
133 print('done')
134
134
135
135
136 def get_anonymous_access():
136 def get_anonymous_access():
137 sa = get_session()
137 sa = get_session()
138 return sa.query(User).filter(User.username == 'default').one().active
138 return sa.query(User).filter(User.username == 'default').one().active
139
139
140
140
141 #==============================================================================
141 #==============================================================================
142 # TESTS
142 # TESTS
143 #==============================================================================
143 #==============================================================================
144 def test_clone_with_credentials(repo=HG_REPO, method=METHOD,
144 def test_clone_with_credentials(repo=HG_REPO, method=METHOD,
145 seq=None, backend='hg', check_output=True):
145 seq=None, backend='hg', check_output=True):
146 cwd = path = jn(TESTS_TMP_PATH, repo)
146 cwd = path = jn(TESTS_TMP_PATH, repo)
147
147
148 if seq is None:
148 if seq is None:
149 seq = _RandomNameSequence().next()
149 seq = next(_RandomNameSequence())
150
150
151 try:
151 try:
152 shutil.rmtree(path, ignore_errors=True)
152 shutil.rmtree(path, ignore_errors=True)
153 os.makedirs(path)
153 os.makedirs(path)
154 except OSError:
154 except OSError:
155 raise
155 raise
156
156
157 clone_url = 'http://%(user)s:%(pass)s@%(host)s/%(cloned_repo)s' % \
157 clone_url = 'http://%(user)s:%(pass)s@%(host)s/%(cloned_repo)s' % \
158 {'user': USER,
158 {'user': USER,
159 'pass': PASS,
159 'pass': PASS,
160 'host': HOST,
160 'host': HOST,
161 'cloned_repo': repo, }
161 'cloned_repo': repo, }
162
162
163 dest = path + seq
163 dest = path + seq
164 if method == 'pull':
164 if method == 'pull':
165 stdout, stderr = Command(cwd).execute(backend, method, '--cwd', dest, clone_url)
165 stdout, stderr = Command(cwd).execute(backend, method, '--cwd', dest, clone_url)
166 else:
166 else:
167 stdout, stderr = Command(cwd).execute(backend, method, clone_url, dest)
167 stdout, stderr = Command(cwd).execute(backend, method, clone_url, dest)
168 if check_output:
168 if check_output:
169 if backend == 'hg':
169 if backend == 'hg':
170 assert """adding file changes""" in stdout, 'no messages about cloning'
170 assert """adding file changes""" in stdout, 'no messages about cloning'
171 assert """abort""" not in stderr, 'got error from clone'
171 assert """abort""" not in stderr, 'got error from clone'
172 elif backend == 'git':
172 elif backend == 'git':
173 assert """Cloning into""" in stdout, 'no messages about cloning'
173 assert """Cloning into""" in stdout, 'no messages about cloning'
174
174
175
175
176 if __name__ == '__main__':
176 if __name__ == '__main__':
177 try:
177 try:
178 create_test_user(force=False)
178 create_test_user(force=False)
179 seq = None
179 seq = None
180 import time
180 import time
181
181
182 try:
182 try:
183 METHOD = sys.argv[3]
183 METHOD = sys.argv[3]
184 except Exception:
184 except Exception:
185 pass
185 pass
186
186
187 try:
187 try:
188 backend = sys.argv[4]
188 backend = sys.argv[4]
189 except Exception:
189 except Exception:
190 backend = 'hg'
190 backend = 'hg'
191
191
192 if METHOD == 'pull':
192 if METHOD == 'pull':
193 seq = _RandomNameSequence().next()
193 seq = next(_RandomNameSequence())
194 test_clone_with_credentials(repo=sys.argv[1], method='clone',
194 test_clone_with_credentials(repo=sys.argv[1], method='clone',
195 seq=seq, backend=backend)
195 seq=seq, backend=backend)
196 s = time.time()
196 s = time.time()
197 for i in range(1, int(sys.argv[2]) + 1):
197 for i in range(1, int(sys.argv[2]) + 1):
198 print('take {}'.format(i))
198 print('take {}'.format(i))
199 test_clone_with_credentials(repo=sys.argv[1], method=METHOD,
199 test_clone_with_credentials(repo=sys.argv[1], method=METHOD,
200 seq=seq, backend=backend)
200 seq=seq, backend=backend)
201 print('time taken %.4f' % (time.time() - s))
201 print('time taken %.4f' % (time.time() - s))
202 except Exception as e:
202 except Exception as e:
203 sys.exit('stop on %s' % e)
203 sys.exit('stop on %s' % e)
@@ -1,193 +1,193 b''
1 # -*- coding: utf-8 -*-
1 # -*- coding: utf-8 -*-
2
2
3 # Copyright (C) 2010-2020 RhodeCode GmbH
3 # Copyright (C) 2010-2020 RhodeCode GmbH
4 #
4 #
5 # This program is free software: you can redistribute it and/or modify
5 # This program is free software: you can redistribute it and/or modify
6 # it under the terms of the GNU Affero General Public License, version 3
6 # it under the terms of the GNU Affero General Public License, version 3
7 # (only), as published by the Free Software Foundation.
7 # (only), as published by the Free Software Foundation.
8 #
8 #
9 # This program is distributed in the hope that it will be useful,
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU General Public License for more details.
12 # GNU General Public License for more details.
13 #
13 #
14 # You should have received a copy of the GNU Affero General Public License
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
16 #
16 #
17 # This program is dual-licensed. If you wish to learn more about the
17 # This program is dual-licensed. If you wish to learn more about the
18 # RhodeCode Enterprise Edition, including its added features, Support services,
18 # RhodeCode Enterprise Edition, including its added features, Support services,
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
19 # and proprietary license terms, please see https://rhodecode.com/licenses/
20
20
21 """
21 """
22 Base for test suite for making push/pull operations.
22 Base for test suite for making push/pull operations.
23
23
24 .. important::
24 .. important::
25
25
26 You must have git >= 1.8.5 for tests to work fine. With 68b939b git started
26 You must have git >= 1.8.5 for tests to work fine. With 68b939b git started
27 to redirect things to stderr instead of stdout.
27 to redirect things to stderr instead of stdout.
28 """
28 """
29
29
30 from os.path import join as jn
30 from os.path import join as jn
31 from subprocess import Popen, PIPE
31 from subprocess import Popen, PIPE
32 import logging
32 import logging
33 import os
33 import os
34 import tempfile
34 import tempfile
35
35
36 from rhodecode.tests import GIT_REPO, HG_REPO
36 from rhodecode.tests import GIT_REPO, HG_REPO
37
37
38 DEBUG = True
38 DEBUG = True
39 RC_LOG = os.path.join(tempfile.gettempdir(), 'rc.log')
39 RC_LOG = os.path.join(tempfile.gettempdir(), 'rc.log')
40 REPO_GROUP = 'a_repo_group'
40 REPO_GROUP = 'a_repo_group'
41 HG_REPO_WITH_GROUP = '%s/%s' % (REPO_GROUP, HG_REPO)
41 HG_REPO_WITH_GROUP = '%s/%s' % (REPO_GROUP, HG_REPO)
42 GIT_REPO_WITH_GROUP = '%s/%s' % (REPO_GROUP, GIT_REPO)
42 GIT_REPO_WITH_GROUP = '%s/%s' % (REPO_GROUP, GIT_REPO)
43
43
44 log = logging.getLogger(__name__)
44 log = logging.getLogger(__name__)
45
45
46
46
47 class Command(object):
47 class Command(object):
48
48
49 def __init__(self, cwd):
49 def __init__(self, cwd):
50 self.cwd = cwd
50 self.cwd = cwd
51 self.process = None
51 self.process = None
52
52
53 def execute(self, cmd, *args):
53 def execute(self, cmd, *args):
54 """
54 """
55 Runs command on the system with given ``args``.
55 Runs command on the system with given ``args``.
56 """
56 """
57
57
58 command = cmd + ' ' + ' '.join(args)
58 command = cmd + ' ' + ' '.join(args)
59 if DEBUG:
59 if DEBUG:
60 log.debug('*** CMD %s ***', command)
60 log.debug('*** CMD %s ***', command)
61
61
62 env = dict(os.environ)
62 env = dict(os.environ)
63 # Delete coverage variables, as they make the test fail for Mercurial
63 # Delete coverage variables, as they make the test fail for Mercurial
64 for key in env.keys():
64 for key in env.keys():
65 if key.startswith('COV_CORE_'):
65 if key.startswith('COV_CORE_'):
66 del env[key]
66 del env[key]
67
67
68 self.process = Popen(command, shell=True, stdout=PIPE, stderr=PIPE,
68 self.process = Popen(command, shell=True, stdout=PIPE, stderr=PIPE,
69 cwd=self.cwd, env=env)
69 cwd=self.cwd, env=env)
70 stdout, stderr = self.process.communicate()
70 stdout, stderr = self.process.communicate()
71 if DEBUG:
71 if DEBUG:
72 log.debug('STDOUT:%s', stdout)
72 log.debug('STDOUT:%s', stdout)
73 log.debug('STDERR:%s', stderr)
73 log.debug('STDERR:%s', stderr)
74 return stdout, stderr
74 return stdout, stderr
75
75
76 def assert_returncode_success(self):
76 def assert_returncode_success(self):
77 assert self.process.returncode == 0
77 assert self.process.returncode == 0
78
78
79
79
80 def _add_files(vcs, dest, clone_url=None, tags=None, target_branch=None, new_branch=False, **kwargs):
80 def _add_files(vcs, dest, clone_url=None, tags=None, target_branch=None, new_branch=False, **kwargs):
81 git_ident = "git config user.name {} && git config user.email {}".format(
81 git_ident = "git config user.name {} && git config user.email {}".format(
82 'Marcin KuΕΊminski', 'me@email.com')
82 'Marcin KuΕΊminski', 'me@email.com')
83 cwd = path = jn(dest)
83 cwd = path = jn(dest)
84
84
85 tags = tags or []
85 tags = tags or []
86 added_file = jn(path, '%s_setup.py' % tempfile._RandomNameSequence().next())
86 added_file = jn(path, '{}_setup.py'.format(next(tempfile._RandomNameSequence())))
87 Command(cwd).execute('touch %s' % added_file)
87 Command(cwd).execute('touch %s' % added_file)
88 Command(cwd).execute('%s add %s' % (vcs, added_file))
88 Command(cwd).execute('%s add %s' % (vcs, added_file))
89 author_str = 'Marcin KuΕΊminski <me@email.com>'
89 author_str = 'Marcin KuΕΊminski <me@email.com>'
90
90
91 for i in range(kwargs.get('files_no', 3)):
91 for i in range(kwargs.get('files_no', 3)):
92 cmd = """echo 'added_line%s' >> %s""" % (i, added_file)
92 cmd = """echo 'added_line%s' >> %s""" % (i, added_file)
93 Command(cwd).execute(cmd)
93 Command(cwd).execute(cmd)
94
94
95 if vcs == 'hg':
95 if vcs == 'hg':
96 cmd = """hg commit -m 'committed new %s' -u '%s' %s """ % (
96 cmd = """hg commit -m 'committed new %s' -u '%s' %s """ % (
97 i, author_str, added_file
97 i, author_str, added_file
98 )
98 )
99 elif vcs == 'git':
99 elif vcs == 'git':
100 cmd = """%s && git commit -m 'committed new %s' %s""" % (
100 cmd = """%s && git commit -m 'committed new %s' %s""" % (
101 git_ident, i, added_file)
101 git_ident, i, added_file)
102 Command(cwd).execute(cmd)
102 Command(cwd).execute(cmd)
103
103
104 for tag in tags:
104 for tag in tags:
105 if vcs == 'hg':
105 if vcs == 'hg':
106 Command(cwd).execute(
106 Command(cwd).execute(
107 'hg tag -m "{}" -u "{}" '.format(tag['commit'], author_str), tag['name'])
107 'hg tag -m "{}" -u "{}" '.format(tag['commit'], author_str), tag['name'])
108 elif vcs == 'git':
108 elif vcs == 'git':
109 if tag['commit']:
109 if tag['commit']:
110 # annotated tag
110 # annotated tag
111 _stdout, _stderr = Command(cwd).execute(
111 _stdout, _stderr = Command(cwd).execute(
112 """%s && git tag -a %s -m "%s" """ % (
112 """%s && git tag -a %s -m "%s" """ % (
113 git_ident, tag['name'], tag['commit']))
113 git_ident, tag['name'], tag['commit']))
114 else:
114 else:
115 # lightweight tag
115 # lightweight tag
116 _stdout, _stderr = Command(cwd).execute(
116 _stdout, _stderr = Command(cwd).execute(
117 """%s && git tag %s""" % (
117 """%s && git tag %s""" % (
118 git_ident, tag['name']))
118 git_ident, tag['name']))
119
119
120
120
121 def _add_files_and_push(vcs, dest, clone_url=None, tags=None, target_branch=None,
121 def _add_files_and_push(vcs, dest, clone_url=None, tags=None, target_branch=None,
122 new_branch=False, **kwargs):
122 new_branch=False, **kwargs):
123 """
123 """
124 Generate some files, add it to DEST repo and push back
124 Generate some files, add it to DEST repo and push back
125 vcs is git or hg and defines what VCS we want to make those files for
125 vcs is git or hg and defines what VCS we want to make those files for
126 """
126 """
127 git_ident = "git config user.name {} && git config user.email {}".format(
127 git_ident = "git config user.name {} && git config user.email {}".format(
128 'Marcin KuΕΊminski', 'me@email.com')
128 'Marcin KuΕΊminski', 'me@email.com')
129 cwd = path = jn(dest)
129 cwd = path = jn(dest)
130
130
131 # commit some stuff into this repo
131 # commit some stuff into this repo
132 _add_files(vcs, dest, clone_url, tags, target_branch, new_branch, **kwargs)
132 _add_files(vcs, dest, clone_url, tags, target_branch, new_branch, **kwargs)
133
133
134 default_target_branch = {
134 default_target_branch = {
135 'git': 'master',
135 'git': 'master',
136 'hg': 'default'
136 'hg': 'default'
137 }.get(vcs)
137 }.get(vcs)
138
138
139 target_branch = target_branch or default_target_branch
139 target_branch = target_branch or default_target_branch
140
140
141 # PUSH it back
141 # PUSH it back
142 stdout = stderr = None
142 stdout = stderr = None
143 if vcs == 'hg':
143 if vcs == 'hg':
144 maybe_new_branch = ''
144 maybe_new_branch = ''
145 if new_branch:
145 if new_branch:
146 maybe_new_branch = '--new-branch'
146 maybe_new_branch = '--new-branch'
147 stdout, stderr = Command(cwd).execute(
147 stdout, stderr = Command(cwd).execute(
148 'hg push --verbose {} -r {} {}'.format(maybe_new_branch, target_branch, clone_url)
148 'hg push --verbose {} -r {} {}'.format(maybe_new_branch, target_branch, clone_url)
149 )
149 )
150 elif vcs == 'git':
150 elif vcs == 'git':
151 stdout, stderr = Command(cwd).execute(
151 stdout, stderr = Command(cwd).execute(
152 """{} &&
152 """{} &&
153 git push --verbose --tags {} {}""".format(git_ident, clone_url, target_branch)
153 git push --verbose --tags {} {}""".format(git_ident, clone_url, target_branch)
154 )
154 )
155
155
156 return stdout, stderr
156 return stdout, stderr
157
157
158
158
159 def _check_proper_git_push(
159 def _check_proper_git_push(
160 stdout, stderr, branch='master', should_set_default_branch=False):
160 stdout, stderr, branch='master', should_set_default_branch=False):
161 # Note: Git is writing most information to stderr intentionally
161 # Note: Git is writing most information to stderr intentionally
162 assert 'fatal' not in stderr
162 assert 'fatal' not in stderr
163 assert 'rejected' not in stderr
163 assert 'rejected' not in stderr
164 assert 'Pushing to' in stderr
164 assert 'Pushing to' in stderr
165 assert '%s -> %s' % (branch, branch) in stderr
165 assert '%s -> %s' % (branch, branch) in stderr
166
166
167 if should_set_default_branch:
167 if should_set_default_branch:
168 assert "Setting default branch to %s" % branch in stderr
168 assert "Setting default branch to %s" % branch in stderr
169 else:
169 else:
170 assert "Setting default branch" not in stderr
170 assert "Setting default branch" not in stderr
171
171
172
172
173 def _check_proper_hg_push(stdout, stderr, branch='default'):
173 def _check_proper_hg_push(stdout, stderr, branch='default'):
174 assert 'pushing to' in stdout
174 assert 'pushing to' in stdout
175 assert 'searching for changes' in stdout
175 assert 'searching for changes' in stdout
176
176
177 assert 'abort:' not in stderr
177 assert 'abort:' not in stderr
178
178
179
179
180 def _check_proper_clone(stdout, stderr, vcs):
180 def _check_proper_clone(stdout, stderr, vcs):
181 if vcs == 'hg':
181 if vcs == 'hg':
182 assert 'requesting all changes' in stdout
182 assert 'requesting all changes' in stdout
183 assert 'adding changesets' in stdout
183 assert 'adding changesets' in stdout
184 assert 'adding manifests' in stdout
184 assert 'adding manifests' in stdout
185 assert 'adding file changes' in stdout
185 assert 'adding file changes' in stdout
186
186
187 assert stderr == ''
187 assert stderr == ''
188
188
189 if vcs == 'git':
189 if vcs == 'git':
190 assert '' == stdout
190 assert '' == stdout
191 assert 'Cloning into' in stderr
191 assert 'Cloning into' in stderr
192 assert 'abort:' not in stderr
192 assert 'abort:' not in stderr
193 assert 'fatal:' not in stderr
193 assert 'fatal:' not in stderr
General Comments 0
You need to be logged in to leave comments. Login now