import binascii
import time
from datetime import datetime, timedelta
from ipaddress import ip_address
from markupsafe import Markup

import flask

from nyaa import email, forms, models
from nyaa.extensions import db, limiter
from nyaa.utils import sha1_hash
from nyaa.views.users import get_activation_link, get_password_reset_link, get_serializer

app = flask.current_app
bp = flask.Blueprint('account', __name__)


@bp.route('/login', methods=['GET', 'POST'])
@limiter.limit('6/hour', methods=['POST'],
               error_message="You've tried logging in too many times, try again in an hour.")
def login():
    if flask.g.user:
        return flask.redirect(redirect_url())

    form = forms.LoginForm(flask.request.form)
    if flask.request.method == 'POST' and form.validate():
        if app.config['MAINTENANCE_MODE'] and not app.config['MAINTENANCE_MODE_LOGINS']:
            flask.flash(Markup('<strong>Logins are currently disabled.</strong>'), 'danger')
            return flask.redirect(flask.url_for('account.login'))

        username = form.username.data.strip()
        if not username.isascii():
            flask.flash('Invalid characters in username.', 'danger')
            return flask.redirect(flask.url_for('account.login'))
        password = form.password.data
        user = models.User.by_username(username)

        if not user:
            user = models.User.by_email(username)

        if not user or password != user.password_hash:
            flask.flash(Markup(
                '<strong>Login failed!</strong> Incorrect username or password.'), 'danger')
            return flask.redirect(flask.url_for('account.login'))

        if user.is_banned:
            ban = models.Ban.banned(user.id, None).first()
            ban_reason = ban.reason if ban else '[No reason provided]'
            ban_str = ('<strong>Login failed!</strong> You are banned. '
                       'Reason: "{0}" If you believe this is a mistake, contact '
                       'a moderator on IRC.'.format(ban_reason))
            flask.flash(Markup(ban_str), 'danger')
            return flask.redirect(flask.url_for('account.login'))

        if user.status != models.UserStatusType.ACTIVE:
            flask.flash(Markup(
                '<strong>Login failed!</strong> Account is not activated.'), 'danger')
            return flask.redirect(flask.url_for('account.login'))

        user.last_login_date = datetime.utcnow()
        user.last_login_ip = ip_address(flask.request.remote_addr).packed
        if not app.config['MAINTENANCE_MODE']:
            db.session.add(user)
            db.session.commit()

        flask.g.user = user
        flask.session['user_id'] = user.id
        flask.session.permanent = True
        flask.session.modified = True

        return flask.redirect(redirect_url())

    return flask.render_template('login.html', form=form)


@bp.route('/logout')
def logout():
    flask.g.user = None
    flask.session.permanent = False
    flask.session.modified = False

    response = flask.make_response(flask.redirect(redirect_url()))
    response.set_cookie(app.config['SESSION_COOKIE_NAME'], expires=0)
    return response


def _check_for_multi_account(ip, cooldown):
    if not cooldown:
        return False
    cooldown_timestamp = datetime.utcnow() - timedelta(seconds=cooldown)
    q = models.User.query.filter(ip == models.User.registration_ip,
                                 models.User.created_time > cooldown_timestamp)
    return db.session.query(q.exists()).scalar()


