import json
import datetime
from werkzeug.security import check_password_hash, generate_password_hash
from your_code_here.MongoUtil import MongoUtil
from your_code_here.RedisUtil import RedisUtil
from util.utils import check_answer_valid, check_question_valid, check_vote, generate_session
from flask import Flask, render_template, request, make_response


app = Flask(__name__)
mongo = MongoUtil()
redis = RedisUtil()


@app.route('/')
def index():
    session_data = redis.fetch_session(request.cookies.get('session'))
    question_list = mongo.query_question()
    return render_template('index.html', question_list=question_list, session=session_data)


@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template('login.html')
    else:
        body = request.json
        user = body.get('user', '')
        password = body.get('password', '')
        if not all([user, password]):
            return json.dumps({'success': False, 'reason': '用户名和密码不能同时为空！'}, ensure_ascii=False)
        user_info = mongo.get_user_info(user)
        if not user_info['success']:
            return json.dumps(user_info)
        password_hash = user_info['password_hash']
        if not check_password_hash(password_hash, password):
            return json.dumps({'success': False, 'reason': '用户名与密码不匹配！'})
        session_id, session_data = generate_session(str(user_info['user_info']['_id']), user)
        response = make_response(json.dumps({'success': True}))
        response.set_cookie('session', value=session_id)
        return response


@app.route('/register', methods=['POST'])
def register():
    body = request.json
    user = body.get('user', '')
    password = body.get('password', '')
    if not all([user, password]):
        return json.dumps({'success': False, 'reason': '用户名和密码不能同时为空！'}, ensure_ascii=False)
    if redis.check_user_registered(user):
        return json.dumps({'success': False, 'reason': '用户名已被注册！请更换'})
    password_hash = generate_password_hash(password)
    user_id = mongo.save_user_info(user, password_hash)
    session_id, session_data = generate_session(user_id, user)
    redis.save_session(session_id, session_data)
    response = make_response(json.dumps({'success': True}))
    response.set_cookie('session', value=session_id)
    return response


@app.route('/question/<question_id>')
def question_detail(question_id=None):
    session_data = redis.fetch_session(request.cookies.get('session'))
    question_answer_dict = mongo.query_answer(question_id)

    could_answer_this_question = True
    if 'user' not in session_data or redis.check_user_answer_question(session_data['user'], question_id):
        could_answer_this_question = False
    session_data['could_answer_this_question'] = could_answer_this_question

    return render_template('answer_list.html', question_answer_dict=question_answer_dict, session=session_data)


@app.route('/post_answer', methods=['POST'])
def post_answer():
    session_data = redis.fetch_session(request.cookies.get('session'))
    if not session_data:
        return json.dumps({'success': False, 'reason': '未登录！'})
    answer = request.json
    is_valid = check_answer_valid(answer)
    if not is_valid['success']:
        return json.dumps(is_valid, ensure_ascii=False)
    now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    question_id = answer['question_id']
    if redis.check_user_answer_question(session_data['user'], question_id):
        return json.dumps({'success': False, 'reason': '已经回答过这个问题，不能回答两次。'})
    author = session_data['user']
    answer = answer['answer']
    mongo.insert_answer(question_id, answer, author, now)
    redis.set_answer_flag(question_id, author)
    return json.dumps({'success': True})


@app.route('/post_question', methods=['POST'])
def post_question():
    session_data = redis.fetch_session(request.cookies.get('session'))
    if not session_data:
        return json.dumps({'success': False, 'reason': '未登录！'})
    question = request.json
    is_valid = check_question_valid(question)
    if not is_valid['success']:
        return json.dumps(is_valid, ensure_ascii=False)
    title = question['title']
    detail = question['detail']
    author = session_data['user']
    now = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
    doc_id = mongo.insert_question(title, detail, author, now)
    return json.dumps({'success': True, 'doc_id': doc_id})


def update_question(info):
    if 'doc_id' not in info or 'title' not in info or 'text' not in info:
        return {'success': False, 'reason': '字段不全，必要字段：doc_id, title, text'}
    mongo.update_question(info['doc_id'], info['title'], info['text'])
    return {'success': True}


def update_answer(info):
    if 'doc_id' not in info or 'text' not in info:
        return {'success': False, 'reason': '字段不全，必要字段：doc_id, text'}
    mongo.update_answer(info['doc_id'], info['text'])
    return {'success': True}


@app.route('/update', methods=['POST'])
def update():
    session_data = redis.fetch_session(request.cookies.get('session'))
    if not session_data:
        return json.dumps({'success': False, 'reason': '未登录！'})
    info = request.json
    update_type = info.get('update_type')
    if update_type == 'question':
        return json.dumps(update_question(info))
    elif update_type == 'answer':
        return json.dumps(update_answer(info))
    else:
        return json.dumps({'success': False, 'reason': 'update_type只能是qustion或者answer'})


@app.route('/vote', methods=['POST'])
def vote():
    session_data = redis.fetch_session(request.cookies.get('session'))
    if not session_data:
        return json.dumps({'success': False, 'reason': '未登录！'})
    vote_data = request.json
    is_valid = check_vote(vote_data)
    if not is_valid['success']:
        return json.dumps(is_valid)
    doc_type = vote_data['doc_type']
    value = vote_data['value']
    doc_id = vote_data['doc_id']
    if doc_type == 'question':
        mongo.vote_for_question(doc_id, value)
    else:
        mongo.vote_for_answer(doc_id, value)
    return json.dumps({'success': True})