Show More
@@ -1,481 +1,481 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | # Copyright (C) 2011-2019 RhodeCode GmbH |
|
3 | # Copyright (C) 2011-2019 RhodeCode GmbH | |
4 | # |
|
4 | # | |
5 | # This program is free software: you can redistribute it and/or modify |
|
5 | # This program is free software: you can redistribute it and/or modify | |
6 | # it under the terms of the GNU Affero General Public License, version 3 |
|
6 | # it under the terms of the GNU Affero General Public License, version 3 | |
7 | # (only), as published by the Free Software Foundation. |
|
7 | # (only), as published by the Free Software Foundation. | |
8 | # |
|
8 | # | |
9 | # This program is distributed in the hope that it will be useful, |
|
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. |
|
12 | # GNU General Public License for more details. | |
13 | # |
|
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License |
|
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | # |
|
16 | # | |
17 | # This program is dual-licensed. If you wish to learn more about the |
|
17 | # This program is dual-licensed. If you wish to learn more about the | |
18 | # RhodeCode Enterprise Edition, including its added features, Support services, |
|
18 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ |
|
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
20 |
|
20 | |||
21 | import inspect |
|
21 | import inspect | |
22 | import logging |
|
22 | import logging | |
23 | import itertools |
|
23 | import itertools | |
24 | import base64 |
|
24 | import base64 | |
25 |
|
25 | |||
26 | from pyramid import compat |
|
26 | from pyramid import compat | |
27 |
|
27 | |||
28 | from rhodecode.api import ( |
|
28 | from rhodecode.api import ( | |
29 | jsonrpc_method, JSONRPCError, JSONRPCForbidden, find_methods) |
|
29 | jsonrpc_method, JSONRPCError, JSONRPCForbidden, find_methods) | |
30 |
|
30 | |||
31 | from rhodecode.api.utils import ( |
|
31 | from rhodecode.api.utils import ( | |
32 | Optional, OAttr, has_superadmin_permission, get_user_or_error) |
|
32 | Optional, OAttr, has_superadmin_permission, get_user_or_error) | |
33 | from rhodecode.lib.utils import repo2db_mapper |
|
33 | from rhodecode.lib.utils import repo2db_mapper | |
34 | from rhodecode.lib import system_info |
|
34 | from rhodecode.lib import system_info | |
35 | from rhodecode.lib import user_sessions |
|
35 | from rhodecode.lib import user_sessions | |
36 | from rhodecode.lib import exc_tracking |
|
36 | from rhodecode.lib import exc_tracking | |
37 | from rhodecode.lib.ext_json import json |
|
37 | from rhodecode.lib.ext_json import json | |
38 | from rhodecode.lib.utils2 import safe_int |
|
38 | from rhodecode.lib.utils2 import safe_int | |
39 | from rhodecode.model.db import UserIpMap |
|
39 | from rhodecode.model.db import UserIpMap | |
40 | from rhodecode.model.scm import ScmModel |
|
40 | from rhodecode.model.scm import ScmModel | |
41 | from rhodecode.model.settings import VcsSettingsModel |
|
41 | from rhodecode.model.settings import VcsSettingsModel | |
42 |
from rhodecode.apps. |
|
42 | from rhodecode.apps.file_store import utils | |
43 |
from rhodecode.apps. |
|
43 | from rhodecode.apps.file_store.exceptions import FileNotAllowedException, \ | |
44 | FileOverSizeException |
|
44 | FileOverSizeException | |
45 |
|
45 | |||
46 | log = logging.getLogger(__name__) |
|
46 | log = logging.getLogger(__name__) | |
47 |
|
47 | |||
48 |
|
48 | |||
49 | @jsonrpc_method() |
|
49 | @jsonrpc_method() | |
50 | def get_server_info(request, apiuser): |
|
50 | def get_server_info(request, apiuser): | |
51 | """ |
|
51 | """ | |
52 | Returns the |RCE| server information. |
|
52 | Returns the |RCE| server information. | |
53 |
|
53 | |||
54 | This includes the running version of |RCE| and all installed |
|
54 | This includes the running version of |RCE| and all installed | |
55 | packages. This command takes the following options: |
|
55 | packages. This command takes the following options: | |
56 |
|
56 | |||
57 | :param apiuser: This is filled automatically from the |authtoken|. |
|
57 | :param apiuser: This is filled automatically from the |authtoken|. | |
58 | :type apiuser: AuthUser |
|
58 | :type apiuser: AuthUser | |
59 |
|
59 | |||
60 | Example output: |
|
60 | Example output: | |
61 |
|
61 | |||
62 | .. code-block:: bash |
|
62 | .. code-block:: bash | |
63 |
|
63 | |||
64 | id : <id_given_in_input> |
|
64 | id : <id_given_in_input> | |
65 | result : { |
|
65 | result : { | |
66 | 'modules': [<module name>,...] |
|
66 | 'modules': [<module name>,...] | |
67 | 'py_version': <python version>, |
|
67 | 'py_version': <python version>, | |
68 | 'platform': <platform type>, |
|
68 | 'platform': <platform type>, | |
69 | 'rhodecode_version': <rhodecode version> |
|
69 | 'rhodecode_version': <rhodecode version> | |
70 | } |
|
70 | } | |
71 | error : null |
|
71 | error : null | |
72 | """ |
|
72 | """ | |
73 |
|
73 | |||
74 | if not has_superadmin_permission(apiuser): |
|
74 | if not has_superadmin_permission(apiuser): | |
75 | raise JSONRPCForbidden() |
|
75 | raise JSONRPCForbidden() | |
76 |
|
76 | |||
77 | server_info = ScmModel().get_server_info(request.environ) |
|
77 | server_info = ScmModel().get_server_info(request.environ) | |
78 | # rhodecode-index requires those |
|
78 | # rhodecode-index requires those | |
79 |
|
79 | |||
80 | server_info['index_storage'] = server_info['search']['value']['location'] |
|
80 | server_info['index_storage'] = server_info['search']['value']['location'] | |
81 | server_info['storage'] = server_info['storage']['value']['path'] |
|
81 | server_info['storage'] = server_info['storage']['value']['path'] | |
82 |
|
82 | |||
83 | return server_info |
|
83 | return server_info | |
84 |
|
84 | |||
85 |
|
85 | |||
86 | @jsonrpc_method() |
|
86 | @jsonrpc_method() | |
87 | def get_repo_store(request, apiuser): |
|
87 | def get_repo_store(request, apiuser): | |
88 | """ |
|
88 | """ | |
89 | Returns the |RCE| repository storage information. |
|
89 | Returns the |RCE| repository storage information. | |
90 |
|
90 | |||
91 | :param apiuser: This is filled automatically from the |authtoken|. |
|
91 | :param apiuser: This is filled automatically from the |authtoken|. | |
92 | :type apiuser: AuthUser |
|
92 | :type apiuser: AuthUser | |
93 |
|
93 | |||
94 | Example output: |
|
94 | Example output: | |
95 |
|
95 | |||
96 | .. code-block:: bash |
|
96 | .. code-block:: bash | |
97 |
|
97 | |||
98 | id : <id_given_in_input> |
|
98 | id : <id_given_in_input> | |
99 | result : { |
|
99 | result : { | |
100 | 'modules': [<module name>,...] |
|
100 | 'modules': [<module name>,...] | |
101 | 'py_version': <python version>, |
|
101 | 'py_version': <python version>, | |
102 | 'platform': <platform type>, |
|
102 | 'platform': <platform type>, | |
103 | 'rhodecode_version': <rhodecode version> |
|
103 | 'rhodecode_version': <rhodecode version> | |
104 | } |
|
104 | } | |
105 | error : null |
|
105 | error : null | |
106 | """ |
|
106 | """ | |
107 |
|
107 | |||
108 | if not has_superadmin_permission(apiuser): |
|
108 | if not has_superadmin_permission(apiuser): | |
109 | raise JSONRPCForbidden() |
|
109 | raise JSONRPCForbidden() | |
110 |
|
110 | |||
111 | path = VcsSettingsModel().get_repos_location() |
|
111 | path = VcsSettingsModel().get_repos_location() | |
112 | return {"path": path} |
|
112 | return {"path": path} | |
113 |
|
113 | |||
114 |
|
114 | |||
115 | @jsonrpc_method() |
|
115 | @jsonrpc_method() | |
116 | def get_ip(request, apiuser, userid=Optional(OAttr('apiuser'))): |
|
116 | def get_ip(request, apiuser, userid=Optional(OAttr('apiuser'))): | |
117 | """ |
|
117 | """ | |
118 | Displays the IP Address as seen from the |RCE| server. |
|
118 | Displays the IP Address as seen from the |RCE| server. | |
119 |
|
119 | |||
120 | * This command displays the IP Address, as well as all the defined IP |
|
120 | * This command displays the IP Address, as well as all the defined IP | |
121 | addresses for the specified user. If the ``userid`` is not set, the |
|
121 | addresses for the specified user. If the ``userid`` is not set, the | |
122 | data returned is for the user calling the method. |
|
122 | data returned is for the user calling the method. | |
123 |
|
123 | |||
124 | This command can only be run using an |authtoken| with admin rights to |
|
124 | This command can only be run using an |authtoken| with admin rights to | |
125 | the specified repository. |
|
125 | the specified repository. | |
126 |
|
126 | |||
127 | This command takes the following options: |
|
127 | This command takes the following options: | |
128 |
|
128 | |||
129 | :param apiuser: This is filled automatically from |authtoken|. |
|
129 | :param apiuser: This is filled automatically from |authtoken|. | |
130 | :type apiuser: AuthUser |
|
130 | :type apiuser: AuthUser | |
131 | :param userid: Sets the userid for which associated IP Address data |
|
131 | :param userid: Sets the userid for which associated IP Address data | |
132 | is returned. |
|
132 | is returned. | |
133 | :type userid: Optional(str or int) |
|
133 | :type userid: Optional(str or int) | |
134 |
|
134 | |||
135 | Example output: |
|
135 | Example output: | |
136 |
|
136 | |||
137 | .. code-block:: bash |
|
137 | .. code-block:: bash | |
138 |
|
138 | |||
139 | id : <id_given_in_input> |
|
139 | id : <id_given_in_input> | |
140 | result : { |
|
140 | result : { | |
141 | "server_ip_addr": "<ip_from_clien>", |
|
141 | "server_ip_addr": "<ip_from_clien>", | |
142 | "user_ips": [ |
|
142 | "user_ips": [ | |
143 | { |
|
143 | { | |
144 | "ip_addr": "<ip_with_mask>", |
|
144 | "ip_addr": "<ip_with_mask>", | |
145 | "ip_range": ["<start_ip>", "<end_ip>"], |
|
145 | "ip_range": ["<start_ip>", "<end_ip>"], | |
146 | }, |
|
146 | }, | |
147 | ... |
|
147 | ... | |
148 | ] |
|
148 | ] | |
149 | } |
|
149 | } | |
150 |
|
150 | |||
151 | """ |
|
151 | """ | |
152 | if not has_superadmin_permission(apiuser): |
|
152 | if not has_superadmin_permission(apiuser): | |
153 | raise JSONRPCForbidden() |
|
153 | raise JSONRPCForbidden() | |
154 |
|
154 | |||
155 | userid = Optional.extract(userid, evaluate_locals=locals()) |
|
155 | userid = Optional.extract(userid, evaluate_locals=locals()) | |
156 | userid = getattr(userid, 'user_id', userid) |
|
156 | userid = getattr(userid, 'user_id', userid) | |
157 |
|
157 | |||
158 | user = get_user_or_error(userid) |
|
158 | user = get_user_or_error(userid) | |
159 | ips = UserIpMap.query().filter(UserIpMap.user == user).all() |
|
159 | ips = UserIpMap.query().filter(UserIpMap.user == user).all() | |
160 | return { |
|
160 | return { | |
161 | 'server_ip_addr': request.rpc_ip_addr, |
|
161 | 'server_ip_addr': request.rpc_ip_addr, | |
162 | 'user_ips': ips |
|
162 | 'user_ips': ips | |
163 | } |
|
163 | } | |
164 |
|
164 | |||
165 |
|
165 | |||
166 | @jsonrpc_method() |
|
166 | @jsonrpc_method() | |
167 | def rescan_repos(request, apiuser, remove_obsolete=Optional(False)): |
|
167 | def rescan_repos(request, apiuser, remove_obsolete=Optional(False)): | |
168 | """ |
|
168 | """ | |
169 | Triggers a rescan of the specified repositories. |
|
169 | Triggers a rescan of the specified repositories. | |
170 |
|
170 | |||
171 | * If the ``remove_obsolete`` option is set, it also deletes repositories |
|
171 | * If the ``remove_obsolete`` option is set, it also deletes repositories | |
172 | that are found in the database but not on the file system, so called |
|
172 | that are found in the database but not on the file system, so called | |
173 | "clean zombies". |
|
173 | "clean zombies". | |
174 |
|
174 | |||
175 | This command can only be run using an |authtoken| with admin rights to |
|
175 | This command can only be run using an |authtoken| with admin rights to | |
176 | the specified repository. |
|
176 | the specified repository. | |
177 |
|
177 | |||
178 | This command takes the following options: |
|
178 | This command takes the following options: | |
179 |
|
179 | |||
180 | :param apiuser: This is filled automatically from the |authtoken|. |
|
180 | :param apiuser: This is filled automatically from the |authtoken|. | |
181 | :type apiuser: AuthUser |
|
181 | :type apiuser: AuthUser | |
182 | :param remove_obsolete: Deletes repositories from the database that |
|
182 | :param remove_obsolete: Deletes repositories from the database that | |
183 | are not found on the filesystem. |
|
183 | are not found on the filesystem. | |
184 | :type remove_obsolete: Optional(``True`` | ``False``) |
|
184 | :type remove_obsolete: Optional(``True`` | ``False``) | |
185 |
|
185 | |||
186 | Example output: |
|
186 | Example output: | |
187 |
|
187 | |||
188 | .. code-block:: bash |
|
188 | .. code-block:: bash | |
189 |
|
189 | |||
190 | id : <id_given_in_input> |
|
190 | id : <id_given_in_input> | |
191 | result : { |
|
191 | result : { | |
192 | 'added': [<added repository name>,...] |
|
192 | 'added': [<added repository name>,...] | |
193 | 'removed': [<removed repository name>,...] |
|
193 | 'removed': [<removed repository name>,...] | |
194 | } |
|
194 | } | |
195 | error : null |
|
195 | error : null | |
196 |
|
196 | |||
197 | Example error output: |
|
197 | Example error output: | |
198 |
|
198 | |||
199 | .. code-block:: bash |
|
199 | .. code-block:: bash | |
200 |
|
200 | |||
201 | id : <id_given_in_input> |
|
201 | id : <id_given_in_input> | |
202 | result : null |
|
202 | result : null | |
203 | error : { |
|
203 | error : { | |
204 | 'Error occurred during rescan repositories action' |
|
204 | 'Error occurred during rescan repositories action' | |
205 | } |
|
205 | } | |
206 |
|
206 | |||
207 | """ |
|
207 | """ | |
208 | if not has_superadmin_permission(apiuser): |
|
208 | if not has_superadmin_permission(apiuser): | |
209 | raise JSONRPCForbidden() |
|
209 | raise JSONRPCForbidden() | |
210 |
|
210 | |||
211 | try: |
|
211 | try: | |
212 | rm_obsolete = Optional.extract(remove_obsolete) |
|
212 | rm_obsolete = Optional.extract(remove_obsolete) | |
213 | added, removed = repo2db_mapper(ScmModel().repo_scan(), |
|
213 | added, removed = repo2db_mapper(ScmModel().repo_scan(), | |
214 | remove_obsolete=rm_obsolete) |
|
214 | remove_obsolete=rm_obsolete) | |
215 | return {'added': added, 'removed': removed} |
|
215 | return {'added': added, 'removed': removed} | |
216 | except Exception: |
|
216 | except Exception: | |
217 | log.exception('Failed to run repo rescann') |
|
217 | log.exception('Failed to run repo rescann') | |
218 | raise JSONRPCError( |
|
218 | raise JSONRPCError( | |
219 | 'Error occurred during rescan repositories action' |
|
219 | 'Error occurred during rescan repositories action' | |
220 | ) |
|
220 | ) | |
221 |
|
221 | |||
222 |
|
222 | |||
223 | @jsonrpc_method() |
|
223 | @jsonrpc_method() | |
224 | def cleanup_sessions(request, apiuser, older_then=Optional(60)): |
|
224 | def cleanup_sessions(request, apiuser, older_then=Optional(60)): | |
225 | """ |
|
225 | """ | |
226 | Triggers a session cleanup action. |
|
226 | Triggers a session cleanup action. | |
227 |
|
227 | |||
228 | If the ``older_then`` option is set, only sessions that hasn't been |
|
228 | If the ``older_then`` option is set, only sessions that hasn't been | |
229 | accessed in the given number of days will be removed. |
|
229 | accessed in the given number of days will be removed. | |
230 |
|
230 | |||
231 | This command can only be run using an |authtoken| with admin rights to |
|
231 | This command can only be run using an |authtoken| with admin rights to | |
232 | the specified repository. |
|
232 | the specified repository. | |
233 |
|
233 | |||
234 | This command takes the following options: |
|
234 | This command takes the following options: | |
235 |
|
235 | |||
236 | :param apiuser: This is filled automatically from the |authtoken|. |
|
236 | :param apiuser: This is filled automatically from the |authtoken|. | |
237 | :type apiuser: AuthUser |
|
237 | :type apiuser: AuthUser | |
238 | :param older_then: Deletes session that hasn't been accessed |
|
238 | :param older_then: Deletes session that hasn't been accessed | |
239 | in given number of days. |
|
239 | in given number of days. | |
240 | :type older_then: Optional(int) |
|
240 | :type older_then: Optional(int) | |
241 |
|
241 | |||
242 | Example output: |
|
242 | Example output: | |
243 |
|
243 | |||
244 | .. code-block:: bash |
|
244 | .. code-block:: bash | |
245 |
|
245 | |||
246 | id : <id_given_in_input> |
|
246 | id : <id_given_in_input> | |
247 | result: { |
|
247 | result: { | |
248 | "backend": "<type of backend>", |
|
248 | "backend": "<type of backend>", | |
249 | "sessions_removed": <number_of_removed_sessions> |
|
249 | "sessions_removed": <number_of_removed_sessions> | |
250 | } |
|
250 | } | |
251 | error : null |
|
251 | error : null | |
252 |
|
252 | |||
253 | Example error output: |
|
253 | Example error output: | |
254 |
|
254 | |||
255 | .. code-block:: bash |
|
255 | .. code-block:: bash | |
256 |
|
256 | |||
257 | id : <id_given_in_input> |
|
257 | id : <id_given_in_input> | |
258 | result : null |
|
258 | result : null | |
259 | error : { |
|
259 | error : { | |
260 | 'Error occurred during session cleanup' |
|
260 | 'Error occurred during session cleanup' | |
261 | } |
|
261 | } | |
262 |
|
262 | |||
263 | """ |
|
263 | """ | |
264 | if not has_superadmin_permission(apiuser): |
|
264 | if not has_superadmin_permission(apiuser): | |
265 | raise JSONRPCForbidden() |
|
265 | raise JSONRPCForbidden() | |
266 |
|
266 | |||
267 | older_then = safe_int(Optional.extract(older_then)) or 60 |
|
267 | older_then = safe_int(Optional.extract(older_then)) or 60 | |
268 | older_than_seconds = 60 * 60 * 24 * older_then |
|
268 | older_than_seconds = 60 * 60 * 24 * older_then | |
269 |
|
269 | |||
270 | config = system_info.rhodecode_config().get_value()['value']['config'] |
|
270 | config = system_info.rhodecode_config().get_value()['value']['config'] | |
271 | session_model = user_sessions.get_session_handler( |
|
271 | session_model = user_sessions.get_session_handler( | |
272 | config.get('beaker.session.type', 'memory'))(config) |
|
272 | config.get('beaker.session.type', 'memory'))(config) | |
273 |
|
273 | |||
274 | backend = session_model.SESSION_TYPE |
|
274 | backend = session_model.SESSION_TYPE | |
275 | try: |
|
275 | try: | |
276 | cleaned = session_model.clean_sessions( |
|
276 | cleaned = session_model.clean_sessions( | |
277 | older_than_seconds=older_than_seconds) |
|
277 | older_than_seconds=older_than_seconds) | |
278 | return {'sessions_removed': cleaned, 'backend': backend} |
|
278 | return {'sessions_removed': cleaned, 'backend': backend} | |
279 | except user_sessions.CleanupCommand as msg: |
|
279 | except user_sessions.CleanupCommand as msg: | |
280 | return {'cleanup_command': msg.message, 'backend': backend} |
|
280 | return {'cleanup_command': msg.message, 'backend': backend} | |
281 | except Exception as e: |
|
281 | except Exception as e: | |
282 | log.exception('Failed session cleanup') |
|
282 | log.exception('Failed session cleanup') | |
283 | raise JSONRPCError( |
|
283 | raise JSONRPCError( | |
284 | 'Error occurred during session cleanup' |
|
284 | 'Error occurred during session cleanup' | |
285 | ) |
|
285 | ) | |
286 |
|
286 | |||
287 |
|
287 | |||
288 | @jsonrpc_method() |
|
288 | @jsonrpc_method() | |
289 | def get_method(request, apiuser, pattern=Optional('*')): |
|
289 | def get_method(request, apiuser, pattern=Optional('*')): | |
290 | """ |
|
290 | """ | |
291 | Returns list of all available API methods. By default match pattern |
|
291 | Returns list of all available API methods. By default match pattern | |
292 | os "*" but any other pattern can be specified. eg *comment* will return |
|
292 | os "*" but any other pattern can be specified. eg *comment* will return | |
293 | all methods with comment inside them. If just single method is matched |
|
293 | all methods with comment inside them. If just single method is matched | |
294 | returned data will also include method specification |
|
294 | returned data will also include method specification | |
295 |
|
295 | |||
296 | This command can only be run using an |authtoken| with admin rights to |
|
296 | This command can only be run using an |authtoken| with admin rights to | |
297 | the specified repository. |
|
297 | the specified repository. | |
298 |
|
298 | |||
299 | This command takes the following options: |
|
299 | This command takes the following options: | |
300 |
|
300 | |||
301 | :param apiuser: This is filled automatically from the |authtoken|. |
|
301 | :param apiuser: This is filled automatically from the |authtoken|. | |
302 | :type apiuser: AuthUser |
|
302 | :type apiuser: AuthUser | |
303 | :param pattern: pattern to match method names against |
|
303 | :param pattern: pattern to match method names against | |
304 | :type pattern: Optional("*") |
|
304 | :type pattern: Optional("*") | |
305 |
|
305 | |||
306 | Example output: |
|
306 | Example output: | |
307 |
|
307 | |||
308 | .. code-block:: bash |
|
308 | .. code-block:: bash | |
309 |
|
309 | |||
310 | id : <id_given_in_input> |
|
310 | id : <id_given_in_input> | |
311 | "result": [ |
|
311 | "result": [ | |
312 | "changeset_comment", |
|
312 | "changeset_comment", | |
313 | "comment_pull_request", |
|
313 | "comment_pull_request", | |
314 | "comment_commit" |
|
314 | "comment_commit" | |
315 | ] |
|
315 | ] | |
316 | error : null |
|
316 | error : null | |
317 |
|
317 | |||
318 | .. code-block:: bash |
|
318 | .. code-block:: bash | |
319 |
|
319 | |||
320 | id : <id_given_in_input> |
|
320 | id : <id_given_in_input> | |
321 | "result": [ |
|
321 | "result": [ | |
322 | "comment_commit", |
|
322 | "comment_commit", | |
323 | { |
|
323 | { | |
324 | "apiuser": "<RequiredType>", |
|
324 | "apiuser": "<RequiredType>", | |
325 | "comment_type": "<Optional:u'note'>", |
|
325 | "comment_type": "<Optional:u'note'>", | |
326 | "commit_id": "<RequiredType>", |
|
326 | "commit_id": "<RequiredType>", | |
327 | "message": "<RequiredType>", |
|
327 | "message": "<RequiredType>", | |
328 | "repoid": "<RequiredType>", |
|
328 | "repoid": "<RequiredType>", | |
329 | "request": "<RequiredType>", |
|
329 | "request": "<RequiredType>", | |
330 | "resolves_comment_id": "<Optional:None>", |
|
330 | "resolves_comment_id": "<Optional:None>", | |
331 | "status": "<Optional:None>", |
|
331 | "status": "<Optional:None>", | |
332 | "userid": "<Optional:<OptionalAttr:apiuser>>" |
|
332 | "userid": "<Optional:<OptionalAttr:apiuser>>" | |
333 | } |
|
333 | } | |
334 | ] |
|
334 | ] | |
335 | error : null |
|
335 | error : null | |
336 | """ |
|
336 | """ | |
337 | if not has_superadmin_permission(apiuser): |
|
337 | if not has_superadmin_permission(apiuser): | |
338 | raise JSONRPCForbidden() |
|
338 | raise JSONRPCForbidden() | |
339 |
|
339 | |||
340 | pattern = Optional.extract(pattern) |
|
340 | pattern = Optional.extract(pattern) | |
341 |
|
341 | |||
342 | matches = find_methods(request.registry.jsonrpc_methods, pattern) |
|
342 | matches = find_methods(request.registry.jsonrpc_methods, pattern) | |
343 |
|
343 | |||
344 | args_desc = [] |
|
344 | args_desc = [] | |
345 | if len(matches) == 1: |
|
345 | if len(matches) == 1: | |
346 | func = matches[matches.keys()[0]] |
|
346 | func = matches[matches.keys()[0]] | |
347 |
|
347 | |||
348 | argspec = inspect.getargspec(func) |
|
348 | argspec = inspect.getargspec(func) | |
349 | arglist = argspec[0] |
|
349 | arglist = argspec[0] | |
350 | defaults = map(repr, argspec[3] or []) |
|
350 | defaults = map(repr, argspec[3] or []) | |
351 |
|
351 | |||
352 | default_empty = '<RequiredType>' |
|
352 | default_empty = '<RequiredType>' | |
353 |
|
353 | |||
354 | # kw arguments required by this method |
|
354 | # kw arguments required by this method | |
355 | func_kwargs = dict(itertools.izip_longest( |
|
355 | func_kwargs = dict(itertools.izip_longest( | |
356 | reversed(arglist), reversed(defaults), fillvalue=default_empty)) |
|
356 | reversed(arglist), reversed(defaults), fillvalue=default_empty)) | |
357 | args_desc.append(func_kwargs) |
|
357 | args_desc.append(func_kwargs) | |
358 |
|
358 | |||
359 | return matches.keys() + args_desc |
|
359 | return matches.keys() + args_desc | |
360 |
|
360 | |||
361 |
|
361 | |||
362 | @jsonrpc_method() |
|
362 | @jsonrpc_method() | |
363 | def store_exception(request, apiuser, exc_data_json, prefix=Optional('rhodecode')): |
|
363 | def store_exception(request, apiuser, exc_data_json, prefix=Optional('rhodecode')): | |
364 | """ |
|
364 | """ | |
365 | Stores sent exception inside the built-in exception tracker in |RCE| server. |
|
365 | Stores sent exception inside the built-in exception tracker in |RCE| server. | |
366 |
|
366 | |||
367 | This command can only be run using an |authtoken| with admin rights to |
|
367 | This command can only be run using an |authtoken| with admin rights to | |
368 | the specified repository. |
|
368 | the specified repository. | |
369 |
|
369 | |||
370 | This command takes the following options: |
|
370 | This command takes the following options: | |
371 |
|
371 | |||
372 | :param apiuser: This is filled automatically from the |authtoken|. |
|
372 | :param apiuser: This is filled automatically from the |authtoken|. | |
373 | :type apiuser: AuthUser |
|
373 | :type apiuser: AuthUser | |
374 |
|
374 | |||
375 | :param exc_data_json: JSON data with exception e.g |
|
375 | :param exc_data_json: JSON data with exception e.g | |
376 | {"exc_traceback": "Value `1` is not allowed", "exc_type_name": "ValueError"} |
|
376 | {"exc_traceback": "Value `1` is not allowed", "exc_type_name": "ValueError"} | |
377 | :type exc_data_json: JSON data |
|
377 | :type exc_data_json: JSON data | |
378 |
|
378 | |||
379 | :param prefix: prefix for error type, e.g 'rhodecode', 'vcsserver', 'rhodecode-tools' |
|
379 | :param prefix: prefix for error type, e.g 'rhodecode', 'vcsserver', 'rhodecode-tools' | |
380 | :type prefix: Optional("rhodecode") |
|
380 | :type prefix: Optional("rhodecode") | |
381 |
|
381 | |||
382 | Example output: |
|
382 | Example output: | |
383 |
|
383 | |||
384 | .. code-block:: bash |
|
384 | .. code-block:: bash | |
385 |
|
385 | |||
386 | id : <id_given_in_input> |
|
386 | id : <id_given_in_input> | |
387 | "result": { |
|
387 | "result": { | |
388 | "exc_id": 139718459226384, |
|
388 | "exc_id": 139718459226384, | |
389 | "exc_url": "http://localhost:8080/_admin/settings/exceptions/139718459226384" |
|
389 | "exc_url": "http://localhost:8080/_admin/settings/exceptions/139718459226384" | |
390 | } |
|
390 | } | |
391 | error : null |
|
391 | error : null | |
392 | """ |
|
392 | """ | |
393 | if not has_superadmin_permission(apiuser): |
|
393 | if not has_superadmin_permission(apiuser): | |
394 | raise JSONRPCForbidden() |
|
394 | raise JSONRPCForbidden() | |
395 |
|
395 | |||
396 | prefix = Optional.extract(prefix) |
|
396 | prefix = Optional.extract(prefix) | |
397 | exc_id = exc_tracking.generate_id() |
|
397 | exc_id = exc_tracking.generate_id() | |
398 |
|
398 | |||
399 | try: |
|
399 | try: | |
400 | exc_data = json.loads(exc_data_json) |
|
400 | exc_data = json.loads(exc_data_json) | |
401 | except Exception: |
|
401 | except Exception: | |
402 | log.error('Failed to parse JSON: %r', exc_data_json) |
|
402 | log.error('Failed to parse JSON: %r', exc_data_json) | |
403 | raise JSONRPCError('Failed to parse JSON data from exc_data_json field. ' |
|
403 | raise JSONRPCError('Failed to parse JSON data from exc_data_json field. ' | |
404 | 'Please make sure it contains a valid JSON.') |
|
404 | 'Please make sure it contains a valid JSON.') | |
405 |
|
405 | |||
406 | try: |
|
406 | try: | |
407 | exc_traceback = exc_data['exc_traceback'] |
|
407 | exc_traceback = exc_data['exc_traceback'] | |
408 | exc_type_name = exc_data['exc_type_name'] |
|
408 | exc_type_name = exc_data['exc_type_name'] | |
409 | except KeyError as err: |
|
409 | except KeyError as err: | |
410 | raise JSONRPCError('Missing exc_traceback, or exc_type_name ' |
|
410 | raise JSONRPCError('Missing exc_traceback, or exc_type_name ' | |
411 | 'in exc_data_json field. Missing: {}'.format(err)) |
|
411 | 'in exc_data_json field. Missing: {}'.format(err)) | |
412 |
|
412 | |||
413 | exc_tracking._store_exception( |
|
413 | exc_tracking._store_exception( | |
414 | exc_id=exc_id, exc_traceback=exc_traceback, |
|
414 | exc_id=exc_id, exc_traceback=exc_traceback, | |
415 | exc_type_name=exc_type_name, prefix=prefix) |
|
415 | exc_type_name=exc_type_name, prefix=prefix) | |
416 |
|
416 | |||
417 | exc_url = request.route_url( |
|
417 | exc_url = request.route_url( | |
418 | 'admin_settings_exception_tracker_show', exception_id=exc_id) |
|
418 | 'admin_settings_exception_tracker_show', exception_id=exc_id) | |
419 | return {'exc_id': exc_id, 'exc_url': exc_url} |
|
419 | return {'exc_id': exc_id, 'exc_url': exc_url} | |
420 |
|
420 | |||
421 |
|
421 | |||
422 | @jsonrpc_method() |
|
422 | @jsonrpc_method() | |
423 | def upload_file(request, apiuser, filename, content): |
|
423 | def upload_file(request, apiuser, filename, content): | |
424 | """ |
|
424 | """ | |
425 | Upload API for the file_store |
|
425 | Upload API for the file_store | |
426 |
|
426 | |||
427 | Example usage from CLI:: |
|
427 | Example usage from CLI:: | |
428 | rhodecode-api --instance-name=enterprise-1 upload_file "{\"content\": \"$(cat image.jpg | base64)\", \"filename\":\"image.jpg\"}" |
|
428 | rhodecode-api --instance-name=enterprise-1 upload_file "{\"content\": \"$(cat image.jpg | base64)\", \"filename\":\"image.jpg\"}" | |
429 |
|
429 | |||
430 |
|
430 | |||
431 | This command can only be run using an |authtoken| with admin rights to |
|
431 | This command can only be run using an |authtoken| with admin rights to | |
432 | the specified repository. |
|
432 | the specified repository. | |
433 |
|
433 | |||
434 | This command takes the following options: |
|
434 | This command takes the following options: | |
435 |
|
435 | |||
436 | :param apiuser: This is filled automatically from the |authtoken|. |
|
436 | :param apiuser: This is filled automatically from the |authtoken|. | |
437 | :type apiuser: AuthUser |
|
437 | :type apiuser: AuthUser | |
438 | :param filename: name of the file uploaded |
|
438 | :param filename: name of the file uploaded | |
439 | :type filename: str |
|
439 | :type filename: str | |
440 | :param content: base64 encoded content of the uploaded file |
|
440 | :param content: base64 encoded content of the uploaded file | |
441 | :type prefix: str |
|
441 | :type prefix: str | |
442 |
|
442 | |||
443 | Example output: |
|
443 | Example output: | |
444 |
|
444 | |||
445 | .. code-block:: bash |
|
445 | .. code-block:: bash | |
446 |
|
446 | |||
447 | id : <id_given_in_input> |
|
447 | id : <id_given_in_input> | |
448 | result: { |
|
448 | result: { | |
449 | "access_path": "/_file_store/download/84d156f7-8323-4ad3-9fce-4a8e88e1deaf-0.jpg", |
|
449 | "access_path": "/_file_store/download/84d156f7-8323-4ad3-9fce-4a8e88e1deaf-0.jpg", | |
450 | "access_path_fqn": "http://server.domain.com/_file_store/download/84d156f7-8323-4ad3-9fce-4a8e88e1deaf-0.jpg", |
|
450 | "access_path_fqn": "http://server.domain.com/_file_store/download/84d156f7-8323-4ad3-9fce-4a8e88e1deaf-0.jpg", | |
451 | "store_fid": "84d156f7-8323-4ad3-9fce-4a8e88e1deaf-0.jpg" |
|
451 | "store_fid": "84d156f7-8323-4ad3-9fce-4a8e88e1deaf-0.jpg" | |
452 | } |
|
452 | } | |
453 | error : null |
|
453 | error : null | |
454 | """ |
|
454 | """ | |
455 | if not has_superadmin_permission(apiuser): |
|
455 | if not has_superadmin_permission(apiuser): | |
456 | raise JSONRPCForbidden() |
|
456 | raise JSONRPCForbidden() | |
457 |
|
457 | |||
458 | storage = utils.get_file_storage(request.registry.settings) |
|
458 | storage = utils.get_file_storage(request.registry.settings) | |
459 |
|
459 | |||
460 | try: |
|
460 | try: | |
461 | file_obj = compat.NativeIO(base64.decodestring(content)) |
|
461 | file_obj = compat.NativeIO(base64.decodestring(content)) | |
462 | except Exception as exc: |
|
462 | except Exception as exc: | |
463 | raise JSONRPCError('File `{}` content decoding error: {}.'.format(filename, exc)) |
|
463 | raise JSONRPCError('File `{}` content decoding error: {}.'.format(filename, exc)) | |
464 |
|
464 | |||
465 | metadata = { |
|
465 | metadata = { | |
466 | 'filename': filename, |
|
466 | 'filename': filename, | |
467 | 'size': '', # filled by save_file |
|
467 | 'size': '', # filled by save_file | |
468 | 'user_uploaded': {'username': apiuser.username, |
|
468 | 'user_uploaded': {'username': apiuser.username, | |
469 | 'user_id': apiuser.user_id, |
|
469 | 'user_id': apiuser.user_id, | |
470 | 'ip': apiuser.ip_addr}} |
|
470 | 'ip': apiuser.ip_addr}} | |
471 | try: |
|
471 | try: | |
472 | store_fid = storage.save_file(file_obj, filename, metadata=metadata) |
|
472 | store_fid, metadata = storage.save_file(file_obj, filename, metadata=metadata) | |
473 | except FileNotAllowedException: |
|
473 | except FileNotAllowedException: | |
474 | raise JSONRPCError('File `{}` is not allowed.'.format(filename)) |
|
474 | raise JSONRPCError('File `{}` is not allowed.'.format(filename)) | |
475 |
|
475 | |||
476 | except FileOverSizeException: |
|
476 | except FileOverSizeException: | |
477 | raise JSONRPCError('File `{}` is exceeding allowed limit.'.format(filename)) |
|
477 | raise JSONRPCError('File `{}` is exceeding allowed limit.'.format(filename)) | |
478 |
|
478 | |||
479 | return {'store_fid': store_fid, |
|
479 | return {'store_fid': store_fid, | |
480 | 'access_path_fqn': request.route_url('download_file', fid=store_fid), |
|
480 | 'access_path_fqn': request.route_url('download_file', fid=store_fid), | |
481 | 'access_path': request.route_path('download_file', fid=store_fid)} |
|
481 | 'access_path': request.route_path('download_file', fid=store_fid)} |
@@ -1,49 +1,49 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | # Copyright (C) 2016-2019 RhodeCode GmbH |
|
3 | # Copyright (C) 2016-2019 RhodeCode GmbH | |
4 | # |
|
4 | # | |
5 | # This program is free software: you can redistribute it and/or modify |
|
5 | # This program is free software: you can redistribute it and/or modify | |
6 | # it under the terms of the GNU Affero General Public License, version 3 |
|
6 | # it under the terms of the GNU Affero General Public License, version 3 | |
7 | # (only), as published by the Free Software Foundation. |
|
7 | # (only), as published by the Free Software Foundation. | |
8 | # |
|
8 | # | |
9 | # This program is distributed in the hope that it will be useful, |
|
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. |
|
12 | # GNU General Public License for more details. | |
13 | # |
|
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License |
|
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | # |
|
16 | # | |
17 | # This program is dual-licensed. If you wish to learn more about the |
|
17 | # This program is dual-licensed. If you wish to learn more about the | |
18 | # RhodeCode Enterprise Edition, including its added features, Support services, |
|
18 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ |
|
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
20 | import os |
|
20 | import os | |
21 |
from rhodecode.apps. |
|
21 | from rhodecode.apps.file_store import config_keys | |
22 | from rhodecode.config.middleware import _bool_setting, _string_setting |
|
22 | from rhodecode.config.middleware import _bool_setting, _string_setting | |
23 |
|
23 | |||
24 |
|
24 | |||
25 | def _sanitize_settings_and_apply_defaults(settings): |
|
25 | def _sanitize_settings_and_apply_defaults(settings): | |
26 | """ |
|
26 | """ | |
27 | Set defaults, convert to python types and validate settings. |
|
27 | Set defaults, convert to python types and validate settings. | |
28 | """ |
|
28 | """ | |
29 | _bool_setting(settings, config_keys.enabled, 'true') |
|
29 | _bool_setting(settings, config_keys.enabled, 'true') | |
30 |
|
30 | |||
31 | _string_setting(settings, config_keys.backend, 'local') |
|
31 | _string_setting(settings, config_keys.backend, 'local') | |
32 |
|
32 | |||
33 | default_store = os.path.join(os.path.dirname(settings['__file__']), 'upload_store') |
|
33 | default_store = os.path.join(os.path.dirname(settings['__file__']), 'upload_store') | |
34 | _string_setting(settings, config_keys.store_path, default_store) |
|
34 | _string_setting(settings, config_keys.store_path, default_store) | |
35 |
|
35 | |||
36 |
|
36 | |||
37 | def includeme(config): |
|
37 | def includeme(config): | |
38 | settings = config.registry.settings |
|
38 | settings = config.registry.settings | |
39 | _sanitize_settings_and_apply_defaults(settings) |
|
39 | _sanitize_settings_and_apply_defaults(settings) | |
40 |
|
40 | |||
41 | config.add_route( |
|
41 | config.add_route( | |
42 | name='upload_file', |
|
42 | name='upload_file', | |
43 | pattern='/_file_store/upload') |
|
43 | pattern='/_file_store/upload') | |
44 | config.add_route( |
|
44 | config.add_route( | |
45 | name='download_file', |
|
45 | name='download_file', | |
46 | pattern='/_file_store/download/{fid}') |
|
46 | pattern='/_file_store/download/{fid}') | |
47 |
|
47 | |||
48 | # Scan module for configuration decorators. |
|
48 | # Scan module for configuration decorators. | |
49 | config.scan('.views', ignore='.tests') |
|
49 | config.scan('.views', ignore='.tests') |
1 | NO CONTENT: file renamed from rhodecode/apps/upload_store/config_keys.py to rhodecode/apps/file_store/config_keys.py |
|
NO CONTENT: file renamed from rhodecode/apps/upload_store/config_keys.py to rhodecode/apps/file_store/config_keys.py |
1 | NO CONTENT: file renamed from rhodecode/apps/upload_store/exceptions.py to rhodecode/apps/file_store/exceptions.py |
|
NO CONTENT: file renamed from rhodecode/apps/upload_store/exceptions.py to rhodecode/apps/file_store/exceptions.py |
1 | NO CONTENT: file renamed from rhodecode/apps/upload_store/extensions.py to rhodecode/apps/file_store/extensions.py |
|
NO CONTENT: file renamed from rhodecode/apps/upload_store/extensions.py to rhodecode/apps/file_store/extensions.py |
@@ -1,171 +1,175 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | # Copyright (C) 2016-2019 RhodeCode GmbH |
|
3 | # Copyright (C) 2016-2019 RhodeCode GmbH | |
4 | # |
|
4 | # | |
5 | # This program is free software: you can redistribute it and/or modify |
|
5 | # This program is free software: you can redistribute it and/or modify | |
6 | # it under the terms of the GNU Affero General Public License, version 3 |
|
6 | # it under the terms of the GNU Affero General Public License, version 3 | |
7 | # (only), as published by the Free Software Foundation. |
|
7 | # (only), as published by the Free Software Foundation. | |
8 | # |
|
8 | # | |
9 | # This program is distributed in the hope that it will be useful, |
|
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. |
|
12 | # GNU General Public License for more details. | |
13 | # |
|
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License |
|
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | # |
|
16 | # | |
17 | # This program is dual-licensed. If you wish to learn more about the |
|
17 | # This program is dual-licensed. If you wish to learn more about the | |
18 | # RhodeCode Enterprise Edition, including its added features, Support services, |
|
18 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ |
|
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
20 |
|
20 | |||
21 | import os |
|
21 | import os | |
22 | import time |
|
22 | import time | |
23 | import shutil |
|
23 | import shutil | |
24 |
|
24 | |||
25 | from rhodecode.lib.ext_json import json |
|
25 | from rhodecode.lib.ext_json import json | |
26 |
from rhodecode.apps. |
|
26 | from rhodecode.apps.file_store import utils | |
27 |
from rhodecode.apps. |
|
27 | from rhodecode.apps.file_store.extensions import resolve_extensions | |
28 |
from rhodecode.apps. |
|
28 | from rhodecode.apps.file_store.exceptions import FileNotAllowedException | |
29 |
|
29 | |||
30 | METADATA_VER = 'v1' |
|
30 | METADATA_VER = 'v1' | |
31 |
|
31 | |||
32 |
|
32 | |||
33 | class LocalFileStorage(object): |
|
33 | class LocalFileStorage(object): | |
34 |
|
34 | |||
35 | @classmethod |
|
35 | @classmethod | |
36 | def resolve_name(cls, name, directory): |
|
36 | def resolve_name(cls, name, directory): | |
37 | """ |
|
37 | """ | |
38 | Resolves a unique name and the correct path. If a filename |
|
38 | Resolves a unique name and the correct path. If a filename | |
39 | for that path already exists then a numeric prefix with values > 0 will be |
|
39 | for that path already exists then a numeric prefix with values > 0 will be | |
40 | added, for example test.jpg -> test-1.jpg etc. initially file would have 0 prefix. |
|
40 | added, for example test.jpg -> test-1.jpg etc. initially file would have 0 prefix. | |
41 |
|
41 | |||
42 | :param name: base name of file |
|
42 | :param name: base name of file | |
43 | :param directory: absolute directory path |
|
43 | :param directory: absolute directory path | |
44 | """ |
|
44 | """ | |
45 |
|
45 | |||
46 | basename, ext = os.path.splitext(name) |
|
46 | basename, ext = os.path.splitext(name) | |
47 | counter = 0 |
|
47 | counter = 0 | |
48 | while True: |
|
48 | while True: | |
49 | name = '%s-%d%s' % (basename, counter, ext) |
|
49 | name = '%s-%d%s' % (basename, counter, ext) | |
50 | path = os.path.join(directory, name) |
|
50 | path = os.path.join(directory, name) | |
51 | if not os.path.exists(path): |
|
51 | if not os.path.exists(path): | |
52 | return name, path |
|
52 | return name, path | |
53 | counter += 1 |
|
53 | counter += 1 | |
54 |
|
54 | |||
55 | def __init__(self, base_path, extension_groups=None): |
|
55 | def __init__(self, base_path, extension_groups=None): | |
56 |
|
56 | |||
57 | """ |
|
57 | """ | |
58 | Local file storage |
|
58 | Local file storage | |
59 |
|
59 | |||
60 | :param base_path: the absolute base path where uploads are stored |
|
60 | :param base_path: the absolute base path where uploads are stored | |
61 | :param extension_groups: extensions string |
|
61 | :param extension_groups: extensions string | |
62 | """ |
|
62 | """ | |
63 |
|
63 | |||
64 | extension_groups = extension_groups or ['any'] |
|
64 | extension_groups = extension_groups or ['any'] | |
65 | self.base_path = base_path |
|
65 | self.base_path = base_path | |
66 | self.extensions = resolve_extensions([], groups=extension_groups) |
|
66 | self.extensions = resolve_extensions([], groups=extension_groups) | |
67 |
|
67 | |||
68 | def store_path(self, filename): |
|
68 | def store_path(self, filename): | |
69 | """ |
|
69 | """ | |
70 | Returns absolute file path of the filename, joined to the |
|
70 | Returns absolute file path of the filename, joined to the | |
71 | base_path. |
|
71 | base_path. | |
72 |
|
72 | |||
73 | :param filename: base name of file |
|
73 | :param filename: base name of file | |
74 | """ |
|
74 | """ | |
75 | return os.path.join(self.base_path, filename) |
|
75 | return os.path.join(self.base_path, filename) | |
76 |
|
76 | |||
77 | def delete(self, filename): |
|
77 | def delete(self, filename): | |
78 | """ |
|
78 | """ | |
79 | Deletes the filename. Filename is resolved with the |
|
79 | Deletes the filename. Filename is resolved with the | |
80 | absolute path based on base_path. If file does not exist, |
|
80 | absolute path based on base_path. If file does not exist, | |
81 | returns **False**, otherwise **True** |
|
81 | returns **False**, otherwise **True** | |
82 |
|
82 | |||
83 | :param filename: base name of file |
|
83 | :param filename: base name of file | |
84 | """ |
|
84 | """ | |
85 | if self.exists(filename): |
|
85 | if self.exists(filename): | |
86 | os.remove(self.store_path(filename)) |
|
86 | os.remove(self.store_path(filename)) | |
87 | return True |
|
87 | return True | |
88 | return False |
|
88 | return False | |
89 |
|
89 | |||
90 | def exists(self, filename): |
|
90 | def exists(self, filename): | |
91 | """ |
|
91 | """ | |
92 | Checks if file exists. Resolves filename's absolute |
|
92 | Checks if file exists. Resolves filename's absolute | |
93 | path based on base_path. |
|
93 | path based on base_path. | |
94 |
|
94 | |||
95 | :param filename: base name of file |
|
95 | :param filename: base name of file | |
96 | """ |
|
96 | """ | |
97 | return os.path.exists(self.store_path(filename)) |
|
97 | return os.path.exists(self.store_path(filename)) | |
98 |
|
98 | |||
99 | def filename_allowed(self, filename, extensions=None): |
|
99 | def filename_allowed(self, filename, extensions=None): | |
100 | """Checks if a filename has an allowed extension |
|
100 | """Checks if a filename has an allowed extension | |
101 |
|
101 | |||
102 | :param filename: base name of file |
|
102 | :param filename: base name of file | |
103 | :param extensions: iterable of extensions (or self.extensions) |
|
103 | :param extensions: iterable of extensions (or self.extensions) | |
104 | """ |
|
104 | """ | |
105 | _, ext = os.path.splitext(filename) |
|
105 | _, ext = os.path.splitext(filename) | |
106 | return self.extension_allowed(ext, extensions) |
|
106 | return self.extension_allowed(ext, extensions) | |
107 |
|
107 | |||
108 | def extension_allowed(self, ext, extensions=None): |
|
108 | def extension_allowed(self, ext, extensions=None): | |
109 | """ |
|
109 | """ | |
110 | Checks if an extension is permitted. Both e.g. ".jpg" and |
|
110 | Checks if an extension is permitted. Both e.g. ".jpg" and | |
111 | "jpg" can be passed in. Extension lookup is case-insensitive. |
|
111 | "jpg" can be passed in. Extension lookup is case-insensitive. | |
112 |
|
112 | |||
113 |
:param ext |
|
113 | :param ext: extension to check | |
|
114 | :param extensions: iterable of extensions to validate against (or self.extensions) | |||
114 | """ |
|
115 | """ | |
115 |
|
116 | |||
116 | extensions = extensions or self.extensions |
|
117 | extensions = extensions or self.extensions | |
117 | if not extensions: |
|
118 | if not extensions: | |
118 | return True |
|
119 | return True | |
119 | if ext.startswith('.'): |
|
120 | if ext.startswith('.'): | |
120 | ext = ext[1:] |
|
121 | ext = ext[1:] | |
121 | return ext.lower() in extensions |
|
122 | return ext.lower() in extensions | |
122 |
|
123 | |||
123 | def save_file(self, file_obj, filename, directory=None, extensions=None, |
|
124 | def save_file(self, file_obj, filename, directory=None, extensions=None, | |
124 | metadata=None, **kwargs): |
|
125 | metadata=None, **kwargs): | |
125 | """ |
|
126 | """ | |
126 | Saves a file object to the uploads location. |
|
127 | Saves a file object to the uploads location. | |
127 | Returns the resolved filename, i.e. the directory + |
|
128 | Returns the resolved filename, i.e. the directory + | |
128 | the (randomized/incremented) base name. |
|
129 | the (randomized/incremented) base name. | |
129 |
|
130 | |||
130 | :param file_obj: **cgi.FieldStorage** object (or similar) |
|
131 | :param file_obj: **cgi.FieldStorage** object (or similar) | |
131 | :param filename: original filename |
|
132 | :param filename: original filename | |
132 | :param directory: relative path of sub-directory |
|
133 | :param directory: relative path of sub-directory | |
133 | :param extensions: iterable of allowed extensions, if not default |
|
134 | :param extensions: iterable of allowed extensions, if not default | |
134 | :param metadata: JSON metadata to store next to the file with .meta suffix |
|
135 | :param metadata: JSON metadata to store next to the file with .meta suffix | |
135 | :returns: modified filename |
|
136 | :returns: modified filename | |
136 | """ |
|
137 | """ | |
137 |
|
138 | |||
138 | extensions = extensions or self.extensions |
|
139 | extensions = extensions or self.extensions | |
139 |
|
140 | |||
140 | if not self.filename_allowed(filename, extensions): |
|
141 | if not self.filename_allowed(filename, extensions): | |
141 | raise FileNotAllowedException() |
|
142 | raise FileNotAllowedException() | |
142 |
|
143 | |||
143 | if directory: |
|
144 | if directory: | |
144 | dest_directory = os.path.join(self.base_path, directory) |
|
145 | dest_directory = os.path.join(self.base_path, directory) | |
145 | else: |
|
146 | else: | |
146 | dest_directory = self.base_path |
|
147 | dest_directory = self.base_path | |
147 |
|
148 | |||
148 | if not os.path.exists(dest_directory): |
|
149 | if not os.path.exists(dest_directory): | |
149 | os.makedirs(dest_directory) |
|
150 | os.makedirs(dest_directory) | |
150 |
|
151 | |||
151 | filename = utils.uid_filename(filename) |
|
152 | filename = utils.uid_filename(filename) | |
152 |
|
153 | |||
153 | filename, path = self.resolve_name(filename, dest_directory) |
|
154 | filename, path = self.resolve_name(filename, dest_directory) | |
154 | filename_meta = filename + '.meta' |
|
155 | filename_meta = filename + '.meta' | |
155 |
|
156 | |||
156 | file_obj.seek(0) |
|
157 | file_obj.seek(0) | |
157 |
|
158 | |||
158 | with open(path, "wb") as dest: |
|
159 | with open(path, "wb") as dest: | |
159 | shutil.copyfileobj(file_obj, dest) |
|
160 | shutil.copyfileobj(file_obj, dest) | |
160 |
|
161 | |||
161 | if metadata: |
|
162 | if metadata: | |
162 | size = os.stat(path).st_size |
|
163 | size = os.stat(path).st_size | |
163 |
metadata.update( |
|
164 | metadata.update( | |
164 | "meta_ver": METADATA_VER}) |
|
165 | {"size": size, | |
|
166 | "time": time.time(), | |||
|
167 | "meta_ver": METADATA_VER}) | |||
|
168 | ||||
165 | with open(os.path.join(dest_directory, filename_meta), "wb") as dest_meta: |
|
169 | with open(os.path.join(dest_directory, filename_meta), "wb") as dest_meta: | |
166 | dest_meta.write(json.dumps(metadata)) |
|
170 | dest_meta.write(json.dumps(metadata)) | |
167 |
|
171 | |||
168 | if directory: |
|
172 | if directory: | |
169 | filename = os.path.join(directory, filename) |
|
173 | filename = os.path.join(directory, filename) | |
170 |
|
174 | |||
171 | return filename |
|
175 | return filename, metadata |
1 | NO CONTENT: file renamed from rhodecode/apps/upload_store/tests/__init__.py to rhodecode/apps/file_store/tests/__init__.py |
|
NO CONTENT: file renamed from rhodecode/apps/upload_store/tests/__init__.py to rhodecode/apps/file_store/tests/__init__.py |
@@ -1,110 +1,110 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | # Copyright (C) 2010-2019 RhodeCode GmbH |
|
3 | # Copyright (C) 2010-2019 RhodeCode GmbH | |
4 | # |
|
4 | # | |
5 | # This program is free software: you can redistribute it and/or modify |
|
5 | # This program is free software: you can redistribute it and/or modify | |
6 | # it under the terms of the GNU Affero General Public License, version 3 |
|
6 | # it under the terms of the GNU Affero General Public License, version 3 | |
7 | # (only), as published by the Free Software Foundation. |
|
7 | # (only), as published by the Free Software Foundation. | |
8 | # |
|
8 | # | |
9 | # This program is distributed in the hope that it will be useful, |
|
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. |
|
12 | # GNU General Public License for more details. | |
13 | # |
|
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License |
|
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | # |
|
16 | # | |
17 | # This program is dual-licensed. If you wish to learn more about the |
|
17 | # This program is dual-licensed. If you wish to learn more about the | |
18 | # RhodeCode Enterprise Edition, including its added features, Support services, |
|
18 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ |
|
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
20 | import os |
|
20 | import os | |
21 | import pytest |
|
21 | import pytest | |
22 |
|
22 | |||
23 | from rhodecode.lib.ext_json import json |
|
23 | from rhodecode.lib.ext_json import json | |
24 | from rhodecode.tests import TestController |
|
24 | from rhodecode.tests import TestController | |
25 |
from rhodecode.apps. |
|
25 | from rhodecode.apps.file_store import utils, config_keys | |
26 |
|
26 | |||
27 |
|
27 | |||
28 | def route_path(name, params=None, **kwargs): |
|
28 | def route_path(name, params=None, **kwargs): | |
29 | import urllib |
|
29 | import urllib | |
30 |
|
30 | |||
31 | base_url = { |
|
31 | base_url = { | |
32 | 'upload_file': '/_file_store/upload', |
|
32 | 'upload_file': '/_file_store/upload', | |
33 | 'download_file': '/_file_store/download/{fid}', |
|
33 | 'download_file': '/_file_store/download/{fid}', | |
34 |
|
34 | |||
35 | }[name].format(**kwargs) |
|
35 | }[name].format(**kwargs) | |
36 |
|
36 | |||
37 | if params: |
|
37 | if params: | |
38 | base_url = '{}?{}'.format(base_url, urllib.urlencode(params)) |
|
38 | base_url = '{}?{}'.format(base_url, urllib.urlencode(params)) | |
39 | return base_url |
|
39 | return base_url | |
40 |
|
40 | |||
41 |
|
41 | |||
42 | class TestFileStoreViews(TestController): |
|
42 | class TestFileStoreViews(TestController): | |
43 |
|
43 | |||
44 | @pytest.mark.parametrize("fid, content, exists", [ |
|
44 | @pytest.mark.parametrize("fid, content, exists", [ | |
45 | ('abcde-0.jpg', "xxxxx", True), |
|
45 | ('abcde-0.jpg', "xxxxx", True), | |
46 | ('abcde-0.exe', "1234567", True), |
|
46 | ('abcde-0.exe', "1234567", True), | |
47 | ('abcde-0.jpg', "xxxxx", False), |
|
47 | ('abcde-0.jpg', "xxxxx", False), | |
48 | ]) |
|
48 | ]) | |
49 | def test_get_files_from_store(self, fid, content, exists, tmpdir): |
|
49 | def test_get_files_from_store(self, fid, content, exists, tmpdir): | |
50 | self.log_user() |
|
50 | self.log_user() | |
51 | store_path = self.app._pyramid_settings[config_keys.store_path] |
|
51 | store_path = self.app._pyramid_settings[config_keys.store_path] | |
52 |
|
52 | |||
53 | if exists: |
|
53 | if exists: | |
54 | status = 200 |
|
54 | status = 200 | |
55 | store = utils.get_file_storage({config_keys.store_path: store_path}) |
|
55 | store = utils.get_file_storage({config_keys.store_path: store_path}) | |
56 | filesystem_file = os.path.join(str(tmpdir), fid) |
|
56 | filesystem_file = os.path.join(str(tmpdir), fid) | |
57 | with open(filesystem_file, 'wb') as f: |
|
57 | with open(filesystem_file, 'wb') as f: | |
58 | f.write(content) |
|
58 | f.write(content) | |
59 |
|
59 | |||
60 | with open(filesystem_file, 'rb') as f: |
|
60 | with open(filesystem_file, 'rb') as f: | |
61 | fid = store.save_file(f, fid, metadata={'filename': fid}) |
|
61 | fid, metadata = store.save_file(f, fid, metadata={'filename': fid}) | |
62 |
|
62 | |||
63 | else: |
|
63 | else: | |
64 | status = 404 |
|
64 | status = 404 | |
65 |
|
65 | |||
66 | response = self.app.get(route_path('download_file', fid=fid), status=status) |
|
66 | response = self.app.get(route_path('download_file', fid=fid), status=status) | |
67 |
|
67 | |||
68 | if exists: |
|
68 | if exists: | |
69 | assert response.text == content |
|
69 | assert response.text == content | |
70 | metadata = os.path.join(store_path, fid + '.meta') |
|
70 | metadata = os.path.join(store_path, fid + '.meta') | |
71 | assert os.path.exists(metadata) |
|
71 | assert os.path.exists(metadata) | |
72 | with open(metadata, 'rb') as f: |
|
72 | with open(metadata, 'rb') as f: | |
73 | json_data = json.loads(f.read()) |
|
73 | json_data = json.loads(f.read()) | |
74 |
|
74 | |||
75 | assert json_data |
|
75 | assert json_data | |
76 | assert 'size' in json_data |
|
76 | assert 'size' in json_data | |
77 |
|
77 | |||
78 | def test_upload_files_without_content_to_store(self): |
|
78 | def test_upload_files_without_content_to_store(self): | |
79 | self.log_user() |
|
79 | self.log_user() | |
80 | response = self.app.post( |
|
80 | response = self.app.post( | |
81 | route_path('upload_file'), |
|
81 | route_path('upload_file'), | |
82 | params={'csrf_token': self.csrf_token}, |
|
82 | params={'csrf_token': self.csrf_token}, | |
83 | status=200) |
|
83 | status=200) | |
84 |
|
84 | |||
85 | assert response.json == { |
|
85 | assert response.json == { | |
86 | u'error': u'store_file data field is missing', |
|
86 | u'error': u'store_file data field is missing', | |
87 | u'access_path': None, |
|
87 | u'access_path': None, | |
88 | u'store_fid': None} |
|
88 | u'store_fid': None} | |
89 |
|
89 | |||
90 | def test_upload_files_bogus_content_to_store(self): |
|
90 | def test_upload_files_bogus_content_to_store(self): | |
91 | self.log_user() |
|
91 | self.log_user() | |
92 | response = self.app.post( |
|
92 | response = self.app.post( | |
93 | route_path('upload_file'), |
|
93 | route_path('upload_file'), | |
94 | params={'csrf_token': self.csrf_token, 'store_file': 'bogus'}, |
|
94 | params={'csrf_token': self.csrf_token, 'store_file': 'bogus'}, | |
95 | status=200) |
|
95 | status=200) | |
96 |
|
96 | |||
97 | assert response.json == { |
|
97 | assert response.json == { | |
98 | u'error': u'filename cannot be read from the data field', |
|
98 | u'error': u'filename cannot be read from the data field', | |
99 | u'access_path': None, |
|
99 | u'access_path': None, | |
100 | u'store_fid': None} |
|
100 | u'store_fid': None} | |
101 |
|
101 | |||
102 | def test_upload_content_to_store(self): |
|
102 | def test_upload_content_to_store(self): | |
103 | self.log_user() |
|
103 | self.log_user() | |
104 | response = self.app.post( |
|
104 | response = self.app.post( | |
105 | route_path('upload_file'), |
|
105 | route_path('upload_file'), | |
106 | upload_files=[('store_file', 'myfile.txt', 'SOME CONTENT')], |
|
106 | upload_files=[('store_file', 'myfile.txt', 'SOME CONTENT')], | |
107 | params={'csrf_token': self.csrf_token}, |
|
107 | params={'csrf_token': self.csrf_token}, | |
108 | status=200) |
|
108 | status=200) | |
109 |
|
109 | |||
110 | assert response.json['store_fid'] |
|
110 | assert response.json['store_fid'] |
@@ -1,47 +1,47 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | # Copyright (C) 2016-2019 RhodeCode GmbH |
|
3 | # Copyright (C) 2016-2019 RhodeCode GmbH | |
4 | # |
|
4 | # | |
5 | # This program is free software: you can redistribute it and/or modify |
|
5 | # This program is free software: you can redistribute it and/or modify | |
6 | # it under the terms of the GNU Affero General Public License, version 3 |
|
6 | # it under the terms of the GNU Affero General Public License, version 3 | |
7 | # (only), as published by the Free Software Foundation. |
|
7 | # (only), as published by the Free Software Foundation. | |
8 | # |
|
8 | # | |
9 | # This program is distributed in the hope that it will be useful, |
|
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. |
|
12 | # GNU General Public License for more details. | |
13 | # |
|
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License |
|
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | # |
|
16 | # | |
17 | # This program is dual-licensed. If you wish to learn more about the |
|
17 | # This program is dual-licensed. If you wish to learn more about the | |
18 | # RhodeCode Enterprise Edition, including its added features, Support services, |
|
18 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ |
|
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
20 |
|
20 | |||
21 |
|
21 | |||
22 | import os |
|
22 | import os | |
23 | import uuid |
|
23 | import uuid | |
24 |
|
24 | |||
25 |
|
25 | |||
26 | def get_file_storage(settings): |
|
26 | def get_file_storage(settings): | |
27 |
from rhodecode.apps. |
|
27 | from rhodecode.apps.file_store.local_store import LocalFileStorage | |
28 |
from rhodecode.apps. |
|
28 | from rhodecode.apps.file_store import config_keys | |
29 | store_path = settings.get(config_keys.store_path) |
|
29 | store_path = settings.get(config_keys.store_path) | |
30 | return LocalFileStorage(base_path=store_path) |
|
30 | return LocalFileStorage(base_path=store_path) | |
31 |
|
31 | |||
32 |
|
32 | |||
33 | def uid_filename(filename, randomized=True): |
|
33 | def uid_filename(filename, randomized=True): | |
34 | """ |
|
34 | """ | |
35 | Generates a randomized or stable (uuid) filename, |
|
35 | Generates a randomized or stable (uuid) filename, | |
36 | preserving the original extension. |
|
36 | preserving the original extension. | |
37 |
|
37 | |||
38 | :param filename: the original filename |
|
38 | :param filename: the original filename | |
39 | :param randomized: define if filename should be stable (sha1 based) or randomized |
|
39 | :param randomized: define if filename should be stable (sha1 based) or randomized | |
40 | """ |
|
40 | """ | |
41 | _, ext = os.path.splitext(filename) |
|
41 | _, ext = os.path.splitext(filename) | |
42 | if randomized: |
|
42 | if randomized: | |
43 | uid = uuid.uuid4() |
|
43 | uid = uuid.uuid4() | |
44 | else: |
|
44 | else: | |
45 | hash_key = '{}.{}'.format(filename, 'store') |
|
45 | hash_key = '{}.{}'.format(filename, 'store') | |
46 | uid = uuid.uuid5(uuid.NAMESPACE_URL, hash_key) |
|
46 | uid = uuid.uuid5(uuid.NAMESPACE_URL, hash_key) | |
47 | return str(uid) + ext.lower() |
|
47 | return str(uid) + ext.lower() |
@@ -1,97 +1,96 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | # Copyright (C) 2016-2019 RhodeCode GmbH |
|
3 | # Copyright (C) 2016-2019 RhodeCode GmbH | |
4 | # |
|
4 | # | |
5 | # This program is free software: you can redistribute it and/or modify |
|
5 | # This program is free software: you can redistribute it and/or modify | |
6 | # it under the terms of the GNU Affero General Public License, version 3 |
|
6 | # it under the terms of the GNU Affero General Public License, version 3 | |
7 | # (only), as published by the Free Software Foundation. |
|
7 | # (only), as published by the Free Software Foundation. | |
8 | # |
|
8 | # | |
9 | # This program is distributed in the hope that it will be useful, |
|
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. |
|
12 | # GNU General Public License for more details. | |
13 | # |
|
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License |
|
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | # |
|
16 | # | |
17 | # This program is dual-licensed. If you wish to learn more about the |
|
17 | # This program is dual-licensed. If you wish to learn more about the | |
18 | # RhodeCode Enterprise Edition, including its added features, Support services, |
|
18 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ |
|
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
20 | import logging |
|
20 | import logging | |
21 |
|
21 | |||
22 | from pyramid.view import view_config |
|
22 | from pyramid.view import view_config | |
23 | from pyramid.response import FileResponse |
|
23 | from pyramid.response import FileResponse | |
24 | from pyramid.httpexceptions import HTTPFound, HTTPNotFound |
|
24 | from pyramid.httpexceptions import HTTPFound, HTTPNotFound | |
25 |
|
25 | |||
26 | from rhodecode.apps._base import BaseAppView |
|
26 | from rhodecode.apps._base import BaseAppView | |
27 |
from rhodecode.apps. |
|
27 | from rhodecode.apps.file_store import utils | |
28 |
from rhodecode.apps. |
|
28 | from rhodecode.apps.file_store.exceptions import ( | |
29 | FileNotAllowedException,FileOverSizeException) |
|
29 | FileNotAllowedException,FileOverSizeException) | |
30 |
|
30 | |||
31 | from rhodecode.lib import helpers as h |
|
31 | from rhodecode.lib import helpers as h | |
32 | from rhodecode.lib import audit_logger |
|
32 | from rhodecode.lib import audit_logger | |
33 | from rhodecode.lib.auth import (CSRFRequired, NotAnonymous) |
|
33 | from rhodecode.lib.auth import (CSRFRequired, NotAnonymous) | |
34 |
|
34 | |||
35 | log = logging.getLogger(__name__) |
|
35 | log = logging.getLogger(__name__) | |
36 |
|
36 | |||
37 |
|
37 | |||
38 | class FileStoreView(BaseAppView): |
|
38 | class FileStoreView(BaseAppView): | |
39 | upload_key = 'store_file' |
|
39 | upload_key = 'store_file' | |
40 |
|
40 | |||
41 | def load_default_context(self): |
|
41 | def load_default_context(self): | |
42 | c = self._get_local_tmpl_context() |
|
42 | c = self._get_local_tmpl_context() | |
43 | self.storage = utils.get_file_storage(self.request.registry.settings) |
|
43 | self.storage = utils.get_file_storage(self.request.registry.settings) | |
44 | return c |
|
44 | return c | |
45 |
|
45 | |||
46 | @NotAnonymous() |
|
46 | @NotAnonymous() | |
47 | @CSRFRequired() |
|
47 | @CSRFRequired() | |
48 | @view_config(route_name='upload_file', request_method='POST', renderer='json_ext') |
|
48 | @view_config(route_name='upload_file', request_method='POST', renderer='json_ext') | |
49 | def upload_file(self): |
|
49 | def upload_file(self): | |
50 | self.load_default_context() |
|
50 | self.load_default_context() | |
51 | file_obj = self.request.POST.get(self.upload_key) |
|
51 | file_obj = self.request.POST.get(self.upload_key) | |
52 |
|
52 | |||
53 | if file_obj is None: |
|
53 | if file_obj is None: | |
54 | return {'store_fid': None, |
|
54 | return {'store_fid': None, | |
55 | 'access_path': None, |
|
55 | 'access_path': None, | |
56 | 'error': '{} data field is missing'.format(self.upload_key)} |
|
56 | 'error': '{} data field is missing'.format(self.upload_key)} | |
57 |
|
57 | |||
58 | if not hasattr(file_obj, 'filename'): |
|
58 | if not hasattr(file_obj, 'filename'): | |
59 | return {'store_fid': None, |
|
59 | return {'store_fid': None, | |
60 | 'access_path': None, |
|
60 | 'access_path': None, | |
61 | 'error': 'filename cannot be read from the data field'} |
|
61 | 'error': 'filename cannot be read from the data field'} | |
62 |
|
62 | |||
63 | filename = file_obj.filename |
|
63 | filename = file_obj.filename | |
64 |
|
64 | |||
65 | metadata = { |
|
65 | metadata = { | |
66 | 'filename': filename, |
|
66 | 'filename': filename, | |
67 | 'size': '', # filled by save_file |
|
67 | 'size': '', # filled by save_file | |
68 | 'user_uploaded': {'username': self._rhodecode_user.username, |
|
68 | 'user_uploaded': {'username': self._rhodecode_user.username, | |
69 | 'user_id': self._rhodecode_user.user_id, |
|
69 | 'user_id': self._rhodecode_user.user_id, | |
70 | 'ip': self._rhodecode_user.ip_addr}} |
|
70 | 'ip': self._rhodecode_user.ip_addr}} | |
71 | try: |
|
71 | try: | |
72 | store_fid = self.storage.save_file(file_obj.file, filename, |
|
72 | store_fid, metadata = self.storage.save_file(file_obj.file, filename, metadata=metadata) | |
73 | metadata=metadata) |
|
|||
74 | except FileNotAllowedException: |
|
73 | except FileNotAllowedException: | |
75 | return {'store_fid': None, |
|
74 | return {'store_fid': None, | |
76 | 'access_path': None, |
|
75 | 'access_path': None, | |
77 | 'error': 'File {} is not allowed.'.format(filename)} |
|
76 | 'error': 'File {} is not allowed.'.format(filename)} | |
78 |
|
77 | |||
79 | except FileOverSizeException: |
|
78 | except FileOverSizeException: | |
80 | return {'store_fid': None, |
|
79 | return {'store_fid': None, | |
81 | 'access_path': None, |
|
80 | 'access_path': None, | |
82 | 'error': 'File {} is exceeding allowed limit.'.format(filename)} |
|
81 | 'error': 'File {} is exceeding allowed limit.'.format(filename)} | |
83 |
|
82 | |||
84 | return {'store_fid': store_fid, |
|
83 | return {'store_fid': store_fid, | |
85 | 'access_path': h.route_path('download_file', fid=store_fid)} |
|
84 | 'access_path': h.route_path('download_file', fid=store_fid)} | |
86 |
|
85 | |||
87 | @view_config(route_name='download_file') |
|
86 | @view_config(route_name='download_file') | |
88 | def download_file(self): |
|
87 | def download_file(self): | |
89 | self.load_default_context() |
|
88 | self.load_default_context() | |
90 | file_uid = self.request.matchdict['fid'] |
|
89 | file_uid = self.request.matchdict['fid'] | |
91 | log.debug('Requesting FID:%s from store %s', file_uid, self.storage) |
|
90 | log.debug('Requesting FID:%s from store %s', file_uid, self.storage) | |
92 | if not self.storage.exists(file_uid): |
|
91 | if not self.storage.exists(file_uid): | |
93 | log.debug('File with FID:%s not found in the store', file_uid) |
|
92 | log.debug('File with FID:%s not found in the store', file_uid) | |
94 | raise HTTPNotFound() |
|
93 | raise HTTPNotFound() | |
95 |
|
94 | |||
96 | file_path = self.storage.store_path(file_uid) |
|
95 | file_path = self.storage.store_path(file_uid) | |
97 | return FileResponse(file_path) |
|
96 | return FileResponse(file_path) |
@@ -1,740 +1,740 b'' | |||||
1 | # -*- coding: utf-8 -*- |
|
1 | # -*- coding: utf-8 -*- | |
2 |
|
2 | |||
3 | # Copyright (C) 2010-2019 RhodeCode GmbH |
|
3 | # Copyright (C) 2010-2019 RhodeCode GmbH | |
4 | # |
|
4 | # | |
5 | # This program is free software: you can redistribute it and/or modify |
|
5 | # This program is free software: you can redistribute it and/or modify | |
6 | # it under the terms of the GNU Affero General Public License, version 3 |
|
6 | # it under the terms of the GNU Affero General Public License, version 3 | |
7 | # (only), as published by the Free Software Foundation. |
|
7 | # (only), as published by the Free Software Foundation. | |
8 | # |
|
8 | # | |
9 | # This program is distributed in the hope that it will be useful, |
|
9 | # This program is distributed in the hope that it will be useful, | |
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
12 | # GNU General Public License for more details. |
|
12 | # GNU General Public License for more details. | |
13 | # |
|
13 | # | |
14 | # You should have received a copy of the GNU Affero General Public License |
|
14 | # You should have received a copy of the GNU Affero General Public License | |
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. |
|
15 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
16 | # |
|
16 | # | |
17 | # This program is dual-licensed. If you wish to learn more about the |
|
17 | # This program is dual-licensed. If you wish to learn more about the | |
18 | # RhodeCode Enterprise Edition, including its added features, Support services, |
|
18 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ |
|
19 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
20 |
|
20 | |||
21 | import os |
|
21 | import os | |
22 | import sys |
|
22 | import sys | |
23 | import logging |
|
23 | import logging | |
24 | import collections |
|
24 | import collections | |
25 | import tempfile |
|
25 | import tempfile | |
26 | import time |
|
26 | import time | |
27 |
|
27 | |||
28 | from paste.gzipper import make_gzip_middleware |
|
28 | from paste.gzipper import make_gzip_middleware | |
29 | import pyramid.events |
|
29 | import pyramid.events | |
30 | from pyramid.wsgi import wsgiapp |
|
30 | from pyramid.wsgi import wsgiapp | |
31 | from pyramid.authorization import ACLAuthorizationPolicy |
|
31 | from pyramid.authorization import ACLAuthorizationPolicy | |
32 | from pyramid.config import Configurator |
|
32 | from pyramid.config import Configurator | |
33 | from pyramid.settings import asbool, aslist |
|
33 | from pyramid.settings import asbool, aslist | |
34 | from pyramid.httpexceptions import ( |
|
34 | from pyramid.httpexceptions import ( | |
35 | HTTPException, HTTPError, HTTPInternalServerError, HTTPFound, HTTPNotFound) |
|
35 | HTTPException, HTTPError, HTTPInternalServerError, HTTPFound, HTTPNotFound) | |
36 | from pyramid.renderers import render_to_response |
|
36 | from pyramid.renderers import render_to_response | |
37 |
|
37 | |||
38 | from rhodecode.model import meta |
|
38 | from rhodecode.model import meta | |
39 | from rhodecode.config import patches |
|
39 | from rhodecode.config import patches | |
40 | from rhodecode.config import utils as config_utils |
|
40 | from rhodecode.config import utils as config_utils | |
41 | from rhodecode.config.environment import load_pyramid_environment |
|
41 | from rhodecode.config.environment import load_pyramid_environment | |
42 |
|
42 | |||
43 | import rhodecode.events |
|
43 | import rhodecode.events | |
44 | from rhodecode.lib.middleware.vcs import VCSMiddleware |
|
44 | from rhodecode.lib.middleware.vcs import VCSMiddleware | |
45 | from rhodecode.lib.request import Request |
|
45 | from rhodecode.lib.request import Request | |
46 | from rhodecode.lib.vcs import VCSCommunicationError |
|
46 | from rhodecode.lib.vcs import VCSCommunicationError | |
47 | from rhodecode.lib.exceptions import VCSServerUnavailable |
|
47 | from rhodecode.lib.exceptions import VCSServerUnavailable | |
48 | from rhodecode.lib.middleware.appenlight import wrap_in_appenlight_if_enabled |
|
48 | from rhodecode.lib.middleware.appenlight import wrap_in_appenlight_if_enabled | |
49 | from rhodecode.lib.middleware.https_fixup import HttpsFixup |
|
49 | from rhodecode.lib.middleware.https_fixup import HttpsFixup | |
50 | from rhodecode.lib.celerylib.loader import configure_celery |
|
50 | from rhodecode.lib.celerylib.loader import configure_celery | |
51 | from rhodecode.lib.plugins.utils import register_rhodecode_plugin |
|
51 | from rhodecode.lib.plugins.utils import register_rhodecode_plugin | |
52 | from rhodecode.lib.utils2 import aslist as rhodecode_aslist, AttributeDict |
|
52 | from rhodecode.lib.utils2 import aslist as rhodecode_aslist, AttributeDict | |
53 | from rhodecode.lib.exc_tracking import store_exception |
|
53 | from rhodecode.lib.exc_tracking import store_exception | |
54 | from rhodecode.subscribers import ( |
|
54 | from rhodecode.subscribers import ( | |
55 | scan_repositories_if_enabled, write_js_routes_if_enabled, |
|
55 | scan_repositories_if_enabled, write_js_routes_if_enabled, | |
56 | write_metadata_if_needed, inject_app_settings) |
|
56 | write_metadata_if_needed, inject_app_settings) | |
57 |
|
57 | |||
58 |
|
58 | |||
59 | log = logging.getLogger(__name__) |
|
59 | log = logging.getLogger(__name__) | |
60 |
|
60 | |||
61 |
|
61 | |||
62 | def is_http_error(response): |
|
62 | def is_http_error(response): | |
63 | # error which should have traceback |
|
63 | # error which should have traceback | |
64 | return response.status_code > 499 |
|
64 | return response.status_code > 499 | |
65 |
|
65 | |||
66 |
|
66 | |||
67 | def should_load_all(): |
|
67 | def should_load_all(): | |
68 | """ |
|
68 | """ | |
69 | Returns if all application components should be loaded. In some cases it's |
|
69 | Returns if all application components should be loaded. In some cases it's | |
70 | desired to skip apps loading for faster shell script execution |
|
70 | desired to skip apps loading for faster shell script execution | |
71 | """ |
|
71 | """ | |
72 | return True |
|
72 | return True | |
73 |
|
73 | |||
74 |
|
74 | |||
75 | def make_pyramid_app(global_config, **settings): |
|
75 | def make_pyramid_app(global_config, **settings): | |
76 | """ |
|
76 | """ | |
77 | Constructs the WSGI application based on Pyramid. |
|
77 | Constructs the WSGI application based on Pyramid. | |
78 |
|
78 | |||
79 | Specials: |
|
79 | Specials: | |
80 |
|
80 | |||
81 | * The application can also be integrated like a plugin via the call to |
|
81 | * The application can also be integrated like a plugin via the call to | |
82 | `includeme`. This is accompanied with the other utility functions which |
|
82 | `includeme`. This is accompanied with the other utility functions which | |
83 | are called. Changing this should be done with great care to not break |
|
83 | are called. Changing this should be done with great care to not break | |
84 | cases when these fragments are assembled from another place. |
|
84 | cases when these fragments are assembled from another place. | |
85 |
|
85 | |||
86 | """ |
|
86 | """ | |
87 |
|
87 | |||
88 | # Allows to use format style "{ENV_NAME}" placeholders in the configuration. It |
|
88 | # Allows to use format style "{ENV_NAME}" placeholders in the configuration. It | |
89 | # will be replaced by the value of the environment variable "NAME" in this case. |
|
89 | # will be replaced by the value of the environment variable "NAME" in this case. | |
90 | start_time = time.time() |
|
90 | start_time = time.time() | |
91 |
|
91 | |||
92 | debug = asbool(global_config.get('debug')) |
|
92 | debug = asbool(global_config.get('debug')) | |
93 | if debug: |
|
93 | if debug: | |
94 | enable_debug() |
|
94 | enable_debug() | |
95 |
|
95 | |||
96 | environ = {'ENV_{}'.format(key): value for key, value in os.environ.items()} |
|
96 | environ = {'ENV_{}'.format(key): value for key, value in os.environ.items()} | |
97 |
|
97 | |||
98 | global_config = _substitute_values(global_config, environ) |
|
98 | global_config = _substitute_values(global_config, environ) | |
99 | settings = _substitute_values(settings, environ) |
|
99 | settings = _substitute_values(settings, environ) | |
100 |
|
100 | |||
101 | sanitize_settings_and_apply_defaults(global_config, settings) |
|
101 | sanitize_settings_and_apply_defaults(global_config, settings) | |
102 |
|
102 | |||
103 | config = Configurator(settings=settings) |
|
103 | config = Configurator(settings=settings) | |
104 |
|
104 | |||
105 | # Apply compatibility patches |
|
105 | # Apply compatibility patches | |
106 | patches.inspect_getargspec() |
|
106 | patches.inspect_getargspec() | |
107 |
|
107 | |||
108 | load_pyramid_environment(global_config, settings) |
|
108 | load_pyramid_environment(global_config, settings) | |
109 |
|
109 | |||
110 | # Static file view comes first |
|
110 | # Static file view comes first | |
111 | includeme_first(config) |
|
111 | includeme_first(config) | |
112 |
|
112 | |||
113 | includeme(config) |
|
113 | includeme(config) | |
114 |
|
114 | |||
115 | pyramid_app = config.make_wsgi_app() |
|
115 | pyramid_app = config.make_wsgi_app() | |
116 | pyramid_app = wrap_app_in_wsgi_middlewares(pyramid_app, config) |
|
116 | pyramid_app = wrap_app_in_wsgi_middlewares(pyramid_app, config) | |
117 | pyramid_app.config = config |
|
117 | pyramid_app.config = config | |
118 |
|
118 | |||
119 | config.configure_celery(global_config['__file__']) |
|
119 | config.configure_celery(global_config['__file__']) | |
120 | # creating the app uses a connection - return it after we are done |
|
120 | # creating the app uses a connection - return it after we are done | |
121 | meta.Session.remove() |
|
121 | meta.Session.remove() | |
122 | total_time = time.time() - start_time |
|
122 | total_time = time.time() - start_time | |
123 | log.info('Pyramid app `%s` created and configured in %.2fs', |
|
123 | log.info('Pyramid app `%s` created and configured in %.2fs', | |
124 | pyramid_app.func_name, total_time) |
|
124 | pyramid_app.func_name, total_time) | |
125 |
|
125 | |||
126 | return pyramid_app |
|
126 | return pyramid_app | |
127 |
|
127 | |||
128 |
|
128 | |||
129 | def not_found_view(request): |
|
129 | def not_found_view(request): | |
130 | """ |
|
130 | """ | |
131 | This creates the view which should be registered as not-found-view to |
|
131 | This creates the view which should be registered as not-found-view to | |
132 | pyramid. |
|
132 | pyramid. | |
133 | """ |
|
133 | """ | |
134 |
|
134 | |||
135 | if not getattr(request, 'vcs_call', None): |
|
135 | if not getattr(request, 'vcs_call', None): | |
136 | # handle like regular case with our error_handler |
|
136 | # handle like regular case with our error_handler | |
137 | return error_handler(HTTPNotFound(), request) |
|
137 | return error_handler(HTTPNotFound(), request) | |
138 |
|
138 | |||
139 | # handle not found view as a vcs call |
|
139 | # handle not found view as a vcs call | |
140 | settings = request.registry.settings |
|
140 | settings = request.registry.settings | |
141 | ae_client = getattr(request, 'ae_client', None) |
|
141 | ae_client = getattr(request, 'ae_client', None) | |
142 | vcs_app = VCSMiddleware( |
|
142 | vcs_app = VCSMiddleware( | |
143 | HTTPNotFound(), request.registry, settings, |
|
143 | HTTPNotFound(), request.registry, settings, | |
144 | appenlight_client=ae_client) |
|
144 | appenlight_client=ae_client) | |
145 |
|
145 | |||
146 | return wsgiapp(vcs_app)(None, request) |
|
146 | return wsgiapp(vcs_app)(None, request) | |
147 |
|
147 | |||
148 |
|
148 | |||
149 | def error_handler(exception, request): |
|
149 | def error_handler(exception, request): | |
150 | import rhodecode |
|
150 | import rhodecode | |
151 | from rhodecode.lib import helpers |
|
151 | from rhodecode.lib import helpers | |
152 |
|
152 | |||
153 | rhodecode_title = rhodecode.CONFIG.get('rhodecode_title') or 'RhodeCode' |
|
153 | rhodecode_title = rhodecode.CONFIG.get('rhodecode_title') or 'RhodeCode' | |
154 |
|
154 | |||
155 | base_response = HTTPInternalServerError() |
|
155 | base_response = HTTPInternalServerError() | |
156 | # prefer original exception for the response since it may have headers set |
|
156 | # prefer original exception for the response since it may have headers set | |
157 | if isinstance(exception, HTTPException): |
|
157 | if isinstance(exception, HTTPException): | |
158 | base_response = exception |
|
158 | base_response = exception | |
159 | elif isinstance(exception, VCSCommunicationError): |
|
159 | elif isinstance(exception, VCSCommunicationError): | |
160 | base_response = VCSServerUnavailable() |
|
160 | base_response = VCSServerUnavailable() | |
161 |
|
161 | |||
162 | if is_http_error(base_response): |
|
162 | if is_http_error(base_response): | |
163 | log.exception( |
|
163 | log.exception( | |
164 | 'error occurred handling this request for path: %s', request.path) |
|
164 | 'error occurred handling this request for path: %s', request.path) | |
165 |
|
165 | |||
166 | error_explanation = base_response.explanation or str(base_response) |
|
166 | error_explanation = base_response.explanation or str(base_response) | |
167 | if base_response.status_code == 404: |
|
167 | if base_response.status_code == 404: | |
168 | error_explanation += " Optionally you don't have permission to access this page." |
|
168 | error_explanation += " Optionally you don't have permission to access this page." | |
169 | c = AttributeDict() |
|
169 | c = AttributeDict() | |
170 | c.error_message = base_response.status |
|
170 | c.error_message = base_response.status | |
171 | c.error_explanation = error_explanation |
|
171 | c.error_explanation = error_explanation | |
172 | c.visual = AttributeDict() |
|
172 | c.visual = AttributeDict() | |
173 |
|
173 | |||
174 | c.visual.rhodecode_support_url = ( |
|
174 | c.visual.rhodecode_support_url = ( | |
175 | request.registry.settings.get('rhodecode_support_url') or |
|
175 | request.registry.settings.get('rhodecode_support_url') or | |
176 | request.route_url('rhodecode_support') |
|
176 | request.route_url('rhodecode_support') | |
177 | ) |
|
177 | ) | |
178 | c.redirect_time = 0 |
|
178 | c.redirect_time = 0 | |
179 | c.rhodecode_name = rhodecode_title |
|
179 | c.rhodecode_name = rhodecode_title | |
180 | if not c.rhodecode_name: |
|
180 | if not c.rhodecode_name: | |
181 | c.rhodecode_name = 'Rhodecode' |
|
181 | c.rhodecode_name = 'Rhodecode' | |
182 |
|
182 | |||
183 | c.causes = [] |
|
183 | c.causes = [] | |
184 | if is_http_error(base_response): |
|
184 | if is_http_error(base_response): | |
185 | c.causes.append('Server is overloaded.') |
|
185 | c.causes.append('Server is overloaded.') | |
186 | c.causes.append('Server database connection is lost.') |
|
186 | c.causes.append('Server database connection is lost.') | |
187 | c.causes.append('Server expected unhandled error.') |
|
187 | c.causes.append('Server expected unhandled error.') | |
188 |
|
188 | |||
189 | if hasattr(base_response, 'causes'): |
|
189 | if hasattr(base_response, 'causes'): | |
190 | c.causes = base_response.causes |
|
190 | c.causes = base_response.causes | |
191 |
|
191 | |||
192 | c.messages = helpers.flash.pop_messages(request=request) |
|
192 | c.messages = helpers.flash.pop_messages(request=request) | |
193 |
|
193 | |||
194 | exc_info = sys.exc_info() |
|
194 | exc_info = sys.exc_info() | |
195 | c.exception_id = id(exc_info) |
|
195 | c.exception_id = id(exc_info) | |
196 | c.show_exception_id = isinstance(base_response, VCSServerUnavailable) \ |
|
196 | c.show_exception_id = isinstance(base_response, VCSServerUnavailable) \ | |
197 | or base_response.status_code > 499 |
|
197 | or base_response.status_code > 499 | |
198 | c.exception_id_url = request.route_url( |
|
198 | c.exception_id_url = request.route_url( | |
199 | 'admin_settings_exception_tracker_show', exception_id=c.exception_id) |
|
199 | 'admin_settings_exception_tracker_show', exception_id=c.exception_id) | |
200 |
|
200 | |||
201 | if c.show_exception_id: |
|
201 | if c.show_exception_id: | |
202 | store_exception(c.exception_id, exc_info) |
|
202 | store_exception(c.exception_id, exc_info) | |
203 |
|
203 | |||
204 | response = render_to_response( |
|
204 | response = render_to_response( | |
205 | '/errors/error_document.mako', {'c': c, 'h': helpers}, request=request, |
|
205 | '/errors/error_document.mako', {'c': c, 'h': helpers}, request=request, | |
206 | response=base_response) |
|
206 | response=base_response) | |
207 |
|
207 | |||
208 | return response |
|
208 | return response | |
209 |
|
209 | |||
210 |
|
210 | |||
211 | def includeme_first(config): |
|
211 | def includeme_first(config): | |
212 | # redirect automatic browser favicon.ico requests to correct place |
|
212 | # redirect automatic browser favicon.ico requests to correct place | |
213 | def favicon_redirect(context, request): |
|
213 | def favicon_redirect(context, request): | |
214 | return HTTPFound( |
|
214 | return HTTPFound( | |
215 | request.static_path('rhodecode:public/images/favicon.ico')) |
|
215 | request.static_path('rhodecode:public/images/favicon.ico')) | |
216 |
|
216 | |||
217 | config.add_view(favicon_redirect, route_name='favicon') |
|
217 | config.add_view(favicon_redirect, route_name='favicon') | |
218 | config.add_route('favicon', '/favicon.ico') |
|
218 | config.add_route('favicon', '/favicon.ico') | |
219 |
|
219 | |||
220 | def robots_redirect(context, request): |
|
220 | def robots_redirect(context, request): | |
221 | return HTTPFound( |
|
221 | return HTTPFound( | |
222 | request.static_path('rhodecode:public/robots.txt')) |
|
222 | request.static_path('rhodecode:public/robots.txt')) | |
223 |
|
223 | |||
224 | config.add_view(robots_redirect, route_name='robots') |
|
224 | config.add_view(robots_redirect, route_name='robots') | |
225 | config.add_route('robots', '/robots.txt') |
|
225 | config.add_route('robots', '/robots.txt') | |
226 |
|
226 | |||
227 | config.add_static_view( |
|
227 | config.add_static_view( | |
228 | '_static/deform', 'deform:static') |
|
228 | '_static/deform', 'deform:static') | |
229 | config.add_static_view( |
|
229 | config.add_static_view( | |
230 | '_static/rhodecode', path='rhodecode:public', cache_max_age=3600 * 24) |
|
230 | '_static/rhodecode', path='rhodecode:public', cache_max_age=3600 * 24) | |
231 |
|
231 | |||
232 |
|
232 | |||
233 | def includeme(config): |
|
233 | def includeme(config): | |
234 | log.debug('Initializing main includeme from %s', os.path.basename(__file__)) |
|
234 | log.debug('Initializing main includeme from %s', os.path.basename(__file__)) | |
235 | settings = config.registry.settings |
|
235 | settings = config.registry.settings | |
236 | config.set_request_factory(Request) |
|
236 | config.set_request_factory(Request) | |
237 |
|
237 | |||
238 | # plugin information |
|
238 | # plugin information | |
239 | config.registry.rhodecode_plugins = collections.OrderedDict() |
|
239 | config.registry.rhodecode_plugins = collections.OrderedDict() | |
240 |
|
240 | |||
241 | config.add_directive( |
|
241 | config.add_directive( | |
242 | 'register_rhodecode_plugin', register_rhodecode_plugin) |
|
242 | 'register_rhodecode_plugin', register_rhodecode_plugin) | |
243 |
|
243 | |||
244 | config.add_directive('configure_celery', configure_celery) |
|
244 | config.add_directive('configure_celery', configure_celery) | |
245 |
|
245 | |||
246 | if asbool(settings.get('appenlight', 'false')): |
|
246 | if asbool(settings.get('appenlight', 'false')): | |
247 | config.include('appenlight_client.ext.pyramid_tween') |
|
247 | config.include('appenlight_client.ext.pyramid_tween') | |
248 |
|
248 | |||
249 | load_all = should_load_all() |
|
249 | load_all = should_load_all() | |
250 |
|
250 | |||
251 | # Includes which are required. The application would fail without them. |
|
251 | # Includes which are required. The application would fail without them. | |
252 | config.include('pyramid_mako') |
|
252 | config.include('pyramid_mako') | |
253 | config.include('pyramid_beaker') |
|
253 | config.include('pyramid_beaker') | |
254 | config.include('rhodecode.lib.rc_cache') |
|
254 | config.include('rhodecode.lib.rc_cache') | |
255 |
|
255 | |||
256 | config.include('rhodecode.apps._base.navigation') |
|
256 | config.include('rhodecode.apps._base.navigation') | |
257 | config.include('rhodecode.apps._base.subscribers') |
|
257 | config.include('rhodecode.apps._base.subscribers') | |
258 | config.include('rhodecode.tweens') |
|
258 | config.include('rhodecode.tweens') | |
259 |
|
259 | |||
260 | config.include('rhodecode.integrations') |
|
260 | config.include('rhodecode.integrations') | |
261 | config.include('rhodecode.authentication') |
|
261 | config.include('rhodecode.authentication') | |
262 |
|
262 | |||
263 | if load_all: |
|
263 | if load_all: | |
264 | from rhodecode.authentication import discover_legacy_plugins |
|
264 | from rhodecode.authentication import discover_legacy_plugins | |
265 | # load CE authentication plugins |
|
265 | # load CE authentication plugins | |
266 | config.include('rhodecode.authentication.plugins.auth_crowd') |
|
266 | config.include('rhodecode.authentication.plugins.auth_crowd') | |
267 | config.include('rhodecode.authentication.plugins.auth_headers') |
|
267 | config.include('rhodecode.authentication.plugins.auth_headers') | |
268 | config.include('rhodecode.authentication.plugins.auth_jasig_cas') |
|
268 | config.include('rhodecode.authentication.plugins.auth_jasig_cas') | |
269 | config.include('rhodecode.authentication.plugins.auth_ldap') |
|
269 | config.include('rhodecode.authentication.plugins.auth_ldap') | |
270 | config.include('rhodecode.authentication.plugins.auth_pam') |
|
270 | config.include('rhodecode.authentication.plugins.auth_pam') | |
271 | config.include('rhodecode.authentication.plugins.auth_rhodecode') |
|
271 | config.include('rhodecode.authentication.plugins.auth_rhodecode') | |
272 | config.include('rhodecode.authentication.plugins.auth_token') |
|
272 | config.include('rhodecode.authentication.plugins.auth_token') | |
273 |
|
273 | |||
274 | # Auto discover authentication plugins and include their configuration. |
|
274 | # Auto discover authentication plugins and include their configuration. | |
275 | discover_legacy_plugins(config) |
|
275 | discover_legacy_plugins(config) | |
276 |
|
276 | |||
277 | # apps |
|
277 | # apps | |
278 | config.include('rhodecode.apps._base') |
|
278 | config.include('rhodecode.apps._base') | |
279 |
|
279 | |||
280 | if load_all: |
|
280 | if load_all: | |
281 | config.include('rhodecode.apps.ops') |
|
281 | config.include('rhodecode.apps.ops') | |
282 | config.include('rhodecode.apps.admin') |
|
282 | config.include('rhodecode.apps.admin') | |
283 | config.include('rhodecode.apps.channelstream') |
|
283 | config.include('rhodecode.apps.channelstream') | |
284 |
config.include('rhodecode.apps. |
|
284 | config.include('rhodecode.apps.file_store') | |
285 | config.include('rhodecode.apps.login') |
|
285 | config.include('rhodecode.apps.login') | |
286 | config.include('rhodecode.apps.home') |
|
286 | config.include('rhodecode.apps.home') | |
287 | config.include('rhodecode.apps.journal') |
|
287 | config.include('rhodecode.apps.journal') | |
288 | config.include('rhodecode.apps.repository') |
|
288 | config.include('rhodecode.apps.repository') | |
289 | config.include('rhodecode.apps.repo_group') |
|
289 | config.include('rhodecode.apps.repo_group') | |
290 | config.include('rhodecode.apps.user_group') |
|
290 | config.include('rhodecode.apps.user_group') | |
291 | config.include('rhodecode.apps.search') |
|
291 | config.include('rhodecode.apps.search') | |
292 | config.include('rhodecode.apps.user_profile') |
|
292 | config.include('rhodecode.apps.user_profile') | |
293 | config.include('rhodecode.apps.user_group_profile') |
|
293 | config.include('rhodecode.apps.user_group_profile') | |
294 | config.include('rhodecode.apps.my_account') |
|
294 | config.include('rhodecode.apps.my_account') | |
295 | config.include('rhodecode.apps.svn_support') |
|
295 | config.include('rhodecode.apps.svn_support') | |
296 | config.include('rhodecode.apps.ssh_support') |
|
296 | config.include('rhodecode.apps.ssh_support') | |
297 | config.include('rhodecode.apps.gist') |
|
297 | config.include('rhodecode.apps.gist') | |
298 | config.include('rhodecode.apps.debug_style') |
|
298 | config.include('rhodecode.apps.debug_style') | |
299 | config.include('rhodecode.api') |
|
299 | config.include('rhodecode.api') | |
300 |
|
300 | |||
301 | config.add_route('rhodecode_support', 'https://rhodecode.com/help/', static=True) |
|
301 | config.add_route('rhodecode_support', 'https://rhodecode.com/help/', static=True) | |
302 | config.add_translation_dirs('rhodecode:i18n/') |
|
302 | config.add_translation_dirs('rhodecode:i18n/') | |
303 | settings['default_locale_name'] = settings.get('lang', 'en') |
|
303 | settings['default_locale_name'] = settings.get('lang', 'en') | |
304 |
|
304 | |||
305 | # Add subscribers. |
|
305 | # Add subscribers. | |
306 | config.add_subscriber(inject_app_settings, |
|
306 | config.add_subscriber(inject_app_settings, | |
307 | pyramid.events.ApplicationCreated) |
|
307 | pyramid.events.ApplicationCreated) | |
308 | config.add_subscriber(scan_repositories_if_enabled, |
|
308 | config.add_subscriber(scan_repositories_if_enabled, | |
309 | pyramid.events.ApplicationCreated) |
|
309 | pyramid.events.ApplicationCreated) | |
310 | config.add_subscriber(write_metadata_if_needed, |
|
310 | config.add_subscriber(write_metadata_if_needed, | |
311 | pyramid.events.ApplicationCreated) |
|
311 | pyramid.events.ApplicationCreated) | |
312 | config.add_subscriber(write_js_routes_if_enabled, |
|
312 | config.add_subscriber(write_js_routes_if_enabled, | |
313 | pyramid.events.ApplicationCreated) |
|
313 | pyramid.events.ApplicationCreated) | |
314 |
|
314 | |||
315 | # request custom methods |
|
315 | # request custom methods | |
316 | config.add_request_method( |
|
316 | config.add_request_method( | |
317 | 'rhodecode.lib.partial_renderer.get_partial_renderer', |
|
317 | 'rhodecode.lib.partial_renderer.get_partial_renderer', | |
318 | 'get_partial_renderer') |
|
318 | 'get_partial_renderer') | |
319 |
|
319 | |||
320 | # Set the authorization policy. |
|
320 | # Set the authorization policy. | |
321 | authz_policy = ACLAuthorizationPolicy() |
|
321 | authz_policy = ACLAuthorizationPolicy() | |
322 | config.set_authorization_policy(authz_policy) |
|
322 | config.set_authorization_policy(authz_policy) | |
323 |
|
323 | |||
324 | # Set the default renderer for HTML templates to mako. |
|
324 | # Set the default renderer for HTML templates to mako. | |
325 | config.add_mako_renderer('.html') |
|
325 | config.add_mako_renderer('.html') | |
326 |
|
326 | |||
327 | config.add_renderer( |
|
327 | config.add_renderer( | |
328 | name='json_ext', |
|
328 | name='json_ext', | |
329 | factory='rhodecode.lib.ext_json_renderer.pyramid_ext_json') |
|
329 | factory='rhodecode.lib.ext_json_renderer.pyramid_ext_json') | |
330 |
|
330 | |||
331 | # include RhodeCode plugins |
|
331 | # include RhodeCode plugins | |
332 | includes = aslist(settings.get('rhodecode.includes', [])) |
|
332 | includes = aslist(settings.get('rhodecode.includes', [])) | |
333 | for inc in includes: |
|
333 | for inc in includes: | |
334 | config.include(inc) |
|
334 | config.include(inc) | |
335 |
|
335 | |||
336 | # custom not found view, if our pyramid app doesn't know how to handle |
|
336 | # custom not found view, if our pyramid app doesn't know how to handle | |
337 | # the request pass it to potential VCS handling ap |
|
337 | # the request pass it to potential VCS handling ap | |
338 | config.add_notfound_view(not_found_view) |
|
338 | config.add_notfound_view(not_found_view) | |
339 | if not settings.get('debugtoolbar.enabled', False): |
|
339 | if not settings.get('debugtoolbar.enabled', False): | |
340 | # disabled debugtoolbar handle all exceptions via the error_handlers |
|
340 | # disabled debugtoolbar handle all exceptions via the error_handlers | |
341 | config.add_view(error_handler, context=Exception) |
|
341 | config.add_view(error_handler, context=Exception) | |
342 |
|
342 | |||
343 | # all errors including 403/404/50X |
|
343 | # all errors including 403/404/50X | |
344 | config.add_view(error_handler, context=HTTPError) |
|
344 | config.add_view(error_handler, context=HTTPError) | |
345 |
|
345 | |||
346 |
|
346 | |||
347 | def wrap_app_in_wsgi_middlewares(pyramid_app, config): |
|
347 | def wrap_app_in_wsgi_middlewares(pyramid_app, config): | |
348 | """ |
|
348 | """ | |
349 | Apply outer WSGI middlewares around the application. |
|
349 | Apply outer WSGI middlewares around the application. | |
350 | """ |
|
350 | """ | |
351 | registry = config.registry |
|
351 | registry = config.registry | |
352 | settings = registry.settings |
|
352 | settings = registry.settings | |
353 |
|
353 | |||
354 | # enable https redirects based on HTTP_X_URL_SCHEME set by proxy |
|
354 | # enable https redirects based on HTTP_X_URL_SCHEME set by proxy | |
355 | pyramid_app = HttpsFixup(pyramid_app, settings) |
|
355 | pyramid_app = HttpsFixup(pyramid_app, settings) | |
356 |
|
356 | |||
357 | pyramid_app, _ae_client = wrap_in_appenlight_if_enabled( |
|
357 | pyramid_app, _ae_client = wrap_in_appenlight_if_enabled( | |
358 | pyramid_app, settings) |
|
358 | pyramid_app, settings) | |
359 | registry.ae_client = _ae_client |
|
359 | registry.ae_client = _ae_client | |
360 |
|
360 | |||
361 | if settings['gzip_responses']: |
|
361 | if settings['gzip_responses']: | |
362 | pyramid_app = make_gzip_middleware( |
|
362 | pyramid_app = make_gzip_middleware( | |
363 | pyramid_app, settings, compress_level=1) |
|
363 | pyramid_app, settings, compress_level=1) | |
364 |
|
364 | |||
365 | # this should be the outer most middleware in the wsgi stack since |
|
365 | # this should be the outer most middleware in the wsgi stack since | |
366 | # middleware like Routes make database calls |
|
366 | # middleware like Routes make database calls | |
367 | def pyramid_app_with_cleanup(environ, start_response): |
|
367 | def pyramid_app_with_cleanup(environ, start_response): | |
368 | try: |
|
368 | try: | |
369 | return pyramid_app(environ, start_response) |
|
369 | return pyramid_app(environ, start_response) | |
370 | finally: |
|
370 | finally: | |
371 | # Dispose current database session and rollback uncommitted |
|
371 | # Dispose current database session and rollback uncommitted | |
372 | # transactions. |
|
372 | # transactions. | |
373 | meta.Session.remove() |
|
373 | meta.Session.remove() | |
374 |
|
374 | |||
375 | # In a single threaded mode server, on non sqlite db we should have |
|
375 | # In a single threaded mode server, on non sqlite db we should have | |
376 | # '0 Current Checked out connections' at the end of a request, |
|
376 | # '0 Current Checked out connections' at the end of a request, | |
377 | # if not, then something, somewhere is leaving a connection open |
|
377 | # if not, then something, somewhere is leaving a connection open | |
378 | pool = meta.Base.metadata.bind.engine.pool |
|
378 | pool = meta.Base.metadata.bind.engine.pool | |
379 | log.debug('sa pool status: %s', pool.status()) |
|
379 | log.debug('sa pool status: %s', pool.status()) | |
380 | log.debug('Request processing finalized') |
|
380 | log.debug('Request processing finalized') | |
381 |
|
381 | |||
382 | return pyramid_app_with_cleanup |
|
382 | return pyramid_app_with_cleanup | |
383 |
|
383 | |||
384 |
|
384 | |||
385 | def sanitize_settings_and_apply_defaults(global_config, settings): |
|
385 | def sanitize_settings_and_apply_defaults(global_config, settings): | |
386 | """ |
|
386 | """ | |
387 | Applies settings defaults and does all type conversion. |
|
387 | Applies settings defaults and does all type conversion. | |
388 |
|
388 | |||
389 | We would move all settings parsing and preparation into this place, so that |
|
389 | We would move all settings parsing and preparation into this place, so that | |
390 | we have only one place left which deals with this part. The remaining parts |
|
390 | we have only one place left which deals with this part. The remaining parts | |
391 | of the application would start to rely fully on well prepared settings. |
|
391 | of the application would start to rely fully on well prepared settings. | |
392 |
|
392 | |||
393 | This piece would later be split up per topic to avoid a big fat monster |
|
393 | This piece would later be split up per topic to avoid a big fat monster | |
394 | function. |
|
394 | function. | |
395 | """ |
|
395 | """ | |
396 |
|
396 | |||
397 | settings.setdefault('rhodecode.edition', 'Community Edition') |
|
397 | settings.setdefault('rhodecode.edition', 'Community Edition') | |
398 |
|
398 | |||
399 | if 'mako.default_filters' not in settings: |
|
399 | if 'mako.default_filters' not in settings: | |
400 | # set custom default filters if we don't have it defined |
|
400 | # set custom default filters if we don't have it defined | |
401 | settings['mako.imports'] = 'from rhodecode.lib.base import h_filter' |
|
401 | settings['mako.imports'] = 'from rhodecode.lib.base import h_filter' | |
402 | settings['mako.default_filters'] = 'h_filter' |
|
402 | settings['mako.default_filters'] = 'h_filter' | |
403 |
|
403 | |||
404 | if 'mako.directories' not in settings: |
|
404 | if 'mako.directories' not in settings: | |
405 | mako_directories = settings.setdefault('mako.directories', [ |
|
405 | mako_directories = settings.setdefault('mako.directories', [ | |
406 | # Base templates of the original application |
|
406 | # Base templates of the original application | |
407 | 'rhodecode:templates', |
|
407 | 'rhodecode:templates', | |
408 | ]) |
|
408 | ]) | |
409 | log.debug( |
|
409 | log.debug( | |
410 | "Using the following Mako template directories: %s", |
|
410 | "Using the following Mako template directories: %s", | |
411 | mako_directories) |
|
411 | mako_directories) | |
412 |
|
412 | |||
413 | # Default includes, possible to change as a user |
|
413 | # Default includes, possible to change as a user | |
414 | pyramid_includes = settings.setdefault('pyramid.includes', [ |
|
414 | pyramid_includes = settings.setdefault('pyramid.includes', [ | |
415 | 'rhodecode.lib.middleware.request_wrapper', |
|
415 | 'rhodecode.lib.middleware.request_wrapper', | |
416 | ]) |
|
416 | ]) | |
417 | log.debug( |
|
417 | log.debug( | |
418 | "Using the following pyramid.includes: %s", |
|
418 | "Using the following pyramid.includes: %s", | |
419 | pyramid_includes) |
|
419 | pyramid_includes) | |
420 |
|
420 | |||
421 | # TODO: johbo: Re-think this, usually the call to config.include |
|
421 | # TODO: johbo: Re-think this, usually the call to config.include | |
422 | # should allow to pass in a prefix. |
|
422 | # should allow to pass in a prefix. | |
423 | settings.setdefault('rhodecode.api.url', '/_admin/api') |
|
423 | settings.setdefault('rhodecode.api.url', '/_admin/api') | |
424 | settings.setdefault('__file__', global_config.get('__file__')) |
|
424 | settings.setdefault('__file__', global_config.get('__file__')) | |
425 |
|
425 | |||
426 | # Sanitize generic settings. |
|
426 | # Sanitize generic settings. | |
427 | _list_setting(settings, 'default_encoding', 'UTF-8') |
|
427 | _list_setting(settings, 'default_encoding', 'UTF-8') | |
428 | _bool_setting(settings, 'is_test', 'false') |
|
428 | _bool_setting(settings, 'is_test', 'false') | |
429 | _bool_setting(settings, 'gzip_responses', 'false') |
|
429 | _bool_setting(settings, 'gzip_responses', 'false') | |
430 |
|
430 | |||
431 | # Call split out functions that sanitize settings for each topic. |
|
431 | # Call split out functions that sanitize settings for each topic. | |
432 | _sanitize_appenlight_settings(settings) |
|
432 | _sanitize_appenlight_settings(settings) | |
433 | _sanitize_vcs_settings(settings) |
|
433 | _sanitize_vcs_settings(settings) | |
434 | _sanitize_cache_settings(settings) |
|
434 | _sanitize_cache_settings(settings) | |
435 |
|
435 | |||
436 | # configure instance id |
|
436 | # configure instance id | |
437 | config_utils.set_instance_id(settings) |
|
437 | config_utils.set_instance_id(settings) | |
438 |
|
438 | |||
439 | return settings |
|
439 | return settings | |
440 |
|
440 | |||
441 |
|
441 | |||
442 | def enable_debug(): |
|
442 | def enable_debug(): | |
443 | """ |
|
443 | """ | |
444 | Helper to enable debug on running instance |
|
444 | Helper to enable debug on running instance | |
445 | :return: |
|
445 | :return: | |
446 | """ |
|
446 | """ | |
447 | import tempfile |
|
447 | import tempfile | |
448 | import textwrap |
|
448 | import textwrap | |
449 | import logging.config |
|
449 | import logging.config | |
450 |
|
450 | |||
451 | ini_template = textwrap.dedent(""" |
|
451 | ini_template = textwrap.dedent(""" | |
452 | ##################################### |
|
452 | ##################################### | |
453 | ### DEBUG LOGGING CONFIGURATION #### |
|
453 | ### DEBUG LOGGING CONFIGURATION #### | |
454 | ##################################### |
|
454 | ##################################### | |
455 | [loggers] |
|
455 | [loggers] | |
456 | keys = root, sqlalchemy, beaker, celery, rhodecode, ssh_wrapper |
|
456 | keys = root, sqlalchemy, beaker, celery, rhodecode, ssh_wrapper | |
457 |
|
457 | |||
458 | [handlers] |
|
458 | [handlers] | |
459 | keys = console, console_sql |
|
459 | keys = console, console_sql | |
460 |
|
460 | |||
461 | [formatters] |
|
461 | [formatters] | |
462 | keys = generic, color_formatter, color_formatter_sql |
|
462 | keys = generic, color_formatter, color_formatter_sql | |
463 |
|
463 | |||
464 | ############# |
|
464 | ############# | |
465 | ## LOGGERS ## |
|
465 | ## LOGGERS ## | |
466 | ############# |
|
466 | ############# | |
467 | [logger_root] |
|
467 | [logger_root] | |
468 | level = NOTSET |
|
468 | level = NOTSET | |
469 | handlers = console |
|
469 | handlers = console | |
470 |
|
470 | |||
471 | [logger_sqlalchemy] |
|
471 | [logger_sqlalchemy] | |
472 | level = INFO |
|
472 | level = INFO | |
473 | handlers = console_sql |
|
473 | handlers = console_sql | |
474 | qualname = sqlalchemy.engine |
|
474 | qualname = sqlalchemy.engine | |
475 | propagate = 0 |
|
475 | propagate = 0 | |
476 |
|
476 | |||
477 | [logger_beaker] |
|
477 | [logger_beaker] | |
478 | level = DEBUG |
|
478 | level = DEBUG | |
479 | handlers = |
|
479 | handlers = | |
480 | qualname = beaker.container |
|
480 | qualname = beaker.container | |
481 | propagate = 1 |
|
481 | propagate = 1 | |
482 |
|
482 | |||
483 | [logger_rhodecode] |
|
483 | [logger_rhodecode] | |
484 | level = DEBUG |
|
484 | level = DEBUG | |
485 | handlers = |
|
485 | handlers = | |
486 | qualname = rhodecode |
|
486 | qualname = rhodecode | |
487 | propagate = 1 |
|
487 | propagate = 1 | |
488 |
|
488 | |||
489 | [logger_ssh_wrapper] |
|
489 | [logger_ssh_wrapper] | |
490 | level = DEBUG |
|
490 | level = DEBUG | |
491 | handlers = |
|
491 | handlers = | |
492 | qualname = ssh_wrapper |
|
492 | qualname = ssh_wrapper | |
493 | propagate = 1 |
|
493 | propagate = 1 | |
494 |
|
494 | |||
495 | [logger_celery] |
|
495 | [logger_celery] | |
496 | level = DEBUG |
|
496 | level = DEBUG | |
497 | handlers = |
|
497 | handlers = | |
498 | qualname = celery |
|
498 | qualname = celery | |
499 |
|
499 | |||
500 |
|
500 | |||
501 | ############## |
|
501 | ############## | |
502 | ## HANDLERS ## |
|
502 | ## HANDLERS ## | |
503 | ############## |
|
503 | ############## | |
504 |
|
504 | |||
505 | [handler_console] |
|
505 | [handler_console] | |
506 | class = StreamHandler |
|
506 | class = StreamHandler | |
507 | args = (sys.stderr, ) |
|
507 | args = (sys.stderr, ) | |
508 | level = DEBUG |
|
508 | level = DEBUG | |
509 | formatter = color_formatter |
|
509 | formatter = color_formatter | |
510 |
|
510 | |||
511 | [handler_console_sql] |
|
511 | [handler_console_sql] | |
512 | # "level = DEBUG" logs SQL queries and results. |
|
512 | # "level = DEBUG" logs SQL queries and results. | |
513 | # "level = INFO" logs SQL queries. |
|
513 | # "level = INFO" logs SQL queries. | |
514 | # "level = WARN" logs neither. (Recommended for production systems.) |
|
514 | # "level = WARN" logs neither. (Recommended for production systems.) | |
515 | class = StreamHandler |
|
515 | class = StreamHandler | |
516 | args = (sys.stderr, ) |
|
516 | args = (sys.stderr, ) | |
517 | level = WARN |
|
517 | level = WARN | |
518 | formatter = color_formatter_sql |
|
518 | formatter = color_formatter_sql | |
519 |
|
519 | |||
520 | ################ |
|
520 | ################ | |
521 | ## FORMATTERS ## |
|
521 | ## FORMATTERS ## | |
522 | ################ |
|
522 | ################ | |
523 |
|
523 | |||
524 | [formatter_generic] |
|
524 | [formatter_generic] | |
525 | class = rhodecode.lib.logging_formatter.ExceptionAwareFormatter |
|
525 | class = rhodecode.lib.logging_formatter.ExceptionAwareFormatter | |
526 | format = %(asctime)s.%(msecs)03d [%(process)d] %(levelname)-5.5s [%(name)s] %(message)s | %(req_id)s |
|
526 | format = %(asctime)s.%(msecs)03d [%(process)d] %(levelname)-5.5s [%(name)s] %(message)s | %(req_id)s | |
527 | datefmt = %Y-%m-%d %H:%M:%S |
|
527 | datefmt = %Y-%m-%d %H:%M:%S | |
528 |
|
528 | |||
529 | [formatter_color_formatter] |
|
529 | [formatter_color_formatter] | |
530 | class = rhodecode.lib.logging_formatter.ColorRequestTrackingFormatter |
|
530 | class = rhodecode.lib.logging_formatter.ColorRequestTrackingFormatter | |
531 | format = %(asctime)s.%(msecs)03d [%(process)d] %(levelname)-5.5s [%(name)s] %(message)s | %(req_id)s |
|
531 | format = %(asctime)s.%(msecs)03d [%(process)d] %(levelname)-5.5s [%(name)s] %(message)s | %(req_id)s | |
532 | datefmt = %Y-%m-%d %H:%M:%S |
|
532 | datefmt = %Y-%m-%d %H:%M:%S | |
533 |
|
533 | |||
534 | [formatter_color_formatter_sql] |
|
534 | [formatter_color_formatter_sql] | |
535 | class = rhodecode.lib.logging_formatter.ColorFormatterSql |
|
535 | class = rhodecode.lib.logging_formatter.ColorFormatterSql | |
536 | format = %(asctime)s.%(msecs)03d [%(process)d] %(levelname)-5.5s [%(name)s] %(message)s |
|
536 | format = %(asctime)s.%(msecs)03d [%(process)d] %(levelname)-5.5s [%(name)s] %(message)s | |
537 | datefmt = %Y-%m-%d %H:%M:%S |
|
537 | datefmt = %Y-%m-%d %H:%M:%S | |
538 | """) |
|
538 | """) | |
539 |
|
539 | |||
540 | with tempfile.NamedTemporaryFile(prefix='rc_debug_logging_', suffix='.ini', |
|
540 | with tempfile.NamedTemporaryFile(prefix='rc_debug_logging_', suffix='.ini', | |
541 | delete=False) as f: |
|
541 | delete=False) as f: | |
542 | log.info('Saved Temporary DEBUG config at %s', f.name) |
|
542 | log.info('Saved Temporary DEBUG config at %s', f.name) | |
543 | f.write(ini_template) |
|
543 | f.write(ini_template) | |
544 |
|
544 | |||
545 | logging.config.fileConfig(f.name) |
|
545 | logging.config.fileConfig(f.name) | |
546 | log.debug('DEBUG MODE ON') |
|
546 | log.debug('DEBUG MODE ON') | |
547 | os.remove(f.name) |
|
547 | os.remove(f.name) | |
548 |
|
548 | |||
549 |
|
549 | |||
550 | def _sanitize_appenlight_settings(settings): |
|
550 | def _sanitize_appenlight_settings(settings): | |
551 | _bool_setting(settings, 'appenlight', 'false') |
|
551 | _bool_setting(settings, 'appenlight', 'false') | |
552 |
|
552 | |||
553 |
|
553 | |||
554 | def _sanitize_vcs_settings(settings): |
|
554 | def _sanitize_vcs_settings(settings): | |
555 | """ |
|
555 | """ | |
556 | Applies settings defaults and does type conversion for all VCS related |
|
556 | Applies settings defaults and does type conversion for all VCS related | |
557 | settings. |
|
557 | settings. | |
558 | """ |
|
558 | """ | |
559 | _string_setting(settings, 'vcs.svn.compatible_version', '') |
|
559 | _string_setting(settings, 'vcs.svn.compatible_version', '') | |
560 | _string_setting(settings, 'git_rev_filter', '--all') |
|
560 | _string_setting(settings, 'git_rev_filter', '--all') | |
561 | _string_setting(settings, 'vcs.hooks.protocol', 'http') |
|
561 | _string_setting(settings, 'vcs.hooks.protocol', 'http') | |
562 | _string_setting(settings, 'vcs.hooks.host', '127.0.0.1') |
|
562 | _string_setting(settings, 'vcs.hooks.host', '127.0.0.1') | |
563 | _string_setting(settings, 'vcs.scm_app_implementation', 'http') |
|
563 | _string_setting(settings, 'vcs.scm_app_implementation', 'http') | |
564 | _string_setting(settings, 'vcs.server', '') |
|
564 | _string_setting(settings, 'vcs.server', '') | |
565 | _string_setting(settings, 'vcs.server.log_level', 'debug') |
|
565 | _string_setting(settings, 'vcs.server.log_level', 'debug') | |
566 | _string_setting(settings, 'vcs.server.protocol', 'http') |
|
566 | _string_setting(settings, 'vcs.server.protocol', 'http') | |
567 | _bool_setting(settings, 'startup.import_repos', 'false') |
|
567 | _bool_setting(settings, 'startup.import_repos', 'false') | |
568 | _bool_setting(settings, 'vcs.hooks.direct_calls', 'false') |
|
568 | _bool_setting(settings, 'vcs.hooks.direct_calls', 'false') | |
569 | _bool_setting(settings, 'vcs.server.enable', 'true') |
|
569 | _bool_setting(settings, 'vcs.server.enable', 'true') | |
570 | _bool_setting(settings, 'vcs.start_server', 'false') |
|
570 | _bool_setting(settings, 'vcs.start_server', 'false') | |
571 | _list_setting(settings, 'vcs.backends', 'hg, git, svn') |
|
571 | _list_setting(settings, 'vcs.backends', 'hg, git, svn') | |
572 | _int_setting(settings, 'vcs.connection_timeout', 3600) |
|
572 | _int_setting(settings, 'vcs.connection_timeout', 3600) | |
573 |
|
573 | |||
574 | # Support legacy values of vcs.scm_app_implementation. Legacy |
|
574 | # Support legacy values of vcs.scm_app_implementation. Legacy | |
575 | # configurations may use 'rhodecode.lib.middleware.utils.scm_app_http', or |
|
575 | # configurations may use 'rhodecode.lib.middleware.utils.scm_app_http', or | |
576 | # disabled since 4.13 'vcsserver.scm_app' which is now mapped to 'http'. |
|
576 | # disabled since 4.13 'vcsserver.scm_app' which is now mapped to 'http'. | |
577 | scm_app_impl = settings['vcs.scm_app_implementation'] |
|
577 | scm_app_impl = settings['vcs.scm_app_implementation'] | |
578 | if scm_app_impl in ['rhodecode.lib.middleware.utils.scm_app_http', 'vcsserver.scm_app']: |
|
578 | if scm_app_impl in ['rhodecode.lib.middleware.utils.scm_app_http', 'vcsserver.scm_app']: | |
579 | settings['vcs.scm_app_implementation'] = 'http' |
|
579 | settings['vcs.scm_app_implementation'] = 'http' | |
580 |
|
580 | |||
581 |
|
581 | |||
582 | def _sanitize_cache_settings(settings): |
|
582 | def _sanitize_cache_settings(settings): | |
583 | temp_store = tempfile.gettempdir() |
|
583 | temp_store = tempfile.gettempdir() | |
584 | default_cache_dir = os.path.join(temp_store, 'rc_cache') |
|
584 | default_cache_dir = os.path.join(temp_store, 'rc_cache') | |
585 |
|
585 | |||
586 | # save default, cache dir, and use it for all backends later. |
|
586 | # save default, cache dir, and use it for all backends later. | |
587 | default_cache_dir = _string_setting( |
|
587 | default_cache_dir = _string_setting( | |
588 | settings, |
|
588 | settings, | |
589 | 'cache_dir', |
|
589 | 'cache_dir', | |
590 | default_cache_dir, lower=False, default_when_empty=True) |
|
590 | default_cache_dir, lower=False, default_when_empty=True) | |
591 |
|
591 | |||
592 | # ensure we have our dir created |
|
592 | # ensure we have our dir created | |
593 | if not os.path.isdir(default_cache_dir): |
|
593 | if not os.path.isdir(default_cache_dir): | |
594 | os.makedirs(default_cache_dir, mode=0o755) |
|
594 | os.makedirs(default_cache_dir, mode=0o755) | |
595 |
|
595 | |||
596 | # exception store cache |
|
596 | # exception store cache | |
597 | _string_setting( |
|
597 | _string_setting( | |
598 | settings, |
|
598 | settings, | |
599 | 'exception_tracker.store_path', |
|
599 | 'exception_tracker.store_path', | |
600 | temp_store, lower=False, default_when_empty=True) |
|
600 | temp_store, lower=False, default_when_empty=True) | |
601 |
|
601 | |||
602 | # cache_perms |
|
602 | # cache_perms | |
603 | _string_setting( |
|
603 | _string_setting( | |
604 | settings, |
|
604 | settings, | |
605 | 'rc_cache.cache_perms.backend', |
|
605 | 'rc_cache.cache_perms.backend', | |
606 | 'dogpile.cache.rc.file_namespace', lower=False) |
|
606 | 'dogpile.cache.rc.file_namespace', lower=False) | |
607 | _int_setting( |
|
607 | _int_setting( | |
608 | settings, |
|
608 | settings, | |
609 | 'rc_cache.cache_perms.expiration_time', |
|
609 | 'rc_cache.cache_perms.expiration_time', | |
610 | 60) |
|
610 | 60) | |
611 | _string_setting( |
|
611 | _string_setting( | |
612 | settings, |
|
612 | settings, | |
613 | 'rc_cache.cache_perms.arguments.filename', |
|
613 | 'rc_cache.cache_perms.arguments.filename', | |
614 | os.path.join(default_cache_dir, 'rc_cache_1'), lower=False) |
|
614 | os.path.join(default_cache_dir, 'rc_cache_1'), lower=False) | |
615 |
|
615 | |||
616 | # cache_repo |
|
616 | # cache_repo | |
617 | _string_setting( |
|
617 | _string_setting( | |
618 | settings, |
|
618 | settings, | |
619 | 'rc_cache.cache_repo.backend', |
|
619 | 'rc_cache.cache_repo.backend', | |
620 | 'dogpile.cache.rc.file_namespace', lower=False) |
|
620 | 'dogpile.cache.rc.file_namespace', lower=False) | |
621 | _int_setting( |
|
621 | _int_setting( | |
622 | settings, |
|
622 | settings, | |
623 | 'rc_cache.cache_repo.expiration_time', |
|
623 | 'rc_cache.cache_repo.expiration_time', | |
624 | 60) |
|
624 | 60) | |
625 | _string_setting( |
|
625 | _string_setting( | |
626 | settings, |
|
626 | settings, | |
627 | 'rc_cache.cache_repo.arguments.filename', |
|
627 | 'rc_cache.cache_repo.arguments.filename', | |
628 | os.path.join(default_cache_dir, 'rc_cache_2'), lower=False) |
|
628 | os.path.join(default_cache_dir, 'rc_cache_2'), lower=False) | |
629 |
|
629 | |||
630 | # cache_license |
|
630 | # cache_license | |
631 | _string_setting( |
|
631 | _string_setting( | |
632 | settings, |
|
632 | settings, | |
633 | 'rc_cache.cache_license.backend', |
|
633 | 'rc_cache.cache_license.backend', | |
634 | 'dogpile.cache.rc.file_namespace', lower=False) |
|
634 | 'dogpile.cache.rc.file_namespace', lower=False) | |
635 | _int_setting( |
|
635 | _int_setting( | |
636 | settings, |
|
636 | settings, | |
637 | 'rc_cache.cache_license.expiration_time', |
|
637 | 'rc_cache.cache_license.expiration_time', | |
638 | 5*60) |
|
638 | 5*60) | |
639 | _string_setting( |
|
639 | _string_setting( | |
640 | settings, |
|
640 | settings, | |
641 | 'rc_cache.cache_license.arguments.filename', |
|
641 | 'rc_cache.cache_license.arguments.filename', | |
642 | os.path.join(default_cache_dir, 'rc_cache_3'), lower=False) |
|
642 | os.path.join(default_cache_dir, 'rc_cache_3'), lower=False) | |
643 |
|
643 | |||
644 | # cache_repo_longterm memory, 96H |
|
644 | # cache_repo_longterm memory, 96H | |
645 | _string_setting( |
|
645 | _string_setting( | |
646 | settings, |
|
646 | settings, | |
647 | 'rc_cache.cache_repo_longterm.backend', |
|
647 | 'rc_cache.cache_repo_longterm.backend', | |
648 | 'dogpile.cache.rc.memory_lru', lower=False) |
|
648 | 'dogpile.cache.rc.memory_lru', lower=False) | |
649 | _int_setting( |
|
649 | _int_setting( | |
650 | settings, |
|
650 | settings, | |
651 | 'rc_cache.cache_repo_longterm.expiration_time', |
|
651 | 'rc_cache.cache_repo_longterm.expiration_time', | |
652 | 345600) |
|
652 | 345600) | |
653 | _int_setting( |
|
653 | _int_setting( | |
654 | settings, |
|
654 | settings, | |
655 | 'rc_cache.cache_repo_longterm.max_size', |
|
655 | 'rc_cache.cache_repo_longterm.max_size', | |
656 | 10000) |
|
656 | 10000) | |
657 |
|
657 | |||
658 | # sql_cache_short |
|
658 | # sql_cache_short | |
659 | _string_setting( |
|
659 | _string_setting( | |
660 | settings, |
|
660 | settings, | |
661 | 'rc_cache.sql_cache_short.backend', |
|
661 | 'rc_cache.sql_cache_short.backend', | |
662 | 'dogpile.cache.rc.memory_lru', lower=False) |
|
662 | 'dogpile.cache.rc.memory_lru', lower=False) | |
663 | _int_setting( |
|
663 | _int_setting( | |
664 | settings, |
|
664 | settings, | |
665 | 'rc_cache.sql_cache_short.expiration_time', |
|
665 | 'rc_cache.sql_cache_short.expiration_time', | |
666 | 30) |
|
666 | 30) | |
667 | _int_setting( |
|
667 | _int_setting( | |
668 | settings, |
|
668 | settings, | |
669 | 'rc_cache.sql_cache_short.max_size', |
|
669 | 'rc_cache.sql_cache_short.max_size', | |
670 | 10000) |
|
670 | 10000) | |
671 |
|
671 | |||
672 |
|
672 | |||
673 | def _int_setting(settings, name, default): |
|
673 | def _int_setting(settings, name, default): | |
674 | settings[name] = int(settings.get(name, default)) |
|
674 | settings[name] = int(settings.get(name, default)) | |
675 | return settings[name] |
|
675 | return settings[name] | |
676 |
|
676 | |||
677 |
|
677 | |||
678 | def _bool_setting(settings, name, default): |
|
678 | def _bool_setting(settings, name, default): | |
679 | input_val = settings.get(name, default) |
|
679 | input_val = settings.get(name, default) | |
680 | if isinstance(input_val, unicode): |
|
680 | if isinstance(input_val, unicode): | |
681 | input_val = input_val.encode('utf8') |
|
681 | input_val = input_val.encode('utf8') | |
682 | settings[name] = asbool(input_val) |
|
682 | settings[name] = asbool(input_val) | |
683 | return settings[name] |
|
683 | return settings[name] | |
684 |
|
684 | |||
685 |
|
685 | |||
686 | def _list_setting(settings, name, default): |
|
686 | def _list_setting(settings, name, default): | |
687 | raw_value = settings.get(name, default) |
|
687 | raw_value = settings.get(name, default) | |
688 |
|
688 | |||
689 | old_separator = ',' |
|
689 | old_separator = ',' | |
690 | if old_separator in raw_value: |
|
690 | if old_separator in raw_value: | |
691 | # If we get a comma separated list, pass it to our own function. |
|
691 | # If we get a comma separated list, pass it to our own function. | |
692 | settings[name] = rhodecode_aslist(raw_value, sep=old_separator) |
|
692 | settings[name] = rhodecode_aslist(raw_value, sep=old_separator) | |
693 | else: |
|
693 | else: | |
694 | # Otherwise we assume it uses pyramids space/newline separation. |
|
694 | # Otherwise we assume it uses pyramids space/newline separation. | |
695 | settings[name] = aslist(raw_value) |
|
695 | settings[name] = aslist(raw_value) | |
696 | return settings[name] |
|
696 | return settings[name] | |
697 |
|
697 | |||
698 |
|
698 | |||
699 | def _string_setting(settings, name, default, lower=True, default_when_empty=False): |
|
699 | def _string_setting(settings, name, default, lower=True, default_when_empty=False): | |
700 | value = settings.get(name, default) |
|
700 | value = settings.get(name, default) | |
701 |
|
701 | |||
702 | if default_when_empty and not value: |
|
702 | if default_when_empty and not value: | |
703 | # use default value when value is empty |
|
703 | # use default value when value is empty | |
704 | value = default |
|
704 | value = default | |
705 |
|
705 | |||
706 | if lower: |
|
706 | if lower: | |
707 | value = value.lower() |
|
707 | value = value.lower() | |
708 | settings[name] = value |
|
708 | settings[name] = value | |
709 | return settings[name] |
|
709 | return settings[name] | |
710 |
|
710 | |||
711 |
|
711 | |||
712 | def _substitute_values(mapping, substitutions): |
|
712 | def _substitute_values(mapping, substitutions): | |
713 | result = {} |
|
713 | result = {} | |
714 |
|
714 | |||
715 | try: |
|
715 | try: | |
716 | for key, value in mapping.items(): |
|
716 | for key, value in mapping.items(): | |
717 | # initialize without substitution first |
|
717 | # initialize without substitution first | |
718 | result[key] = value |
|
718 | result[key] = value | |
719 |
|
719 | |||
720 | # Note: Cannot use regular replacements, since they would clash |
|
720 | # Note: Cannot use regular replacements, since they would clash | |
721 | # with the implementation of ConfigParser. Using "format" instead. |
|
721 | # with the implementation of ConfigParser. Using "format" instead. | |
722 | try: |
|
722 | try: | |
723 | result[key] = value.format(**substitutions) |
|
723 | result[key] = value.format(**substitutions) | |
724 | except KeyError as e: |
|
724 | except KeyError as e: | |
725 | env_var = '{}'.format(e.args[0]) |
|
725 | env_var = '{}'.format(e.args[0]) | |
726 |
|
726 | |||
727 | msg = 'Failed to substitute: `{key}={{{var}}}` with environment entry. ' \ |
|
727 | msg = 'Failed to substitute: `{key}={{{var}}}` with environment entry. ' \ | |
728 | 'Make sure your environment has {var} set, or remove this ' \ |
|
728 | 'Make sure your environment has {var} set, or remove this ' \ | |
729 | 'variable from config file'.format(key=key, var=env_var) |
|
729 | 'variable from config file'.format(key=key, var=env_var) | |
730 |
|
730 | |||
731 | if env_var.startswith('ENV_'): |
|
731 | if env_var.startswith('ENV_'): | |
732 | raise ValueError(msg) |
|
732 | raise ValueError(msg) | |
733 | else: |
|
733 | else: | |
734 | log.warning(msg) |
|
734 | log.warning(msg) | |
735 |
|
735 | |||
736 | except ValueError as e: |
|
736 | except ValueError as e: | |
737 | log.warning('Failed to substitute ENV variable: %s', e) |
|
737 | log.warning('Failed to substitute ENV variable: %s', e) | |
738 | result = mapping |
|
738 | result = mapping | |
739 |
|
739 | |||
740 | return result |
|
740 | return result |
General Comments 0
You need to be logged in to leave comments.
Login now