import sqlite3
#all the imports
from flask import Flask, request, session, g, redirect,\
    url_for, render_template, flash, jsonify
from functools import wraps
import hashlib
from pytz import timezone
import pytz
from docutils import core
from datetime import datetime
import time
from email import utils
from bs4 import BeautifulSoup
import locale
from smartypants import smartypants, Attr
from email_helpers import GmailMessage

locale.setlocale(locale.LC_ALL, 'fr_FR.UTF-8')


def rstify(string):
    result = smartypants(core.publish_parts(string, writer_name='html')['fragment'],
                         Attr.set2 | Attr.w)
    return result

# configuration
app = Flask(__name__)
app.config.from_envvar('CONF')

@app.template_filter('shortify')
def shortify(string):
    soup = BeautifulSoup(string)
    try:
        return soup("p")[0].get_text()[:50]
    except IndexError:
        return string


@app.template_filter('format_date')
def format_date(date_object, format="%a %d %b %Y à %Hh%M"):
    if not date_object:
        return ""
    return (date_object.replace(tzinfo=timezone('utc')).
            astimezone(timezone(session['timezone'])).
            strftime(format))


@app.template_filter('format_ago')
def format_ago(date_object, format):
    if not date_object:
        return ""
    readable = date_object.replace(tzinfo=timezone('utc')).\
        astimezone(timezone(session['timezone'])).strftime(format)
    iso_format = date_object.strftime('%Y-%m-%dT%H:%M:%SZ')
    return '<span class=timeago title="{0}">{1}</span>'.\
        format(iso_format, readable)


@app.template_filter('format_rfc2822')
def format_rfc2822(date_object):
    timestamp = time.mktime(date_object.timetuple())
    return utils.formatdate(timestamp)


@app.template_filter('pluralize')
def pluralize(word, count, plural=None):
    if count == 0:
        return "Aucun {0}".format(word)
    elif count == 1:
        return "1 {0}".format(word)
    elif plural:
        return "{0} {1}".format(count, plural)
    else:
        return "{0} {1}s".format(count, word)


def query_db(query, args=(), one=False):
    cur = get_db().execute(query, args)
    rv = cur.fetchone() if one else cur.fetchall()
    cur.close()
    return rv


def connect_db():
    conn = sqlite3.connect(app.config['DATABASE'],
                           detect_types=sqlite3.PARSE_DECLTYPES)
    conn.row_factory = sqlite3.Row
    conn.execute("PRAGMA foreign_keys=ON")
    return conn


def get_db():
    db = getattr(g, 'sqlite_db', None)
    if db is None:
        db = g.sqlite_db = connect_db()
    return db

@app.teardown_appcontext
def close_db(error):
    """Closes the database again at the end of the request."""
    db = getattr(g, 'sqlite_db', None)
    if db:
        db.close()


def login_required(f):
    @wraps(f)
    def decorated_function(*args, **kwargs):
        if 'user_name' not in session:
            user = query_db('select * from users where id = ?',
                        (request.headers["X-Remote-User"],), True)
            session['user_name'] = user['user_name']
            session['user_id'] = user['id']
            session['timezone'] = user['timezone'] or "UTC"
            db = get_db()
            db.execute("UPDATE users SET last_seen=? WHERE id=?",
                       (datetime.utcnow(), session['user_id']))
            db.commit()
        return f(*args, **kwargs)
    return decorated_function


@app.route('/')
@app.route('/news/')
@login_required
def list_news():
    news = query_db("SELECT news.*, users.user_name, "
                    "count(comments.id) as ncomments FROM news "
                    "LEFT JOIN users ON news.user_id = users.id "
                    "LEFT JOIN comments ON news.id=comments.news_id "
                    "GROUP BY news.id ORDER BY news.date DESC")
    comments = query_db("SELECT * from comments LEFT JOIN users "
                        "ON comments.user_id = users.id "
                        "ORDER BY date DESC LIMIT 5")
    users = query_db("SELECT * from users ORDER BY last_seen DESC")
    return render_template("news/list.html", news=news, comments=comments,
                           users=users)


@app.route('/news/add/', methods=['GET', 'POST'])
@login_required
def add_news():
    if request.method == 'POST':
        content = request.form['content']
        content_cache = rstify(content)
        if 'Add' in request.form:
            db = get_db()
            cur = db.execute("INSERT INTO news "
                             "('title', 'content', 'user_id','content_cache') "
                             "VALUES (?, ?, ?, ?)",
                             (request.form['title'], content,
                              session['user_id'], content_cache))
            news_id = cur.lastrowid
            db.commit()

            # send email
            emails = query_db("SELECT email from users where notify=1")
            emails = [email["email"] for email in emails if email["email"]]
            if emails:
                message = GmailMessage()
                message['to'] = emails
                message['subject'] = request.form['title']
                message['reply-to'] = "news.horel+{}@gmail.com".format(news_id)
                message.set_content(content)
                url = url_for('show_news', news_id=news_id, _external=True)
                html_content = content_cache + \
                               ("<p style='margin-top:2em'>Vous pouvez "
                                "<a href='{0}'>Lire cette nouvelle</a> "
                                "sur le site de la famille.</p>").format(url)
                message.add_alternative(html_content, subtype='html')
                message.send()

            return redirect(url_for('show_news', news_id=news_id))
        else:
            news = {'content': content, 'content_cache': content_cache,
                    'title': request.form['title']}
            return render_template("news/news.html", news=news)
    elif request.method == 'GET':
        return render_template("news/news.html", news=None)


