|
|
# -*- coding: utf-8 -*-
|
|
|
# This program is free software: you can redistribute it and/or modify
|
|
|
# it under the terms of the GNU General Public License as published by
|
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
|
# (at your option) any later version.
|
|
|
#
|
|
|
# This program is distributed in the hope that it will be useful,
|
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
# GNU General Public License for more details.
|
|
|
#
|
|
|
# You should have received a copy of the GNU General Public License
|
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
"""
|
|
|
rhodecode.controllers.api
|
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
|
|
|
|
JSON RPC controller
|
|
|
|
|
|
:created_on: Aug 20, 2011
|
|
|
:author: marcink
|
|
|
:copyright: (c) 2013 RhodeCode GmbH.
|
|
|
:license: GPLv3, see LICENSE for more details.
|
|
|
"""
|
|
|
|
|
|
import inspect
|
|
|
import logging
|
|
|
import types
|
|
|
import urllib
|
|
|
import traceback
|
|
|
import time
|
|
|
|
|
|
from paste.response import replace_header
|
|
|
from pylons.controllers import WSGIController
|
|
|
|
|
|
from webob.exc import HTTPError
|
|
|
|
|
|
from rhodecode.model.db import User
|
|
|
from rhodecode.model import meta
|
|
|
from rhodecode.lib.compat import izip_longest, json
|
|
|
from rhodecode.lib.auth import AuthUser
|
|
|
from rhodecode.lib.base import _get_ip_addr as _get_ip, _get_access_path
|
|
|
from rhodecode.lib.utils2 import safe_unicode, safe_str
|
|
|
|
|
|
log = logging.getLogger('JSONRPC')
|
|
|
|
|
|
|
|
|
class JSONRPCError(BaseException):
|
|
|
|
|
|
def __init__(self, message):
|
|
|
self.message = message
|
|
|
super(JSONRPCError, self).__init__()
|
|
|
|
|
|
def __str__(self):
|
|
|
return safe_str(self.message)
|
|
|
|
|
|
|
|
|
def jsonrpc_error(message, retid=None, code=None):
|
|
|
"""
|
|
|
Generate a Response object with a JSON-RPC error body
|
|
|
|
|
|
:param code:
|
|
|
:param retid:
|
|
|
:param message:
|
|
|
"""
|
|
|
from pylons.controllers.util import Response
|
|
|
return Response(
|
|
|
body=json.dumps(dict(id=retid, result=None, error=message)),
|
|
|
status=code,
|
|
|
content_type='application/json'
|
|
|
)
|
|
|
|
|
|
|
|
|
class JSONRPCController(WSGIController):
|
|
|
"""
|
|
|
A WSGI-speaking JSON-RPC controller class
|
|
|
|
|
|
See the specification:
|
|
|
<http://json-rpc.org/wiki/specification>`.
|
|
|
|
|
|
Valid controller return values should be json-serializable objects.
|
|
|
|
|
|
Sub-classes should catch their exceptions and raise JSONRPCError
|
|
|
if they want to pass meaningful errors to the client.
|
|
|
|
|
|
"""
|
|
|
|
|
|
def _get_ip_addr(self, environ):
|
|
|
return _get_ip(environ)
|
|
|
|
|
|
def _get_method_args(self):
|
|
|
"""
|
|
|
Return `self._rpc_args` to dispatched controller method
|
|
|
chosen by __call__
|
|
|
"""
|
|
|
return self._rpc_args
|
|
|
|
|
|
def __call__(self, environ, start_response):
|
|
|
"""
|
|
|
Parse the request body as JSON, look up the method on the
|
|
|
controller and if it exists, dispatch to it.
|
|
|
"""
|
|
|
try:
|
|
|
return self._handle_request(environ, start_response)
|
|
|
finally:
|
|
|
meta.Session.remove()
|
|
|
|
|
|
def _handle_request(self, environ, start_response):
|
|
|
start = time.time()
|
|
|
ip_addr = self.ip_addr = self._get_ip_addr(environ)
|
|
|
self._req_id = None
|
|
|
if 'CONTENT_LENGTH' not in environ:
|
|
|
log.debug("No Content-Length")
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message="No Content-Length in request")
|
|
|
else:
|
|
|
length = environ['CONTENT_LENGTH'] or 0
|
|
|
length = int(environ['CONTENT_LENGTH'])
|
|
|
log.debug('Content-Length: %s' % length)
|
|
|
|
|
|
if length == 0:
|
|
|
log.debug("Content-Length is 0")
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message="Content-Length is 0")
|
|
|
|
|
|
raw_body = environ['wsgi.input'].read(length)
|
|
|
|
|
|
try:
|
|
|
json_body = json.loads(raw_body)
|
|
|
except ValueError, e:
|
|
|
# catch JSON errors Here
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message="JSON parse error ERR:%s RAW:%r"
|
|
|
% (e, raw_body))
|
|
|
|
|
|
# check AUTH based on API KEY
|
|
|
try:
|
|
|
self._req_api_key = json_body['api_key']
|
|
|
self._req_id = json_body['id']
|
|
|
self._req_method = json_body['method']
|
|
|
self._request_params = json_body['args']
|
|
|
if not isinstance(self._request_params, dict):
|
|
|
self._request_params = {}
|
|
|
|
|
|
log.debug(
|
|
|
'method: %s, params: %s' % (self._req_method,
|
|
|
self._request_params)
|
|
|
)
|
|
|
except KeyError, e:
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message='Incorrect JSON query missing %s' % e)
|
|
|
|
|
|
# check if we can find this session using api_key
|
|
|
try:
|
|
|
u = User.get_by_api_key(self._req_api_key)
|
|
|
if u is None:
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message='Invalid API KEY')
|
|
|
|
|
|
#check if we are allowed to use this IP
|
|
|
auth_u = AuthUser(u.user_id, self._req_api_key, ip_addr=ip_addr)
|
|
|
if not auth_u.ip_allowed:
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message='request from IP:%s not allowed' % (ip_addr,))
|
|
|
else:
|
|
|
log.info('Access for IP:%s allowed' % (ip_addr,))
|
|
|
|
|
|
except Exception, e:
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message='Invalid API KEY')
|
|
|
|
|
|
self._error = None
|
|
|
try:
|
|
|
self._func = self._find_method()
|
|
|
except AttributeError, e:
|
|
|
return jsonrpc_error(retid=self._req_id,
|
|
|
message=str(e))
|
|
|
|
|
|
# now that we have a method, add self._req_params to
|
|
|
# self.kargs and dispatch control to WGIController
|
|
|
argspec = inspect.getargspec(self._func)
|
|
|
arglist = argspec[0][1:]
|
|
|
defaults = map(type, argspec[3] or [])
|
|
|
default_empty = types.NotImplementedType
|
|
|
|
|
|
# kw arguments required by this method
|
|
|
func_kwargs = dict(izip_longest(reversed(arglist), reversed(defaults),
|
|
|
fillvalue=default_empty))
|
|
|
|
|
|
# this is little trick to inject logged in user for
|
|
|
# perms decorators to work they expect the controller class to have
|
|
|
# rhodecode_user attribute set
|
|
|
self.rhodecode_user = auth_u
|
|
|
|
|
|
# This attribute will need to be first param of a method that uses
|
|
|
# api_key, which is translated to instance of user at that name
|
|
|
USER_SESSION_ATTR = 'apiuser'
|
|
|
|
|
|
if USER_SESSION_ATTR not in arglist:
|
|
|
return jsonrpc_error(
|
|
|
retid=self._req_id,
|
|
|
message='This method [%s] does not support '
|
|
|
'authentication (missing %s param)' % (
|
|
|
self._func.__name__, USER_SESSION_ATTR)
|
|
|
)
|
|
|
|
|
|
# get our arglist and check if we provided them as args
|
|
|
for arg, default in func_kwargs.iteritems():
|
|
|
if arg == USER_SESSION_ATTR:
|
|
|
# USER_SESSION_ATTR is something translated from api key and
|
|
|
# this is checked before so we don't need validate it
|
|
|
continue
|
|
|
|
|
|
# skip the required param check if it's default value is
|
|
|
# NotImplementedType (default_empty)
|
|
|
if default == default_empty and arg not in self._request_params:
|
|
|
return jsonrpc_error(
|
|
|
retid=self._req_id,
|
|
|
message=(
|
|
|
'Missing non optional `%s` arg in JSON DATA' % arg
|
|
|
)
|
|
|
)
|
|
|
|
|
|
self._rpc_args = {USER_SESSION_ATTR: u}
|
|
|
|
|
|
self._rpc_args.update(self._request_params)
|
|
|
|
|
|
self._rpc_args['action'] = self._req_method
|
|
|
self._rpc_args['environ'] = environ
|
|
|
self._rpc_args['start_response'] = start_response
|
|
|
|
|
|
status = []
|
|
|
headers = []
|
|
|
exc_info = []
|
|
|
|
|
|
def change_content(new_status, new_headers, new_exc_info=None):
|
|
|
status.append(new_status)
|
|
|
headers.extend(new_headers)
|
|
|
exc_info.append(new_exc_info)
|
|
|
|
|
|
output = WSGIController.__call__(self, environ, change_content)
|
|
|
output = list(output)
|
|
|
headers.append(('Content-Length', str(len(output[0]))))
|
|
|
replace_header(headers, 'Content-Type', 'application/json')
|
|
|
start_response(status[0], headers, exc_info[0])
|
|
|
log.info('IP: %s Request to %s time: %.3fs' % (
|
|
|
self._get_ip_addr(environ),
|
|
|
safe_unicode(_get_access_path(environ)), time.time() - start)
|
|
|
)
|
|
|
return output
|
|
|
|
|
|
def _dispatch_call(self):
|
|
|
"""
|
|
|
Implement dispatch interface specified by WSGIController
|
|
|
"""
|
|
|
raw_response = ''
|
|
|
try:
|
|
|
raw_response = self._inspect_call(self._func)
|
|
|
if isinstance(raw_response, HTTPError):
|
|
|
self._error = str(raw_response)
|
|
|
except JSONRPCError, e:
|
|
|
self._error = safe_str(e)
|
|
|
except Exception, e:
|
|
|
log.error('Encountered unhandled exception: %s'
|
|
|
% (traceback.format_exc(),))
|
|
|
json_exc = JSONRPCError('Internal server error')
|
|
|
self._error = safe_str(json_exc)
|
|
|
|
|
|
if self._error is not None:
|
|
|
raw_response = None
|
|
|
|
|
|
response = dict(id=self._req_id, result=raw_response, error=self._error)
|
|
|
try:
|
|
|
return json.dumps(response)
|
|
|
except TypeError, e:
|
|
|
log.error('API FAILED. Error encoding response: %s' % e)
|
|
|
return json.dumps(
|
|
|
dict(
|
|
|
id=self._req_id,
|
|
|
result=None,
|
|
|
error="Error encoding response"
|
|
|
)
|
|
|
)
|
|
|
|
|
|
def _find_method(self):
|
|
|
"""
|
|
|
Return method named by `self._req_method` in controller if able
|
|
|
"""
|
|
|
log.debug('Trying to find JSON-RPC method: %s' % (self._req_method,))
|
|
|
if self._req_method.startswith('_'):
|
|
|
raise AttributeError("Method not allowed")
|
|
|
|
|
|
try:
|
|
|
func = getattr(self, self._req_method, None)
|
|
|
except UnicodeEncodeError:
|
|
|
raise AttributeError("Problem decoding unicode in requested "
|
|
|
"method name.")
|
|
|
|
|
|
if isinstance(func, types.MethodType):
|
|
|
return func
|
|
|
else:
|
|
|
raise AttributeError("No such method: %s" % (self._req_method,))
|
|
|
|