@bp.route('/register', methods=['GET', 'POST'])
def register():
    if flask.g.user:
        return flask.redirect(redirect_url())

    form = forms.RegisterForm(flask.request.form)
    if flask.request.method == 'POST' and form.validate():
        ip = ip_address(flask.request.remote_addr).packed

        if _check_for_multi_account(ip, app.config.get('PER_IP_ACCOUNT_COOLDOWN', 0)):
            flask.flash('You or somebody else has already registered an account from this IP '
                        'recently. You cannot register another one.', 'danger')
            return flask.render_template('register.html', form=form)

        user = models.User(username=form.username.data.strip(),
                           email=form.email.data.strip(), password=form.password.data)
        user.registration_ip = ip
        user.last_login_ip = ip
        db.session.add(user)
        db.session.commit()

        if app.config['RAID_MODE_LIMIT_REGISTER']:
            flask.flash(Markup(app.config['RAID_MODE_REGISTER_MESSAGE'] + ' '
                                     'Please <a href="{}">ask a moderator</a> to manually '
                                     'activate your account <a href="{}">\'{}\'</a>.'
                                     .format(flask.url_for('site.help') + '#irchelp',
                                             flask.url_for('users.view_user',
                                                           user_name=user.username),
                                             user.username)), 'warning')

        elif models.RangeBan.is_rangebanned(user.registration_ip):
            flask.flash(Markup('Your IP is blocked from creating new accounts. '
                                     'Please <a href="{}">ask a moderator</a> to manually '
                                     'activate your account <a href="{}">\'{}\'</a>.'
                                     .format(flask.url_for('site.help') + '#irchelp',
                                             flask.url_for('users.view_user',
                                                           user_name=user.username),
                                             user.username)), 'warning')
        else:
            if app.config['USE_EMAIL_VERIFICATION']:  # force verification, enable email
                send_verification_email(user)
                return flask.render_template('waiting.html')
            else:  # disable verification, set user as active and auto log in
                user.status = models.UserStatusType.ACTIVE
                db.session.add(user)
                db.session.commit()
                flask.g.user = user
                flask.session['user_id'] = user.id
                flask.session.permanent = True
                flask.session.modified = True
                return flask.redirect(redirect_url())

    return flask.render_template('register.html', form=form)


@bp.route('/password-reset/<payload>', methods=['GET', 'POST'])
@bp.route('/password-reset', methods=['GET', 'POST'])
def password_reset(payload=None):
    if not app.config['ALLOW_PASSWORD_RESET']:
        return flask.abort(404)

    if flask.g.user:
        return flask.redirect(redirect_url())

    if payload is None:
        form = forms.PasswordResetRequestForm(flask.request.form)
        if flask.request.method == 'POST' and form.validate():
            user = models.User.by_email(form.email.data.strip())
            if user:
                send_password_reset_request_email(user)

            flask.flash(Markup(
                'A password reset request was sent to the provided email, '
                'if a matching account was found.'), 'info')
            return flask.redirect(flask.url_for('main.home'))
        return flask.render_template('password_reset_request.html', form=form)

    else:
        s = get_serializer()
        try:
            request_timestamp, pw_hash, user_id = s.loads(payload)
        except Exception:
            return flask.abort(404)

        user = models.User.by_id(user_id)
        if not user:
            return flask.abort(404)

        # Timeout after six hours
        if (time.time() - request_timestamp) > 6 * 3600:
            return flask.abort(404)

        sha1_password_hash_hash = binascii.hexlify(sha1_hash(user.password_hash.hash)).decode()
        if pw_hash != sha1_password_hash_hash:
            return flask.abort(404)

        form = forms.PasswordResetForm(flask.request.form)
        if flask.request.method == 'POST' and form.validate():
            user.password_hash = form.password.data

            db.session.add(user)
            db.session.commit()

            send_password_reset_email(user)

            flask.flash(Markup('Your password was reset. Log in now.'), 'info')
            return flask.redirect(flask.url_for('account.login'))
        return flask.render_template('password_reset.html', form=form)


@bp.route('/profile', methods=['GET', 'POST'])
def profile():
    if not flask.g.user:
        # so we don't get stuck in infinite loop when signing out
        return flask.redirect(flask.url_for('main.home'))

    form = forms.ProfileForm(flask.request.form)

    if flask.request.method == 'POST':
        if form.authorized_submit and form.validate():
            user = flask.g.user
            new_email = form.email.data.strip() if form.email.data else None
            new_password = form.new_password.data

            if new_email and new_email.strip():
                if form.current_password.data != user.password_hash:
                    flask.flash(Markup(
                        '<strong>Email change failed!</strong> Incorrect password.'), 'danger')
                    return flask.redirect('/profile')
                user.email = form.email.data
                flask.flash(Markup(
                    '<strong>Email successfully changed!</strong>'), 'success')

            if new_password:
                if form.current_password.data != user.password_hash:
                    flask.flash(Markup(
                        '<strong>Password change failed!</strong> Incorrect password.'), 'danger')
                    return flask.redirect('/profile')
                user.password_hash = form.new_password.data
                flask.flash(Markup(
                    '<strong>Password successfully changed!</strong>'), 'success')
            db.session.add(user)
            db.session.commit()
            flask.g.user = user
            return flask.redirect('/profile')

        elif form.submit_settings:
            user = flask.g.user
            if user.preferences is None:
                preferences = models.UserPreferences(user.id)
                db.session.add(preferences)
                db.session.commit()
            user.preferences.hide_comments = form.hide_comments.data
            flask.flash(Markup(
                '<strong>Preferences successfully changed!</strong>'), 'success')
            db.session.add(user)
            db.session.commit()
            flask.g.user = user
            return flask.redirect('/profile')

    return flask.render_template('profile.html', form=form)


