#!/usr/bin/python3

import time
import bottle
import pygments
import pygments.formatters
import pygments.lexers
import identigen
import config
from pathlib import Path

application = bottle.default_app() # application used for wsgi mode
pathbase = Path(config.pastedir)
pygment_formater = pygments.formatters.HtmlFormatter(linenos="table")

if not pathbase.exists():
    pathbase.mkdir(mode=0o700, parents=True)

@bottle.route('/')
def route_root():
    return bottle.template('root')

@bottle.route('/', method='POST')
def route_paste_post():
    content = bottle.request.forms.getunicode('content', '') or ''
    pid = identigen.generate(content)
    path = pathbase / pid
    with path.open(mode='wb') as fd:
        fd.write(content.encode('utf8'))
    bottle.redirect('/' + pid)

@bottle.route('/static/<path:path>')
def route_static(path):
    return bottle.static_file(path, root='static')

@bottle.route('/favicon.ico')
def route_favicon():
    return bottle.static_file('favicon.ico', root='static')

@bottle.route('/robots.txt')
def route_robots():
    return bottle.static_file('robots.txt', root='static')

@bottle.route('/<pid>')
@bottle.route('/<pid>/<pformat>')
def route_paste_get(pid, pformat='colored'):
    if pformat != 'colored' and pformat != 'raw':
        return bottle.template('bad_format')
    path = pathbase / pid
    try:
        with path.open(mode='rb') as fd:
            content = fd.read().decode('utf8')
    except IOError:
        # use this template for all file based exception
        bottle.abort(404)
    if pformat == 'colored':
        try:
            lexer = pygments.lexers.guess_lexer(content)
        except pygments.util.ClassNotFound:
            lexer = pygments.lexers.special.TextLexer()
        content = pygments.highlight(content, lexer, pygment_formater)
        timeleft = path.lstat()[9] + config.timeout * 24 * 3600
        return bottle.template('paste',
                               content=content,
                               timeleft=time.ctime(timeleft),
                               pid=pid)
    bottle.response.content_type = 'text/plain; charset=UTF8' # HTTP header
    return content

@bottle.error(404)
def error404(error):
    return bottle.template("not_found")

@bottle.error(400)
def error400(error):
    return bottle.template("bad_request")

if __name__ == '__main__':
    print('I: Starting application with development server')
    bottle.run(host='0.0.0.0', port=8080, debug=True, reloader=True)
else:
    print('I: Starting application as a wsgi application')