import datetime
import json
import shutil
import threading
import time
from typing import Tuple

import xmltodict

from entity.comment import Comment
from entity.contact import Contact
from exporter.avatar_exporter import AvatarExporter
from exporter.emoji_exporter import EmojiExporter
from exporter.image_exporter import ImageExporter
from exporter.video_exporter import VideoExporter
from log import LOG
from entity.moment_msg import MomentMsg
from pathlib import Path


def get_img_div_css(size: int) -> str:
    if size == 1:
        return 'width:10rem; overflow:hidden'
    else:
        return 'width:19rem; overflow:hidden'


def get_img_css(size: int) -> str:
    """object-fit: cover; 预览图居中裁剪
       cursor:pointer; 手形鼠标
    """
    img_style = "object-fit:cover;cursor:pointer;"
    if size == 1:
        return f'width:10rem;height:10rem;{img_style}'
    elif size == 2:
        return f'width:8rem;height:8rem;float:left;margin-bottom:0.2rem;margin-right:0.2rem;{img_style}'
    elif size == 4:
        return f'width:8rem;height:8rem;float:left;margin-bottom:0.2rem;margin-right:0.2rem;{img_style}'
    else:
        return f'width:5rem;height:5rem;float:left;margin-bottom:0.2rem;margin-right:0.2rem;{img_style}'


def is_music_msg(msg: MomentMsg) -> bool:
    """判断一个msg是否为音乐分享
    """
    if msg.timelineObject.ContentObject and msg.timelineObject.ContentObject.mediaList and msg.timelineObject.ContentObject.mediaList.media:
        media = msg.timelineObject.ContentObject.mediaList.media[0]
        if media.type == '5':
            return True
    return False


def get_music_info(msg: MomentMsg) -> Tuple[str, str, str]:
    """获取音乐标题，演唱者，音乐源
    """
    title = ""
    musician = ""
    src = ""
    if msg.timelineObject.ContentObject and msg.timelineObject.ContentObject.mediaList and msg.timelineObject.ContentObject.mediaList.media:
        media = msg.timelineObject.ContentObject.mediaList.media[0]
        title = media.title
        musician = media.description
        if media.url:
            src = media.url.text
    return title, musician, src


