import hashlib
import json,os,time,random,shutil
import re,math

import execjs
from utils.NetUtils import netUtils
from utils.HtmlUtils import htmlUtils
from utils.ImageUtils import imageUtils
from utils.comic.ComicInfo import comicInfo
from utils.CBZUtils import CBZUtils
from utils.downloader import download_images
from utils.Ntfy import ntfy
from utils.VerUtils import verUtils
from PIL import Image

class comicEntity:
    count_chapter = 0
    aid = None
    repeat = 0


    @classmethod
    def downladsComcis(cls,url):
            #漫画名
        comic_href_list = htmlUtils.xpathData("//div[@class='thumb-overlay-albums']/a/@href",url,update=True)
        comics_name = htmlUtils.xpathData("//span[@class='video-title title-truncate m-t-5']/text()")
        len_books = len(comic_href_list)
        base_url = comicInfo.getBaseUrl(url)
        for x in range(0, len_books):
            book_name = comics_name[x]
            comicInfo.setComicName(book_name)
            comic_href = base_url+comic_href_list[x]            
            random_int = random.randint(5,20)
            dir_conf_comic = comicInfo.getDirConfComic()
            if not os.path.exists(dir_conf_comic):
                ntfy.sendMsg(f"{random_int}秒后开始下载 漫画：{book_name}")
                time.sleep(random_int)
            else:
                ntfy.sendMsg(f"已存在 漫画：{book_name}")
            cls.oneComic(comic_href, random.uniform(0,10))
            
        #print(books)
        #for comicHref in comicsHref:
        #    cls.oneComic(comicHref,random.uniform(10,20))
     
    @classmethod
    def oneComic(cls,c_url,sleep=None):
        nums = htmlUtils.xpathData("//div[@class='row']/div[6]/div[1]/div[1]/ul[contains(@class,'btn-toolbar')]/a",url=c_url,update=True)
        book_name = htmlUtils.xpathData("//div[@itemprop='name']/h1[@id='book-name']/text()")
        book_name = re.sub(r'[\\\/\|\(\)\~\?\.\:\：\-\*\<\>]', '', book_name)
        tags = htmlUtils.xpathData("//div[@class='row']/div[@class='col-lg-7']/div[1]/div[@class='tag-block']/span[@data-type='tags']/a[@class='btn btn-sm btn-primary']/text()")
        author = htmlUtils.xpathData("//div[@class='row']/div[@class='col-lg-7']/div[1]/div[@class='tag-block']/span[@data-type='author']/a[@class='btn btn-sm btn-primary']/text()")
        book_msg = htmlUtils.xpathData("//div[@class='row']/div[@class='col-lg-7']/div[1]/div[@class='p-t-5 p-b-5']/text()")
        jmid = book_msg[0]
        dep = str(book_msg[1]).replace("叙述：","")
        icon = htmlUtils.xpathData("//div[@class='row']//div[@class='thumb-overlay']/img/@src",num=0)

        comicInfo.setIcon(icon)
        comicInfo.setHomePage(c_url)
        comicInfo.setComicName(book_name)
        comicInfo.setAuthor(author)
        comicInfo.setDep(dep)
        comicInfo.setTags(tags)
        comicInfo.setTag(tags)
        comicInfo.setCBS("韩漫")
        comicInfo.setLang("zh")
        albumid = re.search(r'/album/(\d+)', c_url).group(1)
        referer = re.search(r'(https://\w+\.\w+)/', c_url).group(1)

        if nums:
            list_chapter_name = []
            list_chapter_href = []
            cls.count_chapter = 0

            for i in nums:
                photo_name_list = i.xpath("li/text()")[0].split()
                photo_date = i.xpath("li/span/text()")[0].split()
                #print(re.findall(r'[\u4E00-\u9FA5]+.*?', i.xpath("li/text()")[0]))
                try:
                    if re.findall(r'[\u4E00-\u9FA5]', photo_name_list[2]):
                        photo_name=re.sub(r'\s','',photo_name_list[0])+' '+photo_name_list[2]
                    else:photo_name=re.sub(r'\s','',photo_name_list[0])
                except Exception as e:
                    photo_name = re.sub(r'\s', '', photo_name_list[0])
                photo_name = re.sub(r'[\\\/\|\(\)\~\?\.\:\：\-\*\<\>\-]', '',photo_name)
                #print(photo_name)
                photoid=i.attrib['data-album']
                cls.aid = photoid
                comicInfo.setChapterName(photo_name)
                comicInfo.setDate(photo_date[0],split='-')
                comicInfo.setWeb(referer+i.attrib['href'])
                is_scramble = False
                if int(photoid) > 220980:
                    is_scramble = True
                if not comicInfo.nextExistsGetPath("done_"):
                    comicEntity.comicChapter(referer+i.attrib['href'],scramble=is_scramble,sleep=random.randint(5,15))
                #存在就校验CBZ包是否完整
                if comicInfo.nextExistsGetPath("done_"):
                    verUtils.verCBZ()
                cls.count_chapter += 1
                #一本漫画下载后等待
        #清空文件夹
        path_dir_comic = comicInfo.getDirComic()
        if os.path.exists(path_dir_comic):
            shutil.rmtree(path_dir_comic)
        if sleep != None:
            time.sleep(sleep)

    '''
    
    读取某章节下所有图片
    ''' 
    @classmethod
    def comicChapter(cls,chapter_url,scramble=None,sleep=None):
        is_next = True
