# -*- coding: utf-8 -*-
import re
from typing import Optional

from lxml import etree

from app.plugins.sitestatistic.siteuserinfo import ISiteUserInfo, SITE_BASE_ORDER
from app.utils.string import StringUtils
from app.utils.types import SiteSchema


class IptSiteUserInfo(ISiteUserInfo):
    schema = SiteSchema.Ipt
    order = SITE_BASE_ORDER + 35

    @classmethod
    def match(cls, html_text: str) -> bool:
        return 'IPTorrents' in html_text

    def _parse_user_base_info(self, html_text: str):
        html_text = self._prepare_html_text(html_text)
        html = etree.HTML(html_text)
        tmps = html.xpath('//a[contains(@href, "/u/")]//text()')
        tmps_id = html.xpath('//a[contains(@href, "/u/")]/@href')
        if tmps:
            self.username = str(tmps[-1])
        if tmps_id:
            user_id_match = re.search(r"/u/(\d+)", tmps_id[0])
            if user_id_match and user_id_match.group().strip():
                self.userid = user_id_match.group(1)
                self._user_detail_page = f"user.php?u={self.userid}"
                self._torrent_seeding_page = f"peers?u={self.userid}"

        tmps = html.xpath('//div[@class = "stats"]/div/div')
        if tmps:
            self.upload = StringUtils.num_filesize(str(tmps[0].xpath('span/text()')[1]).strip())
            self.download = StringUtils.num_filesize(str(tmps[0].xpath('span/text()')[2]).strip())
            self.seeding = StringUtils.str_int(tmps[0].xpath('a')[2].xpath('text()')[0])
            self.leeching = StringUtils.str_int(tmps[0].xpath('a')[2].xpath('text()')[1])
            self.ratio = StringUtils.str_float(str(tmps[0].xpath('span/text()')[0]).strip().replace('-', '0'))
            self.bonus = StringUtils.str_float(tmps[0].xpath('a')[3].xpath('text()')[0])

    def _parse_site_page(self, html_text: str):
        # TODO
        pass

    def _parse_user_detail_info(self, html_text: str):
        html = etree.HTML(html_text)
        if not html:
            return

        user_levels_text = html.xpath('//tr/th[text()="Class"]/following-sibling::td[1]/text()')
        if user_levels_text:
            self.user_level = user_levels_text[0].strip()

        # 加入日期
        join_at_text = html.xpath('//tr/th[text()="Join date"]/following-sibling::td[1]/text()')
        if join_at_text:
            self.join_at = StringUtils.unify_datetime_str(join_at_text[0].split(' (')[0])

    def _parse_user_torrent_seeding_info(self, html_text: str, multi_page: bool = False) -> Optional[str]:
        html = etree.HTML(html_text)
        if not html:
            return
        # seeding start
        seeding_end_pos = 3
        if html.xpath('//tr/td[text() = "Leechers"]'):
            seeding_end_pos = len(html.xpath('//tr/td[text() = "Leechers"]/../preceding-sibling::tr')) + 1
            seeding_end_pos = seeding_end_pos - 3

        page_seeding = 0
        page_seeding_size = 0
        seeding_torrents = html.xpath('//tr/td[text() = "Seeders"]/../following-sibling::tr/td[position()=6]/text()')
        if seeding_torrents:
            page_seeding = seeding_end_pos
            for per_size in seeding_torrents[:seeding_end_pos]:
                if '(' in per_size and ')' in per_size:
                    per_size = per_size.split('(')[-1]
                    per_size = per_size.split(')')[0]

                page_seeding_size += StringUtils.num_filesize(per_size)

        self.seeding = page_seeding
        self.seeding_size = page_seeding_size

    def _parse_user_traffic_info(self, html_text: str):
        # TODO
        pass

    def _parse_message_unread_links(self, html_text: str, msg_links: list) -> Optional[str]:
        return None

    def _parse_message_content(self, html_text):
        return None, None, None