import config
import redis
import logging
import rethinkdb as rdb
import os
import time
from flask import Flask, request, render_template, url_for, redirect
from flask.ext.bcrypt import Bcrypt
from flask.ext.login import LoginManager
from flask.ext.assets import Environment
from session import RedisSessionInterface
from wiki import Wiki
from util import to_canonical, remove_ext
from recaptcha.client import captcha

app = Flask(__name__)
app.config.update(config.flask)
app.debug = (config.ENV is not 'PROD')
app.secret_key = config.secret_key
app.static_path = os.sep + 'static'
app.session_interface = RedisSessionInterface()

bcrypt = Bcrypt(app)

login_manager = LoginManager()
login_manager.init_app(app)

assets = Environment(app)
assets.url = app.static_url_path
assets.directory = app.static_folder

cache = redis.StrictRedis(host=config.cache['host'], port=config.cache['port'])

conn = rdb.connect(config.db['host'], config.db['port'], db=config.db['dbname'])

if not config.db['dbname'] in rdb.db_list().run(conn) and config.ENV is not 'PROD':
    # Create default db and repo
    print "Creating DB %s" % config.db['dbname']
    rdb.db_create(config.db['dbname']).run(conn)
    for tbl in ['sites', 'users', 'pages']:
        rdb.table_create(tbl).run(conn)

repo_dir = config.repo['dir']

from models import Site

w = Wiki(repo_dir)


def redirect_url():
    return request.args.get('next') or request.referrer or url_for('index')


@app.template_filter('datetime')
def _jinja2_filter_datetime(ts):
    return time.strftime('%b %d, %Y %I:%M %p', time.localtime(ts))


@app.errorhandler(404)
def page_not_found(e):
    return render_template('errors/404.html'), 404


@app.errorhandler(500)
def page_error(e):
    logging.exception(e)
    return render_template('errors/500.html'), 500


@app.route("/")
def root():
    return redirect('/home')

@app.route("/commit/<sha>/<name>")
def commit_sha(name, sha):
    cname = to_canonical(name)

    data = w.get_page(cname, sha=sha)
    if data:
        return render_template('page/page.html', page=data)
    else:
        return redirect('/create/'+cname)


@app.route("/register", methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        response = captcha.submit(
            request.form['recaptcha_challenge_field'],
            request.form['recaptcha_response_field'],
            app.config['RECAPTCHA_PRIVATE_KEY'],
            request.remote_addr)
        if not response.is_valid:
            return redirect('/register?fail')
        else:
            return redirect("/")
    else:
        return render_template('account/register.html')


@app.route("/login", methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        pass
    else:
        return render_template('account/login.html')


@app.route("/history/<name>")
def history(name):
    history = w.get_history(name)
    return render_template('page/history.html', name=name, history=history)


@app.route("/edit/<name>", methods=['GET', 'POST'])
def edit(name):
    data = w.get_page(name)
    cname = to_canonical(name)
    if request.method == 'POST':
        edit_cname = to_canonical(request.form['name'])
        if edit_cname != cname:
            w.rename_page(cname, edit_cname)
        w.write_page(edit_cname, request.form['content'], message=request.form['message'])
        return redirect("/" + edit_cname)
    else:
        if data:
            name = remove_ext(data['name'])
            content = data['data']
            return render_template('page/edit.html', name=name, content=content)
        else:
            return redirect('/create/'+cname)


@app.route("/delete/<name>", methods=['POST'])
def delete(name):
    pass


@app.route("/create/", methods=['GET', 'POST'])
@app.route("/create/<name>", methods=['GET', 'POST'])
def create(name=None):
    cname = ""
    if name:
        cname = to_canonical(name)
        if w.get_page(cname):
            # Page exists, edit instead
            return redirect("/edit/" + cname)
    if request.method == 'POST':
        w.write_page(request.form['name'], request.form['content'],  message=request.form['message'], create=True)
        return redirect("/" + cname)
    else:
        return render_template('page/edit.html', name=cname, content="")


@app.route("/<name>")
def render(name):
    cname = to_canonical(name)
    if cname != name:
        return redirect('/' + cname)

    data = w.get_page(cname)
    if data:
        #if data.get('data'):
         #   data['data'] = markdown(data['data'])
        return render_template('page/page.html', name=cname, page=data)
    else:
        return redirect('/create/'+cname)

import ratelimit