#        try:
        is_next = cls.Onechapter(chapter_url,scramble)
            #进入下个阶段
        if comicInfo.nextExistsGetPath("down_"):
        #章节图片全部下载后，调用下载封面
            netUtils.downloadComicIcon()
         #下个阶段
        if comicInfo.nextExistsGetPath("cbz_"):
            time.sleep(0.1)
            #下载后自动打包
            is_next = CBZUtils.packAutoComicChapterCBZ()
        #完成删除原文件
            remove_path = comicInfo.getDirComicChapter()
            shutil.rmtree(remove_path)
            print(f"文件已删除: {remove_path}")
#        except:
#            ntfy.sendMsg(f"{comicInfo.getComicName()} 下载出错了")
#            is_next = False
        #ntfy.sendMsg(f"预计总章节大小：{cls.count_chapter + 1}  / "+ str(comicInfo.getLenChapters()))
        if sleep != None and is_next == True:
            ntfy.sendMsg(f"{sleep} 秒后开始下载下一个章节")
            time.sleep(sleep)

    
    @classmethod
    def Onechapter(cls,chapter_url,scramble=None):
        if not str(chapter_url).startswith("http"):
            chapter_url = comicInfo.getBaseUrl() + chapter_url
        try:
            is_next = cls.comicChapterDownload(chapter_url,scramble)
        except:
            htmlUtils.remove_HtmlCache(chapter_url)
            cls.repeat = 0
            is_next = cls.comicChapterDownload(chapter_url,scramble)
        comicInfo.nextInfoToImgChapter()
        #下载完成后, 开始解密图片
        if scramble:
            #获取章节图片路径
            chapter_dir = comicInfo.getDirComicChapter()
            dirs = os.listdir(chapter_dir)
            for img in dirs:
                is_scramble = str(img).startswith("scramble=")
                if is_scramble:
                    c_path = os.path.join(chapter_dir, img)
                    #imageUtils.getScrambleImage(c_path)
                    cls.encode_scramble_image(c_path)
        #进入下一阶段
        comicInfo.nextImgToDownloadChapter()
        return is_next
    
    @classmethod
    def comicChapterDownload(cls,chapter_url,c_scramble):
        img_list =htmlUtils.xpathData("//div[@class='panel-body']/div/div[contains(@class,'center')]/img/@data-original",url=chapter_url,update=True)
        pages_imgs =htmlUtils.xpathData("//div[@class='center scramble-page']/@id",url=chapter_url)
        comicInfo.setPages(pages_imgs)
        comicInfo.writeComicInfoXML(comicInfo.str_chapter)
        #print("img_list:",len(img_list))
        list_img = []
        list_file_name = []
        for i in img_list:
            img_url= i
            img_name = os.path.basename(img_url).split('.')[0]
            if c_scramble:
                img_name = "scramble="+str(cls.get_scramble_num(cls.aid,img_name))+"_"+img_name
            #path_img = "%s\\%s.jpg" % (cls.aid, img_name)
            path_img = "%s.jpg" % (img_name)
            list_img.append(img_url)
            list_file_name.append(path_img)
        comicInfo.setChapterImgs(list_img)
                #保存图像
        comicInfo.nextSaveInfoChapter(comicInfo.str_chapter, list_img)
        is_next = verUtils.verNextCBZ(list_img) 
        list_shunt = ["?shunt=2","?shunt=1","?shunt=3",""]
        while not is_next:
            time.sleep(10)
            download_images(list_img,comicInfo.getDirComicChapter(), filesName=list_file_name,timeout=1500)
            file_imgs = os.listdir(comicInfo.getDirComicChapter())
            count_jpg = ",".join(file_imgs).split(".jpg")
            is_next = len(count_jpg)-1 == len(list_img)
            cls.repeat += 1
            if cls.repeat > 3:
                url = list_shunt[cls.repeat % len(list_shunt)]
                print("分流中=",url)
                cls.comicChapterDownload(str(chapter_url).split("?")[0]+url,c_scramble)
        return True
         
    @classmethod
    def get_md5(cls,num):
        result1 = hashlib.md5(num.encode()).hexdigest()
        print('get_md5-', result1)
        return result1
        
    @classmethod
    def get_scramble_num(cls,e, t):
        #print(type(e),e, type(t),t)
        a = 10
        try:
            num_dict = {}
            for i in range(10):
                num_dict[i] = i * 2 + 2
            if (int(e) >= 268850):
                n = str(e) + t;
                # switch(n=(n = (n = md5(n)).substr(-1)), n %= 10) {
                #print("n=",n)
                tmp = ord(cls.get_md5(n)[-1])
                result = num_dict[tmp % 10]
                a = result
            return a
        except Exception as e:
            print(e.__traceback__.tb_lineno,e)
            return False

    @classmethod
    def encode_scramble_image(cls,imgpath):
        image = Image.open(imgpath)
        w, h = image.size
        #image.show()
        file_str = str(imgpath).split("=")
        #10_29.jpg
        base_dir = file_str[0].replace("scramble","")
        base_name = file_str[-1]
        base_fn = base_name.split("_") 
        save_name = base_fn[1]
        save_name_delesu = save_name.split(".")[0]
        blocks = int(base_fn[0])
        img_type = os.path.basename(imgpath).split('.')[-1]
        save_path = os.path.join(os.path.dirname(imgpath),save_name_delesu+"."+img_type)
        # print(type(aid),type(img_name))
        if blocks:
            s = blocks  # 随机值
            # print(s)
            l = h % s  # 切割最后多余的值
            box_list = []
            hz = 0
            for i in range(s):
                c = math.floor(h / s)
                g = i * c
                hz += c
                h2 = h - c * (i + 1) - l
                if i == 0:
                    c += l;hz += l
                else:
                    g += l
                box_list.append((0, h2, w, h - g))

            # print(box_list,len(box_list))
            item_width = w
            # box_list.reverse() #还原切图可以倒序列表
            # print(box_list, len(box_list))
            newh = 0
            image_list = [image.crop(box) for box in box_list]
            # print(box_list)
            newimage = Image.new("RGB", (w, h))
            for image in image_list:
                # image.show()
                b_w, b_h = image.size
                newimage.paste(image, (0, newh))

                newh += b_h
            newimage.save(save_path)
            if os.path.exists(imgpath):
                os.remove(imgpath)