import re

import ujson as json

from core.builtins import Bot
from core.builtins import Plain, Image, Url
from core.component import module
from core.utils.http import get_url
from core.utils.i18n import Locale
from core.utils.image import msgchain2image
from modules.wiki.utils.screenshot_image import generate_screenshot_v2
from modules.wiki.utils.wikilib import WikiLib
from .teahouse import get_rss as get_teahouse_rss


async def get_weekly(with_img=False, zh_tw=False):
    locale = Locale('zh_cn' if not zh_tw else 'zh_tw')
    result = json.loads(await get_url(
        'https://minecraft.fandom.com/zh/api.php?action=parse&page=Minecraft_Wiki/weekly&prop=text|revid|images&format=json' +
        ('&variant=zh-tw' if zh_tw else ''),
        200))
    html = result['parse']['text']['*']
    text = re.sub(r'<p>', '\n', html)  # 分段
    text = re.sub(r'<(.*?)>', '', text, flags=re.DOTALL)  # 移除所有 HTML 标签
    text = re.sub(r'\n\n\n', '\n\n', text)  # 移除不必要的空行
    text = re.sub(r'\n*$', '', text)
    img = result['parse']['images']
    page = re.findall(r'(?<=<b><a href=").*?(?=")', html)
    msg_list = [Plain(locale.t("weekly.message.expired") if page[
                                                                0] == '/zh/wiki/%E7%8E%BB%E7%92%83' else locale.t(
        "weekly.message", text=text))]
    imglink = None
    if img:
        get_image = await (WikiLib('https://minecraft.fandom.com/zh/wiki/')).parse_page_info('File:' + img[0])
        if get_image.status:
            imglink = get_image.file
    msg_list.append(Plain(locale.t("weekly.message.link", img=imglink if imglink is not None else locale.t("none"),
                                   article=str(Url(f'https://minecraft.fandom.com{page[0]}')),
                                   link=str(
                                       Url(f'https://minecraft.fandom.com/zh/wiki/?oldid={str(result["parse"]["revid"])}')))))
    if imglink is not None and with_img:
        msg_list.append(Image(path=imglink))

    return msg_list


async def get_weekly_img(with_img=False, zh_tw=False):
    locale = Locale('zh_cn' if not zh_tw else 'zh_tw')
    img = await generate_screenshot_v2('https://minecraft.fandom.com/zh/wiki/Minecraft_Wiki/weekly' +
                                       ('&variant=zh-tw' if zh_tw else ''), content_mode=True, allow_special_page=True)
    msg_ = []
    if img:
        msg_.append(Image(path=img))
    if with_img:
        result = json.loads(await get_url(
            'https://minecraft.fandom.com/zh/api.php?action=parse&page=Minecraft_Wiki/weekly&prop=images&format=json' +
            ('&variant=zh-tw' if zh_tw else ''),
            200))
        img = result['parse']['images']
        if img:
            get_image = await (WikiLib('https://minecraft.fandom.com/zh/wiki/')).parse_page_info('File:' + img[0])
            if get_image.status:
                msg_.append(Plain(locale.t("weekly.message.image", img=get_image.file)))
    return msg_


wky = module('weekly', developers=['Dianliang233'], support_languages=['zh_cn', 'zh_tw'])


@wky.handle('{{weekly.help}}')
async def _(msg: Bot.MessageSession):
    weekly = await get_weekly(True if msg.target.clientName in ['QQ', 'TEST'] else False,
                              zh_tw=True if msg.locale.locale == 'zh_tw' else False)
    await msg.finish(weekly)


@wky.handle('image {{weekly.image.help}}')
async def _(msg: Bot.MessageSession):
    await msg.finish(await get_weekly_img(True if msg.target.clientName in ['QQ', 'TEST'] else False,
                                          zh_tw=True if msg.locale.locale == 'zh_tw' else False))


@wky.handle('teahouse {{weekly.teahouse.help}}')
async def _(msg: Bot.MessageSession):
    weekly = await get_teahouse_rss()
    await msg.finish(weekly)


@wky.handle('teahouse image {{weekly.teahouse.help}}')
async def _(msg: Bot.MessageSession):
    weekly = await get_teahouse_rss()
    await msg.finish(Image(await msgchain2image([Plain(weekly)])))