mirror of
https://github.com/Tygs/0bin.git
synced 2023-08-10 21:13:00 +03:00
142 lines
4.0 KiB
Python
142 lines
4.0 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# vim: ai ts=4 sts=4 et sw=4
|
|
|
|
"""
|
|
Main script including controller, rooting, dependancy management, and
|
|
server run.
|
|
"""
|
|
|
|
import thread
|
|
|
|
from datetime import datetime, timedelta
|
|
|
|
# add project dir and libs dir to the PYTHON PATH to ensure they are
|
|
# importable
|
|
from utils import settings
|
|
|
|
import bottle
|
|
from bottle import (Bottle, run, static_file, view, request)
|
|
|
|
import clize
|
|
|
|
from paste import Paste
|
|
from utils import drop_privileges, dmerge
|
|
|
|
|
|
app = Bottle()
|
|
GLOBAL_CONTEXT = {
|
|
'settings': settings
|
|
}
|
|
|
|
|
|
@app.route('/')
|
|
@view('home')
|
|
def index():
|
|
return GLOBAL_CONTEXT
|
|
|
|
|
|
@app.route('/paste/create', method='POST')
|
|
def create_paste():
|
|
|
|
try:
|
|
content = unicode(request.forms.get('content', ''), 'utf8')
|
|
except UnicodeDecodeError:
|
|
return {'status': 'error',
|
|
'message': u"Encoding error: the paste couldn't be saved."}
|
|
|
|
if '{"iv":' not in content: # reject silently non encrypted content
|
|
return ''
|
|
|
|
if content:
|
|
# check size of the paste. if more than settings return error without saving paste.
|
|
# prevent from unusual use of the system.
|
|
# need to be improved
|
|
if len(content) < settings.MAX_SIZE:
|
|
expiration = request.forms.get('expiration', u'burn_after_reading')
|
|
paste = Paste(expiration=expiration, content=content)
|
|
paste.save()
|
|
return {'status': 'ok',
|
|
'paste': paste.uuid}
|
|
|
|
return {'status': 'error',
|
|
'message': u"Serveur error: the paste couldn't be saved. Please try later."}
|
|
|
|
|
|
@app.route('/paste/:paste_id')
|
|
@view('paste')
|
|
def display_paste(paste_id):
|
|
|
|
now = datetime.now()
|
|
keep_alive = False
|
|
try:
|
|
paste = Paste.load(paste_id)
|
|
# Delete the paste if it expired:
|
|
if 'burn_after_reading' in str(paste.expiration):
|
|
# burn_after_reading contains the paste creation date
|
|
# if this read appends 10 seconds after the creation date
|
|
# we don't delete the paste because it means it's the redirection
|
|
# to the paste that happens during the paste creation
|
|
try:
|
|
keep_alive = paste.expiration.split('#')[1]
|
|
keep_alive = datetime.strptime(keep_alive, '%Y-%m-%d %H:%M:%S.%f')
|
|
keep_alive = now < keep_alive + timedelta(seconds=10)
|
|
except IndexError:
|
|
keep_alive = False
|
|
if not keep_alive:
|
|
paste.delete()
|
|
|
|
elif paste.expiration < now:
|
|
paste.delete()
|
|
raise ValueError()
|
|
|
|
except (TypeError, ValueError):
|
|
#abort(404, u"This paste doesn't exist or has expired")
|
|
return error404(ValueError)
|
|
|
|
context = {'paste': paste, 'keep_alive': keep_alive}
|
|
return dmerge(context, GLOBAL_CONTEXT)
|
|
|
|
|
|
@app.error(404)
|
|
@view('404')
|
|
def error404(code):
|
|
return GLOBAL_CONTEXT
|
|
|
|
|
|
@clize.clize
|
|
def runserver(host='', port='', debug=None, serve_static='', user='',
|
|
group='', settings_file=''):
|
|
|
|
# merge the settings
|
|
if settings_file:
|
|
settings.update_with_file(settings_file)
|
|
|
|
settings.HOST = host or settings.HOST
|
|
settings.PORT = port or settings.PORT
|
|
settings.DEBUG = debug if debug is not None else settings.DEBUG
|
|
settings.STATIC_FILES_ROOT = serve_static or settings.STATIC_FILES_ROOT
|
|
settings.USER = user or settings.USER
|
|
settings.GROUP = group or settings.GROUP
|
|
|
|
# make sure the templates can be loaded
|
|
for d in reversed(settings.TEMPLATE_DIRS):
|
|
bottle.TEMPLATE_PATH.insert(0, d)
|
|
|
|
if serve_static:
|
|
@app.route('/static/<filename:path>')
|
|
def server_static(filename):
|
|
return static_file(filename, root=settings.STATIC_FILES_ROOT)
|
|
|
|
thread.start_new_thread(drop_privileges, (settings.USER, settings.GROUP))
|
|
|
|
if settings.DEBUG:
|
|
bottle.debug(True)
|
|
run(app, host=settings.HOST, port=settings.PORT, reloader=True, server="cherrypy")
|
|
else:
|
|
run(app, host=settings.HOST, port=settings.PORT, server="cherrypy")
|
|
|
|
|
|
def main():
|
|
clize.run(runserver)
|