2019-05-26 00:42:39 -04:00
|
|
|
# cython: language_level=3
|
2019-05-21 07:54:52 -04:00
|
|
|
import logging
|
2019-07-22 08:15:54 -04:00
|
|
|
import time
|
2019-05-21 07:54:52 -04:00
|
|
|
|
2019-06-04 08:14:12 -04:00
|
|
|
import asyncio
|
2019-05-20 06:49:08 -04:00
|
|
|
from aiohttp import web
|
2019-05-21 07:54:52 -04:00
|
|
|
from aiojobs.aiohttp import atomic
|
|
|
|
|
2019-07-24 01:43:50 -04:00
|
|
|
from src.server.misc import clean_input_params
|
|
|
|
from src.server.database import get_claim_comments
|
|
|
|
from src.server.database import get_comments_by_id, get_comment_ids
|
|
|
|
from src.server.database import get_channel_id_from_comment_id
|
|
|
|
from src.server.database import obtain_connection
|
|
|
|
from src.server.misc import is_valid_base_comment
|
|
|
|
from src.server.misc import is_valid_credential_input
|
|
|
|
from src.server.misc import make_error
|
|
|
|
from src.server.writes import delete_comment_if_authorized, write_comment
|
2019-05-21 06:56:27 -04:00
|
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
2019-05-20 06:49:08 -04:00
|
|
|
|
2019-07-19 00:33:34 -04:00
|
|
|
# noinspection PyUnusedLocal
|
|
|
|
def ping(*args):
|
|
|
|
return 'pong'
|
2019-05-20 06:49:08 -04:00
|
|
|
|
2019-07-16 20:20:33 -04:00
|
|
|
|
2019-07-19 00:33:34 -04:00
|
|
|
def handle_get_channel_from_comment_id(app, kwargs: dict):
|
2019-07-24 01:43:50 -04:00
|
|
|
with app['reader'] as conn:
|
2019-07-20 09:10:59 -04:00
|
|
|
return get_channel_id_from_comment_id(conn, **kwargs)
|
2019-05-20 06:49:08 -04:00
|
|
|
|
|
|
|
|
2019-07-16 20:20:33 -04:00
|
|
|
def handle_get_comment_ids(app, kwargs):
|
2019-07-24 01:43:50 -04:00
|
|
|
with app['reader'] as conn:
|
2019-05-21 05:00:01 -04:00
|
|
|
return get_comment_ids(conn, **kwargs)
|
2019-05-20 06:49:08 -04:00
|
|
|
|
|
|
|
|
2019-07-16 20:20:33 -04:00
|
|
|
def handle_get_claim_comments(app, kwargs):
|
2019-07-24 01:43:50 -04:00
|
|
|
with app['reader'] as conn:
|
2019-05-21 05:00:01 -04:00
|
|
|
return get_claim_comments(conn, **kwargs)
|
2019-05-20 06:49:08 -04:00
|
|
|
|
|
|
|
|
2019-07-16 20:20:33 -04:00
|
|
|
def handle_get_comments_by_id(app, kwargs):
|
2019-07-24 01:43:50 -04:00
|
|
|
with app['reader'] as conn:
|
2019-05-21 05:00:01 -04:00
|
|
|
return get_comments_by_id(conn, **kwargs)
|
2019-05-20 06:49:08 -04:00
|
|
|
|
|
|
|
|
2019-07-19 00:33:34 -04:00
|
|
|
async def handle_create_comment(app, params):
|
2019-07-22 08:15:54 -04:00
|
|
|
if is_valid_base_comment(**params) and is_valid_credential_input(**params):
|
|
|
|
job = await app['comment_scheduler'].spawn(write_comment(app, params))
|
|
|
|
return await job.wait()
|
|
|
|
else:
|
|
|
|
raise ValueError('base comment is invalid')
|
2019-07-19 00:33:34 -04:00
|
|
|
|
|
|
|
|
|
|
|
async def handle_delete_comment(app, params):
|
|
|
|
return await delete_comment_if_authorized(app, **params)
|
2019-05-20 06:49:08 -04:00
|
|
|
|
|
|
|
|
|
|
|
METHODS = {
|
|
|
|
'ping': ping,
|
|
|
|
'get_claim_comments': handle_get_claim_comments,
|
|
|
|
'get_comment_ids': handle_get_comment_ids,
|
|
|
|
'get_comments_by_id': handle_get_comments_by_id,
|
2019-07-19 00:33:34 -04:00
|
|
|
'get_channel_from_comment_id': handle_get_channel_from_comment_id,
|
|
|
|
'create_comment': handle_create_comment,
|
2019-07-24 02:26:47 -04:00
|
|
|
# 'delete_comment': handle_delete_comment,
|
|
|
|
# 'abandon_comment': handle_delete_comment,
|
2019-05-20 06:49:08 -04:00
|
|
|
}
|
|
|
|
|
|
|
|
|
2019-05-21 05:00:01 -04:00
|
|
|
async def process_json(app, body: dict) -> dict:
|
2019-05-20 06:49:08 -04:00
|
|
|
response = {'jsonrpc': '2.0', 'id': body['id']}
|
|
|
|
if body['method'] in METHODS:
|
|
|
|
method = body['method']
|
|
|
|
params = body.get('params', {})
|
2019-07-16 20:20:33 -04:00
|
|
|
clean_input_params(params)
|
2019-07-20 09:06:34 -04:00
|
|
|
logger.debug(f'Received Method {method}, params: {params}')
|
2019-05-20 06:49:08 -04:00
|
|
|
try:
|
2019-07-22 11:10:55 -04:00
|
|
|
start = time.time()
|
2019-05-21 05:00:01 -04:00
|
|
|
if asyncio.iscoroutinefunction(METHODS[method]):
|
2019-07-19 00:33:34 -04:00
|
|
|
result = await METHODS[method](app, params)
|
2019-05-20 06:49:08 -04:00
|
|
|
else:
|
2019-07-16 20:20:33 -04:00
|
|
|
result = METHODS[method](app, params)
|
2019-05-20 06:49:08 -04:00
|
|
|
response['result'] = result
|
2019-07-20 09:06:34 -04:00
|
|
|
except Exception as err:
|
|
|
|
logger.exception(f'Got {type(err).__name__}: {err}')
|
|
|
|
if type(err) in (ValueError, TypeError):
|
|
|
|
response['error'] = make_error('INVALID_PARAMS', err)
|
|
|
|
else:
|
|
|
|
response['error'] = make_error('INTERNAL', err)
|
2019-07-22 11:10:55 -04:00
|
|
|
finally:
|
|
|
|
end = time.time()
|
|
|
|
logger.debug(f'Time taken to process {method}: {end - start} secs')
|
2019-05-20 06:49:08 -04:00
|
|
|
else:
|
2019-07-20 09:10:14 -04:00
|
|
|
response['error'] = make_error('METHOD_NOT_FOUND')
|
2019-05-20 06:49:08 -04:00
|
|
|
return response
|
|
|
|
|
|
|
|
|
2019-05-21 05:00:01 -04:00
|
|
|
@atomic
|
|
|
|
async def api_endpoint(request: web.Request):
|
2019-05-20 06:49:08 -04:00
|
|
|
try:
|
2019-06-04 08:14:12 -04:00
|
|
|
body = await request.json()
|
2019-05-20 06:49:08 -04:00
|
|
|
if type(body) is list or type(body) is dict:
|
|
|
|
if type(body) is list:
|
2019-07-19 00:33:34 -04:00
|
|
|
# for batching
|
2019-05-21 05:00:01 -04:00
|
|
|
return web.json_response(
|
|
|
|
[await process_json(request.app, part) for part in body]
|
|
|
|
)
|
2019-05-20 06:49:08 -04:00
|
|
|
else:
|
2019-05-21 05:00:01 -04:00
|
|
|
return web.json_response(await process_json(request.app, body))
|
2019-07-19 00:33:34 -04:00
|
|
|
except Exception as e:
|
2019-07-20 09:06:34 -04:00
|
|
|
logger.exception(f'Exception raised by request from {request.remote}: {e}')
|
|
|
|
logger.debug(f'Request headers: {request.headers}')
|
|
|
|
return make_error('INVALID_REQUEST', e)
|
2019-07-22 08:10:21 -04:00
|
|
|
|
|
|
|
|
|
|
|
async def get_api_endpoint(request: web.Request):
|
|
|
|
return web.json_response({
|
|
|
|
'text': 'OK',
|
|
|
|
'is_running': True,
|
|
|
|
'uptime': int(time.time()) - request.app['START_TIME']
|
|
|
|
})
|