import re

import discord

from bots.discord.client import client
from bots.discord.slash_parser import slash_parser, ctx_to_session
from modules.wiki import WikiLib, WikiTargetInfo

wiki = client.create_group("wiki", "查询Mediawiki的相关信息")


async def auto_search(ctx: discord.AutocompleteContext):
    title = ctx.options["title"]
    iw = ''
    target = WikiTargetInfo(ctx_to_session(ctx))
    iws = target.get_interwikis()
    query_wiki = target.get_start_wiki()
    if match_iw := re.match(r'(.*?):(.*)', title):
        if match_iw.group(1) in iws:
            query_wiki = iws[match_iw.group(1)]
            iw = match_iw.group(1) + ':'
            title = match_iw.group(2)
    if query_wiki is None:
        return []
    wiki = WikiLib(query_wiki)
    if title != "":
        return [iw + x for x in (await wiki.search_page(title))]
    else:
        return [iw + (await wiki.get_json(action='query', list='random', rnnamespace='0'))['query']['random'][0]['title']]


async def auto_get_custom_iw_list(ctx: discord.AutocompleteContext):
    target = WikiTargetInfo(ctx_to_session(ctx)).get_interwikis()
    if not target:
        return []
    else:
        return list(target.keys())


async def default_wiki(ctx: discord.AutocompleteContext):
    if ctx.options["link"] == '':
        return ['https://minecraft.fandom.com/zh/']


@wiki.command(description="根据页面名称查询一个wiki页面")
@discord.option(name="title", description="页面名称", autocomplete=auto_search)
async def query(ctx: discord.ApplicationContext, title: str):
    await slash_parser(ctx, title)


@wiki.command(description="根据页面名称搜索一个wiki页面")
@discord.option(name="title", description="页面名称")
async def search(ctx: discord.ApplicationContext, title: str):
    await slash_parser(ctx, f'search {title}')


@wiki.command(name="id", description="根据页面ID查询一个wiki页面")
@discord.option(name="pid", description="页面ID")
async def page_id(ctx: discord.ApplicationContext, pid: str):
    await slash_parser(ctx, f'-p {pid}')


@wiki.command(name="set", description="设置起始查询wiki")
@discord.option(name="link", description="页面链接", autocomplete=default_wiki)
async def set_base(ctx: discord.ApplicationContext, link: str):
    await slash_parser(ctx, f'set {link}')


iw = wiki.create_subgroup("iw", "设置有关自定义Interwiki的命令")


@iw.command(description="添加自定义Interwiki")
@discord.option(name="interwiki", description="自定义iw名")
@discord.option(name="link", description="页面链接")
async def add(ctx: discord.ApplicationContext, iw: str, link: str):
    await slash_parser(ctx, f'iw add {iw} {link}')


@iw.command(name='remove', description="删除自定义Interwiki")
@discord.option(name="iw", description="自定义iw名", autocomplete=auto_get_custom_iw_list)
async def iw_remove(ctx: discord.ApplicationContext, iw: str):
    await slash_parser(ctx, f'iw rm {iw}')


@iw.command(name="list", description="查看所有已自定义的Interwiki")
async def iw_list(ctx: discord.ApplicationContext):
    await slash_parser(ctx, 'iw list')


@iw.command(description="获取自定义Interwiki的链接")
@discord.option(name="iw", description="自定义interwiki名", autocomplete=auto_get_custom_iw_list)
async def get(ctx: discord.ApplicationContext, iw: str):
    await slash_parser(ctx, f'iw get {iw}')


headers = wiki.create_subgroup("headers", "设置有关自定义header的命令")


@headers.command(name="set", description="添加自定义headers")
@discord.option(name="headers_json", description="自定义headers")
async def set_headers(ctx: discord.ApplicationContext, headers_json: str):
    await slash_parser(ctx, f'headers set {headers_json}')


@headers.command(description="删除一个自定义header")
@discord.option(name="header_key")
async def delete(ctx: discord.ApplicationContext, header_key: str):
    await slash_parser(ctx, f'headers del {header_key}')


@headers.command(name='show', description="查看所有自定义的headers")
async def headers_show(ctx: discord.ApplicationContext):
    await slash_parser(ctx, 'headers show')


@headers.command(name='reset', description="重置所有自定义的headers")
async def headers_reset(ctx: discord.ApplicationContext):
    await slash_parser(ctx, 'headers reset')


@wiki.command(description="是否启用Fandom全局Interwiki查询")
@discord.option(name="_", choices=['enable', 'disable'])
async def fandom(ctx: discord.ApplicationContext, _: str):
    await slash_parser(ctx, f'fandom {_}')