125 lines
4.5 KiB
Python
125 lines
4.5 KiB
Python
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 {_}')
|