class HtmlExporter(threading.Thread):

    def __init__(self, gui: 'Gui', dir_name: str, contacts_map: dict[str, Contact], begin_date: datetime.date,
                 end_date: datetime.date, convert_video: int):
        self.dir_name = dir_name
        if Path(f"output/{self.dir_name}").exists():
            shutil.rmtree(f"output/{self.dir_name}")
        shutil.copytree("resource/template/", f"output/{self.dir_name}")

        self.gui = gui
        self.avatar_exporter = AvatarExporter(dir_name)
        self.image_exporter = ImageExporter(dir_name)
        self.video_exporter = VideoExporter(dir_name)
        self.html_head = None
        self.html_end = None
        self.file = None
        self.contacts_map = contacts_map
        self.begin_date = begin_date
        self.end_date = end_date
        self.convert_video = convert_video
        self.stop_flag = False
        super().__init__()

    def run(self) -> None:

        with open(f"resource/template.html", encoding='utf-8') as f:
            content = f.read()
            self.html_head, self.html_end = content.split('/*内容分割线*/')
        self.file = open(f"output/{self.dir_name}/index.html", 'w', encoding='utf-8')

        if self.gui.account_info and self.gui.account_info.get('wxid'):
            self.avatar_exporter.get_avatar_path(self.gui.account_info.get('wxid'))
            self.html_head = self.html_head.replace("{my_wxid}", f"{self.gui.account_info.get('wxid')}")
            from app.DataBase import micro_msg_db
            my_info = micro_msg_db.get_contact_by_username(self.gui.account_info.get('wxid'))
            self.html_head = self.html_head.replace("{my_name}", f"{my_info[4]}")

        from app.DataBase import sns_db
        cover_url = sns_db.get_cover_url()
        if cover_url:
            cover_path = self.image_exporter.save_image((cover_url, "", ""), 'image')
            self.html_head = self.html_head.replace("{cover_path}", cover_path)

        self.file.write(self.html_head)
        # 加一天
        end_date = self.end_date + datetime.timedelta(days=1)
        begin_time = time.mktime(
            datetime.datetime(self.begin_date.year, self.begin_date.month, self.begin_date.day).timetuple())
        end_time = time.mktime(datetime.datetime(end_date.year, end_date.month, end_date.day).timetuple())

        self.gui.image_decrypter.decrypt_images(self, self.begin_date, end_date, self.dir_name)
        self.gui.video_decrypter.decrypt_videos(self, self.begin_date, end_date, self.dir_name, self.convert_video)


        message_datas = sns_db.get_messages_in_time(begin_time, end_time)
        for index, message_data in enumerate(message_datas):
            if not self.stop_flag:
                if message_data[0] in self.contacts_map:
                    comments_datas = sns_db.get_comment_by_feed_id(message_data[2])
                    comments: list[Comment] = []
                    for c in comments_datas:
                        contact = Comment(c[0], c[1], c[2])
                        comments.append(contact)
                    self.export_msg(message_data[1], comments, self.contacts_map)
                    # 更新进度条 前30%视频处理  后70%其他处理
                    progress = round(index / len(message_datas) * 70)
                    self.gui.update_export_progressbar(30 + progress)
        self.gui.update_export_progressbar(100)
        self.finish_file()
        self.gui.export_succeed()

    def stop(self) -> None:
        self.stop_flag = True

    def export_msg(self, message: str, comments: list[Comment], contacts_map: dict[str, Contact]) -> None:

        LOG.info(message)
        # force_list: 强制要求转media为list
        msg_dict = xmltodict.parse(message, force_list={'media'})
        msg_json = json.dumps(msg_dict)
        msg = MomentMsg.from_json(msg_json)

        # 微信ID
        username = msg.timelineObject.username
        # 头像路径
        avatar_path = self.avatar_exporter.get_avatar_path(username)

        contact = contacts_map.get(username)
        # 备注， 或用户名
        remark = contact.remark if contact.remark else contact.nickName

        # 朋友圈图片
        images = self.image_exporter.get_images(msg)

        # 朋友圈视频
        videos = self.video_exporter.get_videos(msg)

        # 样式   3:链接样式
        content_style = msg.timelineObject.ContentObject.contentStyle

        html = ' <div class="row item">\n'
        html += '    <div class="col-xs-2">\n'
        html += '         <div class="logo01_box">\n'
        html += f'             <img src="{avatar_path}" />\n'
        html += '         </div>\n'
        html += '    </div>\n'
        html += '    <div class="col-xs-10 xs8">\n'
        html += '          <div class="towp">\n'
        html += f'              <p class="p1">{remark}</p>\n'
        if msg.timelineObject.contentDesc:
            content_desc = msg.timelineObject.contentDesc.replace("\n", "<br>")
            content_desc = EmojiExporter.replace_emoji(content_desc)
            html += f'              <p class="p2">{content_desc}</p>\n'
        html += '          </div>\n'

        # 超链接
        if content_style == 3:
            html += f'  <a href="{msg.timelineObject.ContentObject.contentUrl}" target="_blank">\n'
            html += '      <div class ="out_link" >\n'
            if images:
                thumb_path, image_path = images[0]
                html += f'        <img src = "{thumb_path}"/>\n'
            html += f'       <div class ="text" >{msg.timelineObject.ContentObject.title}</div>\n'
            html += '      </div >\n'
            html += '   </a>\n'
        # 音乐
        elif is_music_msg(msg):

            title, musician, src = get_music_info(msg)
            html += f'  <a href="{msg.timelineObject.ContentObject.contentUrl}" target="_blank">\n'
            html += '      <div class ="music_link" >\n'
            html += '        <div class ="music_des" >\n'
            if images:
                thumb_path, image_path = images[0]
                html += f'        <img src = "{thumb_path}"/>\n'
            html += '             <div class = "music_title_musician">\n'
            html += f'               <div class = "music_title">{title}</div>\n'
            html += f'               <div class = "music_musician">{musician}</div>\n'
            html += '             </div>\n'
            html += '         </div>\n'
            html += f'        <audio class = "music_audio" controls>'
            html += f'             <source src="{src}" type="audio/mpeg">'
            html += f'        </audio>'
            html += '      </div >\n'
            html += '   </a>\n'
        # 视频号
        elif msg.timelineObject.ContentObject.finderFeed:
            html += f'     <div style="width:10rem; overflow:hidden">\n'
            # 视频号图片
            thumb_path = self.image_exporter.get_finder_images(msg)
            html += f"""       <img src=\"{thumb_path}\" onclick=\"openWarningOverlay(event)\" 
                               style=\"width:10rem;height:10rem;object-fit:cover;cursor:pointer;\"/>\n"""
            html += '      </div>\n'

            # 视频号说明
            html += '          <div class="texts_box">\n'
            nickname = msg.timelineObject.ContentObject.finderFeed.nickname
            desc = msg.timelineObject.ContentObject.finderFeed.desc
            html += f'            <p class="location">视频号 · {nickname} · {desc}</p>\n'
            html += '         </div>\n'
        # 普通朋友圈
        else:
            html += f'     <div style="{get_img_div_css(len(images))}">\n'
            for thumb_path, image_path in images:
                html += f"""     <img src="{thumb_path}" full_img="{image_path}" onclick="openFullSize(event)" 
                                 style="{get_img_css(len(images))}"/>\n"""
            html += '      </div>\n'

            html += '      <div>\n'
            for video_path in videos:
                html += f'     <video controls height="500">\n'
                html += f'        <source  src="{video_path}" type="video/mp4">\n'
                html += f'     <video>\n'
            html += '      </div>\n'

        html += '          <div class="texts_box">\n'
        if msg.timelineObject.location and msg.timelineObject.location.poiName:
            html += f'        <p class="location">{msg.timelineObject.location.poiName}</p>\n'
        html += f'            <p class="time">{msg.timelineObject.create_time}</p>\n'
        html += '         </div>\n'
        html += '    </div>\n'
        html += '</div>\n'
        self.file.write(html)

    def finish_file(self):
        self.file.write(self.html_end)
        self.file.close()