# coding:utf-8
from django.http import HttpResponse,JsonResponse
from django.conf import settings
from django.views.decorators.csrf import csrf_exempt
from django.contrib.auth.decorators import login_required # 登录需求装饰器
from django.utils.translation import gettext_lazy as _
import datetime,time,json,base64,os,uuid
from app_doc.models import Image,ImageGroup,Attachment
from app_doc.utils import validate_url
from app_admin.models import SysSetting
from loguru import logger
import requests
import random


@login_required()
@csrf_exempt
def upload_ice_img(request):
    ##################
    # 如果需要使用ice自带的多文件上传，请修改ice的js文件中的附件上传部分代码如下：
    # 
    # for(var i=0;i<this.files.length;i++){
    # 	formData.append('file_' + i, this.files[i]);
    # }
    # formData.append('upload_num', i);
    # formData.append('upload_type', "files");
    ##################
    try:
        up_type = request.POST.get('upload_type','')
        up_num = request.POST.get('upload_num','')   
        iceEditor_img = str(request.POST.get('iceEditor-img',''))
    except:
       pass    
    if up_type == "files":
        # 多文件上传功能，需要修改js文件
        res_dic = {'length':int(up_num)}
        for  i in range(0,int(up_num)):
            file_obj = request.FILES.get('file_' + str(i))
            result = ice_save_file(file_obj,request.user)
            res_dic[i] = result
    elif iceEditor_img.lower().startswith('http'):
        res_dic = ice_url_img_upload(iceEditor_img,request.user)       
    else:
        # 粘贴上传和单文件上传
        file_obj = request.FILES.get('file[]')
        result = ice_save_file(file_obj,request.user)           
        res_dic = {0:result,"length":1,'other_msg':iceEditor_img}         #一个文件，直接把文件数量固定了
    return JsonResponse(res_dic)


def ice_save_file(file_obj,user):   
    # 默认保留支持ice单文件上传功能，可以iceEditor中开启
    file_suffix = str(file_obj).split(".")[-1] # 提取图片格式
    # 允许上传文件类型，ice粘贴上传为blob
    if file_suffix.lower() == 'blob':
        # 粘贴上传直接默认变更为png就行
        file_suffix = 'png' 
    # allow_suffix =["jpg", "jpeg", "gif", "png", "bmp", "webp","blob"]
    allow_suffix = settings.ALLOWED_IMG
    # 判断附件格式
    is_images = ["jpg", "jpeg", "gif", "png", "bmp", "webp"]
    if file_suffix.lower() not in allow_suffix:         
        return {"error": _("文件格式不允许")}    
    # 接下来构造一个文件名，时间和随机10位字符串构成
    relative_path = upload_generation_dir()
    name_time = time.strftime("%Y-%m-%d_%H%M%S_")
    name_join = ""
    name_rand = name_join.join(random.sample('zyxwvutsrqponmlkjihgfedcba',10) )

    file_name =  name_time +  name_rand + "." + file_suffix
    path_file = relative_path + file_name
    path_file = settings.MEDIA_ROOT + path_file
    #file_Url 是文件的url下发路径
    file_url = (settings.MEDIA_URL + relative_path + file_name).replace("//","/")
    
    with open(path_file, 'wb') as f:
        for chunk in file_obj.chunks():
            f.write(chunk) # 保存文件
        if file_suffix.lower()  in is_images:
            Image.objects.create(
                user=user,
                file_path=file_url,
                file_name=file_name,
                remark=_("iceEditor上传")
            )
        
        else :
            #文件上传，暂时不屏蔽，如果需要正常使用此功能，是需要在iceeditor中修改的，mrdoc使用的是自定义脚本上传
            Attachment.objects.create(
                user=user,
                file_path=file_url,
                file_name=file_name,
                file_size=str(round(len(chunk)/1024,2))+"KB"
            )
        return  {"error":0, "name": str(file_obj),'url':file_url}

    return {"error": _("文件存储异常")}