@app.route('/news/<int:news_id>/', methods=['GET', 'POST'])
@login_required
def show_news(news_id):
    news = query_db("SELECT * FROM news LEFT JOIN users "
                    "ON news.user_id = users.id "
                    "WHERE news.id = ?", (news_id,), True)
    if request.method == 'GET':
        comments = query_db("SELECT * FROM comments LEFT JOIN users "
                            "ON comments.user_id = users.id "
                            "WHERE comments.news_id = ? "
                            "ORDER BY date", (news_id,))
        return render_template("news/show.html", news=news, comments=comments)
    elif request.method == 'POST':
        user_id = session['user_id']
        content = request.form['content']
        if content:
            content_cache = rstify(content)
            db = get_db()
            db.execute("INSERT INTO comments "
                       "('user_id', 'content', 'news_id', 'content_cache') "
                       "VALUES (?, ?, ?, ?)",
                       (user_id, content, news_id, content_cache))
            db.commit()
        return redirect(url_for('show_news', news_id=news_id))


@app.route('/comment/<int:comment_id>', methods=['POST'])
@login_required
def update_comment(comment_id):
    comment = query_db("SELECT * FROM comments WHERE comments.id = ?",
                       (comment_id,), True)
    if comment["user_id"] == session["user_id"]:
        content = request.form['content']
        content_cache = rstify(content)
        db = get_db()
        db.execute("UPDATE comments set content=?, content_cache=? "
                   "WHERE comments.id=?",
                   (content, content_cache, comment_id))
        db.commit()
        return jsonify(content=content, content_cache=content_cache)


@app.route('/news/<int:news_id>/edit', methods=['GET', 'POST'])
@login_required
def edit_news(news_id):
    if request.method == 'GET':
        news = query_db("SELECT * from news WHERE news.id = ?",
                        (news_id,), True)
        if news["user_id"] == session['user_id']:
            return render_template('news/news.html', news=news)
    elif request.method == 'POST':
        title = request.form['title']
        content = request.form['content']
        content_cache = rstify(content)
        db = get_db()
        db.execute("UPDATE news SET 'title'=?, 'content'=?, 'content_cache'=? "
                   "WHERE news.id =?",
                   (title, content, content_cache, news_id))
        db.commit()
        return redirect(url_for('show_news', news_id=news_id))


@app.route('/user/<int:user_id>/')
@login_required
def view_user(user_id):
    user = query_db('SELECT * FROM users WHERE id= ?', (user_id,), True)
    return render_template("user/show.html", user=user)


@app.route('/articles/')
@login_required
def articles():
    return render_template("articles/list.html")


@app.route('/user/edit/', methods=['GET', 'POST'])
@login_required
def edit_user():
    if request.method == 'GET':
        user = query_db('SELECT * FROM users WHERE id= ?',
                        (session['user_id'],), True)
        user = {k: user[k] for k in user.keys() if user[k]}
        return render_template("user/edit.html", user=user, timezones=pytz.common_timezones)
    elif request.method == 'POST':
        result = {}
        try:
            if request.form['password'] == request.form['password_confirm']:
                result['password'] = (hashlib.
                                      md5(request.form['password'].encode()).
                                      hexdigest())
            else:
                flash("Les deux mots de passe ne coïncident pas")
                return render_template("user/edit.html", user=request.form,
                                       timezones=pytz.common_timezones)
        except KeyError:
            pass

        result['notify'] = "notify" in request.form
        for k in ["birthday", "nameday", "email", "phone", "adress_line1",
                "adress_line2", "adress_city_line", "timezone"]:
            if k in result.form:
                result[k] = result.form[k]
        set_string = ",".join([f"{k}=?" for k in result.keys()])
        sqlstr = f"UPDATE users SET {set_string} WHERE id=?"
        db = get_db()
        args = (*result.values(), session['user_id'])
        db.execute(sqlstr, args)
        db.commit()
        session["timezone"] = request.form["timezone"]
        return redirect(url_for('view_user', user_id=session['user_id']))


@app.route('/logout/')
@login_required
def logout():
    session.pop('user_name', None)
    session.pop('user_id', None)
    return redirect(url_for('list_news'))


@app.route('/rss.xml')
def rss():
    news = query_db("SELECT * FROM news LEFT JOIN users ON "
                    "news.user_id=users.id ORDER BY news.date desc")
    return render_template('rss.xml', news=news)


if __name__ == "__main__":
    app.run()