modernpaste hotfixes for forcing 1wk expiry until we have auth working
Signed-off-by: Ricky Elrod <codeblock@fedoraproject.org>
This commit is contained in:
parent
1652f6776c
commit
f683454f14
3 changed files with 384 additions and 0 deletions
249
roles/modernpaste/files/paste.py
Normal file
249
roles/modernpaste/files/paste.py
Normal file
|
|
@ -0,0 +1,249 @@
|
|||
import flask
|
||||
from flask_login import current_user
|
||||
from modern_paste import app
|
||||
|
||||
import config
|
||||
from uri.main import *
|
||||
from uri.paste import *
|
||||
from util.exception import *
|
||||
from api.decorators import require_form_args
|
||||
from api.decorators import require_login_api
|
||||
from api.decorators import optional_login_api
|
||||
import constants.api
|
||||
import database.attachment
|
||||
import database.paste
|
||||
import database.user
|
||||
import util.cryptography
|
||||
|
||||
import datetime
|
||||
|
||||
@app.route(PasteSubmitURI.path, methods=['POST'])
|
||||
@require_form_args(['contents'])
|
||||
@optional_login_api
|
||||
def submit_paste():
|
||||
"""
|
||||
Endpoint for submitting a new paste.
|
||||
"""
|
||||
if config.REQUIRE_LOGIN_TO_PASTE and not current_user.is_authenticated:
|
||||
return (
|
||||
flask.jsonify(constants.api.UNAUTHENTICATED_PASTES_DISABLED_FAILURE),
|
||||
constants.api.UNAUTHENTICATED_PASTES_DISABLED_FAILURE_CODE,
|
||||
)
|
||||
|
||||
data = flask.request.get_json()
|
||||
|
||||
if not config.ENABLE_PASTE_ATTACHMENTS and len(data.get('attachments', [])) > 0:
|
||||
return (
|
||||
flask.jsonify(constants.api.PASTE_ATTACHMENTS_DISABLED_FAILURE),
|
||||
constants.api.PASTE_ATTACHMENTS_DISABLED_FAILURE_CODE,
|
||||
)
|
||||
|
||||
is_attachment_too_large = [
|
||||
# The data is encoded as a string: each character takes 1 B
|
||||
# The base64-encoded string is at 4/3x larger in size than the raw file
|
||||
len(attachment.get('data', '')) * 3 / 4.0 > config.MAX_ATTACHMENT_SIZE * 1000 * 1000
|
||||
for attachment in data.get('attachments', [])
|
||||
]
|
||||
if any(is_attachment_too_large) and config.MAX_ATTACHMENT_SIZE > 0:
|
||||
return (
|
||||
flask.jsonify(constants.api.PASTE_ATTACHMENT_TOO_LARGE_FAILURE),
|
||||
constants.api.PASTE_ATTACHMENT_TOO_LARGE_FAILURE_CODE,
|
||||
)
|
||||
|
||||
try:
|
||||
new_paste = database.paste.create_new_paste(
|
||||
contents=data.get('contents'),
|
||||
user_id=current_user.user_id if current_user.is_authenticated else None,
|
||||
#expiry_time=data.get('expiry_time'),
|
||||
expiry_time=(datetime.datetime.now() + datetime.timedelta(weeks=1)).strftime('%s'),
|
||||
title=data.get('title'),
|
||||
language=data.get('language'),
|
||||
password=data.get('password'),
|
||||
# The paste is considered an API post if any of the following conditions are met:
|
||||
# (1) The referrer is null.
|
||||
# (2) The Home or PastePostInterface URIs are *not* contained within the referrer string (if a paste was
|
||||
# posted via the web interface, this is where the user should be coming from, unless the client performed
|
||||
# some black magic and spoofed the referrer string or something equally sketchy).
|
||||
is_api_post=not flask.request.referrer or not any(
|
||||
[uri in flask.request.referrer for uri in [HomeURI.full_uri(), PastePostInterfaceURI.full_uri()]]
|
||||
),
|
||||
)
|
||||
new_attachments = [
|
||||
database.attachment.create_new_attachment(
|
||||
paste_id=new_paste.paste_id,
|
||||
file_name=attachment.get('name'),
|
||||
file_size=attachment.get('size'),
|
||||
mime_type=attachment.get('mime_type'),
|
||||
file_data=attachment.get('data'),
|
||||
)
|
||||
for attachment in data.get('attachments', [])
|
||||
]
|
||||
resp_data = new_paste.as_dict().copy()
|
||||
resp_data['attachments'] = [
|
||||
{
|
||||
'name': attachment.file_name,
|
||||
'size': attachment.file_size,
|
||||
'mime_type': attachment.mime_type,
|
||||
}
|
||||
for attachment in new_attachments
|
||||
]
|
||||
return flask.jsonify(resp_data), constants.api.SUCCESS_CODE
|
||||
except:
|
||||
return flask.jsonify(constants.api.UNDEFINED_FAILURE), constants.api.UNDEFINED_FAILURE_CODE
|
||||
|
||||
|
||||
@app.route(PasteDeactivateURI.path, methods=['POST'])
|
||||
@require_form_args(['paste_id'])
|
||||
@optional_login_api
|
||||
def deactivate_paste():
|
||||
"""
|
||||
Endpoint for deactivating an existing paste.
|
||||
The user can deactivate a paste with this endpoint in two ways:
|
||||
(1) Supply a deactivation token in the request, or
|
||||
(2) Be currently logged in, and own the paste.
|
||||
"""
|
||||
data = flask.request.get_json()
|
||||
try:
|
||||
paste = database.paste.get_paste_by_id(util.cryptography.get_decid(data['paste_id']), active_only=True)
|
||||
if (current_user.is_authenticated and current_user.user_id == paste.user_id) or data.get('deactivation_token') == paste.deactivation_token:
|
||||
database.paste.deactivate_paste(paste.paste_id)
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_SUCCESS,
|
||||
constants.api.MESSAGE: None,
|
||||
'paste_id': util.cryptography.get_id_repr(paste.paste_id),
|
||||
}), constants.api.SUCCESS_CODE
|
||||
fail_msg = 'User does not own requested paste' if current_user.is_authenticated else 'Deactivation token is invalid'
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_FAULURE,
|
||||
constants.api.MESSAGE: fail_msg,
|
||||
constants.api.FAILURE: 'auth_failure',
|
||||
'paste_id': util.cryptography.get_id_repr(paste.paste_id),
|
||||
}), constants.api.AUTH_FAILURE_CODE
|
||||
except (PasteDoesNotExistException, InvalidIDException):
|
||||
return flask.jsonify(constants.api.NONEXISTENT_PASTE_FAILURE), constants.api.NONEXISTENT_PASTE_FAILURE_CODE
|
||||
except:
|
||||
return flask.jsonify(constants.api.UNDEFINED_FAILURE), constants.api.UNDEFINED_FAILURE_CODE
|
||||
|
||||
|
||||
@app.route(PasteSetPasswordURI.path, methods=['POST'])
|
||||
@require_form_args(['paste_id', 'password'], allow_blank_values=True)
|
||||
@require_login_api
|
||||
def set_paste_password():
|
||||
"""
|
||||
Modify a paste's password, unset it, or set a new one.
|
||||
"""
|
||||
data = flask.request.get_json()
|
||||
try:
|
||||
paste = database.paste.get_paste_by_id(util.cryptography.get_decid(data['paste_id']), active_only=True)
|
||||
if paste.user_id != current_user.user_id:
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_FAULURE,
|
||||
constants.api.MESSAGE: 'User does not own the specified paste',
|
||||
constants.api.FAILURE: 'auth_failure',
|
||||
'paste_id': util.cryptography.get_id_repr(paste.paste_id),
|
||||
}), constants.api.AUTH_FAILURE_CODE
|
||||
database.paste.set_paste_password(paste.paste_id, data['password'])
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_SUCCESS,
|
||||
constants.api.MESSAGE: None,
|
||||
'paste_id': util.cryptography.get_id_repr(paste.paste_id),
|
||||
}), constants.api.SUCCESS_CODE
|
||||
except (PasteDoesNotExistException, InvalidIDException):
|
||||
return flask.jsonify(constants.api.NONEXISTENT_PASTE_FAILURE), constants.api.NONEXISTENT_PASTE_FAILURE_CODE
|
||||
except:
|
||||
return flask.jsonify(constants.api.UNDEFINED_FAILURE), constants.api.UNDEFINED_FAILURE_CODE
|
||||
|
||||
|
||||
@app.route(PasteDetailsURI.path, methods=['POST'])
|
||||
@require_form_args(['paste_id'])
|
||||
def paste_details():
|
||||
"""
|
||||
Retrieve details for a particular paste ID.
|
||||
"""
|
||||
data = flask.request.get_json()
|
||||
try:
|
||||
paste = database.paste.get_paste_by_id(util.cryptography.get_decid(data['paste_id']), active_only=True)
|
||||
attachments = database.attachment.get_attachments_for_paste(util.cryptography.get_decid(data['paste_id']), active_only=True)
|
||||
paste_details_dict = paste.as_dict()
|
||||
paste_details_dict['poster_username'] = 'Anonymous'
|
||||
paste_details_dict['attachments'] = [
|
||||
attachment.as_dict()
|
||||
for attachment in attachments
|
||||
]
|
||||
if paste.user_id:
|
||||
poster = database.user.get_user_by_id(paste.user_id)
|
||||
paste_details_dict['poster_username'] = poster.username
|
||||
if not paste.password_hash or (data.get('password') and paste.password_hash == util.cryptography.secure_hash(data.get('password'))):
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_SUCCESS,
|
||||
constants.api.MESSAGE: None,
|
||||
'details': paste_details_dict,
|
||||
}), constants.api.SUCCESS_CODE
|
||||
else:
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_FAULURE,
|
||||
constants.api.MESSAGE: 'Password-protected paste: either no password or wrong password supplied',
|
||||
constants.api.FAILURE: 'password_mismatch_failure',
|
||||
'details': {},
|
||||
}), constants.api.AUTH_FAILURE_CODE
|
||||
except (PasteDoesNotExistException, UserDoesNotExistException, InvalidIDException):
|
||||
return flask.jsonify(constants.api.NONEXISTENT_PASTE_FAILURE), constants.api.NONEXISTENT_PASTE_FAILURE_CODE
|
||||
except:
|
||||
return flask.jsonify(constants.api.UNDEFINED_FAILURE), constants.api.UNDEFINED_FAILURE_CODE
|
||||
|
||||
|
||||
@app.route(PastesForUserURI.path, methods=['POST'])
|
||||
@require_login_api
|
||||
def pastes_for_user():
|
||||
"""
|
||||
Get all pastes for the currently logged in user.
|
||||
"""
|
||||
try:
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_SUCCESS,
|
||||
constants.api.MESSAGE: None,
|
||||
'pastes': [
|
||||
paste.as_dict()
|
||||
for paste in database.paste.get_all_pastes_for_user(current_user.user_id, active_only=True)
|
||||
],
|
||||
}), constants.api.SUCCESS_CODE
|
||||
except:
|
||||
return flask.jsonify(constants.api.UNDEFINED_FAILURE), constants.api.UNDEFINED_FAILURE_CODE
|
||||
|
||||
|
||||
@app.route(RecentPastesURI.path, methods=['POST'])
|
||||
@require_form_args(['page_num', 'num_per_page'])
|
||||
def recent_pastes():
|
||||
"""
|
||||
Get details for the most recent pastes.
|
||||
"""
|
||||
try:
|
||||
data = flask.request.get_json()
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_SUCCESS,
|
||||
constants.api.MESSAGE: None,
|
||||
'pastes': [
|
||||
paste.as_dict() for paste in database.paste.get_recent_pastes(data['page_num'], data['num_per_page'])
|
||||
],
|
||||
}), constants.api.SUCCESS_CODE
|
||||
except:
|
||||
return flask.jsonify(constants.api.UNDEFINED_FAILURE), constants.api.UNDEFINED_FAILURE_CODE
|
||||
|
||||
|
||||
@app.route(TopPastesURI.path, methods=['POST'])
|
||||
@require_form_args(['page_num', 'num_per_page'])
|
||||
def top_pastes():
|
||||
"""
|
||||
Get details for the top pastes.
|
||||
"""
|
||||
try:
|
||||
data = flask.request.get_json()
|
||||
return flask.jsonify({
|
||||
constants.api.RESULT: constants.api.RESULT_SUCCESS,
|
||||
constants.api.MESSAGE: None,
|
||||
'pastes': [
|
||||
paste.as_dict() for paste in database.paste.get_top_pastes(data['page_num'], data['num_per_page'])
|
||||
],
|
||||
}), constants.api.SUCCESS_CODE
|
||||
except:
|
||||
return flask.jsonify(constants.api.UNDEFINED_FAILURE), constants.api.UNDEFINED_FAILURE_CODE
|
||||
Loading…
Add table
Add a link
Reference in a new issue