# 引入数据库操作函数
from db import searchUser, insertUser, updateCodebook, ActivateUser, searchUserByUuid
# 引入json
import json
# 引入时间
import datetime
# 引入发送邮件
from mail import sendMail
# 引入md5
from hashlib import md5
# 引入随机数
import random
# 主处理逻辑函数-----------------------------------------------------------

# 处理用户登录 前端发送 mail_addr password update_time
def manageLogin(request):
    # json化，应该能当dict用
    try:
        data_cache = json.loads(request.form['data'])
    except Exception as e:
        return {'errcode': 100, 'errmsg': '校验失败'}
    checked = checkData(data_cache)
    if checked:
        # 校验通过
        user_info = searchUser(data_cache['mail_addr'])
        if user_info['errcode'] == 200:
            # 用户登录
            user_info = user_info['user_info']
            if user_info['active']:
                # 已经激活的账户
                if user_info['password'] == data_cache['password']:
                    # 密码正确
                    return {'errcode': 200, 'errmsg': 'ok', 'update_time': user_info['update_time']}
                else:
                    # 密码错误
                    return {'errcode': 106, 'errmsg': '密码错误'}
            else:
                # 没激活的用户
                time_difference = int(data_cache['update_time']) - int(user_info['update_time'])
                if time_difference > 1000 * 60 * 5:
                    # 超过五分钟，重新发送uuid
                    send_str = '非常感谢您的使用，您的验证码为：' + user_info['uuid']
                    print(send_str)
                    sendMail('欢迎注册Canary Codebook', send_str, [user_info['mail_addr']])
                    print('stop at send')
                    return {'errcode': 108, 'errmsg': '用户未激活，已重新发送邮件'}
                else:
                    return {'errcode': 105, 'errmsg': '用户未激活，未重新发送邮件'}
        elif user_info['errcode'] == 101:
            # 用户注册
            user_data = createUserData(data_cache)
            insert_res = insertUser(user_data)
            if insert_res['errcode'] == 200:
                # 新用户插入成功，发送邮件
                send_str = '非常感谢您的使用，您的验证码为：' + user_data['uuid']
                sendMail('欢迎注册Canary Codebook', send_str, [user_data['mail_addr']])
                return {'errcode': 107, 'errmsg': '注册成功，验证码已下发'}
            else:
                # 用户插入失败
                return insert_res
        else:
            # 查询错误
            return user_info
    else:
        return {'errcode': 104, 'errmsg': '签名失败'}


# 处理激活用户 用户发来uuid mail_addr
def manageActivation(request):
    # json化，应该能当dict用
    try:
        data_cache = json.loads(request.form['data'])
    except Exception as e:
        return {'errcode': 400, 'errmsg': '校验失败'}
    checked = checkData(data_cache)
    if checked:
        # 校验通过
        # 判断用户之前是否未激活
        user_info = searchUserByUuid(data_cache['uuid'])
        if user_info['errcode'] == 200:
            # 用户存在
            user_info = user_info['user_info']
            if user_info['mail_addr'] == data_cache['mail_addr']:
                # 是自己的激活码
                if user_info['active']:
                    return {'errcode': 406, 'errmsg': '该用户已经激活过了', 'update_time': user_info['update_time']}
                else:
                    # 进入激活程序
                    active_res = ActivateUser(data_cache['uuid'])
                    if active_res['errcode'] == 200:
                        # 激活成功
                        return {'errcode': 200, 'errmsg': 'ok', 'update_time': user_info['update_time']}
                    else:
                        # 激活失败
                        return active_res
            else:
                # 输入的不是自己的激活码
                return {'errcode': 407, 'errmsg': '验证码非本人所有'}
        else:
            # 用户不存在或者错误
            return user_info
    else:
        return {'errcode': 403, 'errmsg': '签名失败'}

# 处理云端覆写本地数据 用户发来 mail_addr password
def manageDownload(request):
    # json化，应该能当dict用
    try:
        data_cache = json.loads(request.form['data'])
    except Exception as e:
        return {'errcode': 500, 'errmsg': '校验失败'}
    checked = checkData(data_cache)
    if checked:
        # 校验通过
        # 获取用户信息
        user_info = searchUser(data_cache['mail_addr'])
        if user_info['errcode'] == 200:
            user_info = user_info['user_info']
            if user_info['password'] == data_cache['password']:
                # 密码正确
                return {'errcode': 200, 'errmsg': 'ok', 'codebook': user_info['codebook'], 'update_time': user_info['update_time']}
            else:
                # 密码错误
                return {'errcode': 502, 'errmsg': '密码错误'}
        else:
            return user_info
    else:
        return {'errcode': 501, 'errmsg': '签名失败'}

# 处理本地覆写云端数据 用户发来 mail_addr password codebook update_time
def manageUpload(request):
    # json化，应该能当dict用
    try:
        data_cache = json.loads(request.form['data'])
    except Exception as e:
        return {'errcode': 300, 'errmsg': '校验失败'}
    checked = checkData(data_cache)
    if checked:
        # 校验通过
        # 获取用户信息
        user_info = searchUser(data_cache['mail_addr'])
        if user_info['errcode'] == 200:
            user_info = user_info['user_info']
            if user_info['password'] == data_cache['password']:
                # 密码正确
                return updateCodebook(user_info['mail_addr'], data_cache['codebook'], data_cache['update_time'])
            else:
                # 密码错误
                return {'errcode': 502, 'errmsg': '密码错误'}
        else:
            return user_info
    else:
        return {'errcode': 304, 'errmsg': '签名失败'}

# 工具函数---------------------------------------------------------------

# MD5 校验
def checkData(data):
    d = data.copy()
    try:
        d.pop('sign')
    except KeyError:
        pass
    d = str(d)
    d = d.replace(' ', '')
    md = md5()
    md.update(d.encode('utf-8'))
    r = md.hexdigest().upper()
    return r == data['sign']
    # return True

# 创建用户初始信息
def createUserData(data):
    user_info = {
        'mail_addr' : data['mail_addr'],
        'password': data['password'],
        'update_time': data['update_time'],
        'codebook': '',
        'active': False,
        'uuid': create_uuid()
    }
    return user_info

#生成唯一的名称字符串，防止重名问题
def create_uuid(): 
    # 生成当前时间
    nowTime = datetime.datetime.now().strftime("%Y%m%d%H%M%S")  
    # 生成的随机整数n，其中0<=n<=100
    randomNum = random.randint(0, 99)  
    if randomNum <= 10:
        randomNum = str(0) + str(randomNum)
    uniqueNum = str(nowTime) + str(randomNum)
    return uniqueNum