# ice_url图片上传
def ice_url_img_upload(url,user):

    relative_path = upload_generation_dir()
    name_time = time.strftime("%Y-%m-%d_%H%M%S_")
    name_join = ""
    name_rand = name_join.join(random.sample('zyxwvutsrqponmlkjihgfedcba',10) )
    file_name =  name_time +  name_rand + '.png'  # 日期时间_随机字符串命名
    path_file = os.path.join(relative_path, file_name)
    path_file = settings.MEDIA_ROOT + path_file
    # print('文件路径：', path_file)
    file_url = (settings.MEDIA_URL + relative_path + file_name).replace("//","/")
    # print("文件URL：", file_url)
    header = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.198 Safari/537.36"
    }
    r = requests.get(url, headers=header, stream=True)

    if r.status_code == 200:
        # 判断是否为允许上传的图片类型
        remote_type = r.headers['Content-Type'].split("/")[1]
        if remote_type not in settings.ALLOWED_IMG:
            logger.error("上传了不允许的URL图片：{}".format(url))
            resp_data = {
                'error': 0,
                'name': {},
                'file':{}
            }
            return resp_data
        with open(path_file, 'wb') as f:
            f.write(r.content)  # 保存文件
        Image.objects.create(
            user=user,
            file_path=file_url,
            file_name=file_name,
            remark=_('iceurl粘贴上传'),
        )
    resp_data = {"error":0, "name": file_name,'url':file_url}        
    return resp_data


@login_required()
@csrf_exempt
def upload_img(request):
    ##################
    # {"success": 0, "message": "出错信息"}
    # {"success": 1, "url": "图片地址"}
    ##################
    img = request.FILES.get("editormd-image-file", None) # 编辑器上传
    manage_upload = request.FILES.get('manage_upload',None) # 图片管理上传
    try:
        url_img = json.loads(request.body.decode())['url']
        url_img = validate_url(url_img)
        if url_img is False:
            return JsonResponse({"success": 0, "message": _("无效的URL！")})
    except:
        url_img = None
    dir_name = request.POST.get('dirname','')
    base_img = request.POST.get('base',None)
    group_id = request.POST.get('group_id',0)

    if int(group_id) not in [0,-1]:
        try:
            group_id = ImageGroup.objects.get(id=group_id)
        except:
            group_id = None
    else:
        group_id = None

    # 上传普通图片文件
    if img:
        result = img_upload(img, dir_name,request.user)
    # 图片管理上传
    elif manage_upload:
        result = img_upload(manage_upload, dir_name, request.user, group_id=group_id)
    # 上传base64编码图片
    elif base_img:
        result = base_img_upload(base_img,dir_name,request.user)
    # 上传图片URL地址
    elif url_img:
        if url_img.startswith("data:image"):# 以URL形式上传的BASE64编码图片
            result = base_img_upload(url_img, dir_name, request.user)
        else:
            result = url_img_upload(url_img,dir_name,request.user)
    else:
        result = {"success": 0, "message": _("上传出错")}
    return JsonResponse(result)


# 目录创建
def upload_generation_dir(dir_name=''):
    today = datetime.datetime.today()
    dir_name = dir_name + '/%d%02d/' %(today.year,today.month)
    # print("dir_name:",dir_name)
    if not os.path.exists(settings.MEDIA_ROOT + dir_name):
        # print("创建目录")
        os.makedirs(settings.MEDIA_ROOT + dir_name)
    return dir_name