@bp.route('/trusted/request', methods=['GET', 'POST'])
def request_trusted():
    if not flask.g.user:
        return flask.redirect(flask.url_for('account.login'))
    trusted_form = None
    deny_reasons = []
    if flask.g.user.is_trusted:
        deny_reasons.append('You are already trusted.')
    if not flask.g.user.satisfies_trusted_reqs:
        deny_reasons.append('You do not satisfy the minimum requirements.')
    if (models.TrustedApplication.query.
            filter(models.TrustedApplication.submitter_id == flask.g.user.id).
            filter_by(is_closed=False).first()):
        deny_reasons.append('You already have an open application.')
    last_app = models.TrustedApplication.query \
        .filter(models.TrustedApplication.submitter_id == flask.g.user.id) \
        .filter_by(is_rejected=True) \
        .order_by(models.TrustedApplication.closed_time.desc()) \
        .first()
    if last_app:
        if ((datetime.utcnow() - last_app.closed_time).days <
                app.config['TRUSTED_REAPPLY_COOLDOWN']):
            deny_reasons.append('Your last application was rejected less than {} days ago.'
                                .format(app.config['TRUSTED_REAPPLY_COOLDOWN']))
    if flask.request.method == 'POST':
        trusted_form = forms.TrustedForm(flask.request.form)
        if trusted_form.validate() and not deny_reasons:
            ta = models.TrustedApplication()
            ta.submitter_id = flask.g.user.id
            ta.why_want = trusted_form.why_want_trusted.data.rstrip()
            ta.why_give = trusted_form.why_give_trusted.data.rstrip()
            db.session.add(ta)
            db.session.commit()
            flask.flash('Your trusted application has been submitted. '
                        'You will receive an email when a decision has been made.', 'success')
            return flask.redirect(flask.url_for('site.trusted'))
    else:
        if len(deny_reasons) == 0:
            trusted_form = forms.TrustedForm()
    return flask.render_template('trusted_form.html', trusted_form=trusted_form,
                                 deny_reasons=deny_reasons)


def redirect_url():
    next_url = flask.request.args.get('next', '')
    referrer = flask.request.referrer or ''

    target_url = (
        # Use ?next= param if it's a local (/foo/bar) path
        (next_url.startswith('/') and next_url) or
        # Use referrer if it's on our own host
        (referrer.startswith(flask.request.host_url) and referrer)
    )

    # Return the target, avoiding infinite loops
    if target_url and target_url != flask.request.url:
        return target_url

    # Default to index
    return flask.url_for('main.home')


def send_verification_email(user):
    activation_link = get_activation_link(user)

    tmpl_context = {
        'activation_link': activation_link,
        'user': user
    }

    email_msg = email.EmailHolder(
        subject='Verify your {} account'.format(app.config['GLOBAL_SITE_NAME']),
        recipient=user,
        text=flask.render_template('email/verify.txt', **tmpl_context),
        html=flask.render_template('email/verify.html', **tmpl_context),
    )

    email.send_email(email_msg)


def send_password_reset_email(user):
    ''' Alert user that their password has been successfully reset '''

    email_msg = email.EmailHolder(
        subject='Your {} password has been reset'.format(app.config['GLOBAL_SITE_NAME']),
        recipient=user,
        text=flask.render_template('email/reset.txt', user=user),
        html=flask.render_template('email/reset.html', user=user),
    )

    email.send_email(email_msg)


def send_password_reset_request_email(user):
    ''' Send user a password reset link '''
    reset_link = get_password_reset_link(user)

    tmpl_context = {
        'reset_link': reset_link,
        'user': user
    }

    email_msg = email.EmailHolder(
        subject='{} password reset request'.format(app.config['GLOBAL_SITE_NAME']),
        recipient=user,
        text=flask.render_template('email/reset-request.txt', **tmpl_context),
        html=flask.render_template('email/reset-request.html', **tmpl_context),
    )

    email.send_email(email_msg)