# 普通图片上传
def img_upload(files, dir_name, user, group_id=None):
    # 允许上传文件类型
    # allow_suffix =["jpg", "jpeg", "gif", "png", "bmp", "webp","blob"]
    allow_suffix = settings.ALLOWED_IMG
    file_suffix = files.name.split(".")[-1] # 提取图片格式
    # 判断图片格式
    if file_suffix.lower() not in allow_suffix:
        return {"success": 0, "message": _("图片格式不正确")}

    # 判断图片的大小
    try:
        allow_image_size = SysSetting.objects.get(types='doc', name='img_size')
        allow_img_size = int(allow_image_size.value) * 1048576
    except Exception as e:
        # print(repr(e))
        allow_img_size = 10485760
    if files.size > allow_img_size:
        return {"success": 0, "message": _("图片大小超出{}MB".format(allow_img_size / 1048576))}

    relative_path = upload_generation_dir(dir_name)
    file_name = files.name.replace(file_suffix,'').replace('.','') + '_' +str(int(time.time())) + '.' + file_suffix
    path_file=os.path.join(relative_path, file_name)
    path_file = settings.MEDIA_ROOT + path_file
    # print('文件路径：',path_file)
    file_url = (settings.MEDIA_URL + relative_path + file_name).replace("//",'/')
    # print("文件URL：",file_url)
    with open(path_file, 'wb') as f:
        for chunk in files.chunks():
            f.write(chunk) # 保存文件
    Image.objects.create(
        user=user,
        file_path=file_url,
        file_name=file_name,
        remark=_('本地上传'),
        group = group_id,
    )
    return {"success": 1, "url": file_url,'message':_('上传图片成功')}

# 解析image/png获取扩展名
def getImageExtensionName(temps):
    if len(temps) == 2:
        #image/png
        temps = temps[0].split('image/')
        if len(temps) == 2:
            ## 如果文件传了扩展名，就取扩展名的文件类型，判断图片格式是否允许上传
            if temps[-1] in settings.ALLOWED_IMG:
                return "." + temps[-1]
    return ".png" 

# base64编码图片上传
def base_img_upload(files,dir_name, user):
    temps = files.split(';base64,') # 截取图片正文
    files_str = temps[-1] # 截取图片正文
    extensionName = getImageExtensionName(temps)

    files_base = base64.b64decode(files_str) # 进行base64编码
    relative_path = upload_generation_dir(dir_name)
    file_name = str(datetime.datetime.today()).replace(':', '').replace(' ', '_').replace('.', '_') + str(random.random()) +  extensionName # 日期时间
    path_file = os.path.join(relative_path, file_name)
    path_file = settings.MEDIA_ROOT + path_file
    # print('文件路径：', path_file)
    file_url = (settings.MEDIA_URL + relative_path + file_name).replace("//","/")
    # print("文件URL：", file_url)
    with open(path_file, 'wb') as f:
        f.write(files_base)  # 保存文件
    Image.objects.create(
        user = user,
        file_path = file_url,
        file_name=file_name,
        remark = _('粘贴上传'),
    )
    return {"success": 1, "url": file_url, 'message': _('上传图片成功')}


# url图片上传
def url_img_upload(url,dir_name,user):

    relative_path = upload_generation_dir(dir_name)
    file_name = str(datetime.datetime.today()).replace(':', '').replace(' ', '_').split('.')[0] + str(random.random()) + '.png'  # 日期时间
    path_file = os.path.join(relative_path, file_name)
    path_file = settings.MEDIA_ROOT + path_file
    # print('文件路径：', path_file)
    file_url = (settings.MEDIA_URL + relative_path + file_name).replace("//","/")
    # print("文件URL：", file_url)
    header = {
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/86.0.4240.198 Safari/537.36"
    }
    try:
        r = requests.get(url, headers=header, stream=True)
        if r.status_code == 200:
            # 判断是否为允许上传的图片类型
            remote_type = r.headers['Content-Type'].split("/")[1]
            if remote_type not in settings.ALLOWED_IMG:
                logger.error("上传了不允许的URL图片：{}".format(url))
                resp_data = {
                    'msg': '',
                    'code': 1,
                    'data': {}
                }
                return resp_data
            with open(path_file, 'wb') as f:
                f.write(r.content)  # 保存文件
            Image.objects.create(
                user=user,
                file_path=file_url,
                file_name=file_name,
                remark=_('粘贴上传'),
            )

            resp_data = {
                 'msg': '',
                 'code': 0,
                 'data' : {
                   'originalURL': url,
                   'url': file_url
                 }
                }
        else:
            resp_data = {
                'msg': '',
                'code': 1,
                'data': {}
            }
    except Exception as e:
        logger.error("上传URL图片异常：{}".format(repr(e)))
        resp_data = {
            'msg': '',
            'code': 1,
            'data': {}
        }
    return resp_data
    # return {"success": 1, "url": file_url, 'message': '上传图片成功'}